用自监督模型+大模型,让心电图自动生成报告并回答问题。
Electrocardiogram Report Generation and Question Answering via Retrieval-Augmented Self-Supervised Modeling
- 基于自监督学习的编码器,实现心电图与报告的高效匹配检索。
- 在PTB-XL和MIMIC-IV-ECG数据集上生成报告效果优于现有方法。
- 零样本问答性能媲美有监督模型,适合临床快速辅助决策。
心电图解读与报告生成在心脏病学中仍具挑战性,需专业知识且耗时。为解决此问题,我们提出ECG-ReGen,一种基于检索的心电图转文本报告生成与问答方法。该方法采用自监督学习构建心电图编码器,支持高效相似性搜索与报告检索。通过预训练结合动态检索及大语言模型(LLM)精炼,有效分析心电图数据并回答相关问题。在PTB-XL与MIMIC-IV-ECG数据集上的实验表明,该方法在域内与跨域场景下均表现优异。此外,在ECG-QA数据集上,仅使用现成大模型进行零样本问答,性能已可媲美全监督方法。该方法融合自监督编码器与大语言模型,为精准心电图解读提供了可扩展、高效的解决方案,显著提升临床决策支持能力。
原文摘要 · Abstract (English)
Interpreting electrocardiograms (ECGs) and generating comprehensive reports remain challenging tasks in cardiology, often requiring specialized expertise and significant time investment. To address these critical issues, we propose ECG-ReGen, a retrieval-based approach for ECG-to-text report generation and question answering. Our method leverages a self-supervised learning for the ECG encoder, enabling efficient similarity searches and report retrieval. By combining pre-training with dynamic retrieval and Large Language Model (LLM)-based refinement, ECG-ReGen effectively analyzes ECG data and answers related queries, with the potential of improving patient care. Experiments conducted on the PTB-XL and MIMIC-IV-ECG datasets demonstrate superior performance in both in-domain and cross-domain scenarios for report generation. Furthermore, our approach exhibits competitive performance on ECG-QA dataset compared to fully supervised methods when utilizing off-the-shelf LLMs for zero-shot question answering. This approach, effectively combining self-supervised encoder and LLMs, offers a scalable and efficient solution for accurate ECG interpretation, holding significant potential to enhance clinical decision-making.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。