{"id":"https://openalex.org/W7138089354","doi":"https://doi.org/10.48550/arxiv.2603.14591","title":"FlashHead: Efficient Drop-In Replacement for the Classification Head in Language Model Inference","display_name":"FlashHead: Efficient Drop-In Replacement for the Classification Head in Language Model Inference","publication_year":2026,"publication_date":"2026-03-15","ids":{"openalex":"https://openalex.org/W7138089354","doi":"https://doi.org/10.48550/arxiv.2603.14591"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.14591","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.14591","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.2603.14591","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5062862473","display_name":"Wilhelm Tranheden","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tranheden, Wilhelm","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5002007962","display_name":"Shahnawaz Ahmed","orcid":"https://orcid.org/0000-0002-0913-5177"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ahmed, Shahnawaz","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5073376939","display_name":"Devdatt Dubhashi","orcid":"https://orcid.org/0000-0002-9928-2305"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dubhashi, Devdatt","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5129737100","display_name":"Jonna Matthiesen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Matthiesen, Jonna","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5087957717","display_name":"Hannes von Essen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"von Essen, Hannes","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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.39469999074935913,"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"}},"topics":[{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.39469999074935913,"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/T10028","display_name":"Topic Modeling","score":0.04410000145435333,"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/T12026","display_name":"Explainable Artificial Intelligence (XAI)","score":0.03889999911189079,"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.7379000186920166},{"id":"https://openalex.org/keywords/bottleneck","display_name":"Bottleneck","score":0.603600025177002},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.5216000080108643},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.5134999752044678},{"id":"https://openalex.org/keywords/cluster-analysis","display_name":"Cluster analysis","score":0.4966000020503998},{"id":"https://openalex.org/keywords/vocabulary","display_name":"Vocabulary","score":0.4909000098705292},{"id":"https://openalex.org/keywords/computation","display_name":"Computation","score":0.4187000095844269},{"id":"https://openalex.org/keywords/benchmark","display_name":"Benchmark (surveying)","score":0.41830000281333923}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.8148000240325928},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.7379000186920166},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.6044999957084656},{"id":"https://openalex.org/C2780513914","wikidata":"https://www.wikidata.org/wiki/Q18210350","display_name":"Bottleneck","level":2,"score":0.603600025177002},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.5216000080108643},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.5134999752044678},{"id":"https://openalex.org/C73555534","wikidata":"https://www.wikidata.org/wiki/Q622825","display_name":"Cluster analysis","level":2,"score":0.4966000020503998},{"id":"https://openalex.org/C2777601683","wikidata":"https://www.wikidata.org/wiki/Q6499736","display_name":"Vocabulary","level":2,"score":0.4909000098705292},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.47909998893737793},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.4187000095844269},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.41830000281333923},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.37549999356269836},{"id":"https://openalex.org/C2780312720","wikidata":"https://www.wikidata.org/wiki/Q5689100","display_name":"Head (geology)","level":2,"score":0.3578000068664551},{"id":"https://openalex.org/C49937458","wikidata":"https://www.wikidata.org/wiki/Q2599292","display_name":"Probabilistic logic","level":2,"score":0.35260000824928284},{"id":"https://openalex.org/C9652623","wikidata":"https://www.wikidata.org/wiki/Q190109","display_name":"Field (mathematics)","level":2,"score":0.3393999934196472},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.33009999990463257},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.3190000057220459},{"id":"https://openalex.org/C140779682","wikidata":"https://www.wikidata.org/wiki/Q210868","display_name":"Sampling (signal processing)","level":3,"score":0.31630000472068787},{"id":"https://openalex.org/C155092808","wikidata":"https://www.wikidata.org/wiki/Q182557","display_name":"Computational linguistics","level":2,"score":0.31369999051094055},{"id":"https://openalex.org/C114289077","wikidata":"https://www.wikidata.org/wiki/Q3284399","display_name":"Statistical model","level":2,"score":0.30820000171661377},{"id":"https://openalex.org/C184337299","wikidata":"https://www.wikidata.org/wiki/Q1437428","display_name":"Semantics (computer science)","level":2,"score":0.3059999942779541},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.2838999927043915},{"id":"https://openalex.org/C160234255","wikidata":"https://www.wikidata.org/wiki/Q812535","display_name":"Bayesian inference","level":3,"score":0.2782999873161316},{"id":"https://openalex.org/C43521106","wikidata":"https://www.wikidata.org/wiki/Q2165493","display_name":"Pipeline (software)","level":2,"score":0.2630000114440918},{"id":"https://openalex.org/C44291984","wikidata":"https://www.wikidata.org/wiki/Q1074173","display_name":"Question answering","level":2,"score":0.2565999925136566}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.14591","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.14591","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.2603.14591","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.14591","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":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Language":[0],"models":[1,207],"are":[2],"increasingly":[3],"adopting":[4],"smaller":[5],"architectures":[6],"optimized":[7],"for":[8,36,56,195,208],"consumer":[9,209],"devices.":[10],"In":[11],"this":[12],"setting,":[13],"inference":[14,46,169,197],"efficiency":[15],"is":[16,62],"the":[17,28,51,57,77,90,144,156,181,186],"primary":[18],"constraint.":[19],"Meanwhile,":[20],"vocabulary":[21,105],"sizes":[22],"continue":[23],"to":[24,38,115,123,173,180,203],"grow":[25],"rapidly,":[26],"making":[27],"classification":[29,59,88,187],"head":[30,60,79,188],"a":[31,81,86,99,129,192,200],"critical":[32],"bottleneck":[33],"that":[34,61,74,103,134,165],"accounts":[35],"up":[37,172],"60\\%":[39],"of":[40,45,121,171],"model":[41,117],"parameters,":[42],"and":[43,64,147,162,198],"50\\%":[44],"compute.":[47],"We":[48],"introduce":[49],"FlashHead,":[50],"first":[52],"efficient":[53,196],"drop-in":[54],"replacement":[55],"dense":[58,87],"training-free":[63],"hardware-friendly.":[65],"FlashHead":[66,93,166,190],"builds":[67],"on":[68,159],"principles":[69],"from":[70],"information":[71],"retrieval,":[72],"reframing":[73],"computation":[75,154],"at":[76],"output":[78,177],"as":[80],"retrieval":[82,114,136],"problem":[83],"rather":[84],"than":[85],"over":[89],"full":[91,145],"vocabulary.":[92],"introduces":[94],"four":[95],"key":[96,201],"innovations:":[97],"(1)":[98],"balanced":[100],"clustering":[101],"scheme":[102],"structures":[104],"partitions":[106],"into":[107],"compact":[108],"hardware-efficient":[109],"tensors,":[110],"(2)":[111],"extending":[112],"multiprobe":[113],"language":[116],"heads,":[118],"enabling":[119,140,151],"thousands":[120],"clusters":[122],"be":[124],"scored":[125],"in":[126,155],"parallel,":[127],"(3)":[128],"novel":[130],"inference-time":[131],"sampling":[132,142],"mechanism":[133],"extends":[135],"beyond":[137],"top":[138],"tokens,":[139],"probabilistic":[141],"across":[143],"vocabulary,":[146],"(4)":[148],"selective":[149],"quantization,":[150],"effective":[152],"low-bit":[153],"head.":[157,183],"Experiments":[158],"Llama-3.2,":[160],"Gemma-3,":[161],"Qwen-3":[163],"show":[164],"delivers":[167],"model-level":[168],"speedups":[170],"\\textbf{1.75x}":[174],"which":[175],"maintaining":[176],"accuracy":[178],"compared":[179],"original":[182],"By":[184],"overcoming":[185],"bottleneck,":[189],"establishes":[191],"new":[193],"benchmark":[194],"removes":[199],"barrier":[202],"developing":[204],"smaller,":[205],"capable":[206],"hardware.":[210]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-03-18T00:00:00"}