{"id":"https://openalex.org/W7163252926","doi":"https://doi.org/10.48550/arxiv.2606.00395","title":"PR2: Predictive Routing Replay for MoE-Based LLM Reinforcement Learning","display_name":"PR2: Predictive Routing Replay for MoE-Based LLM Reinforcement Learning","publication_year":2026,"publication_date":"2026-05-29","ids":{"openalex":"https://openalex.org/W7163252926","doi":"https://doi.org/10.48550/arxiv.2606.00395"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.00395","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00395","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.00395","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137707556","display_name":"Daize Dong","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dong, Daize","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137664696","display_name":"Junlin Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Junlin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137699990","display_name":"Haolong Jia","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jia, Haolong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137711085","display_name":"Jiang Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Jiang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137678030","display_name":"Jiawei Wu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Jiawei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137669179","display_name":"Huanwei Di","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Di, Huanwei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102823836","display_name":"Jialian Wu","orcid":"https://orcid.org/0000-0003-2629-3270"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Jialian","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137653701","display_name":"Zhengzhong Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Zhengzhong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137680916","display_name":"Zicheng Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Zicheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137662248","display_name":"Emad Barsoum","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Barsoum, Emad","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137674592","display_name":"Dimitris N. Metaxas","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Metaxas, Dimitris N.","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137675874","display_name":"Hongyi Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Hongyi","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/T11273","display_name":"Advanced Graph Neural Networks","score":0.31360000371932983,"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/T11273","display_name":"Advanced Graph Neural Networks","score":0.31360000371932983,"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.1453000009059906,"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/T11704","display_name":"Mobile Crowdsensing and Crowdsourcing","score":0.07680000364780426,"subfield":{"id":"https://openalex.org/subfields/1706","display_name":"Computer Science Applications"},"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/router","display_name":"Router","score":0.8633000254631042},{"id":"https://openalex.org/keywords/routing","display_name":"Routing (electronic design automation)","score":0.6984000205993652},{"id":"https://openalex.org/keywords/consistency","display_name":"Consistency (knowledge bases)","score":0.6601999998092651},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.5464000105857849},{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.34150001406669617},{"id":"https://openalex.org/keywords/stability","display_name":"Stability (learning theory)","score":0.32829999923706055},{"id":"https://openalex.org/keywords/routing-protocol","display_name":"Routing protocol","score":0.3230000138282776}],"concepts":[{"id":"https://openalex.org/C2775896111","wikidata":"https://www.wikidata.org/wiki/Q642560","display_name":"Router","level":2,"score":0.8633000254631042},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7515000104904175},{"id":"https://openalex.org/C74172769","wikidata":"https://www.wikidata.org/wiki/Q1446839","display_name":"Routing (electronic design automation)","level":2,"score":0.6984000205993652},{"id":"https://openalex.org/C2776436953","wikidata":"https://www.wikidata.org/wiki/Q5163215","display_name":"Consistency (knowledge bases)","level":2,"score":0.6601999998092651},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.5464000105857849},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.48910000920295715},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3813999891281128},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.37700000405311584},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.34150001406669617},{"id":"https://openalex.org/C31258907","wikidata":"https://www.wikidata.org/wiki/Q1301371","display_name":"Computer network","level":1,"score":0.3368000090122223},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.32829999923706055},{"id":"https://openalex.org/C104954878","wikidata":"https://www.wikidata.org/wiki/Q1648707","display_name":"Routing protocol","level":3,"score":0.3230000138282776},{"id":"https://openalex.org/C24856439","wikidata":"https://www.wikidata.org/wiki/Q352483","display_name":"Adaptive routing","level":5,"score":0.30709999799728394},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.2944999933242798},{"id":"https://openalex.org/C195780805","wikidata":"https://www.wikidata.org/wiki/Q1535986","display_name":"Metrics","level":5,"score":0.2799000144004822},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.27810001373291016},{"id":"https://openalex.org/C184896649","wikidata":"https://www.wikidata.org/wiki/Q290066","display_name":"Routing table","level":4,"score":0.27649998664855957},{"id":"https://openalex.org/C2984173633","wikidata":"https://www.wikidata.org/wiki/Q22725","display_name":"Routing algorithm","level":4,"score":0.273499995470047}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.00395","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00395","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.00395","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00395","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":{"Mixture":[0],"of":[1],"Experts":[2],"(MoE)":[3],"Large":[4],"Language":[5],"Models":[6],"(LLMs)":[7],"achieve":[8],"strong":[9],"performance":[10,176],"at":[11],"scale.":[12],"However,":[13],"reinforcement":[14],"learning":[15],"(RL)":[16],"on":[17],"MoE-based":[18],"LLMs":[19],"often":[20],"suffers":[21],"from":[22],"training":[23,46,145],"instability.":[24],"A":[25],"root":[26],"cause":[27],"is":[28],"router":[29,80,88,103,114],"drift,":[30],"i.e.,":[31],"expert":[32],"activations":[33],"can":[34],"change":[35],"drastically":[36],"across":[37,177],"model":[38],"updates":[39,84],"and":[40,45,52,85,162,173],"differ":[41],"between":[42],"disaggregated":[43],"rollout":[44,118],"phases,":[47],"causing":[48],"large":[49],"rollout--training":[50],"mismatch":[51],"unstable":[53],"importance":[54,158],"sampling":[55],"weights":[56],"in":[57],"PPO-style":[58],"RL":[59,171],"algorithms.":[60],"Routing":[61,97],"replay":[62,69,148],"mitigates":[63],"this":[64,92],"issue":[65],"by":[66],"freezing":[67],"the":[68,79,117,122,144,149],"route":[70,152],"within":[71],"each":[72,102],"reasoning":[73,179],"trajectory,":[74],"but":[75],"it":[76],"ignores":[77],"how":[78],"evolves":[81],"under":[82],"off-policy":[83],"thus":[86],"causes":[87],"staleness.":[89],"To":[90],"address":[91],"limitation,":[93],"we":[94,120,147],"propose":[95],"Predictive":[96],"Replay":[98],"(PR2),":[99],"which":[100],"augments":[101],"with":[104],"a":[105],"lightweight":[106],"evolution":[107],"predictor":[108],"that":[109,135,165],"learns":[110],"to":[111,126,132,138,153],"anticipate":[112],"short-horizon":[113],"evolution.":[115],"During":[116,143],"phase,":[119,146],"use":[121],"predictive":[123],"routing":[124],"distribution":[125],"apply":[127],"top-$k$":[128],"routing,":[129],"enabling":[130],"gradients":[131],"reach":[133],"experts":[134],"are":[136],"likely":[137],"become":[139],"active":[140],"after":[141],"updates.":[142],"resulting":[150],"predicted":[151],"retain":[154],"consistency":[155],"for":[156],"stable":[157],"estimation.":[159],"Theoretical":[160],"analysis":[161],"experiments":[163],"support":[164],"PR2":[166],"reduces":[167],"routing-induced":[168],"mismatch,":[169],"improves":[170],"stability,":[172],"yields":[174],"stronger":[175],"various":[178],"benchmarks.":[180]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-06-03T00:00:00"}