用分层检索增强生成,让大模型更准确地写出结构化病历。
CLI-RAG: A Retrieval-Augmented Framework for Clinically Structured and Context Aware Text Generation with LLMs
- 分两阶段检索:先定病历类型,再找关键内容段落。
- 生成病历与真实记录对齐率达87.7%,优于医生基准的80.7%。
- 适合临床文本生成、医疗AI系统研发者使用。
大型语言模型在临床文本生成中展现出潜力,但实际应用面临两大挑战:患者数据高度非结构化、异质且分散在多种病历类型中;临床病历通常冗长且语义密集,直接提示难以实现,受限于上下文长度且易遗漏关键信息。本文提出CLI-RAG(临床感知检索增强生成)框架,针对医疗领域设计,支持结构化和临床相关的文本生成。该框架采用新型分层切块策略,尊重临床文档结构,并引入任务特定的双阶段检索机制:全局阶段通过基于证据的查询识别相关病历类型,局部阶段在指定病历内提取高价值内容,实现在文档与章节层面的双重相关性。我们在MIMIC-III数据集的15种临床病历类型上应用该系统,生成单次住院的结构化病程记录。实验表明,生成内容在时间与语义上保持良好对齐,平均对齐得分达87.7%,高于真实医生记录的80.7%基准。生成结果在不同大模型间具高度一致性,体现确定性行为,对可复现性、可靠性及临床信任至关重要。
原文摘要 · Abstract (English)
Large language models (LLMs), including zero-shot and few-shot paradigms, have shown promising capabilities in clinical text generation. However, real-world applications face two key challenges: (1) patient data is highly unstructured, heterogeneous, and scattered across multiple note types and (2) clinical notes are often long and semantically dense, making naive prompting infeasible due to context length constraints and the risk of omitting clinically relevant information. We introduce CLI-RAG (Clinically Informed Retrieval-Augmented Generation), a domain-specific framework for structured and clinically grounded text generation using LLMs. It incorporates a novel hierarchical chunking strategy that respects clinical document structure and introduces a task-specific dual-stage retrieval mechanism. The global stage identifies relevant note types using evidence-based queries, while the local stage extracts high-value content within those notes creating relevance at both document and section levels. We apply the system to generate structured progress notes for individual hospital visits using 15 clinical note types from the MIMIC-III dataset. Experiments show that it preserves temporal and semantic alignment across visits, achieving an average alignment score of 87.7%, surpassing the 80.7% baseline from real clinician-authored notes. The generated outputs also demonstrate high consistency across LLMs, reinforcing deterministic behavior essential for reproducibility, reliability, and clinical trust.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。