Machine learning (ML) holds great promise in transforming healthcare by enabling early disease detection and improving clinical decision-making. Despite this potential, ML models often face obstacles such as irrelevant features, overfitting, and limited generalization capabilities. This research addresses these challenges by focusing on cardiovascular disease prediction using a well-known dataset from Kaggle. The study implements a structured optimization approach, beginning with feature selection methods—such as correlation filtering and regularization techniques (Lasso and Ridge)—to reduce redundancy and enhance data quality. To improve model accuracy and control overfitting, we evaluate high-performing ensemble algorithms including Random Forest, XGBoost, and LightGBM. Each model undergoes fine-tuning using hyperparameter optimization and stratified k-fold cross-validation to validate performance across subsets of the data. Our findings confirm that gradient boosting methods deliver superior accuracy and consistency compared to more conventional algorithms. This work emphasizes the value of tailoring ML pipelines to the specific challenges of medical data, demonstrating how thoughtful preprocessing and algorithm selection can significantly enhance prediction outcomes. The study provides a practical framework that can be adapted to other healthcare applications, with the goal of advancing precision medicine in the field of cardiovascular care.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Enhancing Machine Learning for Medical Use: A Case Study on Cardiovascular Disease Prediction Via Feature Optimization, Regularization, and Overfitting Control

  • Lamiae Eloutouate,
  • Hicham Gibet Tani,
  • Lotfi Elaachak,
  • Fatiha Elouaai,
  • Mohammed Bouhorma

摘要

Machine learning (ML) holds great promise in transforming healthcare by enabling early disease detection and improving clinical decision-making. Despite this potential, ML models often face obstacles such as irrelevant features, overfitting, and limited generalization capabilities. This research addresses these challenges by focusing on cardiovascular disease prediction using a well-known dataset from Kaggle. The study implements a structured optimization approach, beginning with feature selection methods—such as correlation filtering and regularization techniques (Lasso and Ridge)—to reduce redundancy and enhance data quality. To improve model accuracy and control overfitting, we evaluate high-performing ensemble algorithms including Random Forest, XGBoost, and LightGBM. Each model undergoes fine-tuning using hyperparameter optimization and stratified k-fold cross-validation to validate performance across subsets of the data. Our findings confirm that gradient boosting methods deliver superior accuracy and consistency compared to more conventional algorithms. This work emphasizes the value of tailoring ML pipelines to the specific challenges of medical data, demonstrating how thoughtful preprocessing and algorithm selection can significantly enhance prediction outcomes. The study provides a practical framework that can be adapted to other healthcare applications, with the goal of advancing precision medicine in the field of cardiovascular care.