{"id":"https://openalex.org/W7128805072","doi":"https://doi.org/10.48550/arxiv.2602.11220","title":"Patch the Distribution Mismatch: RL Rewriting Agent for Stable Off-Policy SFT","display_name":"Patch the Distribution Mismatch: RL Rewriting Agent for Stable Off-Policy SFT","publication_year":2026,"publication_date":"2026-02-11","ids":{"openalex":"https://openalex.org/W7128805072","doi":"https://doi.org/10.48550/arxiv.2602.11220"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.11220","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":null,"any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5125919352","display_name":"Jiacheng Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Jiacheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5117865091","display_name":"Ping Jian","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jian, Ping","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5125916426","display_name":"Zhen Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Zhen","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":null,"display_name":"Chen, Zirong","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Zirong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5125963849","display_name":"Keren Liao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liao, Keren","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5102557005","display_name":"Zhongbin Guo","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Guo, Zhongbin","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/T10028","display_name":"Topic Modeling","score":0.40720000863075256,"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/T10028","display_name":"Topic Modeling","score":0.40720000863075256,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.07240000367164612,"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/T13702","display_name":"Machine Learning in Healthcare","score":0.06430000066757202,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/rewriting","display_name":"Rewriting","score":0.8407999873161316},{"id":"https://openalex.org/keywords/leverage","display_name":"Leverage (statistics)","score":0.7304999828338623},{"id":"https://openalex.org/keywords/consistency","display_name":"Consistency (knowledge bases)","score":0.5568000078201294},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.5109000205993652},{"id":"https://openalex.org/keywords/downstream","display_name":"Downstream (manufacturing)","score":0.5008000135421753},{"id":"https://openalex.org/keywords/forgetting","display_name":"Forgetting","score":0.42010000348091125},{"id":"https://openalex.org/keywords/distribution","display_name":"Distribution (mathematics)","score":0.3806000053882599},{"id":"https://openalex.org/keywords/source-code","display_name":"Source code","score":0.362199991941452}],"concepts":[{"id":"https://openalex.org/C154690210","wikidata":"https://www.wikidata.org/wiki/Q1668499","display_name":"Rewriting","level":2,"score":0.8407999873161316},{"id":"https://openalex.org/C153083717","wikidata":"https://www.wikidata.org/wiki/Q6535263","display_name":"Leverage (statistics)","level":2,"score":0.7304999828338623},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7059999704360962},{"id":"https://openalex.org/C2776436953","wikidata":"https://www.wikidata.org/wiki/Q5163215","display_name":"Consistency (knowledge bases)","level":2,"score":0.5568000078201294},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.5109000205993652},{"id":"https://openalex.org/C2776207758","wikidata":"https://www.wikidata.org/wiki/Q5303302","display_name":"Downstream (manufacturing)","level":2,"score":0.5008000135421753},{"id":"https://openalex.org/C7149132","wikidata":"https://www.wikidata.org/wiki/Q1377840","display_name":"Forgetting","level":2,"score":0.42010000348091125},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3903999924659729},{"id":"https://openalex.org/C110121322","wikidata":"https://www.wikidata.org/wiki/Q865811","display_name":"Distribution (mathematics)","level":2,"score":0.3806000053882599},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.362199991941452},{"id":"https://openalex.org/C202615002","wikidata":"https://www.wikidata.org/wiki/Q783507","display_name":"Differentiable function","level":2,"score":0.35260000824928284},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.3393000066280365},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.3370000123977661},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3287000060081482},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.30959999561309814},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.2955000102519989},{"id":"https://openalex.org/C177769412","wikidata":"https://www.wikidata.org/wiki/Q278090","display_name":"Prior probability","level":3,"score":0.28600001335144043},{"id":"https://openalex.org/C2777904410","wikidata":"https://www.wikidata.org/wiki/Q7397","display_name":"Software","level":2,"score":0.2669000029563904},{"id":"https://openalex.org/C162319229","wikidata":"https://www.wikidata.org/wiki/Q175263","display_name":"Data structure","level":2,"score":0.25760000944137573},{"id":"https://openalex.org/C165464430","wikidata":"https://www.wikidata.org/wiki/Q1570441","display_name":"Parameterized complexity","level":2,"score":0.2572000026702881},{"id":"https://openalex.org/C198531522","wikidata":"https://www.wikidata.org/wiki/Q485146","display_name":"Sample (material)","level":2,"score":0.2500999867916107}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.11220","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2602.11220","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.11220","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":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2602.11220","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"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":[0],"language":[1],"models":[2],"(LLMs)":[3],"have":[4],"made":[5],"rapid":[6],"progress,":[7],"yet":[8],"adapting":[9],"them":[10],"to":[11,59,94,138,148,195],"downstream":[12,22,55,183,192],"scenarios":[13],"still":[14],"commonly":[15],"relies":[16],"on":[17,89,201,206],"supervised":[18],"fine-tuning":[19],"(SFT).":[20],"When":[21],"data":[23,44,57,103],"exhibit":[24],"a":[25,50,68,106,112,172,178],"substantial":[26],"distribution":[27,122,152],"shift":[28],"from":[29,67],"the":[30,73,81,118,150],"model's":[31,82],"prior":[32,58],"training":[33,56],"distribution,":[34,71],"SFT":[35,197],"can":[36,92],"induce":[37],"catastrophic":[38],"forgetting.":[39],"To":[40,97],"narrow":[41],"this":[42],"gap,":[43],"rewriting":[45,104,113],"has":[46],"been":[47],"proposed":[48],"as":[49,105],"data-centric":[51],"approach":[52],"that":[53,115,188],"rewrites":[54,66],"SFT.":[60,184],"However,":[61],"existing":[62],"methods":[63],"typically":[64],"sample":[65],"prompt-induced":[69],"conditional":[70],"so":[72],"resulting":[74],"targets":[75],"are":[76,133],"not":[77],"necessarily":[78],"aligned":[79],"with":[80,141],"natural":[83],"QA-style":[84,120,166],"generation":[85,121],"distribution.":[86],"Moreover,":[87],"reliance":[88],"fixed":[90],"templates":[91],"lead":[93],"diversity":[95,129,170],"collapse.":[96],"address":[98],"these":[99],"issues,":[100],"we":[101,144],"cast":[102],"policy":[107,114],"learning":[108,147],"problem":[109],"and":[110,130,156,169],"learn":[111],"better":[116],"matches":[117],"backbone's":[119],"while":[123,198],"preserving":[124],"diversity.":[125],"Since":[126],"distributional":[127,167],"alignment,":[128],"task":[131],"consistency":[132],"automatically":[134],"evaluable":[135],"but":[136],"difficult":[137],"optimize":[139,149],"end-to-end":[140],"differentiable":[142],"objectives,":[143],"leverage":[145],"reinforcement":[146],"rewrite":[151],"under":[153,171],"reward":[154],"feedback":[155],"propose":[157],"an":[158],"RL-based":[159],"data-rewriting":[160],"agent.":[161],"The":[162],"agent":[163],"jointly":[164],"optimizes":[165],"alignment":[168],"hard":[173],"task-consistency":[174],"gate,":[175],"thereby":[176],"constructing":[177],"higher-quality":[179],"rewritten":[180],"dataset":[181],"for":[182],"Extensive":[185],"experiments":[186],"show":[187],"our":[189],"method":[190],"achieves":[191],"gains":[193],"comparable":[194],"standard":[196],"reducing":[199],"forgetting":[200],"non-downstream":[202],"benchmarks":[203],"by":[204],"12.34%":[205],"average.":[207],"Our":[208],"code":[209],"is":[210],"available":[211],"at":[212],"https://anonymous.4open.science/r/Patch-the-Prompt-Gap-4112":[213],".":[214]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-02-14T00:00:00"}