{"id":"https://openalex.org/W4395444087","doi":"https://doi.org/10.48550/arxiv.2404.14469","title":"SnapKV: LLM Knows What You are Looking for Before Generation","display_name":"SnapKV: LLM Knows What You are Looking for Before Generation","publication_year":2024,"publication_date":"2024-04-22","ids":{"openalex":"https://openalex.org/W4395444087","doi":"https://doi.org/10.48550/arxiv.2404.14469"},"language":"en","primary_location":{"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2404.14469","pdf_url":"https://arxiv.org/pdf/2404.14469","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":["Cornell University"],"type":"repository"},"license":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false},"type":"preprint","type_crossref":"posted-content","indexed_in":["arxiv"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2404.14469","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5109036417","display_name":"Yuhong Li","orcid":"https://orcid.org/0000-0002-3769-6772"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Yuhong","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100590370","display_name":"Yingbing Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Yingbing","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100547390","display_name":"Bowen Yang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yang, Bowen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5033303445","display_name":"Bharat Venkitesh","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Venkitesh, Bharat","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5003520377","display_name":"Acyr Locatelli","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Locatelli, Acyr","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5088715551","display_name":"Hanchen Ye","orcid":"https://orcid.org/0000-0002-6646-8146"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ye, Hanchen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101342736","display_name":"Tianle Cai","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Cai, Tianle","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5063377058","display_name":"Patrick Lewis","orcid":"https://orcid.org/0000-0002-2192-9543"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lewis, Patrick","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5115588644","display_name":"Deming Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Deming","raw_affiliation_strings":[],"affiliations":[]}],"institution_assertions":[],"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":{"value":0.0,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":0,"max":84},"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"primary_topic":{"id":"https://openalex.org/T13999","display_name":"Digital Rights Management and Security","score":0.9468,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"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/T13999","display_name":"Digital Rights Management and Security","score":0.9468,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[],"concepts":[{"id":"https://openalex.org/C17744445","wikidata":"https://www.wikidata.org/wiki/Q36442","display_name":"Political science","level":0,"score":0.31053}],"mesh":[],"locations_count":1,"locations":[{"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2404.14469","pdf_url":"https://arxiv.org/pdf/2404.14469","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":["Cornell University"],"type":"repository"},"license":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false}],"best_oa_location":{"is_oa":true,"landing_page_url":"https://arxiv.org/abs/2404.14469","pdf_url":"https://arxiv.org/pdf/2404.14469","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":["Cornell University"],"type":"repository"},"license":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false},"sustainable_development_goals":[],"grants":[],"datasets":[],"versions":[],"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W594353338","https://openalex.org/W4391375266","https://openalex.org/W4390697879","https://openalex.org/W4382466601","https://openalex.org/W2949263084","https://openalex.org/W2922049016","https://openalex.org/W2748952813","https://openalex.org/W2743539335","https://openalex.org/W2724734218","https://openalex.org/W2070214669"],"abstract_inverted_index":{"Large":[0],"Language":[1],"Models":[2],"(LLMs)":[3],"have":[4],"made":[5],"remarkable":[6],"progress":[7],"in":[8,21,32,69,78,154,161,215],"processing":[9,139,168],"extensive":[10],"contexts,":[11],"with":[12,150,206],"the":[13,26,29,79,103,106,131,174,216],"Key-Value":[14],"(KV)":[15],"cache":[16,31,62],"playing":[17],"a":[18,146,151,199,211],"vital":[19],"role":[20],"enhancing":[22],"their":[23],"performance.":[24],"However,":[25],"growth":[27],"of":[28,105,170],"KV":[30,61,115,121],"response":[33],"to":[34,40,165,181,194],"increasing":[35],"input":[36,141],"length":[37],"poses":[38],"challenges":[39],"memory":[41,136,162],"and":[42,55,135,157],"time":[43],"efficiency.":[44],"To":[45],"address":[46],"this":[47,49,91,110],"problem,":[48],"paper":[50],"introduces":[51],"SnapKV,":[52],"an":[53,98,158],"innovative":[54],"fine-tuning-free":[56],"approach":[57,128],"that":[58,74],"efficiently":[59],"minimizes":[60],"size":[63],"while":[64],"still":[65],"delivering":[66],"comparable":[67,179],"performance":[68,180],"real-world":[70],"applications.":[71,227],"We":[72],"discover":[73],"each":[75,124],"attention":[76,86,125],"head":[77],"model":[80],"consistently":[81],"focuses":[82],"on":[83,109,198],"specific":[84],"prompt":[85],"features":[87],"during":[88],"generation.":[89],"Meanwhile,":[90],"robust":[92],"pattern":[93],"can":[94,191],"be":[95],"obtained":[96],"from":[97],"`observation'":[99],"window":[100],"located":[101],"at":[102],"end":[104],"prompts.":[107],"Drawing":[108],"insight,":[111],"SnapKV":[112,144,190],"automatically":[113],"compresses":[114],"caches":[116],"by":[117],"selecting":[118],"clustered":[119],"important":[120],"positions":[122],"for":[123,225],"head.":[126],"Our":[127],"significantly":[129],"reduces":[130],"growing":[132],"computational":[133],"overhead":[134],"footprint":[137],"when":[138,167],"long":[140,186],"sequences.":[142],"Specifically,":[143],"achieves":[145],"consistent":[147],"decoding":[148],"speed":[149,156],"3.6x":[152],"increase":[153],"generation":[155],"8.2x":[159],"enhancement":[160],"efficiency":[163],"compared":[164],"baseline":[166,182],"inputs":[169],"16K":[171],"tokens.":[172],"At":[173],"same":[175],"time,":[176],"it":[177],"maintains":[178],"models":[183],"across":[184],"16":[185],"sequence":[187],"datasets.":[188],"Moreover,":[189],"process":[192],"up":[193],"380K":[195],"context":[196],"tokens":[197],"single":[200],"A100-80GB":[201],"GPU":[202],"using":[203],"HuggingFace":[204],"implementation":[205],"minor":[207],"changes,":[208],"exhibiting":[209],"only":[210],"negligible":[212],"accuracy":[213],"drop":[214],"Needle-in-a-Haystack":[217],"test.":[218],"Further":[219],"comprehensive":[220],"studies":[221],"suggest":[222],"SnapKV's":[223],"potential":[224],"practical":[226]},"cited_by_api_url":"https://api.openalex.org/works?filter=cites:W4395444087","counts_by_year":[],"updated_date":"2024-12-24T02:37:23.348502","created_date":"2024-04-26"}