Multi-Task ECG Model Training with Related Abnormality Data
Find Innovative SolutionsGenerate Solutions
Solution Overview
Problem
Deep learning models for electrocardiogram (ECG) classification require large amounts of balanced training data to achieve optimal performance, especially for multi-classification tasks, which can be challenging due to the scarcity of target abnormality data.
Innovation Solution
A multi-task learning method is employed, where a multi-task model is trained using both target and related task models, with a regularized-loss mechanism to adjust parameters and share features, enabling the transfer of knowledge from abundant related abnormality data to improve the recognition of target abnormalities.
Engineering Contradictions & Design Principles
Engineering Contradiction Analysis
1Extent of automation
If deep learning models with multiple hidden layers and large numbers of parameters are used for multi-classification of electrocardio abnormalities, then automatic feature extraction capability is improved, but the amount of training data required increases significantly
Solution Approach 1:
The patent applies preliminary action by pre-training the deep learning model on abundant related abnormality data before fine-tuning on the target abnormality data. This preliminary training on related tasks (e.g., other ECG abnormalities) establishes initial feature extraction capabilities and parameter configurations that can be transferred to the target task, reducing the amount of target-specific training data needed.
Solution Approach 2:
The patent utilizes parameter changes by adjusting model parameters during transfer learning from related tasks to the target task. The model parameters learned from related abnormality classification are modified and adapted through fine-tuning on the target abnormality data, allowing the model to leverage previously learned features while adapting to the specific characteristics of the target abnormality type.
2Adaptability or versatility
If a multi-classification model is trained to recognize multiple types of electrocardio abnormalities, then classification coverage is improved, but the balance and amount of training data required for each class increases
Solution Approach 1:
The patent applies segmentation by dividing the multi-classification problem into multiple single-task classification models, each dedicated to recognizing a specific type of electrocardio abnormality. Instead of training one large multi-class model that requires balanced data across all classes, the approach segments the task into multiple specialized models that can be trained independently on their respective abnormality types, reducing the data balance requirement for each individual model.
Solution Approach 2:
The patent implements universality through a shared feature extraction backbone that serves multiple classification tasks. The deep learning model uses common layers for feature extraction that are shared across different abnormality types, while having separate classification heads for each specific abnormality. This allows the model to achieve multi-classification coverage while training each task independently, reducing the need for perfectly balanced multi-class training data.
Data Source
AI summary
Model training method, signal recognition method, apparatus, computing and processing device, computer program, and computer-readable medium. The model training method comprises: acquiring a training sample set, training sample set includes sample electrocardio-signals and abnormal labels of sample electrocardio-signals, and abnormal labels include a target abnormal labels and at least one related abnormal labels; inputting sample electrocardio-signals into multi-task model, training multi-task model based on a multi-task learning mechanism according to an output of multi-task model and the abnormal labels; multi-task model includes a target task model and at least one related task model, a target output of the target task model is target abnormality labels of inputted sample electrocardio-signals, and a target output of related task model is related abnormal labels of inputted sample electrocardio-signals; determining target task model after trained as target-abnormality-recognition model, and target-abnormality-recognition model is configured for recognizing target abnormality in the electrocardio-signals inputted into target-abnormality-recognition model.


