{"id":"https://openalex.org/W7160428577","doi":"https://doi.org/10.48550/arxiv.2605.03780","title":"Task Vector Geometry Underlies Dual Modes of Task Inference in Transformers","display_name":"Task Vector Geometry Underlies Dual Modes of Task Inference in Transformers","publication_year":2026,"publication_date":"2026-05-05","ids":{"openalex":"https://openalex.org/W7160428577","doi":"https://doi.org/10.48550/arxiv.2605.03780"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.03780","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.03780","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.03780","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5135501675","display_name":"Hao Yan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yan, Hao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135525482","display_name":"Haolin Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Haolin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5100968975","display_name":"Yiqiao Zhong","orcid":"https://orcid.org/0000-0003-0642-662X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhong, Yiqiao","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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.4442000091075897,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.4442000091075897,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.07000000029802322,"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/T12026","display_name":"Explainable Artificial Intelligence (XAI)","score":0.04969999939203262,"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.6377000212669373},{"id":"https://openalex.org/keywords/interpretability","display_name":"Interpretability","score":0.6287000179290771},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.5501999855041504},{"id":"https://openalex.org/keywords/task","display_name":"Task (project management)","score":0.45089998841285706},{"id":"https://openalex.org/keywords/equivalence","display_name":"Equivalence (formal languages)","score":0.4341000020503998},{"id":"https://openalex.org/keywords/bayesian-inference","display_name":"Bayesian inference","score":0.41839998960494995},{"id":"https://openalex.org/keywords/generalization","display_name":"Generalization","score":0.40630000829696655},{"id":"https://openalex.org/keywords/subspace-topology","display_name":"Subspace topology","score":0.4056999981403351},{"id":"https://openalex.org/keywords/task-analysis","display_name":"Task analysis","score":0.3978999853134155}],"concepts":[{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.6377000212669373},{"id":"https://openalex.org/C2781067378","wikidata":"https://www.wikidata.org/wiki/Q17027399","display_name":"Interpretability","level":2,"score":0.6287000179290771},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5848000049591064},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.5501999855041504},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5501999855041504},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.45089998841285706},{"id":"https://openalex.org/C2780069185","wikidata":"https://www.wikidata.org/wiki/Q7977945","display_name":"Equivalence (formal languages)","level":2,"score":0.4341000020503998},{"id":"https://openalex.org/C160234255","wikidata":"https://www.wikidata.org/wiki/Q812535","display_name":"Bayesian inference","level":3,"score":0.41839998960494995},{"id":"https://openalex.org/C177148314","wikidata":"https://www.wikidata.org/wiki/Q170084","display_name":"Generalization","level":2,"score":0.40630000829696655},{"id":"https://openalex.org/C32834561","wikidata":"https://www.wikidata.org/wiki/Q660730","display_name":"Subspace topology","level":2,"score":0.4056999981403351},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.3978999853134155},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3905999958515167},{"id":"https://openalex.org/C28006648","wikidata":"https://www.wikidata.org/wiki/Q6934509","display_name":"Multi-task learning","level":3,"score":0.3862000107765198},{"id":"https://openalex.org/C107673813","wikidata":"https://www.wikidata.org/wiki/Q812534","display_name":"Bayesian probability","level":2,"score":0.37310001254081726},{"id":"https://openalex.org/C162838799","wikidata":"https://www.wikidata.org/wiki/Q596077","display_name":"Counterexample","level":2,"score":0.33399999141693115},{"id":"https://openalex.org/C112680207","wikidata":"https://www.wikidata.org/wiki/Q714886","display_name":"Regular polygon","level":2,"score":0.3264000117778778},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.32420000433921814},{"id":"https://openalex.org/C2778833722","wikidata":"https://www.wikidata.org/wiki/Q28457372","display_name":"Tact","level":2,"score":0.31189998984336853},{"id":"https://openalex.org/C49937458","wikidata":"https://www.wikidata.org/wiki/Q2599292","display_name":"Probabilistic logic","level":2,"score":0.31060001254081726},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.2919999957084656},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.27880001068115234},{"id":"https://openalex.org/C160697094","wikidata":"https://www.wikidata.org/wiki/Q1233197","display_name":"Oblique case","level":2,"score":0.27630001306533813},{"id":"https://openalex.org/C40506919","wikidata":"https://www.wikidata.org/wiki/Q7452469","display_name":"Sequence learning","level":2,"score":0.27160000801086426},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.2712000012397766},{"id":"https://openalex.org/C207201462","wikidata":"https://www.wikidata.org/wiki/Q182505","display_name":"Bayes' theorem","level":3,"score":0.2612000107765198},{"id":"https://openalex.org/C2778023678","wikidata":"https://www.wikidata.org/wiki/Q554403","display_name":"Duality (order theory)","level":2,"score":0.25290000438690186}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.03780","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.03780","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.03780","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.03780","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":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Transformers":[0],"are":[1,168],"effective":[2],"at":[3],"inferring":[4],"the":[5,68,152],"latent":[6],"task":[7,16,37,124,133,142],"from":[8,31,94],"context":[9],"via":[10],"two":[11,109],"inference":[12,110],"modes:":[13],"recognizing":[14],"a":[15,23,44,86,102,115,147],"seen":[17],"during":[18],"training,":[19],"and":[20,71,165],"adapting":[21],"to":[22,53,60,151],"novel":[24],"one.":[25],"Recent":[26],"interpretability":[27],"studies":[28],"have":[29],"identified":[30],"middle-layer":[32],"representations":[33,52,145],"task-specific":[34],"directions,":[35],"or":[36],"vectors,":[38],"that":[39,108,160],"steer":[40],"model":[41,55],"behavior.":[42],"However,":[43],"lack":[45],"of":[46,131],"rigorous":[47],"foundations":[48],"hinders":[49],"connecting":[50],"internal":[51],"external":[54],"behavior:":[56],"existing":[57],"work":[58],"fails":[59],"explain":[61],"how":[62],"task-vector":[63,153,161],"geometry":[64,73],"is":[65,120],"shaped":[66],"by":[67,90,122,137],"training":[69,91,163],"distribution,":[70],"what":[72],"enables":[74],"out-of-distribution":[75],"(OOD)":[76],"generalization.":[77],"In":[78],"this":[79],"paper,":[80],"we":[81],"study":[82],"these":[83],"questions":[84],"in":[85],"controlled":[87],"synthetic":[88],"setting":[89],"small":[92],"transformers":[93],"scratch":[95],"on":[96],"latent-task":[97],"sequence":[98],"distributions,":[99,164],"which":[100],"allows":[101],"principled":[103],"mathematical":[104],"characterization.":[105],"We":[106],"show":[107],"modes":[111],"can":[112],"coexist":[113],"within":[114],"single":[116],"model.":[117],"In-distribution":[118],"behavior":[119],"governed":[121],"Bayesian":[123],"retrieval,":[125],"implemented":[126],"internally":[127],"through":[128,140],"convex":[129],"combinations":[130],"learned":[132],"vectors.":[134],"OOD":[135],"behavior,":[136],"contrast,":[138],"arises":[139],"extrapolative":[141],"learning,":[143],"whose":[144],"occupy":[146],"subspace":[148],"nearly":[149],"orthogonal":[150],"subspace.":[154],"Taken":[155],"together,":[156],"our":[157],"results":[158],"suggest":[159],"geometry,":[162],"generalization":[166],"behaviors":[167],"closely":[169],"related.":[170]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-07T00:00:00"}