{"id":"https://openalex.org/W7162650298","doi":"https://doi.org/10.48550/arxiv.2605.28600","title":"Transformers Provably Learn to Internalize Chain-of-Thought","display_name":"Transformers Provably Learn to Internalize Chain-of-Thought","publication_year":2026,"publication_date":"2026-05-27","ids":{"openalex":"https://openalex.org/W7162650298","doi":"https://doi.org/10.48550/arxiv.2605.28600"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.28600","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.28600","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":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.28600","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137213125","display_name":"Yixiao Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Yixiao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137193565","display_name":"Hanlin Zhu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhu, Hanlin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137245014","display_name":"Zixuan Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Zixuan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5034192173","display_name":"Jiantao Jiao","orcid":"https://orcid.org/0000-0003-3766-8031"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Jiao, Jiantao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137255145","display_name":"Stuart Russell","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Russell, Stuart","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5072786751","display_name":"Somayeh Sojoudi","orcid":"https://orcid.org/0000-0001-7177-7712"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sojoudi, Somayeh","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137196583","display_name":"Song Mei","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mei, Song","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":7,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.16820000112056732,"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"}},"topics":[{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.16820000112056732,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.1137000024318695,"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.06750000268220901,"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/inference","display_name":"Inference","score":0.6952000260353088},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.6273000240325928},{"id":"https://openalex.org/keywords/sample-complexity","display_name":"Sample complexity","score":0.42980000376701355},{"id":"https://openalex.org/keywords/parity","display_name":"Parity (physics)","score":0.3869999945163727},{"id":"https://openalex.org/keywords/exponential-function","display_name":"Exponential function","score":0.3801000118255615},{"id":"https://openalex.org/keywords/rule-of-inference","display_name":"Rule of inference","score":0.3052000105381012}],"concepts":[{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.6952000260353088},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.6273000240325928},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5810999870300293},{"id":"https://openalex.org/C2778445095","wikidata":"https://www.wikidata.org/wiki/Q18354077","display_name":"Sample complexity","level":2,"score":0.42980000376701355},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4287000000476837},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.40450000762939453},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.39480000734329224},{"id":"https://openalex.org/C2777151079","wikidata":"https://www.wikidata.org/wiki/Q141160","display_name":"Parity (physics)","level":2,"score":0.3869999945163727},{"id":"https://openalex.org/C151376022","wikidata":"https://www.wikidata.org/wiki/Q168698","display_name":"Exponential function","level":2,"score":0.3801000118255615},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.33500000834465027},{"id":"https://openalex.org/C3746660","wikidata":"https://www.wikidata.org/wiki/Q1068763","display_name":"Rule of inference","level":2,"score":0.3052000105381012},{"id":"https://openalex.org/C3018263672","wikidata":"https://www.wikidata.org/wiki/Q1296251","display_name":"Efficient algorithm","level":2,"score":0.28949999809265137},{"id":"https://openalex.org/C120936955","wikidata":"https://www.wikidata.org/wiki/Q2155640","display_name":"Empirical research","level":2,"score":0.2892000079154968},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.27320000529289246},{"id":"https://openalex.org/C195344581","wikidata":"https://www.wikidata.org/wiki/Q2555318","display_name":"Automated reasoning","level":2,"score":0.2694000005722046},{"id":"https://openalex.org/C179799912","wikidata":"https://www.wikidata.org/wiki/Q205084","display_name":"Computational complexity theory","level":2,"score":0.2648000121116638}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.28600","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.28600","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":"doi:10.48550/arxiv.2605.28600","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.28600","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":"article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Chain-of-Thought":[0,37],"(CoT)":[1],"prompting":[2],"substantially":[3],"improves":[4],"the":[5,11,23,66,97,136,151],"sample":[6,98],"efficiency":[7,99],"of":[8,13,70,100,138],"transformers,":[9],"reducing":[10,135],"complexity":[12],"tasks":[14],"like":[15],"parity":[16,112],"learning":[17],"from":[18,140],"exponential":[19],"to":[20,49,114,118,144],"polynomial":[21],"in":[22,132,142],"input":[24],"length.":[25],"However,":[26],"generating":[27],"explicit":[28,101],"reasoning":[29,156],"steps":[30,52],"at":[31,126],"inference":[32,106],"is":[33,157],"computationally":[34],"expensive.":[35],"Implicit":[36],"(ICoT)":[38],"has":[39],"emerged":[40],"as":[41],"a":[42,127],"promising":[43],"empirical":[44],"remedy":[45],"that":[46,73],"trains":[47],"models":[48],"internalize":[50],"intermediate":[51],"within":[53],"their":[54],"hidden":[55],"states,":[56],"but":[57],"its":[58,105],"theoretical":[59,68],"foundations":[60],"remain":[61],"poorly":[62],"understood.":[63],"We":[64],"give":[65],"first":[67],"analysis":[69],"ICoT,":[71,120],"proving":[72],"an":[74],"$L$-layer":[75],"transformer":[76],"trained":[77],"under":[78],"our":[79],"proposed":[80],"Log-ICoT":[81,129],"curriculum":[82],"learns":[83],"$k$-parity":[84],"with":[85],"$\\mathsf{poly}(n)$":[86],"samples":[87],"and":[88,108,153],"$L":[89],"=":[90],"\\log_2":[91],"k$":[92],"training":[93],"stages.":[94],"This":[95],"matches":[96],"CoT":[102],"while":[103],"eliminating":[104],"overhead,":[107],"extends":[109],"prior":[110],"one-layer":[111],"guarantees":[113],"multi-layer":[115,148],"architectures.":[116],"Compared":[117],"standard":[119],"which":[121],"removes":[122,130],"thinking":[123],"tokens":[124],"one":[125],"time,":[128],"them":[131],"geometric":[133],"chunks,":[134],"number":[137],"stages":[139],"linear":[141],"$k$":[143],"logarithmic.":[145],"Experiments":[146],"on":[147],"transformers":[149],"confirm":[150],"theory":[152],"visualize":[154],"how":[155],"progressively":[158],"absorbed":[159],"into":[160],"deeper":[161],"layers.":[162]},"counts_by_year":[],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2026-05-29T00:00:00"}
