arXiv:2511.13293cs.AI2025-11被引 1

用分层智能体让大模型在医疗预测中更准、更懂何时该查资料。

Grounded by Experience: Generative Healthcare Prediction Augmented with Hierarchical Agentic Retrieval

  • 设计双智能体架构,自动判断何时检索外部知识
  • 在三个数据集上显著优于现有方法,提升预测准确率
  • 适合医疗AI研发者和需要高可靠性的临床决策系统

精准的医疗预测对改善患者结果和降低运营成本至关重要。尽管大型语言模型(LLMs)凭借其丰富的参数化知识展现出增强医疗预测的潜力,但其嵌入知识的可靠性与覆盖度有限,易产生事实性错误。虽有检索增强生成(RAG)框架如GraphRAG等试图缓解此问题,但在医疗场景下仍面临两大挑战:一是如何识别临床需求以触发检索机制;二是如何实现检索器与生成器间的协同优化。为此,本文提出GHAR——一种生成式分层智能体检索增强框架,同时解决何时检索及模块间协作优化问题。具体而言,设计双智能体结构:顶层智能体(Agent-Top)模拟主诊医生,迭代判断是否依赖参数知识或启动检索;底层智能体(Agent-Low)作为会诊助手,在触发检索后整合所有相关知识。为应对第二项挑战,创新性地将两智能体优化统一于马尔可夫决策过程,设计多样化奖励函数,使二者共享准确预测目标的同时保持角色分工。在三个基准数据集上的三项任务实验表明,本方法显著优于当前最优基线,验证了分层智能体RAG在推动医疗系统发展中的潜力。

原文摘要 · Abstract (English)

Accurate healthcare prediction is critical for improving patient outcomes and reducing operational costs. Bolstered by growing reasoning capabilities, large language models (LLMs) offer a promising path to enhance healthcare predictions by drawing on their rich parametric knowledge. However, LLMs are prone to factual inaccuracies due to limitations in the reliability and coverage of their embedded knowledge. While retrieval-augmented generation (RAG) frameworks, such as GraphRAG and its variants, have been proposed to mitigate these issues by incorporating external knowledge, they face two key challenges in the healthcare scenario: (1) identifying the clinical necessity to activate the retrieval mechanism, and (2) achieving synergy between the retriever and the generator to craft contextually appropriate retrievals. To address these challenges, we propose GHAR, a \underline{g}enerative \underline{h}ierarchical \underline{a}gentic \underline{R}AG framework that simultaneously resolves when to retrieve and how to optimize the collaboration between submodules in healthcare. Specifically, for the first challenge, we design a dual-agent architecture comprising Agent-Top and Agent-Low. Agent-Top acts as the primary physician, iteratively deciding whether to rely on parametric knowledge or to initiate retrieval, while Agent-Low acts as the consulting service, summarising all task-relevant knowledge once retrieval was triggered. To tackle the second challenge, we innovatively unify the optimization of both agents within a formal Markov Decision Process, designing diverse rewards to align their shared goal of accurate prediction while preserving their distinct roles. Extensive experiments on three benchmark datasets across three popular tasks demonstrate our superiority over state-of-the-art baselines, highlighting the potential of hierarchical agentic RAG in advancing healthcare systems.

医疗AI智能体RAG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。