Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Modeling Uncertainty Trends for Timely Retrieval in Dynamic RAG

About

Dynamic retrieval-augmented generation (RAG) allows large language models (LLMs) to fetch external knowledge on demand, offering greater adaptability than static RAG. A central challenge in this setting lies in determining the optimal timing for retrieval. Existing methods often trigger retrieval based on low token-level confidence, which may lead to delayed intervention after errors have already propagated. We introduce Entropy-Trend Constraint (ETC), a training-free method that determines optimal retrieval timing by modeling the dynamics of token-level uncertainty. Specifically, ETC utilizes first- and second-order differences of the entropy sequence to detect emerging uncertainty trends, enabling earlier and more precise retrieval. Experiments on six QA benchmarks with three LLM backbones demonstrate that ETC consistently outperforms strong baselines while reducing retrieval frequency. ETC is particularly effective in domain-specific scenarios, exhibiting robust generalization capabilities. Ablation studies and qualitative analyses further confirm that trend-aware uncertainty modeling yields more effective retrieval timing. The method is plug-and-play, model-agnostic, and readily integrable into existing decoding pipelines. Implementation code is included in the supplementary materials.

Bo Li, Tian Tian, Zhenghua Xu, Hao Cheng, Shikun Zhang, Wei Ye• 2025

Related benchmarks

TaskDatasetResultRank
Multi-hop Question Answering2WikiMultihopQA
EM36
387
Multi-hop Question AnsweringHotpotQA (test)--
255
Multi-hop Question Answering2WikiMultiHopQA (test)
EM31.5
195
Question Answering2WikiMultiHopQA (test)
F140.2
81
Multi-hop Question AnsweringHotpotQA
F142.2
79
Question AnsweringWebQ
EM18.9
27
Question AnsweringPopQA
EM30.5
17
Question AnsweringTriviaQA
EM52.9
17
Question AnsweringPubMedQA
Acc55
6
Showing 9 of 9 rows

Other info

Follow for update