{"id":"https://openalex.org/W7166651948","doi":"https://doi.org/10.48550/arxiv.2606.29959","title":"Know Before You Fetch: Calibrated Retrieval-Budget Allocation for Retrieval-Augmented Generation","display_name":"Know Before You Fetch: Calibrated Retrieval-Budget Allocation for Retrieval-Augmented Generation","publication_year":2026,"publication_date":"2026-06-29","ids":{"openalex":"https://openalex.org/W7166651948","doi":"https://doi.org/10.48550/arxiv.2606.29959"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.29959","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.29959","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.29959","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5139662471","display_name":"Zhe Dong","orcid":null},"institutions":[{"id":"https://openalex.org/I160136153","display_name":"University of Maine at Presque Isle","ror":"https://ror.org/00nk17n43","country_code":"US","type":"education","lineage":["https://openalex.org/I160136153","https://openalex.org/I2802397601"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Dong, Zhe","raw_affiliation_strings":["University of Maine at Presque Isle"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"University of Maine at Presque Isle","institution_ids":["https://openalex.org/I160136153"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100605395","display_name":"Fang Qin","orcid":"https://orcid.org/0009-0004-7152-1142"},"institutions":[{"id":"https://openalex.org/I97018004","display_name":"Stanford University","ror":"https://ror.org/00f54p054","country_code":"US","type":"education","lineage":["https://openalex.org/I97018004"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Qin, Fang","raw_affiliation_strings":["Stanford University"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Stanford University","institution_ids":["https://openalex.org/I97018004"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5020383228","display_name":"M Kabiruzzaman Shah","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shah, Manish","raw_affiliation_strings":["Independent Researcher"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Independent Researcher","institution_ids":[]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5139643024","display_name":"Yicheng Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Yicheng","raw_affiliation_strings":["Independent Researcher"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Independent Researcher","institution_ids":[]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":2,"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/T10286","display_name":"Information Retrieval and Search Behavior","score":0.9018999934196472,"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/T10286","display_name":"Information Retrieval and Search Behavior","score":0.9018999934196472,"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.03999999910593033,"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/T10918","display_name":"Memory Processes and Influences","score":0.007600000128149986,"subfield":{"id":"https://openalex.org/subfields/2805","display_name":"Cognitive Neuroscience"},"field":{"id":"https://openalex.org/fields/28","display_name":"Neuroscience"},"domain":{"id":"https://openalex.org/domains/1","display_name":"Life Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.6715999841690063},{"id":"https://openalex.org/keywords/calibration","display_name":"Calibration","score":0.5823000073432922},{"id":"https://openalex.org/keywords/sequence","display_name":"Sequence (biology)","score":0.5060999989509583},{"id":"https://openalex.org/keywords/binary-number","display_name":"Binary number","score":0.4758000075817108},{"id":"https://openalex.org/keywords/task","display_name":"Task (project management)","score":0.4496999979019165},{"id":"https://openalex.org/keywords/interface","display_name":"Interface (matter)","score":0.4334000051021576},{"id":"https://openalex.org/keywords/probability-distribution","display_name":"Probability distribution","score":0.3815999925136566},{"id":"https://openalex.org/keywords/latency","display_name":"Latency (audio)","score":0.38019999861717224}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7013999819755554},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.6715999841690063},{"id":"https://openalex.org/C165838908","wikidata":"https://www.wikidata.org/wiki/Q736777","display_name":"Calibration","level":2,"score":0.5823000073432922},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.5060999989509583},{"id":"https://openalex.org/C48372109","wikidata":"https://www.wikidata.org/wiki/Q3913","display_name":"Binary number","level":2,"score":0.4758000075817108},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.4496999979019165},{"id":"https://openalex.org/C113843644","wikidata":"https://www.wikidata.org/wiki/Q901882","display_name":"Interface (matter)","level":4,"score":0.4334000051021576},{"id":"https://openalex.org/C149441793","wikidata":"https://www.wikidata.org/wiki/Q200726","display_name":"Probability distribution","level":2,"score":0.3815999925136566},{"id":"https://openalex.org/C82876162","wikidata":"https://www.wikidata.org/wiki/Q17096504","display_name":"Latency (audio)","level":2,"score":0.38019999861717224},{"id":"https://openalex.org/C141603448","wikidata":"https://www.wikidata.org/wiki/Q134830","display_name":"Prefix","level":2,"score":0.3677000105381012},{"id":"https://openalex.org/C194544171","wikidata":"https://www.wikidata.org/wiki/Q21105679","display_name":"Gating","level":2,"score":0.3653999865055084},{"id":"https://openalex.org/C2779530757","wikidata":"https://www.wikidata.org/wiki/Q1207505","display_name":"Quality (philosophy)","level":2,"score":0.36329999566078186},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.3582000136375427},{"id":"https://openalex.org/C2779843651","wikidata":"https://www.wikidata.org/wiki/Q7390335","display_name":"SIGNAL (programming language)","level":2,"score":0.3447999954223633},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3257000148296356},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.3140000104904175},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.305400013923645},{"id":"https://openalex.org/C23123220","wikidata":"https://www.wikidata.org/wiki/Q816826","display_name":"Information retrieval","level":1,"score":0.2847000062465668},{"id":"https://openalex.org/C41661131","wikidata":"https://www.wikidata.org/wiki/Q220764","display_name":"Interrupt","level":3,"score":0.27489998936653137},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.26980000734329224},{"id":"https://openalex.org/C2164484","wikidata":"https://www.wikidata.org/wiki/Q5170150","display_name":"Core (optical fiber)","level":2,"score":0.26570001244544983},{"id":"https://openalex.org/C2778067643","wikidata":"https://www.wikidata.org/wiki/Q166507","display_name":"Interval (graph theory)","level":2,"score":0.26179999113082886},{"id":"https://openalex.org/C53833338","wikidata":"https://www.wikidata.org/wiki/Q1061424","display_name":"Context switch","level":2,"score":0.2578999996185303}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.29959","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.29959","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.29959","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.29959","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":{"Retrieval-augmented":[0],"generation":[1],"(RAG)":[2],"typically":[3],"retrieves":[4],"a":[5,46,54,59,68,73,181,197,225,237],"fixed":[6],"number":[7],"of":[8,88,228],"passages":[9,33],"for":[10,94,130,160,240],"every":[11],"query.":[12],"This":[13],"is":[14,67,179,204,233],"wasteful":[15],"when":[16,28],"the":[17,21,35],"reader":[18],"already":[19],"knows":[20],"answer,":[22],"and":[23,82,100,112,119,146,157,164,246],"it":[24,208],"can":[25],"be":[26],"harmful":[27],"irrelevant":[29],"or":[30,63],"partially":[31],"relevant":[32],"distract":[34],"reader.":[36],"We":[37,78],"formulate":[38],"adaptive":[39,229],"RAG":[40],"as":[41,236],"calibrated":[42,231],"retrieval-budget":[43],"allocation:":[44],"given":[45],"query,":[47],"decide":[48],"whether":[49],"to":[50,137,142,148],"answer":[51],"closed-book,":[52],"retrieve":[53,58],"compact":[55],"context":[56,61,96],"(k=1),":[57],"full":[60],"(k=5),":[62],"abstain.":[64],"The":[65],"contribution":[66],"probability":[69,127],"interface":[70,239],"rather":[71],"than":[72],"new":[74],"raw":[75],"uncertainty":[76,84],"signal.":[77],"calibrate":[79],"sequence":[80,131],"log-probability":[81],"prefix-logit":[83],"signals":[85],"into":[86],"probabilities":[87,93],"correctness,":[89],"then":[90],"use":[91],"these":[92],"graded":[95],"selection,":[97],"selective":[98],"abstention,":[99],"explicit":[101],"latency/token":[102],"trade-offs.":[103],"Across":[104],"core":[105],"QA":[106],"experiments":[107,187],"on":[108,139,144,150,214,220],"TriviaQA,":[109,140],"Natural":[110],"Questions,":[111],"MS":[113,151],"MARCO,":[114],"with":[115,174],"auxiliary":[116],"PopQA":[117],"motivation":[118],"Qwen/Llama":[120],"family":[121],"checks,":[122],"diagnostic":[123],"out-of-fold":[124],"calibration":[125],"improves":[126,155],"quality":[128],"dramatically:":[129],"log-probability,":[132],"ECE":[133],"drops":[134],"from":[135],"0.275":[136],"0.062":[138],"0.643":[141],"0.009":[143],"NQ,":[145],"0.711":[147],"0.031":[149],"MARCO.":[152],"Graded":[153],"retrieval":[154,182,242],"full-context":[156],"passage-budget":[158],"frontiers":[159],"both":[161],"our":[162],"signal":[163],"TARG-style":[165],"prefix":[166],"entropy/margin,":[167],"while":[168],"retrieval-call":[169],"AUC":[170],"remains":[171],"essentially":[172],"tied":[173],"binary":[175],"gating":[176,203],"because":[177],"k=1":[178],"still":[180],"call.":[183],"Held-out":[184],"train/validation/test":[185],"threshold":[186],"report":[188],"deployable":[189],"operating":[190,195],"points.":[191],"At":[192],"matched-accuracy":[193],"frontier":[194],"points,":[196],"measured":[198],"cost":[199],"model":[200],"reveals":[201],"that":[202],"not":[205],"universally":[206],"faster:":[207],"increases":[209],"latency":[210],"by":[211],"about":[212,218],"27%":[213],"Qwen3-8B":[215],"but":[216],"saves":[217],"8%":[219],"Qwen3-32B.":[221],"These":[222],"results":[223],"support":[224],"nuanced":[226],"view":[227],"RAG:":[230],"confidence":[232],"best":[234],"understood":[235],"reusable":[238],"allocating":[241],"budget":[243],"under":[244],"task":[245],"system":[247],"constraints.":[248]},"counts_by_year":[],"updated_date":"2026-07-01T06:29:00.853634","created_date":"2026-07-01T00:00:00"}