{"id":"https://openalex.org/W4406023743","doi":"https://doi.org/10.48550/arxiv.2408.08146","title":"KOALA: Enhancing Speculative Decoding for LLM via Multi-Layer Draft Heads with Adversarial Learning","display_name":"KOALA: Enhancing Speculative Decoding for LLM via Multi-Layer Draft Heads with Adversarial Learning","publication_year":2024,"publication_date":"2024-08-15","ids":{"openalex":"https://openalex.org/W4406023743","doi":"https://doi.org/10.48550/arxiv.2408.08146"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2408.08146","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2408.08146","pdf_url":"https://arxiv.org/pdf/2408.08146","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2408.08146","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5101894184","display_name":"Kaiqi Zhang","orcid":"https://orcid.org/0000-0003-4097-4750"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Zhang, Kaiqi","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5107342072","display_name":"Jing Zhao","orcid":"https://orcid.org/0009-0006-6839-9355"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhao, Jing","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5100403647","display_name":"Rui Chen","orcid":"https://orcid.org/0000-0002-4711-4545"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Rui","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":3,"corresponding_author_ids":["https://openalex.org/A5101894184"],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"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.9786999821662903,"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.9786999821662903,"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/T13999","display_name":"Digital Rights Management and Security","score":0.9128999710083008,"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":[{"id":"https://openalex.org/keywords/layer","display_name":"Layer (electronics)","score":0.7667981386184692},{"id":"https://openalex.org/keywords/adversarial-system","display_name":"Adversarial system","score":0.7633035778999329},{"id":"https://openalex.org/keywords/decoding-methods","display_name":"Decoding methods","score":0.7498593330383301},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.5691304206848145},{"id":"https://openalex.org/keywords/hull","display_name":"Hull","score":0.45450741052627563},{"id":"https://openalex.org/keywords/computer-security","display_name":"Computer security","score":0.37417662143707275},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.3100595474243164},{"id":"https://openalex.org/keywords/telecommunications","display_name":"Telecommunications","score":0.20964547991752625},{"id":"https://openalex.org/keywords/engineering","display_name":"Engineering","score":0.16260558366775513},{"id":"https://openalex.org/keywords/marine-engineering","display_name":"Marine engineering","score":0.10805484652519226},{"id":"https://openalex.org/keywords/chemistry","display_name":"Chemistry","score":0.0917990505695343}],"concepts":[{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.7667981386184692},{"id":"https://openalex.org/C37736160","wikidata":"https://www.wikidata.org/wiki/Q1801315","display_name":"Adversarial system","level":2,"score":0.7633035778999329},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.7498593330383301},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5691304206848145},{"id":"https://openalex.org/C37423430","wikidata":"https://www.wikidata.org/wiki/Q6750281","display_name":"Hull","level":2,"score":0.45450741052627563},{"id":"https://openalex.org/C38652104","wikidata":"https://www.wikidata.org/wiki/Q3510521","display_name":"Computer security","level":1,"score":0.37417662143707275},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3100595474243164},{"id":"https://openalex.org/C76155785","wikidata":"https://www.wikidata.org/wiki/Q418","display_name":"Telecommunications","level":1,"score":0.20964547991752625},{"id":"https://openalex.org/C127413603","wikidata":"https://www.wikidata.org/wiki/Q11023","display_name":"Engineering","level":0,"score":0.16260558366775513},{"id":"https://openalex.org/C199104240","wikidata":"https://www.wikidata.org/wiki/Q118291","display_name":"Marine engineering","level":1,"score":0.10805484652519226},{"id":"https://openalex.org/C185592680","wikidata":"https://www.wikidata.org/wiki/Q2329","display_name":"Chemistry","level":0,"score":0.0917990505695343},{"id":"https://openalex.org/C178790620","wikidata":"https://www.wikidata.org/wiki/Q11351","display_name":"Organic chemistry","level":1,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2408.08146","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2408.08146","pdf_url":"https://arxiv.org/pdf/2408.08146","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},{"id":"doi:10.48550/arxiv.2408.08146","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2408.08146","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":"article"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2408.08146","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2408.08146","pdf_url":"https://arxiv.org/pdf/2408.08146","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4406023743.pdf"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W2502115930","https://openalex.org/W2467926385","https://openalex.org/W2482350142","https://openalex.org/W4246396837","https://openalex.org/W3126451824","https://openalex.org/W1561927205","https://openalex.org/W3191453585","https://openalex.org/W4297672492","https://openalex.org/W4310988119","https://openalex.org/W4285226279"],"abstract_inverted_index":{"Large":[0],"Language":[1],"Models":[2],"(LLMs)":[3],"exhibit":[4],"high":[5],"inference":[6],"latency":[7,129],"due":[8],"to":[9,43],"their":[10],"autoregressive":[11,119],"decoding":[12,20],"nature.":[13],"While":[14],"the":[15,44,49,63,70,73,84,93,103,140],"draft":[16,45,52,74,104,122,142],"head":[17,53,75],"in":[18,76],"speculative":[19,109],"mitigates":[21],"this":[22,30,89],"issue,":[23],"its":[24],"full":[25],"potential":[26],"remains":[27],"unexplored.":[28],"In":[29],"paper,":[31],"we":[32],"introduce":[33],"KOALA":[34,67,100],"(K-layer":[35],"Optimized":[36],"Adversarial":[37],"Learning":[38],"Architecture),":[39],"an":[40],"orthogonal":[41],"approach":[42],"head.":[46],"By":[47],"transforming":[48],"conventional":[50],"single-layer":[51],"into":[54,62],"a":[55,128],"multi-layer":[56],"architecture":[57],"and":[58,120],"incorporating":[59],"adversarial":[60],"learning":[61],"traditional":[64],"supervised":[65],"training,":[66],"significantly":[68],"improves":[69],"accuracy":[71],"of":[72,86,95,115,133],"predicting":[77],"subsequent":[78],"tokens,":[79],"thus":[80],"more":[81],"closely":[82],"mirroring":[83],"functionality":[85],"LLMs.":[87],"Although":[88],"improvement":[90,132],"comes":[91],"at":[92],"cost":[94],"slightly":[96],"increased":[97],"drafting":[98],"overhead,":[99],"substantially":[101],"unlocks":[102],"head's":[105],"potential,":[106],"greatly":[107],"enhancing":[108],"decoding.":[110],"We":[111],"conducted":[112],"comprehensive":[113],"evaluations":[114],"KOALA,":[116],"including":[117],"both":[118],"non-autoregressive":[121],"heads":[123],"across":[124],"various":[125],"tasks,":[126],"demonstrating":[127],"speedup":[130],"ratio":[131],"0.24x-0.41x,":[134],"which":[135],"is":[136],"10.57%-14.09%":[137],"faster":[138],"than":[139],"original":[141],"heads.":[143]},"counts_by_year":[],"updated_date":"2026-03-11T14:59:36.786465","created_date":"2025-01-03T00:00:00"}