{"id":"https://openalex.org/W7134813685","doi":"https://doi.org/10.48550/arxiv.2603.07079","title":"Entropy-Aware On-Policy Distillation of Language Models","display_name":"Entropy-Aware On-Policy Distillation of Language Models","publication_year":2026,"publication_date":"2026-03-07","ids":{"openalex":"https://openalex.org/W7134813685","doi":"https://doi.org/10.48550/arxiv.2603.07079"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2603.07079","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","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":null,"display_name":"Jin, Woogyeol","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Jin, Woogyeol","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128654858","display_name":"Taywon Min","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Min, Taywon","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128677853","display_name":"Yongjin Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Yongjin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5053049612","display_name":"Swanand Kadhe","orcid":"https://orcid.org/0000-0003-2006-1030"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kadhe, Swanand Ravindra","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128688278","display_name":"Yi Zhou","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhou, Yi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5103053820","display_name":"Dennis Wei","orcid":"https://orcid.org/0000-0002-6510-1537"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wei, Dennis","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5010677582","display_name":"Nathalie Baracaldo","orcid":"https://orcid.org/0000-0001-9469-045X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Baracaldo, Nathalie","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5128640613","display_name":"Kimin Lee","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lee, Kimin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":8,"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.3028999865055084,"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.3028999865055084,"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.12110000103712082,"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"}},{"id":"https://openalex.org/T11902","display_name":"Intelligent Tutoring Systems and Adaptive Learning","score":0.09809999912977219,"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/distillation","display_name":"Distillation","score":0.6820999979972839},{"id":"https://openalex.org/keywords/robustness","display_name":"Robustness (evolution)","score":0.6291000247001648},{"id":"https://openalex.org/keywords/entropy","display_name":"Entropy (arrow of time)","score":0.5745999813079834},{"id":"https://openalex.org/keywords/property","display_name":"Property (philosophy)","score":0.5333999991416931},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.45179998874664307},{"id":"https://openalex.org/keywords/range","display_name":"Range (aeronautics)","score":0.38940000534057617},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.37279999256134033}],"concepts":[{"id":"https://openalex.org/C204030448","wikidata":"https://www.wikidata.org/wiki/Q101017","display_name":"Distillation","level":2,"score":0.6820999979972839},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6639000177383423},{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.6291000247001648},{"id":"https://openalex.org/C106301342","wikidata":"https://www.wikidata.org/wiki/Q4117933","display_name":"Entropy (arrow of time)","level":2,"score":0.5745999813079834},{"id":"https://openalex.org/C189950617","wikidata":"https://www.wikidata.org/wiki/Q937228","display_name":"Property (philosophy)","level":2,"score":0.5333999991416931},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.45179998874664307},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3950999975204468},{"id":"https://openalex.org/C204323151","wikidata":"https://www.wikidata.org/wiki/Q905424","display_name":"Range (aeronautics)","level":2,"score":0.38940000534057617},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.37279999256134033},{"id":"https://openalex.org/C171752962","wikidata":"https://www.wikidata.org/wiki/Q255166","display_name":"Kullback\u2013Leibler divergence","level":2,"score":0.3725999891757965},{"id":"https://openalex.org/C9679016","wikidata":"https://www.wikidata.org/wiki/Q1417473","display_name":"Principle of maximum entropy","level":2,"score":0.36649999022483826},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.34630000591278076},{"id":"https://openalex.org/C126388530","wikidata":"https://www.wikidata.org/wiki/Q1131737","display_name":"Imitation","level":2,"score":0.3352000117301941},{"id":"https://openalex.org/C12725497","wikidata":"https://www.wikidata.org/wiki/Q810247","display_name":"Baseline (sea)","level":2,"score":0.3325999975204468},{"id":"https://openalex.org/C2781316041","wikidata":"https://www.wikidata.org/wiki/Q1230584","display_name":"Diversity (politics)","level":2,"score":0.28209999203681946},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.2524999976158142}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2603.07079","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2603.07079","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.07079","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":"article"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2603.07079","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[{"score":0.7453622221946716,"id":"https://metadata.un.org/sdg/4","display_name":"Quality Education"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"On-policy":[0],"distillation":[1,161],"is":[2,76,89,171],"a":[3,13],"promising":[4],"approach":[5],"for":[6,148,151,155,168,173],"transferring":[7],"knowledge":[8,179],"between":[9],"language":[10],"models,":[11],"where":[12],"student":[14,33],"learns":[15],"from":[16],"dense":[17],"token-level":[18,124],"signals":[19,57],"along":[20],"its":[21],"own":[22],"trajectories.":[23],"This":[24],"framework":[25],"typically":[26],"uses":[27],"reverse":[28,48,80],"KL":[29,49,81,85,132],"divergence,":[30],"encouraging":[31],"the":[32,36,44,59,78,92],"to":[34,158],"match":[35],"teacher's":[37],"high-confidence":[38],"predictions.":[39],"However,":[40],"we":[41,68],"show":[42,116],"that":[43,117,166],"mode-seeking":[45,105],"property":[46],"of":[47,95,146],"reduces":[50],"generation":[51,121],"diversity":[52,122,175],"and":[53,126,153,176],"yields":[54,142],"unstable":[55],"learning":[56],"when":[58,86],"teacher":[60,87,169],"distribution":[61],"has":[62],"high":[63],"entropy.":[64],"To":[65],"address":[66],"this,":[67],"introduce":[69],"Entropy-Aware":[70],"On-Policy":[71],"Distillation.":[72],"Our":[73],"key":[74],"idea":[75],"augmenting":[77],"standard":[79],"objective":[82],"with":[83,107],"forward":[84,131],"entropy":[88],"high,":[90],"capturing":[91],"full":[93],"range":[94],"plausible":[96],"outputs":[97],"while":[98],"retaining":[99],"precise":[100],"imitation":[101],"elsewhere.":[102],"It":[103],"balances":[104],"precision":[106],"mode-covering":[108],"robustness":[109],"without":[110],"sacrificing":[111],"on-policy":[112,160],"training":[113],"efficiency.":[114],"Experiments":[115],"our":[118],"method":[119],"maintains":[120],"(sustained":[123],"entropy)":[125],"improves":[127],"student-teacher":[128],"alignment":[129],"(lower":[130],"on":[133],"high-entropy":[134],"tokens).":[135],"Across":[136],"six":[137],"math":[138],"reasoning":[139],"benchmarks,":[140],"this":[141],"Pass@8":[143],"accuracy":[144],"gains":[145],"+1.37":[147],"Qwen3-0.6B-Base,":[149],"+2.39":[150],"Qwen3-1.7B-Base,":[152],"+5.05":[154],"Qwen3-4B-Base":[156],"compared":[157],"baseline":[159],"methods.":[162],"These":[163],"results":[164],"demonstrate":[165],"accounting":[167],"uncertainty":[170],"essential":[172],"maintaining":[174],"achieving":[177],"effective":[178],"transfer.":[180]},"counts_by_year":[],"updated_date":"2026-05-05T08:41:31.759640","created_date":"2026-03-11T00:00:00"}