{"id":"https://openalex.org/W7163160246","doi":"https://doi.org/10.48550/arxiv.2606.00183","title":"Agentic Transformers Provably Learn to Search via Reinforcement Learning","display_name":"Agentic Transformers Provably Learn to Search via Reinforcement Learning","publication_year":2026,"publication_date":"2026-05-29","ids":{"openalex":"https://openalex.org/W7163160246","doi":"https://doi.org/10.48550/arxiv.2606.00183"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.00183","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00183","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":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.00183","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137665054","display_name":"Tong Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Tong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137628760","display_name":"Yu Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Yu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137619224","display_name":"Yingbin Liang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liang, Yingbin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137685480","display_name":"Yuejie Chi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chi, Yuejie","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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.28679999709129333,"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.28679999709129333,"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.10700000077486038,"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/T12090","display_name":"Language and cultural evolution","score":0.065700002014637,"subfield":{"id":"https://openalex.org/subfields/3316","display_name":"Cultural Studies"},"field":{"id":"https://openalex.org/fields/33","display_name":"Social Sciences"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.8281999826431274},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.4950999915599823},{"id":"https://openalex.org/keywords/reinforcement","display_name":"Reinforcement","score":0.3937000036239624},{"id":"https://openalex.org/keywords/action","display_name":"Action (physics)","score":0.3416999876499176},{"id":"https://openalex.org/keywords/markov-decision-process","display_name":"Markov decision process","score":0.33709999918937683},{"id":"https://openalex.org/keywords/temporal-difference-learning","display_name":"Temporal difference learning","score":0.32899999618530273},{"id":"https://openalex.org/keywords/construct","display_name":"Construct (python library)","score":0.32690000534057617},{"id":"https://openalex.org/keywords/search-algorithm","display_name":"Search algorithm","score":0.3264000117778778},{"id":"https://openalex.org/keywords/abstraction","display_name":"Abstraction","score":0.32420000433921814}],"concepts":[{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.8281999826431274},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.675599992275238},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.6148999929428101},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.5299000144004822},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.4950999915599823},{"id":"https://openalex.org/C67203356","wikidata":"https://www.wikidata.org/wiki/Q1321905","display_name":"Reinforcement","level":2,"score":0.3937000036239624},{"id":"https://openalex.org/C2780791683","wikidata":"https://www.wikidata.org/wiki/Q846785","display_name":"Action (physics)","level":2,"score":0.3416999876499176},{"id":"https://openalex.org/C106189395","wikidata":"https://www.wikidata.org/wiki/Q176789","display_name":"Markov decision process","level":3,"score":0.33709999918937683},{"id":"https://openalex.org/C196340769","wikidata":"https://www.wikidata.org/wiki/Q7698910","display_name":"Temporal difference learning","level":3,"score":0.32899999618530273},{"id":"https://openalex.org/C2780801425","wikidata":"https://www.wikidata.org/wiki/Q5164392","display_name":"Construct (python library)","level":2,"score":0.32690000534057617},{"id":"https://openalex.org/C125583679","wikidata":"https://www.wikidata.org/wiki/Q755673","display_name":"Search algorithm","level":2,"score":0.3264000117778778},{"id":"https://openalex.org/C124304363","wikidata":"https://www.wikidata.org/wiki/Q673661","display_name":"Abstraction","level":2,"score":0.32420000433921814},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.3224000036716461},{"id":"https://openalex.org/C46149586","wikidata":"https://www.wikidata.org/wiki/Q11785332","display_name":"Monte Carlo tree search","level":3,"score":0.31940001249313354},{"id":"https://openalex.org/C6177178","wikidata":"https://www.wikidata.org/wiki/Q10998070","display_name":"Discounting","level":2,"score":0.3167000114917755},{"id":"https://openalex.org/C207024777","wikidata":"https://www.wikidata.org/wiki/Q621673","display_name":"Search tree","level":3,"score":0.2924000024795532},{"id":"https://openalex.org/C166109690","wikidata":"https://www.wikidata.org/wiki/Q4677422","display_name":"Action selection","level":3,"score":0.2904999852180481},{"id":"https://openalex.org/C37404715","wikidata":"https://www.wikidata.org/wiki/Q380679","display_name":"Dynamic programming","level":2,"score":0.2741999924182892},{"id":"https://openalex.org/C84525736","wikidata":"https://www.wikidata.org/wiki/Q831366","display_name":"Decision tree","level":2,"score":0.26820001006126404},{"id":"https://openalex.org/C136197465","wikidata":"https://www.wikidata.org/wiki/Q1729295","display_name":"Variety (cybernetics)","level":2,"score":0.2646999955177307},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.25690001249313354},{"id":"https://openalex.org/C13662910","wikidata":"https://www.wikidata.org/wiki/Q193139","display_name":"Trajectory","level":2,"score":0.2558000087738037},{"id":"https://openalex.org/C113174947","wikidata":"https://www.wikidata.org/wiki/Q2859736","display_name":"Tree (set theory)","level":2,"score":0.2547000050544739},{"id":"https://openalex.org/C169258074","wikidata":"https://www.wikidata.org/wiki/Q245748","display_name":"Random forest","level":2,"score":0.2508000135421753}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.00183","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00183","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":"doi:10.48550/arxiv.2606.00183","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.00183","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":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"id":"https://metadata.un.org/sdg/16","display_name":"Peace, Justice and strong institutions","score":0.770898699760437}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Tree":[0],"search":[1,36,89],"is":[2],"a":[3,27,51,69,74,82,115,167,179],"central":[4],"abstraction":[5],"behind":[6],"many":[7],"language-agent":[8],"reasoning":[9],"and":[10,19,67,102,145,191,199],"decision-making":[11],"tasks:":[12],"agents":[13],"must":[14],"explore":[15],"actions,":[16,95],"remember":[17],"failures,":[18],"backtrack":[20],"toward":[21],"promising":[22],"alternatives.":[23],"Yet,":[24],"we":[25],"lack":[26],"theoretical":[28],"understanding":[29],"of":[30,42,111],"how":[31],"transformer-based":[32,184],"policies":[33],"acquire":[34],"such":[35],"capabilities":[37],"from":[38,127,197],"the":[39,97,108,163],"training":[40,109,141],"dynamics":[41,110],"reinforcement":[43,129],"learning":[44],"(RL).":[45],"We":[46,79,105,154],"study":[47],"this":[48,120],"question":[49],"in":[50,125,186],"stochastic":[52],"$k$-ary":[53],"tree":[54],"environment,":[55],"where":[56],"an":[57],"agentic":[58,203],"transformer":[59,84],"observes":[60],"only":[61,142],"its":[62],"trajectory":[63],"history":[64],"through":[65],"interaction":[66],"receives":[68],"terminal":[70],"reward":[71],"for":[72,183],"reaching":[73],"hidden":[75],"leaf":[76],"goal":[77,160],"node.":[78],"first":[80],"construct":[81],"two-head":[83],"that":[85,119,171],"implements":[86],"randomized":[87],"depth-first":[88],"(DFS):":[90],"one":[91],"head":[92],"tracks":[93],"previous":[94],"while":[96],"other":[98],"detects":[99],"failure":[100],"outcomes":[101],"triggers":[103],"backtracking.":[104],"then":[106],"analyze":[107],"policy":[112,136,170],"gradient":[113],"under":[114,158],"depth-wise":[116],"curriculum,":[117],"showing":[118],"same":[121],"DFS":[122,169],"mechanism":[123],"emerges":[124],"stages":[126],"sparse":[128],"feedback":[130],"without":[131],"expert":[132],"demonstrations.":[133],"The":[134],"resulting":[135],"exhibits":[137],"depth":[138],"generalization:":[139],"after":[140],"on":[143,150],"depth-$1$":[144],"depth-$2$":[146],"trees,":[147],"it":[148],"succeeds":[149],"deeper":[151],"full":[152],"trees.":[153],"further":[155],"show":[156],"that,":[157],"imbalanced":[159],"distributions,":[161],"discounting":[162],"return":[164],"leads":[165],"to":[166,193],"ranked":[168],"prioritizes":[172],"higher-probability":[173],"branches.":[174],"Overall,":[175],"our":[176],"results":[177],"identify":[178],"mechanistic":[180],"normal":[181],"form":[182],"search,":[185],"which":[187],"attention":[188],"heads":[189],"specialize":[190],"cooperate":[192],"extract":[194],"decision-relevant":[195],"traces":[196],"context":[198],"convert":[200],"them":[201],"into":[202],"action":[204],"selection":[205],"via":[206],"RL":[207],"training.":[208]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-03T00:00:00"}