{"id":"https://openalex.org/W7161675408","doi":"https://doi.org/10.48550/arxiv.2605.17575","title":"UniAlign: A Model-Agnostic Framework for Robust Network Traffic Classification under Distribution Shifts","display_name":"UniAlign: A Model-Agnostic Framework for Robust Network Traffic Classification under Distribution Shifts","publication_year":2026,"publication_date":"2026-05-17","ids":{"openalex":"https://openalex.org/W7161675408","doi":"https://doi.org/10.48550/arxiv.2605.17575"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.17575","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.17575","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.17575","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5136494730","display_name":"Tongze Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Tongze","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136487615","display_name":"Xiaohui Xie","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xie, Xiaohui","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5078217759","display_name":"Wenduo Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Wenduo","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136488190","display_name":"Chuyi Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Chuyi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136471089","display_name":"Yong Cui","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Cui, Yong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11598","display_name":"Internet Traffic Analysis and Secure E-voting","score":0.9365000128746033,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T11598","display_name":"Internet Traffic Analysis and Secure E-voting","score":0.9365000128746033,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10400","display_name":"Network Security and Intrusion Detection","score":0.032999999821186066,"subfield":{"id":"https://openalex.org/subfields/1705","display_name":"Computer Networks and Communications"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T14258","display_name":"Legal and Policy Issues","score":0.0052999998442828655,"subfield":{"id":"https://openalex.org/subfields/3308","display_name":"Law"},"field":{"id":"https://openalex.org/fields/33","display_name":"Social Sciences"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/robustness","display_name":"Robustness (evolution)","score":0.8295000195503235},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.6589000225067139},{"id":"https://openalex.org/keywords/training-set","display_name":"Training set","score":0.4620000123977661},{"id":"https://openalex.org/keywords/baseline","display_name":"Baseline (sea)","score":0.4440000057220459},{"id":"https://openalex.org/keywords/feature","display_name":"Feature (linguistics)","score":0.3837999999523163},{"id":"https://openalex.org/keywords/data-modeling","display_name":"Data modeling","score":0.37529999017715454},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.3407000005245209},{"id":"https://openalex.org/keywords/network-model","display_name":"Network model","score":0.31520000100135803}],"concepts":[{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.8295000195503235},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.6589000225067139},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.644599974155426},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.573199987411499},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5340999960899353},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.48420000076293945},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.4620000123977661},{"id":"https://openalex.org/C12725497","wikidata":"https://www.wikidata.org/wiki/Q810247","display_name":"Baseline (sea)","level":2,"score":0.4440000057220459},{"id":"https://openalex.org/C2776401178","wikidata":"https://www.wikidata.org/wiki/Q12050496","display_name":"Feature (linguistics)","level":2,"score":0.3837999999523163},{"id":"https://openalex.org/C67186912","wikidata":"https://www.wikidata.org/wiki/Q367664","display_name":"Data modeling","level":2,"score":0.37529999017715454},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.3407000005245209},{"id":"https://openalex.org/C104122410","wikidata":"https://www.wikidata.org/wiki/Q1416406","display_name":"Network model","level":2,"score":0.31520000100135803},{"id":"https://openalex.org/C67226441","wikidata":"https://www.wikidata.org/wiki/Q1665389","display_name":"Robust statistics","level":3,"score":0.3133000135421753},{"id":"https://openalex.org/C133462117","wikidata":"https://www.wikidata.org/wiki/Q4929239","display_name":"Data collection","level":2,"score":0.31040000915527344},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.28610000014305115},{"id":"https://openalex.org/C2779714256","wikidata":"https://www.wikidata.org/wiki/Q25305062","display_name":"Multiple Models","level":2,"score":0.2851000130176544},{"id":"https://openalex.org/C148730421","wikidata":"https://www.wikidata.org/wiki/Q141090","display_name":"Encryption","level":2,"score":0.28459998965263367},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.2759000062942505},{"id":"https://openalex.org/C9357733","wikidata":"https://www.wikidata.org/wiki/Q6878417","display_name":"Missing data","level":2,"score":0.272599995136261},{"id":"https://openalex.org/C2776145971","wikidata":"https://www.wikidata.org/wiki/Q30673951","display_name":"Labeled data","level":2,"score":0.2718000113964081},{"id":"https://openalex.org/C149441793","wikidata":"https://www.wikidata.org/wiki/Q200726","display_name":"Probability distribution","level":2,"score":0.2678999900817871},{"id":"https://openalex.org/C136389625","wikidata":"https://www.wikidata.org/wiki/Q334384","display_name":"Supervised learning","level":3,"score":0.2581000030040741},{"id":"https://openalex.org/C83665646","wikidata":"https://www.wikidata.org/wiki/Q42139305","display_name":"Feature vector","level":2,"score":0.2565000057220459},{"id":"https://openalex.org/C52622490","wikidata":"https://www.wikidata.org/wiki/Q1026626","display_name":"Feature extraction","level":2,"score":0.2556000053882599}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.17575","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.17575","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.17575","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.17575","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"score":0.5026636719703674,"id":"https://metadata.un.org/sdg/14","display_name":"Life below water"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Network":[0],"traffic":[1,83],"classification":[2,163],"(NTC)":[3],"models":[4,68,115,153],"often":[5],"suffer":[6],"severe":[7],"performance":[8],"degradation":[9],"when":[10],"deployed":[11],"in":[12,179,183],"real-world":[13],"environments":[14],"due":[15],"to":[16,30,38,40],"distribution":[17,70,136],"shifts":[18],"caused":[19],"by":[20,97,165,171,177],"changing":[21],"network":[22,87],"conditions.":[23],"Existing":[24],"robustness-enhancing":[25],"approaches":[26],"are":[27],"commonly":[28],"coupled":[29],"specific":[31,118],"model":[32,91],"architectures":[33],"or":[34,45,121],"data":[35,141],"settings,":[36],"fail":[37],"generalize":[39],"state-of-the-art":[41],"raw-byte-based":[42],"NTC":[43,67,114,152],"models,":[44],"incur":[46],"significant":[47],"training":[48,125,192],"overhead.":[49],"In":[50],"this":[51],"paper,":[52],"we":[53],"propose":[54],"UniAlign,":[55],"a":[56,101],"novel":[57],"model-agnostic":[58],"framework":[59,106],"that":[60],"improves":[61,161],"the":[62,79,174,191],"robustness":[63,96],"of":[64,81,190,194],"deep":[65],"learning-based":[66],"under":[69],"shifts.":[71],"UniAlign":[72,129,160],"combines":[73],"\\emph{domain":[74],"alignment":[75],"fine-tuning},":[76],"which":[77,93],"encourages":[78],"learning":[80],"domain-invariant":[82],"representations":[84],"across":[85],"heterogeneous":[86],"conditions,":[88],"with":[89,157],"\\emph{stable":[90],"ensembling},":[92],"enhances":[94],"inference":[95],"aggregating":[98],"checkpoints":[99],"within":[100],"flat":[102],"loss":[103],"region.":[104],"The":[105],"can":[107],"be":[108],"seamlessly":[109],"integrated":[110],"into":[111],"existing":[112],"supervised":[113],"without":[116],"requiring":[117,187],"feature":[119],"modalities":[120],"introducing":[122],"non-constant":[123],"additional":[124],"costs.":[126],"We":[127],"evaluate":[128],"on":[130,149],"three":[131],"public":[132],"datasets":[133],"covering":[134],"diverse":[135],"shifts,":[137],"including":[138],"encryption":[139],"schemes,":[140],"collection":[142],"devices,":[143],"and":[144,167,181],"attack":[145],"behaviors.":[146],"Experimental":[147],"results":[148],"two":[150],"representative":[151],"demonstrate":[154],"that,":[155],"compared":[156],"standard":[158],"training,":[159],"average":[162,168],"accuracy":[164,180],"2.51\\%":[166],"F1":[169,184],"score":[170],"2.71\\%,":[172],"outperforming":[173],"strongest":[175],"baseline":[176],"1.45\\%":[178],"1.69\\%":[182],"score,":[185],"while":[186],"only":[188],"12.4\\%--53.9\\%":[189],"time":[193],"all":[195],"NTC-specific":[196],"baselines.":[197]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-20T00:00:00"}