{"id":"https://openalex.org/W7160877530","doi":"https://doi.org/10.48550/arxiv.2605.07333","title":"Beyond Linear Attention: Softmax Transformers Implement In-Context Reinforcement Learning","display_name":"Beyond Linear Attention: Softmax Transformers Implement In-Context Reinforcement Learning","publication_year":2026,"publication_date":"2026-05-08","ids":{"openalex":"https://openalex.org/W7160877530","doi":"https://doi.org/10.48550/arxiv.2605.07333"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.07333","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07333","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.2605.07333","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5135857734","display_name":"Zixuan Xie","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xie, Zixuan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135848873","display_name":"Xinyu Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Xinyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135847498","display_name":"Claire Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Claire","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5061588035","display_name":"Shuze Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Shuze Daniel","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5016704715","display_name":"Rohan Chandra","orcid":"https://orcid.org/0000-0003-4843-6375"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chandra, Rohan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5135832492","display_name":"Shangtong Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Shangtong","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.7317000031471252,"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.7317000031471252,"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.05920000001788139,"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/T12026","display_name":"Explainable Artificial Intelligence (XAI)","score":0.0357000008225441,"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/softmax-function","display_name":"Softmax function","score":0.9950000047683716},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.8095999956130981},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.5913000106811523},{"id":"https://openalex.org/keywords/reinforcement","display_name":"Reinforcement","score":0.32760000228881836},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.31049999594688416},{"id":"https://openalex.org/keywords/contraction","display_name":"Contraction (grammar)","score":0.3075000047683716}],"concepts":[{"id":"https://openalex.org/C188441871","wikidata":"https://www.wikidata.org/wiki/Q7554146","display_name":"Softmax function","level":3,"score":0.9950000047683716},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.8095999956130981},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.5913000106811523},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.4887999892234802},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.46219998598098755},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3831999897956848},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.3743000030517578},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3504999876022339},{"id":"https://openalex.org/C67203356","wikidata":"https://www.wikidata.org/wiki/Q1321905","display_name":"Reinforcement","level":2,"score":0.32760000228881836},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.31049999594688416},{"id":"https://openalex.org/C163415756","wikidata":"https://www.wikidata.org/wiki/Q126473","display_name":"Contraction (grammar)","level":2,"score":0.3075000047683716},{"id":"https://openalex.org/C163175372","wikidata":"https://www.wikidata.org/wiki/Q3339222","display_name":"Linear model","level":2,"score":0.30309998989105225},{"id":"https://openalex.org/C14036430","wikidata":"https://www.wikidata.org/wiki/Q3736076","display_name":"Function (biology)","level":2,"score":0.29319998621940613},{"id":"https://openalex.org/C188116033","wikidata":"https://www.wikidata.org/wiki/Q2664563","display_name":"Q-learning","level":3,"score":0.2858999967575073},{"id":"https://openalex.org/C159694833","wikidata":"https://www.wikidata.org/wiki/Q2321565","display_name":"Iterative method","level":2,"score":0.2824999988079071},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.2768000066280365},{"id":"https://openalex.org/C196340769","wikidata":"https://www.wikidata.org/wiki/Q7698910","display_name":"Temporal difference learning","level":3,"score":0.26489999890327454}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.07333","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07333","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.2605.07333","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07333","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":[{"score":0.5425890684127808,"id":"https://metadata.un.org/sdg/16","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":{"In-context":[0],"reinforcement":[1],"learning":[2,100],"(ICRL)":[3],"studies":[4],"agents":[5],"that,":[6,73],"after":[7],"pretraining,":[8],"adapt":[9],"to":[10,90],"new":[11,108],"tasks":[12],"by":[13],"conditioning":[14],"on":[15,28],"additional":[16],"context":[17],"without":[18,53],"parameter":[19],"updates.":[20],"Existing":[21],"theoretical":[22,49],"analyses":[23],"of":[24,51,81,93,146,164],"ICRL":[25,52],"largely":[26],"rely":[27],"linear":[29,57,121],"attention,":[30],"which":[31],"replaces":[32],"the":[33,37,47,55,64,77,138,144,150],"softmax":[34,66,86,96,104],"function":[35],"in":[36,69,115,171],"standard":[38,65],"attention":[39,58,67,87],"with":[40,74,84,149],"an":[41],"identity":[42],"mapping.":[43],"This":[44],"paper":[45],"provides":[46],"first":[48],"understanding":[50],"making":[54],"unrealistic":[56],"simplification.":[59],"In":[60],"particular,":[61],"we":[62,155],"consider":[63],"used":[68],"practice.":[70],"We":[71,129],"show":[72],"certain":[75,135],"parameters,":[76],"layerwise":[78],"forward":[79],"pass":[80],"a":[82,94,107,134,161,165],"Transformer":[83],"such":[85],"is":[88,106],"equivalent":[89],"iterative":[91],"updates":[92],"weighted":[95,103],"temporal":[97],"difference":[98],"(TD)":[99],"algorithm.":[101],"Here,":[102],"TD":[105,122,125],"RL":[109],"algorithm":[110],"that":[111,132,157],"performs":[112],"policy":[113,139],"evaluation":[114,140],"kernel":[116],"space":[117],"and":[118,123],"adopts":[119],"both":[120],"tabular":[124],"as":[126,143],"special":[127],"cases.":[128],"also":[130],"prove":[131,156],"under":[133],"contraction":[136],"condition,":[137],"error":[141],"decays":[142],"number":[145],"layers":[147],"grows,":[148],"identified":[151],"parameters":[152,159],"above.":[153],"Finally,":[154],"those":[158],"are":[160],"global":[162],"minimizer":[163],"pretraining":[166],"loss,":[167],"explaining":[168],"their":[169],"emergence":[170],"our":[172],"numerical":[173],"experiments.":[174]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-12T00:00:00"}