2025-11-16T09:46:12.577001

ICA-RAG: Information Completeness Guided Adaptive Retrieval-Augmented Generation for Disease Diagnosis

He, Jia, Jia et al.

Retrieval-Augmented Large Language Models (LLMs), which integrate external knowledge, have shown remarkable performance in medical domains, including clinical diagnosis. However, existing RAG methods often struggle to tailor retrieval strategies to diagnostic difficulty and input sample informativeness. This limitation leads to excessive and often unnecessary retrieval, impairing computational efficiency and increasing the risk of introducing noise that can degrade diagnostic accuracy. To address this, we propose ICA-RAG (\textbf{I}nformation \textbf{C}ompleteness Guided \textbf{A}daptive \textbf{R}etrieval-\textbf{A}ugmented \textbf{G}eneration), a novel framework for enhancing RAG reliability in disease diagnosis. ICA-RAG utilizes an adaptive control module to assess the necessity of retrieval based on the input's information completeness. By optimizing retrieval and incorporating knowledge filtering, ICA-RAG better aligns retrieval operations with clinical requirements. Experiments on three Chinese electronic medical record datasets demonstrate that ICA-RAG significantly outperforms baseline methods, highlighting its effectiveness in clinical diagnosis.

academic

ICA-RAG: रोग निदान के लिए सूचना पूर्णता निर्देशित अनुकूली पुनर्प्राप्ति-संवर्धित पीढ़ी

मूल जानकारी

पेपर ID: 2502.14614
शीर्षक: ICA-RAG: Information Completeness Guided Adaptive Retrieval-Augmented Generation for Disease Diagnosis
लेखक: Jiawei He, Mingyi Jia, Zhihao Jia, Junwen Duan, Yan Song, Jianxin Wang
वर्गीकरण: cs.CL (कम्प्यूटेशन और भाषा)
प्रकाशन समय: arXiv प्रीप्रिंट (25 अक्टूबर 2025 का नवीनतम संस्करण)
पेपर लिंक: https://arxiv.org/abs/2502.14614

सारांश

पुनर्प्राप्ति-संवर्धित बड़े भाषा मॉडल (RAG-LLMs) बाहरी ज्ञान को एकीकृत करके चिकित्सा क्षेत्र में असाधारण प्रदर्शन प्रदर्शित करते हैं, विशेष रूप से नैदानिक निदान में। हालांकि, मौजूदा RAG विधियां निदान की कठिनाई और इनपुट नमूने की सूचना मात्रा के अनुसार पुनर्प्राप्ति रणनीति को अनुकूलित करने में विफल रहती हैं, जिससे अत्यधिक और अनावश्यक पुनर्प्राप्ति होती है, जो कम्प्यूटेशनल दक्षता को नुकसान पहुंचाती है और शोर परिचय का जोखिम बढ़ाती है, जिससे निदान सटीकता में कमी आती है। इस समस्या को हल करने के लिए, यह पेपर ICA-RAG (सूचना पूर्णता निर्देशित अनुकूली पुनर्प्राप्ति-संवर्धित पीढ़ी) प्रस्तावित करता है, जो रोग निदान में RAG की विश्वसनीयता को बढ़ाने के लिए एक नया ढांचा है। ICA-RAG इनपुट की सूचना पूर्णता के आधार पर पुनर्प्राप्ति की आवश्यकता का मूल्यांकन करने के लिए अनुकूली नियंत्रण मॉड्यूल का उपयोग करता है, पुनर्प्राप्ति और ज्ञान फ़िल्टरिंग को अनुकूलित करके, पुनर्प्राप्ति संचालन को नैदानिक आवश्यकताओं के साथ बेहतर ढंग से संरेखित करता है। तीन चीनी इलेक्ट्रॉनिक चिकित्सा रिकॉर्ड डेटासेट पर प्रयोग दर्शाते हैं कि ICA-RAG आधारभूत विधियों से काफी बेहतर है, जो नैदानिक निदान में इसकी प्रभावशीलता को उजागर करता है।

अनुसंधान पृष्ठभूमि और प्रेरणा

समस्या की पृष्ठभूमि

बड़े भाषा मॉडल चिकित्सा कार्यों में दो प्रमुख चुनौतियों का सामना करते हैं:

भ्रम समस्या: ऐसी जानकारी उत्पन्न करना जो उचित लगे लेकिन वास्तव में गलत हो
ज्ञान अद्यतन लागत: चिकित्सा ज्ञान को वर्तमान रखने की संसाधन-गहन प्रकृति

मौजूदा RAG विधियों की सीमाएं

चयनात्मक पुनर्प्राप्ति तर्क की कमी: सभी प्रश्नों के लिए भेदभाव रहित पुनर्प्राप्ति, कम्प्यूटेशनल और समय लागत में वृद्धि
निम्न गुणवत्ता की पुनर्प्राप्ति का परिचय: असंबंधित जानकारी के माध्यम से प्रदर्शन को कम करने की संभावना
चिकित्सा क्षेत्र की विशेषता: कई सामान्य रोग या हल्के लक्षण वाले मामले स्पष्ट निदान के साथ पुनर्प्राप्ति के बिना सटीक निदान के लिए

मौजूदा अनुकूली RAG की कमियां

LLM आउटपुट वितरण पर आधारित विधियां: LLM अत्यधिक आत्मविश्वास प्रदर्शित करते हैं, संबंधित ज्ञान की कमी के बावजूद उच्च आत्मविश्वास वितरण उत्पन्न करते हैं
वर्गीकरण मॉडल पर आधारित विधियां: चिकित्सा क्षेत्र में, इनपुट पाठ आमतौर पर स्पष्ट संरचनात्मक पैटर्न नहीं रखते हैं, छोटे भाषा मॉडल प्रश्न कठिनाई को समझने में विफल रहते हैं

मुख्य योगदान

ICA-RAG ढांचा प्रस्तावित करना: एक अनुकूली पुनर्प्राप्ति-संवर्धित रोग निदान ढांचा जिसमें बैकबोन LLM को ट्यून करने की आवश्यकता नहीं है
नवीन डेटा एनोटेशन विधि: मास्किंग संचालन पर आधारित एनोटेशन रणनीति डिजाइन करना, LLM की विभिन्न प्रतिक्रियाओं को प्रेरित करके लेबल जानकारी प्राप्त करना
पुनर्प्राप्ति प्रक्रिया को अनुकूलित करना: जटिल संदर्भ वाले नैदानिक परिदृश्यों के लिए पुनर्प्राप्ति प्रवाह को अनुकूलित करना
प्रायोगिक सत्यापन: तीन चीनी EMR डेटासेट पर व्यापक प्रयोग, ढांचे की प्रभावशीलता को साबित करना

विधि विवरण

कार्य परिभाषा

प्रत्यक्ष रोग निदान: इनपुट पाठ का प्रतिनिधित्व करने वाले टोकन अनुक्रम $x = [x_1, x_2, ..., x_n]$ दिया गया है, LLM पाठ पीढ़ी को निम्नानुसार औपचारिक किया जा सकता है: $\hat{D} = \text{LLM}(Q, \text{prompt})$

RAG रोग निदान: बाहरी ज्ञान स्रोत से संबंधित ज्ञान पुनर्प्राप्त करना और एकीकृत करना: $\hat{D} = \text{LLM}(Q, d, \text{prompt})$ जहां $d = \text{Retriever}(K, Q)$

अनुकूली RAG रोग निदान: इनपुट Q का मूल्यांकन करने के लिए नियंत्रण फ़ंक्शन F का परिचय: