Breast cancer remains one of the leading causes of death in women across the world, especially considering the vital role of early and accurate diagnosis in disease management. In this paper, a proposal for a hybrid model based on deep learning that combines Convolutional Neural Networks (CNN) with Extreme Gradient Boosting (XGBoost) towards breast cancer detection in mammography images is proposed. Taking advantage of the intrinsic power of CNNs in automatic feature learning and classification capability of XGBoost, the new model is designed to perform better than traditional approaches in diagnostic efficiency. The Augmented INbreast database with thorough preprocessing via normalization and on-the-fly data augmentation is utilized by the paper for inducing dataset diversity as well as overfitting avoidance. A pre-trained CNN model is utilized as a feature extractor and features are passed through an XGBoost classifier for prediction. The experimental result yields much better classification performance with 96.24% test accuracy and satisfactory recall rate in cases of malignancy. The model is good enough for other metrics such as precision, F1-score, and AUC to qualify as a decision-support system for radiologists. This work illustrates the utility of deep learning and ensemble methods for accurate, interpretable, and scalable breast cancer classification from mammography images.

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

MammoCheck: Intelligent Mammogram Analysis for Breast Cancer Diagnosis Using CNNs and XGBoost

  • Shivpratap Singh Kushwah,
  • Pradeep Gupta,
  • Sonam Gupta

摘要

Breast cancer remains one of the leading causes of death in women across the world, especially considering the vital role of early and accurate diagnosis in disease management. In this paper, a proposal for a hybrid model based on deep learning that combines Convolutional Neural Networks (CNN) with Extreme Gradient Boosting (XGBoost) towards breast cancer detection in mammography images is proposed. Taking advantage of the intrinsic power of CNNs in automatic feature learning and classification capability of XGBoost, the new model is designed to perform better than traditional approaches in diagnostic efficiency. The Augmented INbreast database with thorough preprocessing via normalization and on-the-fly data augmentation is utilized by the paper for inducing dataset diversity as well as overfitting avoidance. A pre-trained CNN model is utilized as a feature extractor and features are passed through an XGBoost classifier for prediction. The experimental result yields much better classification performance with 96.24% test accuracy and satisfactory recall rate in cases of malignancy. The model is good enough for other metrics such as precision, F1-score, and AUC to qualify as a decision-support system for radiologists. This work illustrates the utility of deep learning and ensemble methods for accurate, interpretable, and scalable breast cancer classification from mammography images.