{"id":"https://openalex.org/W7162567160","doi":"https://doi.org/10.48550/arxiv.2605.26849","title":"Uncertainty-Aware Budget Allocation for Adaptive Test-Time Reasoning","display_name":"Uncertainty-Aware Budget Allocation for Adaptive Test-Time Reasoning","publication_year":2026,"publication_date":"2026-05-26","ids":{"openalex":"https://openalex.org/W7162567160","doi":"https://doi.org/10.48550/arxiv.2605.26849"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.26849","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26849","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.2605.26849","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137134195","display_name":"Manh Nguyen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nguyen, Manh","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137112851","display_name":"Sunil Gupta","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gupta, Sunil","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137123171","display_name":"Hung Le","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Le, Hung","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/T10028","display_name":"Topic Modeling","score":0.7414000034332275,"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/T10028","display_name":"Topic Modeling","score":0.7414000034332275,"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.06790000200271606,"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.025499999523162842,"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/sampling","display_name":"Sampling (signal processing)","score":0.6491000056266785},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.5981000065803528},{"id":"https://openalex.org/keywords/budget-constraint","display_name":"Budget constraint","score":0.5925999879837036},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.5009999871253967},{"id":"https://openalex.org/keywords/integer","display_name":"Integer (computer science)","score":0.40950000286102295},{"id":"https://openalex.org/keywords/preference","display_name":"Preference","score":0.3626999855041504},{"id":"https://openalex.org/keywords/adaptive-sampling","display_name":"Adaptive sampling","score":0.353300005197525},{"id":"https://openalex.org/keywords/importance-sampling","display_name":"Importance sampling","score":0.34709998965263367}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6802999973297119},{"id":"https://openalex.org/C140779682","wikidata":"https://www.wikidata.org/wiki/Q210868","display_name":"Sampling (signal processing)","level":3,"score":0.6491000056266785},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.5981000065803528},{"id":"https://openalex.org/C8505890","wikidata":"https://www.wikidata.org/wiki/Q605095","display_name":"Budget constraint","level":2,"score":0.5925999879837036},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.5192000269889832},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.5009999871253967},{"id":"https://openalex.org/C97137487","wikidata":"https://www.wikidata.org/wiki/Q729138","display_name":"Integer (computer science)","level":2,"score":0.40950000286102295},{"id":"https://openalex.org/C2781249084","wikidata":"https://www.wikidata.org/wiki/Q908656","display_name":"Preference","level":2,"score":0.3626999855041504},{"id":"https://openalex.org/C2781395549","wikidata":"https://www.wikidata.org/wiki/Q4680762","display_name":"Adaptive sampling","level":3,"score":0.353300005197525},{"id":"https://openalex.org/C52740198","wikidata":"https://www.wikidata.org/wiki/Q1539564","display_name":"Importance sampling","level":3,"score":0.34709998965263367},{"id":"https://openalex.org/C2986053828","wikidata":"https://www.wikidata.org/wiki/Q355217","display_name":"Time budget","level":2,"score":0.33230000734329224},{"id":"https://openalex.org/C56086750","wikidata":"https://www.wikidata.org/wiki/Q6042592","display_name":"Integer programming","level":2,"score":0.3255999982357025},{"id":"https://openalex.org/C29202148","wikidata":"https://www.wikidata.org/wiki/Q287260","display_name":"Resource allocation","level":2,"score":0.3163999915122986},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.2921000123023987},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.289900004863739},{"id":"https://openalex.org/C44280652","wikidata":"https://www.wikidata.org/wiki/Q104837","display_name":"Phase (matter)","level":2,"score":0.27869999408721924},{"id":"https://openalex.org/C198531522","wikidata":"https://www.wikidata.org/wiki/Q485146","display_name":"Sample (material)","level":2,"score":0.2750000059604645},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.2689000070095062},{"id":"https://openalex.org/C73602740","wikidata":"https://www.wikidata.org/wiki/Q7795822","display_name":"Thompson sampling","level":3,"score":0.26350000500679016}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.26849","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26849","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.2605.26849","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26849","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":{"Sampling":[0],"multiple":[1],"responses":[2],"improves":[3],"language":[4],"model":[5,161],"reasoning,":[6],"but":[7],"uniform":[8],"compute":[9],"allocation":[10],"is":[11,86,167],"inefficient:":[12],"easy":[13],"questions":[14,19,100,107],"are":[15],"over-sampled":[16],"while":[17,72,105],"hard":[18],"remain":[20],"under-explored.":[21],"We":[22],"propose":[23],"Uncertainty-Aware":[24],"Budget":[25],"Allocation":[26],"(UAB),":[27],"a":[28,35,69,89,94],"concave":[29,95],"integer":[30],"optimization":[31],"framework":[32],"that":[33,92],"reallocates":[34],"fixed":[36],"sampling":[37,103],"budget":[38,85,104],"based":[39],"on":[40,113,148],"per-question":[41],"uncertainty":[42],"estimated":[43],"at":[44,170],"no":[45,159],"additional":[46,110,163],"inference":[47],"cost.":[48],"In":[49,80],"Phase":[50,81],"1,":[51],"every":[52],"question":[53],"receives":[54],"one":[55],"generation;":[56],"its":[57],"average":[58,142],"negative":[59],"log-likelihood":[60],"(ANLL),":[61],"extracted":[62],"directly":[63],"from":[64],"output":[65],"log-probabilities,":[66],"serves":[67],"as":[68],"difficulty":[70],"signal":[71],"the":[73,77,83,152],"generation":[74],"contributes":[75],"to":[76,121,139,146],"final":[78],"vote.":[79],"2,":[82],"remaining":[84],"allocated":[87],"by":[88,137],"marginal-greedy":[90],"algorithm":[91],"solves":[93],"coverage-maximization":[96],"surrogate":[97],"exactly:":[98],"uncertain":[99],"receive":[101,108],"more":[102],"confident":[106],"fewer":[109],"samples.":[111],"Evaluated":[112],"six":[114],"open-weight":[115],"and":[116,124,131,144],"black-box":[117],"models":[118],"spanning":[119],"1.5B":[120],"27B":[122],"parameters":[123],"five":[125],"reasoning":[126],"benchmarks":[127],"covering":[128],"math,":[129],"logic,":[130],"preference":[132],"tasks,":[133],"UAB":[134],"outperforms":[135],"baselines":[136],"up":[138,145],"+3%":[140],"in":[141,155],"accuracy":[143],"+5%":[147],"individual":[149],"benchmarks,":[150],"with":[151],"largest":[153],"gains":[154],"low-resource":[156],"settings,":[157],"requiring":[158],"auxiliary":[160],"or":[162],"LLM":[164],"call.":[165],"Code":[166],"publicly":[168],"available":[169],"https://github.com/manhitv/UAB.":[171]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-28T00:00:00"}