Automated multi-class ECG arrhythmia detection using VMD and multi-task optimization

Y Murali Krishna1, K Padma Vasavi2, M Krishna Chaitanya3

  • 1ECE, QIS College of Engineering and Technology, Vengamukalapalem, Ongole, 523272, Andhra Pradesh, India.

Scientific Reports
|April 1, 2026
PubMed

Insights

This study introduces an advanced framework for classifying cardiac arrhythmias from ECG signals. The optimized feature set significantly improved detection accuracy for conditions like Atrial Fibrillation.

Area of Science:

  • Cardiology
  • Biomedical Engineering
  • Machine Learning

Background:

  • Accurate Electrocardiogram (ECG) classification is crucial for diagnosing heart rhythm disorders.
  • Existing methods may face challenges in distinguishing between various complex arrhythmias.
  • The need for robust and efficient ECG analysis frameworks remains high in clinical practice.

Purpose of the Study:

  • To develop and evaluate a novel multi-class ECG classification framework.
  • To identify four key cardiac conditions: Atrial Fibrillation (AF), Ventricular Fibrillation (VF), Normal Rhythm (NR), and Ventricular Tachycardia (VT).
  • To enhance feature discriminability and reduce computational load through optimized feature selection.

Main Methods:

  • ECG signals were processed using Variational Mode Decomposition (VMD).
  • Higher-order statistics and entropy-based features were extracted from decomposed modes.
  • Multi-task Particle Swarm Optimization (MT-PSO) was utilized for feature selection and reduction.
  • Several machine learning models, including LightGBM, HistGradientBoost, XGBoost, and ExtraTrees, were evaluated.

Main Results:

  • The optimized feature set significantly improved classification performance across all evaluated models.
  • LightGBM achieved the highest accuracy (0.993), followed closely by HistGradientBoost (0.991), XGBoost (0.990), and ExtraTrees (0.990).
  • Execution time was reduced for several models post-optimization, indicating increased efficiency.
  • Confusion matrix and ROC analyses confirmed reliable detection of all four cardiac classes.

Conclusions:

  • The proposed VMD-based feature extraction and MT-PSO optimization framework offers a highly effective approach for multi-class ECG classification.
  • The framework demonstrates competitive or superior performance compared to existing methods for detecting cardiac arrhythmias.
  • This approach holds promise for improving the accuracy and efficiency of automated cardiac diagnosis.

Related Concept Videos

Correlation between ECG and Cardiac Cycle01:25

Correlation between ECG and Cardiac Cycle

The electrical signals recorded on an electrocardiogram (ECG) occur before the mechanical processes of contraction and relaxation during the cardiac cycle.
A cardiac action potential originates in the SA node and spreads throughout the atria and the AV node in approximately 0.03 seconds. This results in the P wave in an ECG and triggers atrial contraction. The action potential is then briefly slowed at the AV node, allowing the atria to contract and fill the ventricles with blood before...
15.1K
ECG Interpretation of Arrhythmias II: Atrial, Junctional and Ventricular Arrhythmias01:25

ECG Interpretation of Arrhythmias II: Atrial, Junctional and Ventricular Arrhythmias

Arrhythmia is a condition characterized by an irregular heart rhythm, with ECG changes that differ based on its origin and nature. The types of arrhythmias discussed below include atrial, junctional, and ventricular arrhythmias.Atrial ArrhythmiasPremature Atrial Complexes (PACs): PACs are early atrial beats caused by stress, caffeine, alcohol, electrolyte imbalances, hypoxia, hyperthyroidism, or certain medications (e.g., bronchodilators and decongestants). The ECG shows early P waves with an...
983
Electrocardiogram01:29

Electrocardiogram

An electrocardiogram (ECG or EKG) is a critical diagnostic tool that records the electrical signals produced by the heart during each heartbeat. This recording is achieved through electrodes placed strategically on the arms, legs, and chest. The electrocardiograph amplifies these signals and produces 12 distinct tracings, offering a comprehensive understanding of the heart's electrical activity.
Three major waveforms are present in a typical ECG recording: the P wave, the QRS complex, and...
8.1K
ECG Interpretation of Rhythms01:24

ECG Interpretation of Rhythms

An electrocardiogram (ECG)graphically represents the heart's electrical activity on ECG paper or a monitor.
Components of the Electrocardiogram
The primary components of a normal ECG waveform in Normal sinus rhythm(NSR) include the P wave, PR interval, QRS complex, ST segment, T wave, and occasionally a U wave.
ECG waveforms are divided by vertical and horizontal lines at standard intervals.
The horizontal axis measures time and rate, and the vertical axis measures amplitude or voltage....
18.8K
Pulse rhythm01:30

Pulse rhythm

Pulse rhythm refers to the pattern of pulsations within specific intervals, offering valuable insights into the regularity or irregularity of the heart's beats as observed through the pattern of pulsation within specific intervals. A regular pulse exhibits a consistent heart rate with uniform waveforms and pulsation force, variations of which can be classified as normal, weak, or bounding.
Conversely, an irregular pulse pattern is termed dysrhythmia, stemming from disruptions in cardiac...
1.6K
Dysrhythmias II: Classification of Tachyarrhythmias01:28

Dysrhythmias II: Classification of Tachyarrhythmias

Tachyarrhythmias are a type of dysrhythmia where the heart rate exceeds 100 beats per minute. Here are some common types of tachyarrhythmias:Sinus TachycardiaSinus tachycardia originates from increased impulses from the sinus node, leading to an elevated heart rate. It is often triggered by stress, fever, or exercise.Patients may experience palpitations, a sensation of a racing heart, dizziness, and chest discomfort.Causes and Risk Factors: Common causes include physical exertion, emotional...
772