Related Experiment Videos
Generative adversarial networks and hyperparameter-optimized XGBoost for enhanced heart disease prediction
Shaik Sajeera Begum1, Amit Swamy1, Sanjay Dhanka2
1School of Technology, Woxsen University, Hyderabad, India.
Insights
A new hybrid framework, GAN-XO, improves early heart disease (HD) detection by addressing data quality issues. This system enhances diagnostic accuracy for better clinical decision-making and reliable patient identification.
Area of Science:
- Cardiology
- Artificial Intelligence
- Data Science
Background:
- Heart disease (HD) is a leading global cause of mortality, necessitating advanced diagnostic tools.
- Clinical data often suffers from bias and noise, hindering the development of reliable predictive models for early HD detection.
Purpose of the Study:
- To propose and evaluate a novel hybrid framework, GAN-XO, for early detection of heart disease.
- To address data imbalance and noise in clinical datasets for improved predictive model performance.
Main Methods:
- Utilized a Generative Adversarial Network (GAN) for synthetic oversampling to balance imbalanced datasets.
- Implemented a two-phase outlier detection and elimination process (z-score and IQR) for data cleaning.
- Employed an XGBoost classifier with hyperparameter optimization via Optuna for accurate prediction.
Main Results:
- The GAN-XO system achieved a high accuracy of 96.60% and an F1-Score of 0.9649 on a clean, balanced test dataset.
- Ablation studies confirmed the contribution of each component, highlighting the critical role of outlier elimination.
- The framework demonstrated significant performance enhancement, improving trustworthiness and model reliability.
Conclusions:
- Accurate healthcare predictions depend on integrated modeling systems and rigorous data quality management.
- The GAN-XO framework offers a promising approach to enhance clinical decision-making for early heart disease diagnosis.
Abstract:
Heart Disease (HD) serves as the primary reason for worldwide deaths which creates an urgent requirement to create diagnostic intelligent systems that can identify patients at their earliest stage. The development of strong predictive models faces a significant obstacle because clinical data contains both biased information and noisy elements which create unreliable results. This study proposed a hybrid framework, GAN-XO, for early HD detection. The methodology addressed fundamental data quality issues through a multi-step preprocessing pipeline. First, a Generative Adversarial Network (GAN) is employed for synthetic oversampling of the minority class, balancing the imbalanced PKIOHD and Framingham datasets. Second, a two-phase outlier detection and elimination process using z-score and Interquartile Range (IQR) methods is applied to obtain clean data. Finally, an XGBoost classifier, whose hyperparameters are optimally tuned using Optuna, is utilized for prediction. The designed GAN-XO system achieved its best accuracy score of 96.60% and F1-Score value of 0.9649 through testing on an evenly distributed and completely clean test dataset. The ablation research demonstrated that all system elements provided distinct benefits while the outlier elimination process turned out to be vital for improving model performance and trustworthiness. The research results demonstrate that accurate healthcare predictions require both integrated modelling systems and precise management of data quality. The substantial performance enhancement shows that the framework improves clinical decision-making while enabling correct diagnosis of HD.
Related Concept Videos
Blood Studies for Cardiovascular System I: Cardiac Biomarkers
The essential diagnostic tools for detecting myocardial necrosis and monitoring individuals suspected of having acute coronary syndrome (ACS) include:
Troponins
Troponins, particularly cardiac troponins I and T, are the most precise and sensitive markers of myocardial injury. They are detectable within 4-6 hours of myocardial injury and remain...
Heart Failure Drugs: β-Blockers
Heart Failure II: Pathophysiology