{"id":"https://openalex.org/W7160596962","doi":"https://doi.org/10.48550/arxiv.2605.05971","title":"Training Transformers for KV Cache Compressibility","display_name":"Training Transformers for KV Cache Compressibility","publication_year":2026,"publication_date":"2026-05-07","ids":{"openalex":"https://openalex.org/W7160596962","doi":"https://doi.org/10.48550/arxiv.2605.05971"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.05971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.05971","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.05971","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5135662019","display_name":"Yoav Gelberg","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gelberg, Yoav","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5083372935","display_name":"Yam Eitan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Eitan, Yam","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135642934","display_name":"Michael Bronstein","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Bronstein, Michael","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5029186201","display_name":"Yarin Gal","orcid":"https://orcid.org/0000-0002-2733-2078"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gal, Yarin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5135658978","display_name":"Haggai Maron","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Maron, Haggai","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/T10181","display_name":"Natural Language Processing Techniques","score":0.33970001339912415,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.33970001339912415,"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/T11269","display_name":"Algorithms and Data Compression","score":0.08799999952316284,"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.08380000293254852,"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/compressibility","display_name":"Compressibility","score":0.7383999824523926},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.6204000115394592},{"id":"https://openalex.org/keywords/prefix","display_name":"Prefix","score":0.5187000036239624},{"id":"https://openalex.org/keywords/bottleneck","display_name":"Bottleneck","score":0.5160999894142151},{"id":"https://openalex.org/keywords/cache","display_name":"Cache","score":0.4553000032901764},{"id":"https://openalex.org/keywords/embedding","display_name":"Embedding","score":0.3675000071525574},{"id":"https://openalex.org/keywords/compression","display_name":"Compression (physics)","score":0.35409998893737793},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.3472999930381775}],"concepts":[{"id":"https://openalex.org/C84655787","wikidata":"https://www.wikidata.org/wiki/Q8067817","display_name":"Compressibility","level":2,"score":0.7383999824523926},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6815999746322632},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.6204000115394592},{"id":"https://openalex.org/C141603448","wikidata":"https://www.wikidata.org/wiki/Q134830","display_name":"Prefix","level":2,"score":0.5187000036239624},{"id":"https://openalex.org/C2780513914","wikidata":"https://www.wikidata.org/wiki/Q18210350","display_name":"Bottleneck","level":2,"score":0.5160999894142151},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.4553000032901764},{"id":"https://openalex.org/C41608201","wikidata":"https://www.wikidata.org/wiki/Q980509","display_name":"Embedding","level":2,"score":0.3675000071525574},{"id":"https://openalex.org/C180016635","wikidata":"https://www.wikidata.org/wiki/Q2712821","display_name":"Compression (physics)","level":2,"score":0.35409998893737793},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.3472999930381775},{"id":"https://openalex.org/C99821215","wikidata":"https://www.wikidata.org/wiki/Q1136583","display_name":"Swap (finance)","level":2,"score":0.32659998536109924},{"id":"https://openalex.org/C177774035","wikidata":"https://www.wikidata.org/wiki/Q1246948","display_name":"Granularity","level":2,"score":0.2971000075340271},{"id":"https://openalex.org/C170858558","wikidata":"https://www.wikidata.org/wiki/Q1394144","display_name":"Automatic summarization","level":2,"score":0.29339998960494995},{"id":"https://openalex.org/C14036430","wikidata":"https://www.wikidata.org/wiki/Q3736076","display_name":"Function (biology)","level":2,"score":0.2883000075817108},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2842000126838684},{"id":"https://openalex.org/C204323151","wikidata":"https://www.wikidata.org/wiki/Q905424","display_name":"Range (aeronautics)","level":2,"score":0.27959999442100525},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.2728999853134155},{"id":"https://openalex.org/C96147967","wikidata":"https://www.wikidata.org/wiki/Q190686","display_name":"Subroutine","level":2,"score":0.25780001282691956},{"id":"https://openalex.org/C2778755073","wikidata":"https://www.wikidata.org/wiki/Q10858537","display_name":"Scale (ratio)","level":2,"score":0.2533999979496002},{"id":"https://openalex.org/C44291984","wikidata":"https://www.wikidata.org/wiki/Q1074173","display_name":"Question answering","level":2,"score":0.25290000438690186}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.05971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.05971","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.05971","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.05971","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":{"Long-context":[0],"language":[1],"modeling":[2],"is":[3,57,84],"increasingly":[4],"constrained":[5],"by":[6,60,125],"the":[7,20,46,63,75,88,94,114,139,159,182],"Key-Value":[8],"(KV)":[9],"cache,":[10],"whose":[11],"memory":[12],"and":[13,80,108,166,194],"decode-time":[14],"access":[15],"costs":[16],"scale":[17],"linearly":[18],"with":[19],"prefix":[21],"length.":[22],"This":[23,157],"bottleneck":[24],"has":[25],"motivated":[26],"a":[27,50,85,133,146],"range":[28],"of":[29,49,77,87,93,141,185,197],"context-compression":[30],"methods,":[31],"from":[32],"token-level":[33],"summarization":[34],"to":[35,116,161,169,173],"recent":[36],"optimization-based":[37],"KV":[38,47,78,148,153,164],"compression":[39,187],"methods.":[40],"These":[41],"post-hoc":[42,174],"methods":[43,188],"operate":[44],"on":[45],"cache":[48],"fixed":[51],"pretrained":[52],"model,":[53],"so":[54],"their":[55],"effectiveness":[56],"fundamentally":[58],"limited":[59],"how":[61],"well":[62],"model's":[64],"internal":[65],"representations":[66,121,171],"can":[67],"be":[68],"compressed.":[69],"In":[70],"this":[71],"work,":[72],"we":[73,127,177],"formalize":[74],"notion":[76],"compressibility":[79],"show":[81,178],"that":[82,99,137,151,179],"it":[83,168],"property":[86],"learned":[89],"representations,":[90],"rather":[91],"than":[92],"context":[95],"alone.":[96],"We":[97,144],"prove":[98],"almost":[100],"any":[101],"sequence-to-vector":[102],"function":[103],"admits":[104],"both":[105],"highly":[106],"compressible":[107,120,142],"inherently":[109],"non-compressible":[110],"transformer":[111],"implementations,":[112],"highlighting":[113],"need":[115],"guide":[117],"transformers":[118],"toward":[119],"during":[122,155],"training.":[123,156],"Motivated":[124],"this,":[126],"propose":[128],"KV-Compression":[129],"Aware":[130],"Training":[131],"(KV-CAT),":[132],"continued":[134],"pretraining":[135],"procedure":[136],"incentivizes":[138],"emergence":[140],"representations.":[143],"introduce":[145],"train-time":[147],"sparsification":[149],"policy":[150],"masks":[152],"slots":[154,165],"forces":[158],"model":[160],"use":[162],"fewer":[163],"encourages":[167],"learn":[170],"amenable":[172],"compression.":[175],"Empirically,":[176],"KV-CAT":[180],"improves":[181],"quality-budget":[183],"tradeoff":[184],"downstream":[186],"across":[189],"retrieval,":[190],"long-context":[191],"question":[192],"answering,":[193],"perplexity-based":[195],"evaluation":[196],"compressed-prefix":[198],"continuation.":[199]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-09T00:00:00"}