Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tang, Jialu, Xia, Tong, Lu, Yuan, Mascolo, Cecilia, Saeed, Aaqib
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:https://arxiv.org/abs/2409.08788
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913499241775104
author Tang, Jialu
Xia, Tong
Lu, Yuan
Mascolo, Cecilia
Saeed, Aaqib
author_facet Tang, Jialu
Xia, Tong
Lu, Yuan
Mascolo, Cecilia
Saeed, Aaqib
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2409_08788
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Electrocardiogram Report Generation and Question Answering via Retrieval-Augmented Self-Supervised Modeling
Tang, Jialu
Xia, Tong
Lu, Yuan
Mascolo, Cecilia
Saeed, Aaqib
Machine Learning
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.
title Electrocardiogram Report Generation and Question Answering via Retrieval-Augmented Self-Supervised Modeling
topic Machine Learning
url https://arxiv.org/abs/2409.08788