{"id":"https://openalex.org/W7128032302","doi":"https://doi.org/10.48550/arxiv.2602.04380","title":"Beyond KL Divergence: Policy Optimization with Flexible Bregman Divergences for LLM Reasoning","display_name":"Beyond KL Divergence: Policy Optimization with Flexible Bregman Divergences for LLM Reasoning","publication_year":2026,"publication_date":"2026-02-04","ids":{"openalex":"https://openalex.org/W7128032302","doi":"https://doi.org/10.48550/arxiv.2602.04380"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.04380","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/A5115749176","display_name":"Rui Yuan","orcid":"https://orcid.org/0000-0002-4956-1672"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Yuan, Rui","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101337903","display_name":"Mykola Khandoga","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Khandoga, Mykola","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5114731590","display_name":"Vinay Kumar Sankarapu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sankarapu, Vinay Kumar","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":3,"corresponding_author_ids":["https://openalex.org/A5115749176"],"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/T12535","display_name":"Machine Learning and Data Classification","score":0.18410000205039978,"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/T12535","display_name":"Machine Learning and Data Classification","score":0.18410000205039978,"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.18359999358654022,"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/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.07620000094175339,"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/initialization","display_name":"Initialization","score":0.7232000231742859},{"id":"https://openalex.org/keywords/bregman-divergence","display_name":"Bregman divergence","score":0.6958000063896179},{"id":"https://openalex.org/keywords/divergence","display_name":"Divergence (linguistics)","score":0.6665999889373779},{"id":"https://openalex.org/keywords/dimension","display_name":"Dimension (graph theory)","score":0.4880000054836273},{"id":"https://openalex.org/keywords/optimization-problem","display_name":"Optimization problem","score":0.46299999952316284},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.44290000200271606},{"id":"https://openalex.org/keywords/variance","display_name":"Variance (accounting)","score":0.4059999883174896},{"id":"https://openalex.org/keywords/function","display_name":"Function (biology)","score":0.3885999917984009}],"concepts":[{"id":"https://openalex.org/C114466953","wikidata":"https://www.wikidata.org/wiki/Q6034165","display_name":"Initialization","level":2,"score":0.7232000231742859},{"id":"https://openalex.org/C149073432","wikidata":"https://www.wikidata.org/wiki/Q4960382","display_name":"Bregman divergence","level":2,"score":0.6958000063896179},{"id":"https://openalex.org/C207390915","wikidata":"https://www.wikidata.org/wiki/Q1230525","display_name":"Divergence (linguistics)","level":2,"score":0.6665999889373779},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.541700005531311},{"id":"https://openalex.org/C33676613","wikidata":"https://www.wikidata.org/wiki/Q13415176","display_name":"Dimension (graph theory)","level":2,"score":0.4880000054836273},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.4690999984741211},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.46299999952316284},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.44290000200271606},{"id":"https://openalex.org/C196083921","wikidata":"https://www.wikidata.org/wiki/Q7915758","display_name":"Variance (accounting)","level":2,"score":0.4059999883174896},{"id":"https://openalex.org/C14036430","wikidata":"https://www.wikidata.org/wiki/Q3736076","display_name":"Function (biology)","level":2,"score":0.3885999917984009},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.37369999289512634},{"id":"https://openalex.org/C55660270","wikidata":"https://www.wikidata.org/wiki/Q5164377","display_name":"Constrained optimization","level":2,"score":0.3716000020503998},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.3659000098705292},{"id":"https://openalex.org/C171752962","wikidata":"https://www.wikidata.org/wiki/Q255166","display_name":"Kullback\u2013Leibler divergence","level":2,"score":0.305400013923645},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.29739999771118164},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.29350000619888306},{"id":"https://openalex.org/C2987595161","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Optimization algorithm","level":2,"score":0.2858999967575073},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.28279998898506165},{"id":"https://openalex.org/C2776401178","wikidata":"https://www.wikidata.org/wiki/Q12050496","display_name":"Feature (linguistics)","level":2,"score":0.2824000120162964},{"id":"https://openalex.org/C2778334786","wikidata":"https://www.wikidata.org/wiki/Q1586270","display_name":"Variation (astronomy)","level":2,"score":0.27639999985694885},{"id":"https://openalex.org/C68781425","wikidata":"https://www.wikidata.org/wiki/Q2052203","display_name":"Multi-objective optimization","level":2,"score":0.2702000141143799},{"id":"https://openalex.org/C193254401","wikidata":"https://www.wikidata.org/wiki/Q2160088","display_name":"Robust optimization","level":2,"score":0.2700999975204468},{"id":"https://openalex.org/C14646407","wikidata":"https://www.wikidata.org/wiki/Q1430750","display_name":"Bellman equation","level":2,"score":0.266400009393692},{"id":"https://openalex.org/C2776291640","wikidata":"https://www.wikidata.org/wiki/Q2912517","display_name":"Value (mathematics)","level":2,"score":0.2619999945163727},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.25360000133514404}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.04380","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.04380","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.04380","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":"article"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2602.04380","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":{"Policy":[0,6,55],"optimization":[1,64,172],"methods":[2,36],"like":[3],"Group":[4],"Relative":[5],"Optimization":[7,56],"(GRPO)":[8],"and":[9,19,30,76,135],"its":[10,125],"variants":[11],"have":[12],"achieved":[13],"strong":[14],"results":[15,158],"on":[16,141],"mathematical":[17,83],"reasoning":[18],"code":[20,100],"generation":[21],"tasks.":[22],"Despite":[23],"extensive":[24],"exploration":[25],"of":[26,47,114,147],"reward":[27],"processing":[28],"strategies":[29,119],"training":[31],"dynamics,":[32],"all":[33],"existing":[34],"group-based":[35,62,170],"exclusively":[37],"use":[38],"KL":[39],"divergence":[40,48,160],"for":[41,153,173],"policy":[42,63,171],"regularization,":[43],"leaving":[44],"the":[45,94,115],"choice":[46,161],"function":[49],"unexplored.":[50],"We":[51],"introduce":[52],"Group-Based":[53],"Mirror":[54],"(GBMPO),":[57],"a":[58,163],"framework":[59],"that":[60,144],"extends":[61],"to":[65],"flexible":[66],"Bregman":[67],"divergences,":[68],"including":[69],"hand-designed":[70,85],"alternatives":[71],"(L2":[72],"in":[73,129,169],"probability":[74],"space)":[75],"learned":[77],"neural":[78,102,148],"mirror":[79,103,149],"maps.":[80],"On":[81,98],"GSM8K":[82],"reasoning,":[84],"ProbL2-GRPO":[86],"achieves":[87],"86.7%":[88],"accuracy,":[89],"improving":[90],"+5.5":[91],"points":[92],"over":[93],"Dr.":[95],"GRPO":[96],"baseline.":[97],"MBPP":[99],"generation,":[101],"maps":[104,150],"reach":[105],"60.1-60.8%":[106],"pass@1,":[107],"with":[108],"random":[109,145],"initialization":[110,146],"already":[111],"capturing":[112],"most":[113,154],"benefit.":[116],"While":[117],"evolutionary":[118],"meta-learning":[120],"provides":[121],"marginal":[122],"accuracy":[123],"improvements,":[124],"primary":[126],"value":[127],"lies":[128],"variance":[130],"reduction":[131],"($\\pm$0.2":[132],"versus":[133],"$\\pm$0.6)":[134],"efficiency":[136],"gains":[137],"(15%":[138],"shorter":[139],"responses":[140],"MBPP),":[142],"suggesting":[143],"is":[151],"sufficient":[152],"practical":[155],"applications.":[156],"These":[157],"establish":[159],"as":[162],"critical,":[164],"previously":[165],"unexplored":[166],"design":[167],"dimension":[168],"LLM":[174],"reasoning.":[175]},"counts_by_year":[],"updated_date":"2026-04-04T16:13:02.066488","created_date":"2026-02-07T00:00:00"}