{"id":"https://openalex.org/W7163556042","doi":"https://doi.org/10.48550/arxiv.2606.04101","title":"UltraEP: Unleash MoE Training and Inference on Rack-Scale Nodes with Near-Optimal Load Balancing","display_name":"UltraEP: Unleash MoE Training and Inference on Rack-Scale Nodes with Near-Optimal Load Balancing","publication_year":2026,"publication_date":"2026-06-02","ids":{"openalex":"https://openalex.org/W7163556042","doi":"https://doi.org/10.48550/arxiv.2606.04101"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.04101","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.04101","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2606.04101","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137904664","display_name":"Xinming Wei","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wei, Xinming","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137815926","display_name":"Chao Jin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jin, Chao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137817497","display_name":"Tuo Dai","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dai, Tuo","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5000644890","display_name":"Yinmin Zhong","orcid":"https://orcid.org/0000-0002-2504-7652"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhong, Yinmin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137895139","display_name":"Shan Yu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yu, Shan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132695431","display_name":"Chengxu Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Chengxu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137824366","display_name":"Bingyang Wu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Bingyang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137897687","display_name":"Zili Zhang","orcid":"https://orcid.org/0000-0002-2072-4626"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Zili","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5042195167","display_name":"Jing Mai","orcid":"https://orcid.org/0000-0002-3787-1534"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mai, Jing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137843335","display_name":"Qianchao Zhu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhu, Qianchao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137882822","display_name":"Zhouyang Li","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Zhouyang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137846560","display_name":"Yuliang Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Yuliang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5023468643","display_name":"Guojie Luo","orcid":"https://orcid.org/0000-0003-4932-3655"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Luo, Guojie","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/T10847","display_name":"Advanced Optical Network Technologies","score":0.38690000772476196,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10847","display_name":"Advanced Optical Network Technologies","score":0.38690000772476196,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10714","display_name":"Software-Defined Networks and 5G","score":0.08380000293254852,"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/T10232","display_name":"Optical Network Technologies","score":0.07580000162124634,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/load-balancing","display_name":"Load balancing (electrical power)","score":0.6696000099182129},{"id":"https://openalex.org/keywords/scalability","display_name":"Scalability","score":0.6694999933242798},{"id":"https://openalex.org/keywords/robustness","display_name":"Robustness (evolution)","score":0.6201000213623047},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.546500027179718},{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.5058000087738037},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.42829999327659607},{"id":"https://openalex.org/keywords/flexibility","display_name":"Flexibility (engineering)","score":0.4043000042438507}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6901999711990356},{"id":"https://openalex.org/C138959212","wikidata":"https://www.wikidata.org/wiki/Q1806783","display_name":"Load balancing (electrical power)","level":3,"score":0.6696000099182129},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.6694999933242798},{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.6201000213623047},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.546500027179718},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.5058000087738037},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.46889999508857727},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.42829999327659607},{"id":"https://openalex.org/C31258907","wikidata":"https://www.wikidata.org/wiki/Q1301371","display_name":"Computer network","level":1,"score":0.42340001463890076},{"id":"https://openalex.org/C2780598303","wikidata":"https://www.wikidata.org/wiki/Q65921492","display_name":"Flexibility (engineering)","level":2,"score":0.4043000042438507},{"id":"https://openalex.org/C58328972","wikidata":"https://www.wikidata.org/wiki/Q184609","display_name":"Expert system","level":2,"score":0.3643999993801117},{"id":"https://openalex.org/C2779370713","wikidata":"https://www.wikidata.org/wiki/Q357554","display_name":"Load management","level":2,"score":0.3582000136375427},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.33219999074935913},{"id":"https://openalex.org/C2777904410","wikidata":"https://www.wikidata.org/wiki/Q7397","display_name":"Software","level":2,"score":0.30329999327659607},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.298799991607666},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.2978000044822693},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.2858000099658966},{"id":"https://openalex.org/C82876162","wikidata":"https://www.wikidata.org/wiki/Q17096504","display_name":"Latency (audio)","level":2,"score":0.2743000090122223},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.2678000032901764}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.04101","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.04101","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2606.04101","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.04101","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Large-scale":[0],"expert":[1,19,97],"parallelism":[2],"(EP)":[3],"is":[4],"becoming":[5],"pivotal":[6],"for":[7,43,58],"training":[8,61,157],"and":[9,28,62,84,96,118,130,158],"serving":[10,63],"frontier":[11],"MoE":[12,60,148],"models,":[13],"but":[14],"it":[15],"also":[16],"amplifies":[17],"device-level":[18],"load":[20,48,112],"imbalance":[21,178],"into":[22],"compute":[23],"stragglers,":[24],"token":[25],"all-to-all":[26],"bottlenecks,":[27],"activation-memory":[29],"spikes.":[30],"Existing":[31],"balancers":[32],"redistribute":[33],"experts":[34],"periodically":[35],"based":[36],"on":[37,65,86],"historical":[38],"load,":[39],"which":[40,89],"becomes":[41],"unreliable":[42],"production":[44],"deployments":[45],"with":[46,113,125],"non-stationary":[47],"patterns.":[49],"We":[50,134],"present":[51],"UltraEP,":[52],"the":[53,70,120,164,175],"first":[54],"exact-load,":[55],"real-time":[56],"balancer":[57],"large-EP":[59],"prefill":[64],"rack-scale":[66],"nodes":[67],"(RSNs).":[68],"Leveraging":[69],"extended":[71],"scale-up":[72],"connectivity":[73],"among":[74],"dozens":[75],"of":[76,93,141,163],"GPUs":[77],"within":[78],"RSNs,":[79],"UltraEP":[80,107,136,160],"rebalances":[81],"every":[82],"microbatch":[83],"layer":[85],"critical":[87],"paths,":[88],"requires":[90],"nontrivial":[91],"co-design":[92],"plan":[94],"solving":[95],"replication":[98],"communication":[99],"to":[100,110,143,152,181],"minimize":[101],"exposed":[102],"overhead.":[103],"To":[104],"this":[105],"end,":[106],"eagerly":[108],"reacts":[109],"post-gating":[111],"an":[114],"efficient":[115],"quota-driven":[116],"planner,":[117],"executes":[119],"resulting":[121],"irregular":[122],"expert-state":[123],"transfers":[124],"RSN-native":[126],"persistent":[127],"tile":[128],"streaming":[129],"relay-based":[131],"fan-out":[132],"mitigation.":[133],"evaluate":[135],"in":[137],"a":[138],"multi-RSN":[139],"deployment":[140],"up":[142],"256":[144],"GPUs,":[145],"using":[146],"cutting-edge":[147],"models":[149],"from":[150,179],"106B":[151],"671B":[153],"parameters.":[154],"Averaged":[155],"across":[156],"serving,":[159],"achieves":[161],"94.3%":[162],"force-balanced":[165],"ideal":[166],"throughput,":[167],"delivering":[168],"1.49$\\times$":[169],"improvement":[170],"over":[171],"no-balancing,":[172],"while":[173],"reducing":[174],"final":[176],"inter-rank":[177],"1.30$-$4.01":[180],"1.01$-$1.04.":[182]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-06-05T00:00:00"}