{"id":"https://openalex.org/W7161997570","doi":"https://doi.org/10.48550/arxiv.2605.20730","title":"Distributional Alignment as a Criterion for Designing Task Vectors in In-Context Learning","display_name":"Distributional Alignment as a Criterion for Designing Task Vectors in In-Context Learning","publication_year":2026,"publication_date":"2026-05-20","ids":{"openalex":"https://openalex.org/W7161997570","doi":"https://doi.org/10.48550/arxiv.2605.20730"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.20730","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20730","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.20730","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5136697810","display_name":"Jihoon Kwon","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kwon, Jihoon","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136719472","display_name":"Jiwon Choi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Choi, Jiwon","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136702827","display_name":"Jy-yong Sohn","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sohn, Jy-yong","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.4081000089645386,"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.4081000089645386,"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/T10028","display_name":"Topic Modeling","score":0.09229999780654907,"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/T13702","display_name":"Machine Learning in Healthcare","score":0.08649999648332596,"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/task","display_name":"Task (project management)","score":0.673799991607666},{"id":"https://openalex.org/keywords/metric","display_name":"Metric (unit)","score":0.6132000088691711},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.5911999940872192},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.5773000121116638},{"id":"https://openalex.org/keywords/support-vector-machine","display_name":"Support vector machine","score":0.43299999833106995},{"id":"https://openalex.org/keywords/task-analysis","display_name":"Task analysis","score":0.34880000352859497},{"id":"https://openalex.org/keywords/linear-model","display_name":"Linear model","score":0.3294999897480011},{"id":"https://openalex.org/keywords/quality","display_name":"Quality (philosophy)","score":0.3231000006198883}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.720300018787384},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.673799991607666},{"id":"https://openalex.org/C176217482","wikidata":"https://www.wikidata.org/wiki/Q860554","display_name":"Metric (unit)","level":2,"score":0.6132000088691711},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.5921000242233276},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.5911999940872192},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.5773000121116638},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.557699978351593},{"id":"https://openalex.org/C12267149","wikidata":"https://www.wikidata.org/wiki/Q282453","display_name":"Support vector machine","level":2,"score":0.43299999833106995},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.34880000352859497},{"id":"https://openalex.org/C163175372","wikidata":"https://www.wikidata.org/wiki/Q3339222","display_name":"Linear model","level":2,"score":0.3294999897480011},{"id":"https://openalex.org/C2779530757","wikidata":"https://www.wikidata.org/wiki/Q1207505","display_name":"Quality (philosophy)","level":2,"score":0.3231000006198883},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.31540000438690186},{"id":"https://openalex.org/C28006648","wikidata":"https://www.wikidata.org/wiki/Q6934509","display_name":"Multi-task learning","level":3,"score":0.29580000042915344},{"id":"https://openalex.org/C179799912","wikidata":"https://www.wikidata.org/wiki/Q205084","display_name":"Computational complexity theory","level":2,"score":0.2935999929904938},{"id":"https://openalex.org/C2780898871","wikidata":"https://www.wikidata.org/wiki/Q860554","display_name":"Performance metric","level":2,"score":0.2849999964237213},{"id":"https://openalex.org/C311688","wikidata":"https://www.wikidata.org/wiki/Q2393193","display_name":"Time complexity","level":2,"score":0.28209999203681946},{"id":"https://openalex.org/C83546350","wikidata":"https://www.wikidata.org/wiki/Q1139051","display_name":"Regression","level":2,"score":0.2752000093460083},{"id":"https://openalex.org/C61272859","wikidata":"https://www.wikidata.org/wiki/Q7834031","display_name":"Transferability","level":3,"score":0.2736999988555908},{"id":"https://openalex.org/C114289077","wikidata":"https://www.wikidata.org/wiki/Q3284399","display_name":"Statistical model","level":2,"score":0.26249998807907104},{"id":"https://openalex.org/C136389625","wikidata":"https://www.wikidata.org/wiki/Q334384","display_name":"Supervised learning","level":3,"score":0.2549999952316284},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.2549000084400177}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.20730","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20730","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.20730","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20730","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"In-context":[0],"learning":[1],"(ICL)":[2],"allows":[3],"large":[4],"language":[5],"models":[6],"(LLMs)":[7],"to":[8,10,58,137],"adapt":[9],"new":[11,227],"tasks":[12],"through":[13,46,149],"demonstrations,":[14],"yet":[15],"it":[16],"suffers":[17],"from":[18,213],"escalating":[19],"inference":[20,72,172],"costs":[21],"as":[22,113],"context":[23],"length":[24],"increases.":[25],"While":[26],"task":[27,48,62,74,101,162,203,211,231],"vectors":[28,75,212],"offer":[29],"a":[30,91,114,118,134,141,214,219,226],"promising":[31],"alternative":[32],"by":[33,126,168,223],"compressing":[34],"demonstrations":[35],"into":[36,56],"compact":[37],"hidden-state":[38],"representations,":[39],"their":[40,78],"quality":[41],"has":[42,199],"been":[43],"evaluated":[44],"only":[45],"downstream":[47,123],"accuracy.":[49,124],"This":[50],"indirect":[51],"criterion":[52],"provides":[53],"limited":[54],"insight":[55],"how":[57],"design":[59],"more":[60],"effective":[61],"vector":[63,163,204],"extraction":[64],"methods.":[65],"In":[66],"this":[67],"paper,":[68],"we":[69,88,128,186,207],"posit":[70],"that":[71,82,93,110,145,177,198,210],"using":[73],"should":[76],"align":[77],"predictive":[79],"distribution":[80],"with":[81,122],"of":[83,190],"ICL.":[84],"To":[85],"quantify":[86],"this,":[87,127],"introduce":[89],"$d_{\\text{NTP}}$,":[90],"metric":[92],"measures":[94],"the":[95,180,188],"discrepancy":[96],"in":[97,202],"next-token":[98],"probabilities":[99],"between":[100],"vector-based":[102],"and":[103,155],"ICL-based":[104],"inference.":[105],"Our":[106],"empirical":[107],"analysis":[108],"reveals":[109],"$d_{\\text{NTP}}$":[111,139],"serves":[112],"performance":[115,222],"proxy,":[116],"exhibiting":[117],"strong":[119],"negative":[120],"correlation":[121],"Motivated":[125],"develop":[129],"Linear":[130],"Task":[131],"Vector":[132],"(LTV),":[133],"method":[135],"designed":[136],"minimize":[138],"via":[140],"closed-form":[142],"linear":[143],"mapping":[144],"estimates":[146],"demonstration":[147],"effects":[148],"regression.":[150],"Across":[151],"eight":[152],"classification":[153],"benchmarks":[154],"five":[156],"LLMs,":[157],"LTV":[158,178,191],"consistently":[159],"outperforms":[160,179],"existing":[161],"baselines,":[164],"improving":[165],"average":[166],"accuracy":[167],"9.2\\%":[169],"while":[170],"reducing":[171],"latency.":[173],"We":[174],"further":[175],"show":[176,209],"baselines":[181],"on":[182],"regression":[183],"tasks.":[184],"Moreover,":[185],"investigate":[187],"transferability":[189],"across":[192],"different":[193],"model":[194,216],"scales;":[195],"an":[196],"aspect":[197],"remained":[200],"nascent":[201],"research.":[205],"Specifically,":[206],"empirically":[208],"larger":[215],"can":[217],"enhance":[218],"smaller":[220],"model's":[221],"6.4\\%,":[224],"suggesting":[225],"utility":[228],"for":[229],"extracted":[230],"representations.":[232]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-22T00:00:00"}