Genetic Algorithm-based Convolutional Neural Network Feature Engineering for Optimizing Coronary Heart Disease Prediction Performance
1
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
Objectives: This study aimed to optimize early coronary heart disease (CHD) prediction using a genetic algorithm (GA)-based convolutional neural network (CNN) feature engineering approach. We sought to overcome the limitations of traditional hyperparameter optimization techniques by leveraging a GA for superior predictive performance in CHD detection. Methods: Utilizing a GA for hyperparameter optimization, we navigated a complex combinatorial space to identify optimal configurations for a CNN model. We also employed information gain for feature selection optimization, transforming the CHD datasets into an image-like input for the CNN architecture. The efficacy of this method was benchmarked against traditional optimization strategies. Results: The advanced GA-based CNN model outperformed traditional methods, achieving a substantial increase in accuracy. The optimized model delivered a promising accuracy range, with a peak of 85% in hyperparameter optimization and 100% accuracy when integrated with machine learning algorithms, namely naïve Bayes, support vector machine, decision tree, logistic regression, and random forest, for both binary and multiclass CHD prediction tasks. Conclusions: The integration of a GA into CNN feature engineering is a powerful technique for improving the accuracy of CHD predictions. This approach results in a high degree of predictive reliability and can significantly contribute to the field of AI-driven healthcare, with the possibility of clinical deployment for early CHD detection. Future work will focus on expanding the approach to encompass a wider set of CHD data and potential integration with wearable technology for continuous health monitoring.
Abstract
Objectives: This study aimed to optimize early coronary heart disease (CHD) prediction using a genetic algorithm (GA)-based convolutional neural network (CNN) feature engineering approach. We sought to overcome the limitations of traditional hyperparameter optimization techniques by leveraging a GA for superior predictive performance in CHD detection. Methods: Utilizing a GA for hyperparameter optimization, we navigated a complex combinatorial space to identify optimal configurations for a CNN model. We also employed information gain for feature selection optimization, transforming the CHD datasets into an image-like input for the CNN architecture. The efficacy of this method was benchmarked against traditional optimization strategies. Results: The advanced GA-based CNN model outperformed traditional methods, achieving a substantial increase in accuracy. The optimized model delivered a promising accuracy range, with a peak of 85% in hyperparameter optimization and 100% accuracy when integrated with machine learning algorithms, namely naïve Bayes, support vector machine, decision tree, logistic regression, and random forest, for both binary and multiclass CHD prediction tasks. Conclusions: The integration of a GA into CNN feature engineering is a powerful technique for improving the accuracy of CHD predictions. This approach results in a high degree of predictive reliability and can significantly contribute to the field of AI-driven healthcare, with the possibility of clinical deployment for early CHD detection. Future work will focus on expanding the approach to encompass a wider set of CHD data and potential integration with wearable technology for continuous health monitoring.
참고문헌 (27)
- Muhammad Y, "Early and accurate detection and diagnosis of heart disease using intelligent computational model", Sci Rep, 10(1), pp.19747~ (2020)[DOI]
- Zhang P, "Effect of AI deep learning techniques on possible complications and clinical nursing quality of patients with coronary heart disease", Food Sci Technol, 42, pp.e42020~ (2022)[DOI]
- Elwahsh H, "A new smart healthcare framework for real-time heart disease detection based on deep and machine learning", PeerJ Comput Sci, 7, pp.e646~ (2021)[DOI]
- Al-Alshaikh HA, "Comprehensive evaluation and performance analysis of machine learning in heart disease prediction", Sci Rep, 14(1), pp.7819~ (2024)[DOI]
- Kanagarathinam K, "Machine learning-based risk prediction model for cardiovascular disease using a hybrid dataset", Data Knowl Eng, 140, pp.102042~ (2022)[DOI]
- Theng D, "Feature selection techniques for machine learning : a survey of more than two decades of research", Knowl Inf Syst, 66, pp.1575~1637 (2024)[DOI]
- Kurniabudi, "CICIDS-2017 dataset feature analysis with information gain for anomaly detection", IEEE Access, 8, pp.132911~132921 (2020)[DOI]
- Kaur S, "Hyper-parameter optimization of deep learning model for prediction of Parkinson’s disease", Mach Vis Appl, 31, pp.32~ (2020)[DOI]
- Valarmathi R, "Heart disease prediction using hyper parameter optimization(HPO)tuning", Biomed Signal Process Control, 70, pp.103033~ (2021)[DOI]
- Montesinos-Lopez OA, "A comparison between three tuning strategies for gaussian kernels in the context of univariate genomic prediction", Genes(Basel), 13(12), pp.2282~ (2022)[DOI]
- Ghawi R, "Efficient hyperparameter tuning with grid search for text categorization using kNN approach with BM25 similarity", Open Comput Sci, 9(1), pp.160~180 (2019)[DOI]
- Kumar S, "Multi-objective hyperparameter tuning of classifiers for disease diagnosis", Indian J Comput Sci Eng, 12(5), pp.1334~1352 (2021)[DOI]
- Rusch T, "Structure-based hyperparameter selection with Bayesian optimization in multidimensional scaling", Stat Comput, 33(1), pp.28~ (2023)[DOI]
- "Neural architecture search using particle swarm and ant colony optimization" (2024)[URL]
- Tayebi M, "Proceedings of the International Conference on Artificial Intelligence and Computer Vision (AICV2021)", pp.288~297 (2021), Springer
- Alzubaidi L, "Review of deep learning : concepts, CNN architectures, challenges, applications, future directions", J Big Data, 8(1), pp.53~ (2021)[DOI]
- Kong Y, "Multi-stage convolutional broad learning with block diagonal constraint for hyperspectral image classification", Remote Sens, 13(17), pp.3412~ (2021)[DOI]
- Ayan E, "Genetic algorithm-based hyperparameter optimization for convolutional neural networks in the classification of crop pests", Arab J Sci Eng, 49, pp.3079~3093 (2024)[DOI]
- Vetrimani E, "Building convolutional neural network parameters using genetic algorithm for the croup cough classification problem", Measur Sens, 27, pp.100717~ (2023)[DOI]
- Montecino DA, "Two-level genetic algorithm for evolving convolutional neural networks for pattern recognition", IEEE Access, 9, pp.126856~126872 (2021)[DOI]
- Salam A, "Classification of deep learning convolutional neural network feature extraction for student graduation prediction", Indones J Electr Eng Comput Sci, 32(1), pp.335~ (2023)[DOI]
- Zhu Y, "Converting tabular data into images for deep learning with convolutional neural networks", Sci Rep, 11(1), pp.11325~ (2021)[DOI]
- Hassan D, "Heart disease prediction based on pre-trained deep neural networks combined with principal component analysis", Biomed Signal Proc Contr, 79(Part 1), pp.104019~ (2023)[DOI]
- El Sherbiny MM, "Classification of chronic kidney disease based on machine learning techniques", Indones J Electr Eng Comput Sci, 32(2), pp.945~ (2023)[DOI]
- Narisetty N, "Ameliorating heart diseases prediction using machine learning technique for optimal solution", Int J Online Biomed Eng(iJOE), 19(16), pp.156~165 (2023)[DOI]
- El-Ibrahimi A, "Optimizing machine learning algorithms for heart disease classification and prediction", Int J Online Biomed Eng(iJOE), 19(15), pp.61~76 (2023)[DOI]
- Najafi A, "Multiple-criteria decision making, feature selection, and deep learning : a golden triangle for heart disease identification", Eng Appl Artif Intell, 125, pp.106662~ (2023)[DOI]