{"id":"https://openalex.org/W4393760033","doi":"https://doi.org/10.48550/arxiv.2404.00242","title":"DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference","display_name":"DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference","publication_year":2024,"publication_date":"2024-03-30","ids":{"openalex":"https://openalex.org/W4393760033","doi":"https://doi.org/10.48550/arxiv.2404.00242"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2404.00242","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2404.00242","pdf_url":"https://arxiv.org/pdf/2404.00242","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":null,"raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2404.00242","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5100937960","display_name":"Jinwei Yao","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Yao, Jinwei","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5042416229","display_name":"Kaiqi Chen","orcid":"https://orcid.org/0000-0001-7505-0764"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Kaiqi","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5053856461","display_name":"Kexun Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Kexun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5003491365","display_name":"Jiaxuan You","orcid":"https://orcid.org/0000-0003-1812-7598"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"You, Jiaxuan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5002684888","display_name":"Binhang Yuan","orcid":"https://orcid.org/0000-0002-3188-2769"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yuan, Binhang","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5004837462","display_name":"Zeke Wang","orcid":"https://orcid.org/0000-0001-8550-9241"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Zeke","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5112570961","display_name":"Tao Lin","orcid":"https://orcid.org/0000-0003-4282-5433"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lin, Tao","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":7,"corresponding_author_ids":["https://openalex.org/A5100937960"],"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/T12016","display_name":"Web Data Mining and Analysis","score":0.9467999935150146,"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/T12016","display_name":"Web Data Mining and Analysis","score":0.9467999935150146,"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"}},{"id":"https://openalex.org/T13523","display_name":"Mathematics, Computing, and Information Processing","score":0.9301000237464905,"subfield":{"id":"https://openalex.org/subfields/1703","display_name":"Computational Theory and Mathematics"},"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/T11181","display_name":"Advanced Data Storage Technologies","score":0.9251000285148621,"subfield":{"id":"https://openalex.org/subfields/1705","display_name":"Computer Networks and Communications"},"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.7307502031326294},{"id":"https://openalex.org/keywords/tree","display_name":"Tree (set theory)","score":0.6882342100143433},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.6087021827697754},{"id":"https://openalex.org/keywords/flash","display_name":"Flash (photography)","score":0.48076412081718445},{"id":"https://openalex.org/keywords/search-tree","display_name":"Search tree","score":0.4586624205112457},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.34013450145721436},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.2914273142814636},{"id":"https://openalex.org/keywords/algorithm","display_name":"Algorithm","score":0.21772369742393494},{"id":"https://openalex.org/keywords/search-algorithm","display_name":"Search algorithm","score":0.18373560905456543},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.15636789798736572},{"id":"https://openalex.org/keywords/combinatorics","display_name":"Combinatorics","score":0.06290242075920105}],"concepts":[{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.7307502031326294},{"id":"https://openalex.org/C113174947","wikidata":"https://www.wikidata.org/wiki/Q2859736","display_name":"Tree (set theory)","level":2,"score":0.6882342100143433},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6087021827697754},{"id":"https://openalex.org/C2777526259","wikidata":"https://www.wikidata.org/wiki/Q221836","display_name":"Flash (photography)","level":2,"score":0.48076412081718445},{"id":"https://openalex.org/C207024777","wikidata":"https://www.wikidata.org/wiki/Q621673","display_name":"Search tree","level":3,"score":0.4586624205112457},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.34013450145721436},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.2914273142814636},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.21772369742393494},{"id":"https://openalex.org/C125583679","wikidata":"https://www.wikidata.org/wiki/Q755673","display_name":"Search algorithm","level":2,"score":0.18373560905456543},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.15636789798736572},{"id":"https://openalex.org/C114614502","wikidata":"https://www.wikidata.org/wiki/Q76592","display_name":"Combinatorics","level":1,"score":0.06290242075920105},{"id":"https://openalex.org/C153349607","wikidata":"https://www.wikidata.org/wiki/Q36649","display_name":"Visual arts","level":1,"score":0.0},{"id":"https://openalex.org/C142362112","wikidata":"https://www.wikidata.org/wiki/Q735","display_name":"Art","level":0,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2404.00242","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2404.00242","pdf_url":"https://arxiv.org/pdf/2404.00242","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":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2404.00242","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2404.00242","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:2404.00242","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2404.00242","pdf_url":"https://arxiv.org/pdf/2404.00242","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":null,"raw_type":"text"},"sustainable_development_goals":[],"awards":[{"id":"https://openalex.org/G324441156","display_name":null,"funder_award_id":"2022ZD0115101","funder_id":"https://openalex.org/F4320329860","funder_display_name":"National Science and Technology Major Project"}],"funders":[{"id":"https://openalex.org/F4320318567","display_name":"Westlake University","ror":null},{"id":"https://openalex.org/F4320329860","display_name":"National Science and Technology Major Project","ror":null}],"has_content":{"pdf":true,"grobid_xml":false},"content_urls":{"pdf":"https://content.openalex.org/works/W4393760033.pdf"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W3128587426","https://openalex.org/W2227684440","https://openalex.org/W2739550867","https://openalex.org/W2156833871","https://openalex.org/W2359807192","https://openalex.org/W2174962524","https://openalex.org/W2157743338","https://openalex.org/W2083751087","https://openalex.org/W48124421","https://openalex.org/W2915699561"],"abstract_inverted_index":{"Large":[0],"language":[1],"models":[2],"(LLMs)":[3],"are":[4,39],"increasingly":[5],"employed":[6],"for":[7,36,67,188],"complex":[8],"tasks":[9],"that":[10,137,158],"process":[11],"multiple":[12],"generation":[13],"calls":[14],"in":[15,146,200],"a":[16,60,78,108,135,156],"tree":[17],"structure":[18],"with":[19,94,105,112,168],"shared":[20,71,91,144],"prefixes":[21,145],"of":[22,45,62,70,123,126,143,162],"tokens,":[23],"including":[24],"few-shot":[25],"prompting,":[26],"multi-step":[27],"reasoning,":[28],"speculative":[29],"decoding,":[30],"etc.":[31],"However,":[32],"existing":[33],"inference":[34],"systems":[35],"tree-based":[37,207],"applications":[38],"inefficient":[40],"due":[41],"to":[42,55,197,210],"improper":[43],"partitioning":[44],"queries":[46],"and":[47,73,90,114,184],"KV":[48,68,83,116,127,141,154,164,181],"cache":[49,69,84,117,128,142,165,182],"during":[50,129,175,191],"attention":[51,110,130,147,176,192,212],"calculation.":[52],"This":[53],"leads":[54],"two":[56],"main":[57],"issues:":[58],"(1)":[59],"lack":[61],"memory":[63,89],"access":[64],"(IO)":[65],"reuse":[66],"prefixes,":[72],"(2)":[74],"poor":[75],"load":[76],"balancing.As":[77],"result,":[79],"there":[80],"is":[81,216],"redundant":[82],"IO":[85,183,187],"between":[86],"GPU":[87,96,173],"global":[88],"memory,":[92],"along":[93],"low":[95],"utilization.":[97],"To":[98],"address":[99],"these":[100],"challenges,":[101],"we":[102,150],"propose":[103,151],"DeFT(Decoding":[104],"Flash":[106],"Tree-Attention),":[107],"hardware-efficient":[109],"algorithm":[111],"prefix-aware":[113],"load-balanced":[115],"partitions.":[118],"DeFT":[119,194],"reduces":[120],"the":[121,163,201],"number":[122],"read/write":[124],"operations":[125],"calculation":[131],"through":[132],"KV-Guided":[133],"Grouping,":[134],"method":[136],"avoids":[138],"repeatedly":[139],"loading":[140],"computation.":[148],"Additionally,":[149],"Flattened":[152],"Tree":[153],"Splitting,":[155],"mechanism":[157],"ensures":[159],"even":[160],"distribution":[161],"across":[166,204],"partitions":[167],"little":[169],"computation":[170],"redundancy,":[171],"enhancing":[172],"utilization":[174],"computations.":[177],"By":[178],"reducing":[179],"73-99%":[180],"nearly":[185],"100%":[186],"partial":[189],"results":[190],"calculation,":[193],"achieves":[195],"up":[196],"2.23/3.59x":[198],"speedup":[199],"end-to-end/attention":[202],"latency":[203],"three":[205],"practical":[206],"workloads":[208],"compared":[209],"state-of-the-art":[211],"algorithms.":[213],"Our":[214],"code":[215],"available":[217],"at":[218],"https://github.com/LINs-lab/DeFT.":[219]},"counts_by_year":[],"updated_date":"2026-04-10T15:06:20.359241","created_date":"2024-04-03T00:00:00"}