跳到正文
arXiv cs.AI· Kyungkeon Chung, Se Un Park, Hyejin Lee, Junyoung Park·· 3 天前AI 评分22

ARMED:自适应检索增强的药物推荐框架

ARMED: Adaptive Retrieval-Augmented Medication Recommendation

AI 导读

ARMED 是一个自适应临床 RAG 框架,可根据检索充分性和当次就诊相关性动态扩展检索,在证据不足时选择性执行补充或回退多查询检索。该框架引入相关性门控,仅激活临床相关的同一患者病史,并将证据组织为结构化的查询级临床证据。在 MIMIC-IV 上的实验表明,ARMED 持续优于此前的静态检索基线,并提升药物推荐质量;它结合本地可运行的小语言模型,无需依赖外部语言模型 API 或任务特定微调。

正文 · 原文

View PDF HTML (experimental)

Abstract:Medication recommendation requires jointly considering diagnoses, comorbidities, longitudinal patient history, and treatment interactions. Although recent retrieval-augmented generation approaches using language models have shown promise for clinical recommendation, most existing systems rely on fixed retrieval pipelines that cannot adapt to diverse clinical retrieval scenarios and may introduce noisy or irrelevant evidence. We propose ARMED, an adaptive clinical RAG framework that dynamically expands retrieval according to retrieval sufficiency and current-admission relevance. ARMED first observes the initial retrieval outcome and selectively performs supplementary or fallback multi-query retrieval when additional evidence is needed. The framework also incorporates a relevance gate that activates only clinically relevant same-patient history and organizes retrieved evidence into structured query-level clinical evidence. Experiments on MIMIC-IV demonstrate that ARMED consistently outperforms previous static retrieval baselines and improves overall medication recommendation quality. By combining locally runnable small language models with adaptive retrieval, ARMED supports medication recommendation without relying on external language model APIs or task-specific fine-tuning.
Comments: Accepted at ICTAI 2026
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL); Information Retrieval (cs.IR)
Cite as: arXiv:2610.16440 [cs.LG]
  (or arXiv:2610.16440v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2610.16440

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Se Un Park [view email]
[v1] Mon, 12 Oct 2026 09:07:30 UTC (325 KB)

来源:arXiv cs.AI · arxiv.org