This paper presents a transparent and interpretable machine learning pipeline for heart disease prediction based on structured clinical data. The proposed methodology emphasizes three essential properties for medical AI systems: interpretability, robustness to outliers, and generalizability. Multiple supervised learning models were developed and rigorously evaluated, including Logistic Regression, Random Forest, and Decision Trees. While Random Forest achieved strong predictive accuracy, its black-box nature limited its transparency. Logistic Regression offered better interpretability but required advanced preprocessing techniques such as winsorization, feature standardization, and polynomial expansion. Ultimately, a Decision Tree classifier—optimized through pre-pruning strategies (e.g., max_depth = 10) and using the Gini impurity criterion—was selected for its optimal trade-off between performance and clinical comprehensibility. The resulting pipeline provides a practical and trustworthy foundation for developing interpretable and deployable AI solutions in healthcare contexts.

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

Interpretable Machine Learning Pipeline for Heart Disease Prediction Using Clinical Data

  • Mounir Hemam,
  • Meriem Djezzar,
  • Abdelmouaine Benzaim

摘要

This paper presents a transparent and interpretable machine learning pipeline for heart disease prediction based on structured clinical data. The proposed methodology emphasizes three essential properties for medical AI systems: interpretability, robustness to outliers, and generalizability. Multiple supervised learning models were developed and rigorously evaluated, including Logistic Regression, Random Forest, and Decision Trees. While Random Forest achieved strong predictive accuracy, its black-box nature limited its transparency. Logistic Regression offered better interpretability but required advanced preprocessing techniques such as winsorization, feature standardization, and polynomial expansion. Ultimately, a Decision Tree classifier—optimized through pre-pruning strategies (e.g., max_depth = 10) and using the Gini impurity criterion—was selected for its optimal trade-off between performance and clinical comprehensibility. The resulting pipeline provides a practical and trustworthy foundation for developing interpretable and deployable AI solutions in healthcare contexts.