Novel Approaches to Heart Disease Detection with TabNet and FT Transformer Models
Heart disease has been one of the leading causes of mortality worldwide, necessitating accurate predictive models to support early diagnosis and intervention. Traditional ML models have been widely used for heart disease detection due to their interpretability and efficiency on tabular datasets. Recent advancements in deep learning introduced novel architectures, such as TabNet and FT-Transformer, which leveraged attention mechanisms and feature tokenization to capture complex, high-order feature interactions directly from tabular data. This paper applied and compared traditional ML models, TabNet, and FT-Transformer on a heart disease dataset to evaluate their effectiveness in predicting heart disease outcomes. The approach combined the strengths of traditional ML models with the enhanced feature representation and interpretability offered by deep learning-based methods, particularly TabNet's sequential attention mechanism and FT-Transformer's selfattention architecture. Model performance was analyzed across various metrics, including accuracy, F1-score, and AUC-ROC, and feature importance was evaluated to gain insights into the key predictors of heart disease. Results demonstrated that while traditional ML models remained competitive, TabNet and FT-Transformer provided promising alternatives, capturing complex feature dependencies that might otherwise have gone unnoticed.
Authors
- Nagagopiraju Vullam
- A. Lakshmanarao (ORCID: https://orcid.org/0000-0002-8068-4857)
- Grandhi Siva Sankar
- E. Sreedevi
- Sravana Kumar Komma
- Pedapudi Vijaya Bhaskar
Institutions
- Sri Venkateswara University (IN)
- International Institute of Information Technology (IN)
- Aditya Birla (India) (IN)
Publication Details
- Published
- 2025-03-04
- DOI
- https://doi.org/10.1109/icvadv63329.2025.10961899
- Citations
- 2
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 5.88