TAC-ECG: A task-adaptive classification method for electrocardiogram based on cross-modal contrastive learning and low-rank convolutional adapter

Abstract

Background and Objective:

Cardiovascular diseases are one of the major health threats to humans. Researchers have proposed numerous deep learning-based methods for the automatic analysis of electrocardiogram (ECG), achieving encouraging results. However, many existing methods are limited to task-specific model training and require retraining or full fine-tuning when confronted with a new ECG classification task, thus lacking flexibility in clinical applications.

Methods:

In this study, we propose a Task-Adaptive Classification method for ECG (TAC-ECG) based on cross-modal contrastive learning and low-rank convolutional adapters. TAC-ECG comprises two main phases. In the first phase, inspired by the Contrastive Language-Image Pre-training, we design the Contrastive ECG-Text Pre-training (CETP) to pre-train a robust ECG encoder. In the second phase, the pre-trained ECG encoder is frozen and integrated with a lightweight plug-in, the Low-Rank Convolutional Adapter (LRC-Adapter), forming an extensible ECG classification model. The frozen encoder extracts more discriminative features from the ECG signal, while the LRC-Adapter enables task-specific adaptation. For diverse ECG classification tasks, TAC-ECG only requires training the LRC-Adapter. This mechanism enables TAC-ECG to efficiently perform different ECG classification tasks, significantly reducing resource consumption and deployment costs in multi-tasking scenarios compared to traditional fully fine-tuned methods.

Results:

We conducted extensive experiments using six different network architectures as ECG encoders. Specifically, we performed ECG classification experiments on four datasets: CPSC2018, Cinc2017, PTB-XL, and Chapman, targeting 9-category, 3-category, 5-category, and 4-category classifications respectively. The TAC-ECG achieved highly competitive results using only approximately 3% of the trainable parameters and approximately 25% of the total parameters compared to the fully fine-tuned method. These results demonstrates the effectiveness and practicality of the TAC-ECG method.

Conclusion:

The TAC-ECG offers a flexible and efficient method for ECG classification, enabling rapid adaptation to diverse tasks and enhancing clinical diagnostic practicality.