arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.09548cs.LGcs.AIcs.CYq-bio.QM

使用梯度提升算法进行乳腺癌分类:聚焦于降低假阴性及基于SHAP的可解释性

Breast Cancer Classification Using Gradient Boosting Algorithms Focusing on Reducing the False Negative and SHAP for Explainability

João Manoel Herrera Pinheiro, Marcelo Becker

AI总结:

本研究首次结合AdaBoost、XGBoost、CatBoost、LightGBM与Optuna超参数优化及SHAP可解释性方法预测乳腺癌,重点提升召回率并降低假阴性,最终所有模型AUC均超过99.41%。

摘要:

癌症是世界上导致女性死亡最多的疾病之一,其中乳腺癌造成的癌症病例数和死亡人数最高。然而,通过早期检测并进而早期治疗,乳腺癌是可以预防的。任何用于检测或预测这类癌症的进展都对更健康的生活具有重要意义。许多研究关注在癌症预测中具有高准确率的模型,但有时仅凭准确率并不总是一个可靠的指标。本研究采用一种探究性方法,研究基于提升(boosting)的不同机器学习算法在预测乳腺癌时的性能,并重点关注召回率指标。提升类机器学习算法已被证明是检测医学疾病的有效工具。本研究使用加州大学欧文分校(UCI)仓库中的数据集来训练和测试包含其属性的模型分类器。本研究的主要目标是使用最先进的提升算法,如AdaBoost、XGBoost、CatBoost和LightGBM,来预测和诊断乳腺癌,并根据召回率、ROC-AUC和混淆矩阵找出最有效的指标。此外,本研究首次将上述四种提升算法与Optuna(一个用于超参数优化的库)以及SHAP方法结合使用,以提高模型的可解释性,从而可用于辅助识别和预测乳腺癌。我们能够改善所有模型的AUC或召回率,并降低AdaBoost和LightGBM的假阴性,所有模型的最终AUC均超过99.41%。

英文摘要:

Cancer is one of the diseases that kill the most women in the world, with breast cancer being responsible for the highest number of cancer cases and consequently deaths. However, it can be prevented by early detection and, consequently, early treatment. Any development for detection or perdition this kind of cancer is important for a better healthy life. Many studies focus on a model with high accuracy in cancer prediction, but sometimes accuracy alone may not always be a reliable metric. This study implies an investigative approach to studying the performance of different machine learning algorithms based on boosting to predict breast cancer focusing on the recall metric. Boosting machine learning algorithms has been proven to be an effective tool for detecting medical diseases. The dataset of the University of California, Irvine (UCI) repository has been utilized to train and test the model classifier that contains their attributes. The main objective of this study is to use state-of-the-art boosting algorithms such as AdaBoost, XGBoost, CatBoost and LightGBM to predict and diagnose breast cancer and to find the most effective metric regarding recall, ROC-AUC, and confusion matrix. Furthermore, our study is the first to use these four boosting algorithms with Optuna, a library for hyperparameter optimization, and the SHAP method to improve the interpretability of our model, which can be used as a support to identify and predict breast cancer. We were able to improve AUC or recall for all the models and reduce the False Negative for AdaBoost and LigthGBM the final AUC were more than 99.41\% for all models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑