1. Deep attention model for arrhythmia signal classification based on multi-objective crayfish optimization algorithmic variational mode decomposition
- Author
-
Yihang Zhang and Hang Zhao
- Subjects
Arrthythmia signal classification ,Attention scheme ,Finite element method ,Crayfish optimization algorithm ,Variational mode decomposition ,Bayesian optimization ,Medicine ,Science - Abstract
Abstract The detection and classification of arrhythmia play a vital role in the diagnosis and management of cardiac disorders. Many deep learning techniques are utilized for arrhythmia classification in current research but only based on ECG data, lacking the mathematical foundations of cardiac electrophysiology. A finite element model (FEM) of the human heart based on the FitzHugh–Nagumo (FHN) model was established for cardiac electrophysiology simulation and the ECG signals were acquired from the FEM results of representative points. Two different kinds of arrhythmia characterized by major anomalies of parameters a and $$\varepsilon$$ in the FHN model were simulated, and the synthetic ECG signals were obtained respectively. A multi-objective optimization method based on non-dominated sorting was incorporated into the crayfish optimization algorithm to optimize the key parameters in VMD, then a variational mode decomposition technique for ECG signal processing based on a multi-objective crayfish optimization algorithm (MOCOA-VMD) was proposed, wherein the spectral kurtosis and KL divergence were determined as the indicators for decomposition. The Pareto optimal front was generated by MOCOA and the intrinsic mode functions of VMD with the best combination of K and $$\alpha$$ were obtained. A deep attention model based on MOCOA-VMD was constructed for ECG signal classification. The ablation study was implemented to verify the effectiveness of the proposed signal decomposition method and deep attention modules. The performance of the model based on MOCOA-VMD achieves the best accuracy of 94.35%, much higher than the model constructed by modules of EEMD, VMD and CNN. Moreover, Bayesian optimization was carried out to fine-tune the hyperparameters batch size, learning rate, epochs, and momentum. After TPE optimization, the deep model’s performance achieved a maximum accuracy of 95.91%. The MIT-BIH arrhythmia database was further utilized for model validation, ascertaining its robustness and generalizability. The proposed deep attention modeling and classification strategy can help in arrhythmia signal processing and may offer inspiration for other signal processing fields as well.
- Published
- 2025
- Full Text
- View/download PDF