{"id":"https://openalex.org/W7138975341","doi":"https://doi.org/10.48550/arxiv.2603.17484","title":"Learning When to Attend: Conditional Memory Access for Long-Context LLMs","display_name":"Learning When to Attend: Conditional Memory Access for Long-Context LLMs","publication_year":2026,"publication_date":"2026-03-18","ids":{"openalex":"https://openalex.org/W7138975341","doi":"https://doi.org/10.48550/arxiv.2603.17484"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.17484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.17484","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":null,"license_id":null,"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.2603.17484","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5129963208","display_name":"Sakshi Choudhary","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Choudhary, Sakshi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5130080473","display_name":"Aditya Chattopadhyay","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chattopadhyay, Aditya","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5036782268","display_name":"Luca Zancato","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zancato, Luca","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5129872481","display_name":"Elvis Nunez","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nunez, Elvis","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5030073760","display_name":"Matthew Trager","orcid":"https://orcid.org/0000-0001-9304-6427"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Trager, Matthew","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5129829549","display_name":"Wei Xia","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xia, Wei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5038328783","display_name":"Stefano Soatto","orcid":"https://orcid.org/0000-0003-2902-6362"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Soatto, Stefano","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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.20340000092983246,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.20340000092983246,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10036","display_name":"Advanced Neural Network Applications","score":0.1421000063419342,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.09480000287294388,"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/context","display_name":"Context (archaeology)","score":0.6227999925613403},{"id":"https://openalex.org/keywords/pruning","display_name":"Pruning","score":0.49059998989105225},{"id":"https://openalex.org/keywords/limiting","display_name":"Limiting","score":0.4449000060558319},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.3790000081062317},{"id":"https://openalex.org/keywords/sequence","display_name":"Sequence (biology)","score":0.36309999227523804},{"id":"https://openalex.org/keywords/cache","display_name":"Cache","score":0.3116999864578247},{"id":"https://openalex.org/keywords/sequence-learning","display_name":"Sequence learning","score":0.30239999294281006}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6776000261306763},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.6227999925613403},{"id":"https://openalex.org/C108010975","wikidata":"https://www.wikidata.org/wiki/Q500094","display_name":"Pruning","level":2,"score":0.49059998989105225},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.47600001096725464},{"id":"https://openalex.org/C188198153","wikidata":"https://www.wikidata.org/wiki/Q1613840","display_name":"Limiting","level":2,"score":0.4449000060558319},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.40610000491142273},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.3790000081062317},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3765999972820282},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.36309999227523804},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.3116999864578247},{"id":"https://openalex.org/C40506919","wikidata":"https://www.wikidata.org/wiki/Q7452469","display_name":"Sequence learning","level":2,"score":0.30239999294281006},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.2994999885559082},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.2944999933242798},{"id":"https://openalex.org/C100660578","wikidata":"https://www.wikidata.org/wiki/Q18733","display_name":"Recall","level":2,"score":0.2854999899864197},{"id":"https://openalex.org/C2164484","wikidata":"https://www.wikidata.org/wiki/Q5170150","display_name":"Core (optical fiber)","level":2,"score":0.2734000086784363},{"id":"https://openalex.org/C22367795","wikidata":"https://www.wikidata.org/wiki/Q7625208","display_name":"Structured prediction","level":2,"score":0.27239999175071716},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.2639999985694885},{"id":"https://openalex.org/C183322885","wikidata":"https://www.wikidata.org/wiki/Q17007702","display_name":"Context model","level":3,"score":0.2578999996185303},{"id":"https://openalex.org/C173801870","wikidata":"https://www.wikidata.org/wiki/Q201413","display_name":"Heuristic","level":2,"score":0.2529999911785126}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.17484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.17484","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":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2603.17484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.17484","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":null,"license_id":null,"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":{"Language":[0],"models":[1],"struggle":[2],"to":[3,25,72,93,104,124,135,162],"generalize":[4],"beyond":[5],"pretraining":[6,15],"context":[7,89],"lengths,":[8],"limiting":[9],"long-horizon":[10],"reasoning":[11],"and":[12,45,82,141],"retrieval.":[13],"Continued":[14],"on":[16,48,52,79,131],"long-context":[17,102],"data":[18],"can":[19,46],"help":[20],"but":[21],"is":[22,170],"expensive":[23],"due":[24],"the":[26,42,98],"quadratic":[27],"scaling":[28],"of":[29,100,113,150],"Attention.":[30],"We":[31,76,118],"observe":[32],"that":[33,62],"most":[34],"tokens":[35],"do":[36],"not":[37],"require":[38],"(Global)":[39],"Attention":[40,110,130,154],"over":[41,143],"entire":[43],"sequence":[44],"rely":[47],"local":[49],"context.":[50],"Based":[51],"this,":[53],"we":[54],"propose":[55],"L2A":[56,78,96,146],"(Learning":[57],"To":[58],"Attend),":[59],"a":[60],"layer":[61],"enables":[63,147],"conditional":[64,129],"(token-wise)":[65],"long-range":[66],"memory":[67,159],"access":[68],"by":[69,160],"deciding":[70],"when":[71],"invoke":[73],"global":[74],"attention.":[75],"evaluate":[77],"Qwen":[80,83],"2.5":[81],"3":[84],"models,":[85],"extending":[86],"their":[87],"effective":[88],"length":[90],"from":[91],"32K":[92],"128K":[94],"tokens.":[95],"matches":[97],"performance":[99,166],"standard":[101],"training":[103,139],"within":[105],"3\\%":[106],"while":[107],"skipping":[108],"Global":[109,153],"for":[111],"$\\sim$80\\%":[112],"tokens,":[114],"outperforming":[115],"prior":[116],"baselines.":[117],"also":[119],"design":[120],"custom":[121],"Triton":[122],"kernels":[123],"efficiently":[125],"implement":[126],"this":[127],"token-wise":[128],"GPUs,":[132],"achieving":[133],"up":[134,161],"$\\sim$2$\\times$":[136],"improvements":[137],"in":[138],"throughput":[140],"time-to-first-token":[142],"FlashAttention.":[144],"Moreover,":[145],"post-training":[148],"pruning":[149],"highly":[151],"sparse":[152],"layers,":[155],"reducing":[156],"KV":[157],"cache":[158],"50\\%":[163],"with":[164],"negligible":[165],"loss.":[167],"Our":[168],"code":[169],"released":[171],"under":[172],"Apache":[173],"2.0":[174],"at":[175],"https://github.com/awslabs/hybrid-model-factory/tree/main/examples/research/L2A.":[176]},"counts_by_year":[],"updated_date":"2026-07-08T06:17:01.165560","created_date":"2026-03-20T00:00:00"}