An Explainable Sparse Autoencoder–CNN Framework for Robust Cardiovascular Disease Prediction Using Enhanced Feature Representations

Since cardiovascular diseases (CVDs) are the world's leading cause of death, it is critical to develop prediction frameworks that are reliable, accurate, and easy to understand in order to facilitate prompt clinical decision making. Although many studies have been conducted on traditional machine learning techniques for cardiac risk assessment, their efficacy is frequently constrained by their dependence on manually created features and their restricted capacity to identify intricate non-linear relationships in clinical data. Although deep learning techniques provide better representation learning capabilities, overfitting and interpretability issues limit their efficacy on structured, low-dimensional clinical data.
This paper proposes a novel deep learning framework that uses sparse autoencoders as feature augmentation in conjunction with CNN classification to provide robust prediction of heart disease. The sparse autoencoder allows for generation of enriched latent representations, due to the application of sparsity constraints helping to reveal hidden clinically relevant patterns in tabular patient records. This augmented representation is then reshaped into a structured sequence and passed through a CNN to capture higher-order feature interactions. Furthermore, a multitask learning strategy optimally trains the model to simultaneously optimally reconstruct and classify disease, ultimately improving the model's generalization capability and predictive stability.
The proposed framework has been successfully validated through ten-fold cross-validation with a benchmark dataset established for predicting heart disease. The experimental results demonstrated that the framework produced a classification accuracy of 92%; the proposed framework exceeds both traditional machine learning methods and each of the individual neural networks used previously by other authors. Furthermore, statistical method analysis showed that the improvement seen with the proposed framework was statistically significant. Additionally, the explainability analysis identified risk factors that are clinically relevant for predicting the outcome of the model and will therefore enhance transparency and clinical confidence.
The proposed method provides a scalable, easy to understand, and clinically relevant way of detecting Early Heart Disease (CVD) and supporting decisions related to it.

Prediction of heart disease, Sparse AutoEncoder, Convolutional Neural Networks, Feature Augmentation, Explanatory Artificial Intelligence

1. M. N. Alam et al., “A comprehensive evaluation and performance analysis of machine learning in heart disease prediction,” Sci. Rep., vol. 14, no. 1, Art. 58489, 2024. Nature [Google Scholar] [Crossref]

2. M. A. Naser, A. A. M. Majeed, M. Alsabah, T. R. Al-Shaikhli, and K. M. Kaky, “A review of machine learning’s role in cardiovascular disease prediction: Recent advances and future challenges,” Algorithms, vol. 17, no. 2, p. 78, 2024, doi:10.3390/a17020078. MDPI [Google Scholar] [Crossref]

3. R. Kumar et al., “A comprehensive review of machine learning for heart disease prediction: Challenges, trends, ethical considerations, and future directions,” Front. Artif. Intell., vol. 8, p. 1583459, 2025. PMC [Google Scholar] [Crossref]

4. B. Xia et al., “Intelligent cardiovascular disease diagnosis using deep learning,” Sci. Rep., vol. 14, 2024. PubMed [Google Scholar] [Crossref]

5. A. Guldogan et al., “A proposed tree-based explainable artificial intelligence approach for the prediction of angina pectoris,” Sci. Rep., vol. 13, p. 22189, 2023. IJNRD [Google Scholar] [Crossref]

6. H. Tjoa and C. Guan, “A survey on explainable artificial intelligence in healthcare,” IEEE Trans. Neural Netw. Learn. Syst., vol. 32, no. 11, pp. 4793–4813, 2021. [Google Scholar] [Crossref]

7. F. Khan et al., “Explainable deep learning framework for cardiovascular disease prediction,” Diagnostics, vol. 12, no. 9, p. 2022, 2022. [Google Scholar] [Crossref]

8. M. Hossain et al., “cardiovascular disease identification using a hybrid CNN-LSTM model with explainable AI,” Informatics Med. Unlocked, vol. 42, p. 101370, 2023. IJNRD [Google Scholar] [Crossref]

9. H. Zhang, L. Chen, and Q. Liu, “Optimizing latent space dimensionality for clinical deep learning models,” Artificial Intelligence in Medicine, vol. 123, p. 102213, 2022. [Google Scholar] [Crossref]

10. X. Li, Y. Zhang, and S. Wang, “Sparse autoencoder-based feature learning for medical diagnosis,” Biomed. Signal Process. Control, vol. 68, p. 102675, 2021. [Google Scholar] [Crossref]

11. R. Kumar, A. Garg, and R. Kaur, “Machine learning models and algorithms for healthcare: Applications in heart disease prediction,” Front. Artif. Intell., 2025. Frontiers [Google Scholar] [Crossref]

12. A. Shankar, “Prediction of cardiovascular diseases using machine learning: A systematic review,” J. Med. Syst., vol. 44, no. 8, p. 162, 2020. [Google Scholar] [Crossref]

13. S. Ahmed et al., “Explainable AI-reduct: Accuracy preservation despite dimensionality reduction for heart disease classification using explainable AI,” J. Supercomput., vol. 79, no. 16, pp. 18167–18197, 2023. IJNRD [Google Scholar] [Crossref]

14. C. Carter et al., “Deciphering simultaneous heart conditions with spectrogram and explainable-AI approach,” Biomed. Signal Process. Control, vol. 85, p. 104990, 2023. IJNRD [Google Scholar] [Crossref]

15. N. Parveen et al., “ECG based one-dimensional residual deep convolutional auto-encoder model for heart disease classification,” Multimedia Tools Appl., 2024. Bohrium [Google Scholar] [Crossref]

16. H. Hou et al., “Deep learning-based 12-lead electrocardiogram for low ejection fraction detection in patients,” Can. J. Cardiol., vol. 41, pp. 278–290, 2024. PubMed [Google Scholar] [Crossref]

17. H. Lee et al., “Electrocardiographic-Driven AI model for one-year mortality prediction in heart failure,” Int. J. Med. Inform., vol. 197, p. 105843, 2025. PubMed [Google Scholar] [Crossref]

18. S. Campanioni et al., “cardiovascular disease identification using hybrid models,” 2024. OUCI H. Sheng, “Predicting heart disease using machine learning,” Appl. Comp. Eng., vol. 71, pp. 19–23, 2024. Ewa Direct [Google Scholar] [Crossref]

19. R. Jothi Prakash et al., “Attention-Based Cross-Modal transfer learning for CVD prediction,” Front. Artif. Intell., 2024. [Google Scholar] [Crossref]

• The Role of Artificial Intelligence in Revolutionizing Library Services in Nairobi: Ethical Implications and Future Trends in User Interaction
• ESPYREAL: A Mobile Based Multi-Currency Identifier for Visually Impaired Individuals Using Convolutional Neural Network
• Comparative Analysis of AI-Driven IoT-Based Smart Agriculture Platforms with Blockchain-Enabled Marketplaces
• AI-Based Dish Recommender System for Reducing Fruit Waste through Spoilage Detection and Ripeness Assessment
• SEA-TALK: An AI-Powered Voice Translator and Southeast Asian Dialects Recognition