{"id":"https://openalex.org/W7160916286","doi":"https://doi.org/10.48550/arxiv.2605.08639","title":"ReLibra: Routing-Replay-Guided Load Balancing for MoE Training in Reinforcement Learning","display_name":"ReLibra: Routing-Replay-Guided Load Balancing for MoE Training in Reinforcement Learning","publication_year":2026,"publication_date":"2026-05-09","ids":{"openalex":"https://openalex.org/W7160916286","doi":"https://doi.org/10.48550/arxiv.2605.08639"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.08639","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.08639","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.2605.08639","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5135917111","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/A5135963214","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/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/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/A5135925332","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/A5101852647","display_name":"Ruidong Zhu","orcid":"https://orcid.org/0000-0003-1389-7884"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhu, Ruidong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135922751","display_name":"Zili Zhang","orcid":null},"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/A5135971503","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/A5135990876","display_name":"Xin Jin","orcid":"https://orcid.org/0000-0001-7394-0210"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jin, Xin","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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.2612000107765198,"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.2612000107765198,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.1264999955892563,"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/T10036","display_name":"Advanced Neural Network Applications","score":0.061799999326467514,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"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.7139000296592712},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.6241999864578247},{"id":"https://openalex.org/keywords/oracle","display_name":"Oracle","score":0.6140999794006348},{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.6140000224113464},{"id":"https://openalex.org/keywords/matching","display_name":"Matching (statistics)","score":0.5037999749183655},{"id":"https://openalex.org/keywords/throughput","display_name":"Throughput","score":0.4973999857902527},{"id":"https://openalex.org/keywords/exploit","display_name":"Exploit","score":0.4927000105381012},{"id":"https://openalex.org/keywords/load-management","display_name":"Load management","score":0.42829999327659607}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7666000127792358},{"id":"https://openalex.org/C138959212","wikidata":"https://www.wikidata.org/wiki/Q1806783","display_name":"Load balancing (electrical power)","level":3,"score":0.7139000296592712},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.6241999864578247},{"id":"https://openalex.org/C55166926","wikidata":"https://www.wikidata.org/wiki/Q2892946","display_name":"Oracle","level":2,"score":0.6140999794006348},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.6140000224113464},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.5037999749183655},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.4973999857902527},{"id":"https://openalex.org/C165696696","wikidata":"https://www.wikidata.org/wiki/Q11287","display_name":"Exploit","level":2,"score":0.4927000105381012},{"id":"https://openalex.org/C2779370713","wikidata":"https://www.wikidata.org/wiki/Q357554","display_name":"Load management","level":2,"score":0.42829999327659607},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.4099000096321106},{"id":"https://openalex.org/C74172769","wikidata":"https://www.wikidata.org/wiki/Q1446839","display_name":"Routing (electronic design automation)","level":2,"score":0.4074999988079071},{"id":"https://openalex.org/C24856439","wikidata":"https://www.wikidata.org/wiki/Q352483","display_name":"Adaptive routing","level":5,"score":0.4000999927520752},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3871000111103058},{"id":"https://openalex.org/C67203356","wikidata":"https://www.wikidata.org/wiki/Q1321905","display_name":"Reinforcement","level":2,"score":0.36719998717308044},{"id":"https://openalex.org/C58328972","wikidata":"https://www.wikidata.org/wiki/Q184609","display_name":"Expert system","level":2,"score":0.362199991941452},{"id":"https://openalex.org/C62611344","wikidata":"https://www.wikidata.org/wiki/Q1062658","display_name":"Node (physics)","level":2,"score":0.3537999987602234},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.34389999508857727},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.3422999978065491},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3118000030517578},{"id":"https://openalex.org/C31258907","wikidata":"https://www.wikidata.org/wiki/Q1301371","display_name":"Computer network","level":1,"score":0.3100000023841858},{"id":"https://openalex.org/C2777904410","wikidata":"https://www.wikidata.org/wiki/Q7397","display_name":"Software","level":2,"score":0.28929999470710754},{"id":"https://openalex.org/C12590798","wikidata":"https://www.wikidata.org/wiki/Q3933199","display_name":"Replication (statistics)","level":2,"score":0.28360000252723694},{"id":"https://openalex.org/C2984173633","wikidata":"https://www.wikidata.org/wiki/Q22725","display_name":"Routing algorithm","level":4,"score":0.27730000019073486}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.08639","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.08639","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.2605.08639","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.08639","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":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Load":[0],"imbalance":[1],"is":[2,11,180],"a":[3,57,143],"long-standing":[4],"challenge":[5],"in":[6,13,60],"Mixture-of-Experts":[7],"(MoE)":[8],"training":[9,29,53,77,94,162],"and":[10,76,107,155,170],"exacerbated":[12],"reinforcement":[14],"learning":[15],"(RL)":[16],"for":[17,129],"LLMs,":[18],"where":[19],"hot":[20],"experts":[21,128],"can":[22],"shift":[23],"frequently":[24],"across":[25],"micro-batches.":[26],"Existing":[27],"MoE":[28,51,85,102,153],"systems":[30],"rely":[31],"on":[32,151],"historical":[33],"loads":[34],"to":[35,66,114,126,145,166,173],"predict":[36],"future":[37],"expert":[38,124,140],"demand,":[39],"making":[40],"them":[41],"less":[42],"effective":[43],"under":[44],"sharp":[45],"fluctuations.":[46,149],"We":[47],"propose":[48],"ReLibra,":[49],"an":[50,193],"RL":[52,156],"system":[54],"that":[55,159],"exploits":[56],"unique":[58],"opportunity":[59],"RL's":[61],"rollout-training":[62],"workflow,":[63],"routing":[64,89],"replay,":[65],"enable":[67],"fine-grained":[68],"load":[69,148],"balancing":[70],"at":[71,105,133],"micro-batch":[72],"granularity.":[73],"Because":[74],"rollout":[75],"process":[78],"the":[79,83,87,119,134,190],"same":[80,84],"tokens":[81],"with":[82],"parameters,":[86],"token-to-expert":[88],"decisions":[90],"are":[91],"known":[92],"before":[93],"starts.":[95],"Leveraging":[96],"this":[97],"information,":[98],"ReLibra":[99,122,160,185],"places":[100],"two":[101],"load-balancing":[103],"mechanisms":[104],"inter-":[106],"intra-batch":[108,135],"timescales,":[109],"matching":[110],"their":[111],"communication":[112],"patterns":[113],"hierarchical":[115],"network":[116],"bandwidths.":[117],"At":[118],"inter-batch":[120],"timescale,":[121,136],"performs":[123,139],"reordering":[125],"redistribute":[127],"batch-level":[130],"cross-node":[131],"balancing;":[132],"it":[137],"dynamically":[138],"replication":[141],"within":[142,187],"node":[144],"absorb":[146],"micro-batch-level":[147],"Experiments":[150],"diverse":[152],"LLMs":[154],"workloads":[157],"show":[158],"improves":[161],"throughput":[163,191],"by":[164,171],"up":[165,172],"1.6$\\times$":[167],"over":[168,175],"Megatron-LM":[169],"1.2$\\times$":[174],"EPLB,":[176],"even":[177],"when":[178],"EPLB":[179],"given":[181],"oracle":[182],"loads.":[183],"Moreover,":[184],"remains":[186],"6%-10%":[188],"of":[189,192],"idealized":[194],"balanced":[195],"baseline.":[196]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-13T00:00:00"}