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: الاسترجاع المعزز التكيفي الموجه بإكمال المعلومات لتشخيص الأمراض

المعلومات الأساسية

معرّف الورقة: 2502.14614
العنوان: ICA-RAG: الاسترجاع المعزز التكيفي الموجه بإكمال المعلومات لتشخيص الأمراض
المؤلفون: Jiawei He, Mingyi Jia, Zhihao Jia, Junwen Duan, Yan Song, Jianxin Wang
التصنيف: cs.CL (الحوسبة واللغة)
وقت النشر: طبعة arXiv (النسخة الأحدث في 15 أكتوبر 2025)
رابط الورقة: https://arxiv.org/abs/2502.14614

الملخص

تُظهر نماذج اللغة الكبيرة المعززة بالاسترجاع (RAG-LLMs) أداءً متفوقاً في المجال الطبي من خلال دمج المعرفة الخارجية، خاصة في التشخيص السريري. ومع ذلك، تواجه طرق RAG الحالية صعوبة في تخصيص استراتيجيات الاسترجاع وفقاً لصعوبة التشخيص وكمية المعلومات في العينات المدخلة، مما يؤدي إلى استرجاع مفرط وغير ضروري يضر بالكفاءة الحسابية ويزيد من خطر إدخال الضوضاء، وبالتالي يقلل من دقة التشخيص. لمعالجة هذه المشكلة، نقترح ICA-RAG (الاسترجاع المعزز التكيفي الموجه بإكمال المعلومات)، وهو إطار عمل جديد لتعزيز موثوقية RAG في تشخيص الأمراض. يستفيد ICA-RAG من وحدة تحكم تكيفية لتقييم ضرورة الاسترجاع بناءً على إكمال المعلومات للمدخلات، من خلال تحسين الاسترجاع وتصفية المعرفة، ما يحقق محاذاة أفضل بين عمليات الاسترجاع والاحتياجات السريرية. تُظهر التجارب على ثلاث مجموعات بيانات للسجلات الطبية الإلكترونية الصينية أن ICA-RAG يتفوق بشكل ملحوظ على الطرق الأساسية، مما يبرز فعاليته في التشخيص السريري.

الخلفية البحثية والدافع

خلفية المشكلة

تواجه نماذج اللغة الكبيرة تحديين رئيسيين في المهام الطبية:

مشكلة الهلوسة: توليد معلومات تبدو معقولة لكنها خاطئة فعلياً
تكلفة تحديث المعرفة: الطبيعة كثيفة الموارد للحفاظ على تحديث المعرفة الطبية

حدود طرق RAG الحالية

افتقار المنطق الانتقائي للاسترجاع: تنفيذ الاسترجاع بدون تمييز لجميع الاستعلامات، مما يزيد من التكاليف الحسابية والزمنية
إدخال استرجاعات منخفضة الجودة: قد تقلل الأداء بدلاً من تحسينها من خلال معلومات غير ذات صلة
خصوصية المجال الطبي: العديد من الأمراض الشائعة أو الحالات ذات الأعراض الخفيفة والتشخيص الواضح لا تتطلب استرجاعاً للتشخيص الدقيق

قصور RAG التكيفي الحالي

الطرق المستندة إلى توزيع مخرجات LLM: تميل نماذج اللغة الكبيرة إلى الثقة المفرطة، وتولد توزيعات ثقة عالية حتى في غياب المعرفة ذات الصلة
الطرق المستندة إلى نماذج التصنيف: في المجال الطبي، عادة ما تفتقر النصوص المدخلة إلى أنماط هيكلية واضحة، وتواجه نماذج اللغة الصغيرة صعوبة في فهم صعوبة الإجابة

المساهمات الأساسية

اقتراح إطار عمل ICA-RAG: إطار استرجاع معزز تكيفي لتشخيص الأمراض بدون الحاجة إلى ضبط نموذج LLM الأساسي
طريقة تسمية البيانات المبتكرة: تصميم استراتيجية تسمية قائمة على عمليات الإخفاء، من خلال استحضار استجابات مختلفة من LLM للحصول على معلومات التسمية
تحسين عملية الاسترجاع: تحسين عملية الاسترجاع للسيناريوهات السريرية ذات السياق المعقد
التحقق التجريبي: إجراء تجارب واسعة النطاق على ثلاث مجموعات بيانات للسجلات الطبية الإلكترونية الصينية لإثبات فعالية الإطار

شرح الطريقة

تعريف المهمة

التشخيص المباشر للمرض: بالنظر إلى تسلسل الرموز الذي يمثل النص المدخل $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 التكيفي: إدخال دالة تحكم F لتقييم المدخل Q: