An Adaptive Multi-Source Retrieval-Augmented Generation Framework Integrating Query Complexity Awareness and Confidence-Aware Fusion

Retrieval-Augmented Generation (RAG) has been observed to encounter challenges in heterogeneous query scenarios characterised by varying evidence requirements and reasoning depths. In order to address this limitation, the present paper puts forward a proposal for an Adaptive Multi-Source RAG framework (AMSRAG) that integrates query complexity awareness with confidence-aware fusion. The framework performs query complexity classification with a pretrained language model, calibrates the classification confidence to guide the dynamic scheduling of retrieval paths and the adjustment of fusion weights, and enables a controllable balance between answer quality and retrieval efficiency through hierarchical path selection and cross-source weighting. The experiments conducted on multiple open-domain question-answering datasets demonstrate that the query complexity classifier achieves an accuracy of 85.9% and a Macro-F1 score of 85.4%. These outcomes indicate the potential for the classifier to generate a reliable decision signal, which can subsequently be utilised to guide the process of adaptive retrieval and fusion. The proposed framework demonstrates a marked improvement in terms of both answer accuracy and retrieval relevance when compared to the fixed-pipeline RAG. In scenarios involving high-confidence queries, the system has been shown to effectively avoid redundant retrieval, thereby reducing the average number of retrievals. In instances of low-confidence complex queries, the system has been shown to enhance evidence coverage and completeness of answers through multi-source retrieval and confidence-weighted fusion. This study proposes a novel methodology for enhancing the adaptability and resource efficiency of RAG systems in response to heterogeneous query conditions.

Authors

Institutions

Publication Details

Journal
Applied Sciences
Published
2026-03-05
DOI
https://doi.org/10.3390/app16052495
Primary Topic
Information Retrieval and Search Behavior
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

An Adaptive Multi-Source Retrieval-Augmented Generation Framework Integrating Query Complexity Awareness and Confidence-Aware Fusion

Dong Wei, Meiqi Yang, Mingguang Diao
Applied Sciences
Information Retrieval and Search Behavior
article

An Adaptive Multi-Source Retrieval-Augmented Generation Framework Integrating Query Complexity Awareness and Confidence-Aware Fusion

Dong Wei, Meiqi Yang, Mingguang Diao
article en

Abstract

Retrieval-Augmented Generation (RAG) has been observed to encounter challenges in heterogeneous query scenarios characterised by varying evidence requirements and reasoning depths. In order to address this limitation, the present paper puts forward a proposal for an Adaptive Multi-Source RAG framework (AMSRAG) that integrates query complexity awareness with confidence-aware fusion. The framework performs query complexity classification with a pretrained language model, calibrates the classification confidence to guide the dynamic scheduling of retrieval paths and the adjustment of fusion weights, and enables a controllable balance between answer quality and retrieval efficiency through hierarchical path selection and cross-source weighting. The experiments conducted on multiple open-domain question-answering datasets demonstrate that the query complexity classifier achieves an accuracy of 85.9% and a Macro-F1 score of 85.4%. These outcomes indicate the potential for the classifier to generate a reliable decision signal, which can subsequently be utilised to guide the process of adaptive retrieval and fusion. The proposed framework demonstrates a marked improvement in terms of both answer accuracy and retrieval relevance when compared to the fixed-pipeline RAG. In scenarios involving high-confidence queries, the system has been shown to effectively avoid redundant retrieval, thereby reducing the average number of retrievals. In instances of low-confidence complex queries, the system has been shown to enhance evidence coverage and completeness of answers through multi-source retrieval and confidence-weighted fusion. This study proposes a novel methodology for enhancing the adaptability and resource efficiency of RAG systems in response to heterogeneous query conditions.

Applied SciencesVol. 16(5)
China University of Geosciences (Beijing) (CN), Beijing Academy of Artificial Intelligence (CN)
China University of Geosciences, Beijing, China University of Geosciences, National College Students Innovation and Entrepreneurship Training Program, Division of Undergraduate Education
Decent work and economic growth
Openalex Percentile: Top 53%
Information Retrieval and Search Behavior
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.