{"id":"https://openalex.org/W7161825119","doi":"https://doi.org/10.48550/arxiv.2605.20022","title":"FlexDraft: Flexible Speculative Decoding via Attention Tuning and Bonus-Guided Calibration","display_name":"FlexDraft: Flexible Speculative Decoding via Attention Tuning and Bonus-Guided Calibration","publication_year":2026,"publication_date":"2026-05-19","ids":{"openalex":"https://openalex.org/W7161825119","doi":"https://doi.org/10.48550/arxiv.2605.20022"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.20022","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20022","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":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.20022","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5136580388","display_name":"Yaojie Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Yaojie","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101392493","display_name":"Jianuo Huang","orcid":"https://orcid.org/0009-0001-6017-4147"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Jianuo","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136581408","display_name":"Junlong Ke","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ke, Junlong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136545218","display_name":"Yuhang Han","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Han, Yuhang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5134410252","display_name":"Yongji Long","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Long, Yongji","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136529812","display_name":"Tianchen Zhao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhao, Tianchen","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136610380","display_name":"Biqing Qi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Qi, Biqing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136515825","display_name":"Linfeng Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Linfeng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":8,"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.40610000491142273,"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.40610000491142273,"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/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.20489999651908875,"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/T10036","display_name":"Advanced Neural Network Applications","score":0.041600000113248825,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/decoding-methods","display_name":"Decoding methods","score":0.8427000045776367},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.7796000242233276},{"id":"https://openalex.org/keywords/throughput","display_name":"Throughput","score":0.5501000285148621},{"id":"https://openalex.org/keywords/deblocking-filter","display_name":"Deblocking filter","score":0.4447999894618988},{"id":"https://openalex.org/keywords/calibration","display_name":"Calibration","score":0.44269999861717224},{"id":"https://openalex.org/keywords/quality","display_name":"Quality (philosophy)","score":0.41749998927116394},{"id":"https://openalex.org/keywords/block","display_name":"Block (permutation group theory)","score":0.4041999876499176},{"id":"https://openalex.org/keywords/sequential-decoding","display_name":"Sequential decoding","score":0.38519999384880066}],"concepts":[{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.8427000045776367},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7950000166893005},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.7796000242233276},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.5501000285148621},{"id":"https://openalex.org/C143184774","wikidata":"https://www.wikidata.org/wiki/Q3020846","display_name":"Deblocking filter","level":2,"score":0.4447999894618988},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4431000053882599},{"id":"https://openalex.org/C165838908","wikidata":"https://www.wikidata.org/wiki/Q736777","display_name":"Calibration","level":2,"score":0.44269999861717224},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.430400013923645},{"id":"https://openalex.org/C2779530757","wikidata":"https://www.wikidata.org/wiki/Q1207505","display_name":"Quality (philosophy)","level":2,"score":0.41749998927116394},{"id":"https://openalex.org/C2777210771","wikidata":"https://www.wikidata.org/wiki/Q4927124","display_name":"Block (permutation group theory)","level":2,"score":0.4041999876499176},{"id":"https://openalex.org/C193969084","wikidata":"https://www.wikidata.org/wiki/Q7452500","display_name":"Sequential decoding","level":4,"score":0.38519999384880066},{"id":"https://openalex.org/C2779679103","wikidata":"https://www.wikidata.org/wiki/Q5251805","display_name":"Degradation (telecommunications)","level":2,"score":0.352400004863739},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.3504999876022339},{"id":"https://openalex.org/C144745244","wikidata":"https://www.wikidata.org/wiki/Q4927286","display_name":"Blocking (statistics)","level":2,"score":0.33660000562667847},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.3343999981880188},{"id":"https://openalex.org/C2777735758","wikidata":"https://www.wikidata.org/wiki/Q817765","display_name":"Path (computing)","level":2,"score":0.32600000500679016},{"id":"https://openalex.org/C159877910","wikidata":"https://www.wikidata.org/wiki/Q2202883","display_name":"Autoregressive model","level":2,"score":0.32580000162124634},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.3163999915122986},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.3005000054836273},{"id":"https://openalex.org/C9390403","wikidata":"https://www.wikidata.org/wiki/Q3966","display_name":"Computer hardware","level":1,"score":0.289000004529953},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.2732999920845032},{"id":"https://openalex.org/C147764199","wikidata":"https://www.wikidata.org/wiki/Q6865248","display_name":"Minification","level":2,"score":0.25929999351501465},{"id":"https://openalex.org/C204397858","wikidata":"https://www.wikidata.org/wiki/Q4437907","display_name":"List decoding","level":5,"score":0.25119999051094055}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.20022","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20022","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":"article"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.20022","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20022","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":"article"},"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":{"Speculative":[0],"decoding":[1,32,54,90,148],"accelerates":[2],"memory-bound":[3],"LLM":[4],"inference":[5],"without":[6],"quality":[7,98,196],"degradation":[8,99],"by":[9,58,168,224],"using":[10],"a":[11,64,145,206],"fast":[12],"drafter":[13],"to":[14,23,72,124,132,153,188,215,259],"propose":[15],"multiple":[16],"candidate":[17],"tokens":[18],"and":[19,39,41,61,119,128,193,236,242,251],"the":[20,116,120,171,175,184,190,211],"target":[21,66,191],"model":[22],"verify":[24,237,246],"them":[25],"in":[26,114],"parallel.":[27],"However,":[28],"conventional":[29],"sequential":[30,243],"speculative":[31,53,89,147],"suffers":[33,111],"from":[34,102,112],"mutual":[35],"waiting":[36],"between":[37,233],"drafting":[38,60,167],"verification,":[40],"repeated":[42],"exchange":[43],"of":[44,174],"intermediate":[45],"states":[46],"further":[47],"increases":[48],"memory":[49],"access":[50],"overhead.":[51],"Parallel":[52],"addresses":[55],"this":[56,108],"limitation":[57],"performing":[59],"verification":[62,126,221,253],"within":[63],"single":[65],"forward":[67],"pass,":[68],"allowing":[69],"future":[70],"drafts":[71,197],"be":[73],"prepared":[74],"while":[75,182],"current":[76],"candidates":[77],"are":[78],"being":[79],"verified.":[80],"Although":[81],"effective":[82],"at":[83,134,238,247],"small":[84,239],"batch":[85,136,155,240,249],"sizes,":[86,250],"existing":[87],"parallel":[88,234],"methods":[91],"either":[92],"require":[93],"costly":[94],"continual":[95],"pretraining":[96],"with":[97,198],"or":[100],"suffer":[101],"low":[103],"acceptance":[104],"rates.":[105],"More":[106],"importantly,":[107],"paradigm":[109],"inherently":[110],"uncertainty":[113],"both":[115],"bonus":[117,213,225],"token":[118,214,226],"accepted":[121],"length,":[122],"leading":[123],"draft":[125,217,220,235,244,257],"mismatch":[127,222],"causing":[129],"throughput":[130],"gains":[131],"collapse":[133],"large":[135,248],"sizes.":[137],"To":[138],"address":[139],"these":[140],"limitations,":[141],"we":[142],"introduce":[143],"FlexDraft,":[144],"lossless":[146],"framework":[149],"that":[150],"flexibly":[151],"adapts":[152],"varying":[154],"sizes":[156,241],"through":[157],"three":[158],"key":[159],"designs.":[160],"(1)":[161],"Attention":[162],"Tuning":[163],"enables":[164],"block":[165],"diffusion":[166],"tuning":[169],"only":[170],"attention":[172],"projectors":[173],"final":[176],"few":[177],"layers":[178],"on":[179,210,256],"mask":[180],"tokens,":[181],"keeping":[183],"autoregressive":[185],"path":[186],"frozen":[187],"preserve":[189],"distribution":[192],"produce":[194],"high":[195],"minimal":[199],"trainable":[200],"parameters.":[201],"(2)":[202],"Bonus-guided":[203],"Calibration":[204],"uses":[205],"lightweight":[207],"MLP":[208],"conditioned":[209],"resolved":[212],"calibrate":[216],"logits,":[218],"mitigating":[219],"caused":[223],"uncertainty.":[227],"(3)":[228],"Flex":[229],"Decoding":[230],"dynamically":[231],"switches":[232],"then":[245],"adjusts":[252],"length":[254],"based":[255],"confidence":[258],"eliminate":[260],"redundant":[261],"computation.":[262]},"counts_by_year":[],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2026-05-21T00:00:00"}