{"id":"https://openalex.org/W7164146140","doi":"https://doi.org/10.48550/arxiv.2606.11164","title":"ReasonAlloc: Hierarchical Decoding-Time KV Cache Budget Allocation for Reasoning Models","display_name":"ReasonAlloc: Hierarchical Decoding-Time KV Cache Budget Allocation for Reasoning Models","publication_year":2026,"publication_date":"2026-06-09","ids":{"openalex":"https://openalex.org/W7164146140","doi":"https://doi.org/10.48550/arxiv.2606.11164"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.11164","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.11164","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":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.2606.11164","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5138302202","display_name":"Wenhao Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Wenhao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138306153","display_name":"Hao Shi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shi, Hao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101823426","display_name":"Yunhe Li","orcid":"https://orcid.org/0000-0002-5656-0766"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Yunhe","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138337007","display_name":"Weizhi Fei","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Fei, Weizhi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127998090","display_name":"Xiangyuan Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Xiangyuan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102808478","display_name":"M. Ruan","orcid":"https://orcid.org/0009-0008-0458-5751"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ruan, Mengzhe","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5039863234","display_name":"Hanxu Hou","orcid":"https://orcid.org/0000-0001-7328-9341"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hou, Hanxu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138302404","display_name":"Peisong Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Peisong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5138329128","display_name":"Linqi Song","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Song, Linqi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5138344810","display_name":"Shuang Qiu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Qiu, Shuang","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/T14347","display_name":"Big Data and Digital Economy","score":0.14669999480247498,"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/T14347","display_name":"Big Data and Digital Economy","score":0.14669999480247498,"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/T10028","display_name":"Topic Modeling","score":0.14069999754428864,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.1257999986410141,"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/cache","display_name":"Cache","score":0.7106000185012817},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.5827000141143799},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.48820000886917114},{"id":"https://openalex.org/keywords/bridge","display_name":"Bridge (graph theory)","score":0.48730000853538513},{"id":"https://openalex.org/keywords/budget-constraint","display_name":"Budget constraint","score":0.4645000100135803},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.44350001215934753},{"id":"https://openalex.org/keywords/decoding-methods","display_name":"Decoding methods","score":0.44029998779296875},{"id":"https://openalex.org/keywords/layer","display_name":"Layer (electronics)","score":0.36719998717308044},{"id":"https://openalex.org/keywords/autoregressive-model","display_name":"Autoregressive model","score":0.3628000020980835}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.817300021648407},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.7106000185012817},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.5827000141143799},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.48820000886917114},{"id":"https://openalex.org/C100776233","wikidata":"https://www.wikidata.org/wiki/Q2532492","display_name":"Bridge (graph theory)","level":2,"score":0.48730000853538513},{"id":"https://openalex.org/C8505890","wikidata":"https://www.wikidata.org/wiki/Q605095","display_name":"Budget constraint","level":2,"score":0.4645000100135803},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.44350001215934753},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.44029998779296875},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.36719998717308044},{"id":"https://openalex.org/C159877910","wikidata":"https://www.wikidata.org/wiki/Q2202883","display_name":"Autoregressive model","level":2,"score":0.3628000020980835},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.35899999737739563},{"id":"https://openalex.org/C206729178","wikidata":"https://www.wikidata.org/wiki/Q2271896","display_name":"Scheduling (production processes)","level":2,"score":0.35420000553131104},{"id":"https://openalex.org/C189783530","wikidata":"https://www.wikidata.org/wiki/Q352090","display_name":"CPU cache","level":3,"score":0.34769999980926514},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.3449999988079071},{"id":"https://openalex.org/C184337299","wikidata":"https://www.wikidata.org/wiki/Q1437428","display_name":"Semantics (computer science)","level":2,"score":0.328000009059906},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.32510000467300415},{"id":"https://openalex.org/C38556500","wikidata":"https://www.wikidata.org/wiki/Q13404475","display_name":"Cache algorithms","level":4,"score":0.323199987411499},{"id":"https://openalex.org/C2986053828","wikidata":"https://www.wikidata.org/wiki/Q355217","display_name":"Time budget","level":2,"score":0.31940001249313354},{"id":"https://openalex.org/C12725497","wikidata":"https://www.wikidata.org/wiki/Q810247","display_name":"Baseline (sea)","level":2,"score":0.31769999861717224},{"id":"https://openalex.org/C29202148","wikidata":"https://www.wikidata.org/wiki/Q287260","display_name":"Resource allocation","level":2,"score":0.28529998660087585},{"id":"https://openalex.org/C112313634","wikidata":"https://www.wikidata.org/wiki/Q7886648","display_name":"Complement (music)","level":5,"score":0.26809999346733093},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.2655999958515167},{"id":"https://openalex.org/C78548338","wikidata":"https://www.wikidata.org/wiki/Q2493","display_name":"Data compression","level":2,"score":0.2653000056743622},{"id":"https://openalex.org/C180016635","wikidata":"https://www.wikidata.org/wiki/Q2712821","display_name":"Compression (physics)","level":2,"score":0.2605000138282776},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.2581999897956848},{"id":"https://openalex.org/C183322885","wikidata":"https://www.wikidata.org/wiki/Q17007702","display_name":"Context model","level":3,"score":0.25209999084472656},{"id":"https://openalex.org/C93682380","wikidata":"https://www.wikidata.org/wiki/Q2025226","display_name":"Static timing analysis","level":2,"score":0.2517000138759613}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.11164","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.11164","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":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2606.11164","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.11164","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":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":{"Long":[0],"chain-of-thought":[1],"(CoT)":[2],"trajectories":[3],"in":[4],"large":[5],"language":[6],"model":[7],"(LLM)":[8],"reasoning":[9,132],"cause":[10],"severe":[11],"inference":[12],"bottlenecks":[13],"due":[14],"to":[15,122],"rapid":[16],"key-value":[17],"(KV)":[18],"cache":[19],"growth.":[20],"Current":[21],"decoding-time":[22,83],"compression":[23,85],"methods":[24,49],"mitigate":[25],"this":[26,73],"issue":[27],"via":[28],"token":[29],"eviction,":[30],"but":[31],"typically":[32],"assume":[33],"a":[34,78,87,153],"uniform":[35],"budget":[36,47,89],"distribution":[37],"across":[38],"all":[39],"layers":[40],"and":[41,59,139,148,175],"heads.":[42],"In":[43],"contrast,":[44],"existing":[45,172],"non-uniform":[46],"allocation":[48,90],"are":[50],"predominantly":[51],"designed":[52],"for":[53],"the":[54,64,160],"static":[55],"prompt":[56],"prefill":[57],"phase,":[58],"they":[60],"do":[61],"not":[62],"capture":[63],"stepwise":[65],"context":[66],"demands":[67],"of":[68],"autoregressive":[69],"reasoning.":[70],"To":[71],"bridge":[72],"gap,":[74],"we":[75,109],"propose":[76],"ReasonAlloc,":[77],"training-free":[79],"framework":[80],"that":[81,142],"recasts":[82],"KV":[84],"as":[86],"hierarchical":[88],"problem.":[91],"ReasonAlloc":[92,143,168],"operates":[93],"at":[94,163],"two":[95],"complementary":[96],"levels:":[97],"an":[98,104,114],"offline":[99],"layer-wise":[100],"preallocation":[101],"strategy":[102,117],"captures":[103],"architecture-driven":[105],"demand":[106],"pattern":[107],"which":[108],"call":[110],"``\\textit{Reasoning":[111],"Wave}'',":[112],"while":[113],"online":[115],"head-wise":[116],"reallocates":[118],"resources":[119],"during":[120],"decoding":[121],"information-rich":[123],"heads":[124],"based":[125],"on":[126,130],"real-time":[127],"utility.":[128],"Evaluations":[129],"mathematical":[131],"benchmarks":[133],"(MATH-500,":[134],"AIME~2024)":[135],"using":[136],"DeepSeek-R1-Distill-Llama-8B,":[137],"DeepSeek-R1-Distill-Qwen-14B,":[138],"AceReason-14B":[140],"show":[141],"outperforms":[144],"uniform-budget":[145],"R-KV,":[146],"SnapKV,":[147],"Pyramid-RKV":[149],"(a":[150],"baseline":[151],"enforcing":[152],"static,":[154],"monotonically":[155],"decreasing":[156],"layer":[157],"budget),":[158],"with":[159,171],"largest":[161],"gains":[162],"small":[164],"budgets":[165],"(128-512":[166],"tokens).":[167],"is":[169],"plug-and-play":[170],"token-eviction":[173],"policies":[174],"introduces":[176],"negligible":[177],"inference-time":[178],"overhead.":[179]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-11T00:00:00"}