{"id":"https://openalex.org/W7163206692","doi":"https://doi.org/10.48550/arxiv.2606.00628","title":"Robust Reasoning via Dynamic Token Selection for Distribution-Aligned Self-Distillation","display_name":"Robust Reasoning via Dynamic Token Selection for Distribution-Aligned Self-Distillation","publication_year":2026,"publication_date":"2026-05-30","ids":{"openalex":"https://openalex.org/W7163206692","doi":"https://doi.org/10.48550/arxiv.2606.00628"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.00628","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00628","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.2606.00628","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137706696","display_name":"Ruiqi Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Ruiqi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137710945","display_name":"Lingxiang Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Lingxiang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137670178","display_name":"Hainan Zhang Zhiming Zheng","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zheng, Hainan Zhang Zhiming","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.4945000112056732,"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.4945000112056732,"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.11959999799728394,"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.04769999906420708,"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/security-token","display_name":"Security token","score":0.684499979019165},{"id":"https://openalex.org/keywords/rewriting","display_name":"Rewriting","score":0.656000018119812},{"id":"https://openalex.org/keywords/robustness","display_name":"Robustness (evolution)","score":0.5871000289916992},{"id":"https://openalex.org/keywords/encode","display_name":"ENCODE","score":0.5590999722480774},{"id":"https://openalex.org/keywords/generative-grammar","display_name":"Generative grammar","score":0.48919999599456787},{"id":"https://openalex.org/keywords/commonsense-reasoning","display_name":"Commonsense reasoning","score":0.48489999771118164},{"id":"https://openalex.org/keywords/planner","display_name":"Planner","score":0.43790000677108765},{"id":"https://openalex.org/keywords/logical-consequence","display_name":"Logical consequence","score":0.4036000072956085}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7732999920845032},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.684499979019165},{"id":"https://openalex.org/C154690210","wikidata":"https://www.wikidata.org/wiki/Q1668499","display_name":"Rewriting","level":2,"score":0.656000018119812},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5960000157356262},{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.5871000289916992},{"id":"https://openalex.org/C66746571","wikidata":"https://www.wikidata.org/wiki/Q1134833","display_name":"ENCODE","level":3,"score":0.5590999722480774},{"id":"https://openalex.org/C39890363","wikidata":"https://www.wikidata.org/wiki/Q36108","display_name":"Generative grammar","level":2,"score":0.48919999599456787},{"id":"https://openalex.org/C193221554","wikidata":"https://www.wikidata.org/wiki/Q5153664","display_name":"Commonsense reasoning","level":2,"score":0.48489999771118164},{"id":"https://openalex.org/C2776999362","wikidata":"https://www.wikidata.org/wiki/Q2349274","display_name":"Planner","level":2,"score":0.43790000677108765},{"id":"https://openalex.org/C134752490","wikidata":"https://www.wikidata.org/wiki/Q374182","display_name":"Logical consequence","level":2,"score":0.4036000072956085},{"id":"https://openalex.org/C42058472","wikidata":"https://www.wikidata.org/wiki/Q810214","display_name":"Base (topology)","level":2,"score":0.376800000667572},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.36010000109672546},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3596999943256378},{"id":"https://openalex.org/C4554734","wikidata":"https://www.wikidata.org/wiki/Q593744","display_name":"Knowledge base","level":2,"score":0.35010001063346863},{"id":"https://openalex.org/C167966045","wikidata":"https://www.wikidata.org/wiki/Q5532625","display_name":"Generative model","level":3,"score":0.34779998660087585},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.3028999865055084},{"id":"https://openalex.org/C21847791","wikidata":"https://www.wikidata.org/wiki/Q191081","display_name":"Logical conjunction","level":2,"score":0.3027999997138977},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.2971000075340271},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.29269999265670776},{"id":"https://openalex.org/C43971567","wikidata":"https://www.wikidata.org/wiki/Q3142865","display_name":"Logical reasoning","level":2,"score":0.27410000562667847},{"id":"https://openalex.org/C20693621","wikidata":"https://www.wikidata.org/wiki/Q6667502","display_name":"Logical framework","level":2,"score":0.27070000767707825},{"id":"https://openalex.org/C2776135515","wikidata":"https://www.wikidata.org/wiki/Q17143721","display_name":"Regularization (linguistics)","level":2,"score":0.2621000111103058},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.25589999556541443},{"id":"https://openalex.org/C150189527","wikidata":"https://www.wikidata.org/wiki/Q356674","display_name":"Reference model","level":2,"score":0.2524999976158142}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.00628","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00628","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.2606.00628","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00628","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":[{"id":"https://metadata.un.org/sdg/4","score":0.7359344959259033,"display_name":"Quality Education"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Self-distillation":[0],"improves":[1,152],"learning":[2],"efficiency":[3],"by":[4,68],"rewriting":[5,44],"reference":[6,19,69,103],"answers":[7,20],"as":[8],"training":[9],"data":[10,45],"that":[11,42,122,142],"better":[12],"matches":[13],"the":[14,27,43,78,115],"model's":[15,80,117],"own":[16],"distribution.":[17],"However,":[18],"also":[21],"introduce":[22],"strong":[23],"stylistic":[24,65],"biases,":[25],"causing":[26],"generative":[28],"model":[29,104],"to":[30,105,114],"imitate":[31],"surface":[32],"forms":[33],"rather":[34],"than":[35],"learn":[36],"useful":[37,124],"reasoning":[38,89,139],"patterns.":[39],"We":[40],"observe":[41],"contains":[46],"a":[47],"large":[48],"number":[49],"of":[50,156],"high-perplexity":[51],"(PPL)":[52],"tokens,":[53,150],"coming":[54],"from":[55],"two":[56],"distinct":[57],"sources:":[58],"beneficial":[59],"knowledge-enhancing":[60],"logical":[61,125],"corrections,":[62],"and":[63,83,109,137,151],"harmful":[64],"drift":[66],"induced":[67],"imitation.":[70],"Treating":[71],"all":[72],"such":[73],"tokens":[74,108,121],"equally":[75],"can":[76],"disrupt":[77],"base":[79,116],"original":[81],"distribution":[82],"degrade":[84],"performance,":[85],"especially":[86],"on":[87,134],"difficult":[88],"tasks.":[90],"To":[91],"address":[92],"this,":[93],"we":[94],"propose":[95],"Distribution-Aligned":[96],"Self-Distillation":[97],"(DASD),":[98],"which":[99],"uses":[100],"an":[101],"answer-aware":[102],"generate":[106],"candidate":[107],"dynamically":[110],"filters":[111],"them":[112],"according":[113],"confidence.":[118],"DASD":[119,143],"preserves":[120],"encode":[123],"knowledge":[126],"while":[127],"suppressing":[128],"distributionally":[129],"misaligned":[130],"style":[131],"noise.":[132],"Experiments":[133],"math,":[135],"code,":[136],"commonsense":[138],"benchmarks":[140],"show":[141],"consistently":[144],"outperforms":[145],"competitive":[146],"baselines,":[147],"reduces":[148],"high-PPL":[149],"robustness":[153],"across":[154],"tasks":[155],"varying":[157],"difficulty.":[158]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-03T00:00:00"}