An investigation of XGBoost-based algorithm for breast cancer classification

被引:46
|
作者
Liew, Xin Yu [1 ]
Hameed, Nazia [1 ]
Clos, Jeremie [1 ]
机构
[1] Univ Nottingham, Jubilee Campus,Wollaton Rd, Nottingham NG8 1BB, England
来源
关键词
Deep learning; Extreme gradient boosting; XGBoost; Machine learning; Computer-aided diagnosis; Breast cancer; Histopathology images; Classification; NEURAL-NETWORKS; DATASET;
D O I
10.1016/j.mlwa.2021.100154
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Breast cancer is one of the leading cancers affecting women around the world. The Computer -Aided Diagnosis (CAD) system is a powerful tool to assist pathologists during the process of diagnosing cancer, which effectively identifies the presence of cancerous cells. A standard CAD system includes processes of pre-processing, feature extraction, feature selection and classification. In this paper, we propose an enhanced breast cancer classification technique called Deep Learning and eXtreme Gradient Boosting (DLXGB) on histopathology breast cancer images using the BreaKHis dataset. This method first applies data augmentation and stain normalization for pre-processing, then pre -trained DenseNet201 will automatically learn features within an image and combine with a powerful gradient boosting classifier. The proposed classification technique is designed to classify breast cancer histology images into binary benign and malignant, and additionally one of eight non-overlapping/overlapping categories: i.e., Adenosis (A), Fibroadenoma (F), Phyllodes Tumour (PT), And Tubular Adenoma (TA) Ductal Carcinoma (DC), Lobular Carcinoma (LC), Mucinous Carcinoma (MC), And Papillary Carcinoma (PC). With DLXGB, we have obtained an accuracy of 97% for both binary and multiclassification improving the exiting work done by researchers using the BreaKHis dataset. The results indicated that this method could produce a powerful prediction for breast cancer image classification.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] An XGBoost-Based Knowledge Tracing Model
    Su, Wei
    Jiang, Fan
    Shi, Chunyan
    Wu, Dongqing
    Liu, Lei
    Li, Shihua
    Yuan, Yongna
    Shi, Juntai
    [J]. INTERNATIONAL JOURNAL OF COMPUTATIONAL INTELLIGENCE SYSTEMS, 2023, 16 (01)
  • [2] A molecular classification of gastric cancer associated with distinct clinical outcomes and validated by an XGBoost-based prediction model
    Li, Bing
    Zhang, Fengbin
    Niu, Qikai
    Liu, Jun
    Yu, Yanan
    Wang, Pengqian
    Zhang, Siqi
    Zhang, Huamin
    Wang, Zhong
    [J]. MOLECULAR THERAPY NUCLEIC ACIDS, 2023, 31 : 224 - 240
  • [3] XGBoost-based and tumor-immune characterized gene signature for the prediction of metastatic status in breast cancer
    Qingqing Li
    Hui Yang
    Peipei Wang
    Xiaocen Liu
    Kun Lv
    Mingquan Ye
    [J]. Journal of Translational Medicine, 20
  • [4] XGBoost-Based Android Malware Detection
    Wang, Jiong
    Li, Boquan
    Zeng, Yuwei
    [J]. 2017 13TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL INTELLIGENCE AND SECURITY (CIS), 2017, : 268 - 272
  • [5] An XGBoost-Based Knowledge Tracing Model
    Wei Su
    Fan Jiang
    Chunyan Shi
    Dongqing Wu
    Lei Liu
    Shihua Li
    Yongna Yuan
    Juntai Shi
    [J]. International Journal of Computational Intelligence Systems, 16
  • [6] Hybrid classification of XGBoost-based ADAM optimization for coronary artery disease diagnosis
    Nagamani, T.
    Logeswari, S.
    [J]. Journal of Intelligent and Fuzzy Systems, 2024, 46 (04): : 10035 - 10044
  • [7] XGBoost-based and tumor-immune characterized gene signature for the prediction of metastatic status in breast cancer
    Li, Qingqing
    Yang, Hui
    Wang, Peipei
    Liu, Xiaocen
    Lv, Kun
    Ye, Mingquan
    [J]. JOURNAL OF TRANSLATIONAL MEDICINE, 2022, 20 (01)
  • [8] XGBoost-based Algorithm for Post-fault Transient Stability Status Prediction
    Chen M.
    Liu Q.
    Zhang J.
    Chen S.
    Zhang C.
    [J]. Dianwang Jishu/Power System Technology, 2020, 44 (03): : 1026 - 1033
  • [9] A Hybrid Meta-heuristics Algorithm: XGBoost-Based Approach for IDS in IoT
    Soumya Bajpai
    Kapil Sharma
    Brijesh Kumar Chaurasia
    [J]. SN Computer Science, 5 (5)
  • [10] XGBFEMF: An XGBoost-Based Framework for Essential Protein Prediction
    Zhong, Jiancheng
    Sun, Yusui
    Peng, Wei
    Xie, Minzhu
    Yang, Jiahong
    Tang, Xiwei
    [J]. IEEE TRANSACTIONS ON NANOBIOSCIENCE, 2018, 17 (03) : 243 - 250