{"id":"https://openalex.org/W7133342383","doi":"https://doi.org/10.48550/arxiv.2603.01335","title":"Provable and Practical In-Context Policy Optimization for Self-Improvement","display_name":"Provable and Practical In-Context Policy Optimization for Self-Improvement","publication_year":2026,"publication_date":"2026-03-02","ids":{"openalex":"https://openalex.org/W7133342383","doi":"https://doi.org/10.48550/arxiv.2603.01335"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.01335","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.01335","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":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2603.01335","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5127923048","display_name":"Tianrun Yu","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Yu, Tianrun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101908109","display_name":"Yuxiao Yang","orcid":"https://orcid.org/0000-0003-1387-8464"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Yuxiao","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127960650","display_name":"Zhaoyang Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Zhaoyang","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127901348","display_name":"Kaixiang Zhao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhao, Kaixiang","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5060178403","display_name":"Porter Jenkins","orcid":"https://orcid.org/0009-0001-3213-8333"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jenkins, Porter","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128036320","display_name":"Xuchao Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Xuchao","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5122950530","display_name":"Chetan Bansal","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Bansal, Chetan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127985829","display_name":"Huaxiu Yao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yao, Huaxiu","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5127887665","display_name":"Weitong Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Weitong","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":9,"corresponding_author_ids":["https://openalex.org/A5127923048"],"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.5295000076293945,"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.5295000076293945,"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/T12072","display_name":"Machine Learning and Algorithms","score":0.05909999832510948,"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/T12101","display_name":"Advanced Bandit Algorithms Research","score":0.046799998730421066,"subfield":{"id":"https://openalex.org/subfields/1803","display_name":"Management Science and Operations Research"},"field":{"id":"https://openalex.org/fields/18","display_name":"Decision Sciences"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.7071999907493591},{"id":"https://openalex.org/keywords/robustness","display_name":"Robustness (evolution)","score":0.6963000297546387},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.4778999984264374},{"id":"https://openalex.org/keywords/linear-programming","display_name":"Linear programming","score":0.3305000066757202},{"id":"https://openalex.org/keywords/optimization-problem","display_name":"Optimization problem","score":0.323199987411499},{"id":"https://openalex.org/keywords/linear-model","display_name":"Linear model","score":0.3091999888420105}],"concepts":[{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.7071999907493591},{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.6963000297546387},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6236000061035156},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.49059998989105225},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.4778999984264374},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3668999969959259},{"id":"https://openalex.org/C41045048","wikidata":"https://www.wikidata.org/wiki/Q202843","display_name":"Linear programming","level":2,"score":0.3305000066757202},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.32589998841285706},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.323199987411499},{"id":"https://openalex.org/C163175372","wikidata":"https://www.wikidata.org/wiki/Q3339222","display_name":"Linear model","level":2,"score":0.3091999888420105},{"id":"https://openalex.org/C193254401","wikidata":"https://www.wikidata.org/wiki/Q2160088","display_name":"Robust optimization","level":2,"score":0.29260000586509705},{"id":"https://openalex.org/C99844830","wikidata":"https://www.wikidata.org/wiki/Q102441924","display_name":"Scaling","level":2,"score":0.2847999930381775},{"id":"https://openalex.org/C2779793024","wikidata":"https://www.wikidata.org/wiki/Q17299941","display_name":"Indirect Inference","level":3,"score":0.2727000117301941},{"id":"https://openalex.org/C3018263672","wikidata":"https://www.wikidata.org/wiki/Q1296251","display_name":"Efficient algorithm","level":2,"score":0.2637999951839447},{"id":"https://openalex.org/C2987595161","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Optimization algorithm","level":2,"score":0.2567000091075897},{"id":"https://openalex.org/C55660270","wikidata":"https://www.wikidata.org/wiki/Q5164377","display_name":"Constrained optimization","level":2,"score":0.25459998846054077}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.01335","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.01335","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":"doi:10.48550/arxiv.2603.01335","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.01335","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":"article"},"sustainable_development_goals":[{"id":"https://metadata.un.org/sdg/16","score":0.4022584557533264,"display_name":"Peace, Justice and strong institutions"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"We":[0,15],"study":[1],"test-time":[2,155],"scaling,":[3],"where":[4],"a":[5,53,58,80,143],"model":[6,62],"improves":[7],"its":[8,26,38,86,93],"answer":[9],"through":[10],"multi-round":[11],"self-reflection":[12,147],"at":[13,96],"inference.":[14],"introduce":[16],"In-Context":[17],"Policy":[18],"Optimization":[19],"(ICPO),":[20],"in":[21,28,148],"which":[22],"an":[23],"agent":[24],"optimizes":[25],"response":[27,87,94],"context":[29],"using":[30],"self-assessed":[31,89,115],"or":[32],"externally":[33],"observed":[34],"rewards":[35,105,116],"without":[36],"modifying":[37],"parameters.":[39],"To":[40],"explain":[41],"this":[42,73],"ICPO":[43,78,141],"process,":[44],"we":[45,75],"theoretically":[46],"show":[47],"that":[48,83],"with":[49,106,136],"sufficient":[50],"pretraining":[51],"under":[52],"novel":[54],"Fisher-weighted":[55],"logit-matching":[56],"objective,":[57],"single-layer":[59],"linear":[60,69],"self-attention":[61],"can":[63],"provably":[64],"imitate":[65],"policy-optimization":[66],"algorithm":[67,82],"for":[68,154,157],"bandits.":[70],"Building":[71],"on":[72],"theory,":[74],"propose":[76],"Minimum-Entropy":[77],"(ME-ICPO),":[79],"practical":[81,152],"iteratively":[84],"uses":[85],"and":[88,103,150],"reward":[90],"to":[91],"refine":[92],"in-context":[95],"inference":[97,132],"time.":[98],"By":[99],"selecting":[100],"the":[101,111,114],"responses":[102],"their":[104],"minimum":[107],"entropy,":[108],"ME-ICPO":[109,125],"ensures":[110],"robustness":[112],"of":[113,146],"via":[117],"majority":[118],"voting.":[119],"Across":[120],"standard":[121],"mathematical":[122,158],"reasoning":[123],"tasks,":[124],"attains":[126],"competitive,":[127],"top-tier":[128],"performance":[129],"while":[130],"keeping":[131],"costs":[133],"affordable":[134],"compared":[135],"other":[137],"inference-time":[138],"algorithms.":[139],"Overall,":[140],"provides":[142],"principled":[144],"understanding":[145],"LLMs":[149],"yields":[151],"benefits":[153],"scaling":[156],"reasoning.":[159]},"counts_by_year":[],"updated_date":"2026-03-04T07:09:34.246503","created_date":"2026-03-04T00:00:00"}