跳到正文
arXiv:cs.AI· Kyungkeon Chung, Se Un Park, Hyejin Lee, Junyoung Park·· 22 小时前

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

ARMED: Adaptive Retrieval-Augmented Medication Recommendation

AI 导读

研究者提出 ARMED,一个自适应临床 RAG 框架,可根据检索充分性和当前就诊相关性动态扩展检索,仅在需要时执行补充或多查询回退检索,并通过相关性门控只激活临床相关的同患者历史。在 MIMIC-IV 上,ARMED 持续优于此前的静态检索基线,提升了药物推荐质量。该框架结合本地可运行的小语言模型,无需依赖外部语言模型 API 或任务特定微调,已被 ICTAI 2026 接收。

正文

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