StackFBAs: Detection of fetal brain abnormalities using CNN with stacking strategy from MRI images

被引:2
作者
Chowdhury, Anjir Ahmed [1 ,2 ]
Mahmud, S. M. Hasan [1 ,2 ]
Hoque, Khadija Kubra Shahjalal [1 ]
Ahmed, Kawsar [3 ,4 ]
Bui, Francis M. [3 ]
Lio, Pietro [6 ]
Moni, Mohammad Ali [7 ]
Al-Zahrani, Fahad Ahmed [5 ]
机构
[1] Amer Int Univ Bangladesh AIUB, Dept Comp Sci, Dhaka 1229, Bangladesh
[2] Ctr Adv Machine Learning & Applicat CAMLAs, Dhaka 1229, Bangladesh
[3] Univ Saskatchewan, Dept Elect & Comp Engn, 57 Campus Dr, Saskatoon, SK S7N 5A9, Canada
[4] Mawlana Bhashani Sci & Technol Univ MBSTU, Dept Informat & Commun Technol ICT, Grp Biophotomati 5, Tangail 1902, Bangladesh
[5] Umm Al Qura Univ, Dept Comp Engn, Mecca 24381, Saudi Arabia
[6] Univ Cambridge, Dept Comp Sci & Technol, Cambridge CB3 0FD, England
[7] Univ Queensland, Fac Hlth & Behav Sci, St Lucia, Qld 4072, Australia
关键词
Brain abnormalities detection; Deep learning; Federated learning; Neural architecture search; Stacking strategy; Transfer learning; SEMANTIC SEGMENTATION; PREDICTION; LOCALIZATION;
D O I
10.1016/j.jksuci.2023.101647
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Predicting fetal brain abnormalities (FBAs) is an urgent global problem, as nearly three of every thousand women are pregnant with neurological abnormalities. Therefore, early detection of FBAs using deep learning (DL) can help to enhance the planning and quality of diagnosis and treatment for pregnant women. Most of the research papers focused on brain abnormalities of newborns and premature infants, but fewer studies concentrated on fetuses. This study proposed a deep learning-CNN-based framework named StackFBAs that utilized the stacking strategy to classify fetus brain abnormalities more accurately using MRI images at an early stage. We considered the Greedy-based Neural architecture search (NAS) method to identify the best CNN architectures to solve this problem utilizing brain MRI images. A total of 94 CNN architectures were generated from the NAS method, and the best 5 CNN models were selected to build the baseline models. Subsequently, the probabilistic scores of these baseline models were com-bined to construct the final meta-model (KNN) utilizing the stacking strategy. The experimental results demonstrated that StackFBAs outperform pre-trained CNN Models (e.g., VGG16, VGG19, ResNet50, DenseNet121, and ResNet152) with transfer learning (TL) and existing models with the 5-fold cross-validation tests. StackFBAs achieved an overall accuracy of 80%, an F1-score of 78%, 76% sensitivity, and a specificity of 78%. Moreover, we employed the federated learning technique that protects sensitive fetal MRI data, combines results, and finds common patterns from many users, making the model more robust for the privacy and security of user-sensitive data. We believe that our novel framework could be used as a helpful tool for detecting brain abnormalities at an early stage.(c) 2023 The Author(s). Published by Elsevier B.V. on behalf of King Saud University. This is an open access article under the CC BY-NC-ND license (http://creativecommons.org/licenses/by-nc-nd/4.0/).
引用
收藏
页数:14
相关论文
共 77 条
  • [1] Deep Learning with Differential Privacy
    Abadi, Martin
    Chu, Andy
    Goodfellow, Ian
    McMahan, H. Brendan
    Mironov, Ilya
    Talwar, Kunal
    Zhang, Li
    [J]. CCS'16: PROCEEDINGS OF THE 2016 ACM SIGSAC CONFERENCE ON COMPUTER AND COMMUNICATIONS SECURITY, 2016, : 308 - 318
  • [2] Adams R.P., 2012, 25 INT C NEURAL INFP, P2951, DOI DOI 10.5555/2999325.2999464.47
  • [3] Automatic Brain Localization in Fetal MRI Using Superpixel Graphs
    Alansary, Amir
    Lee, Matthew
    Keraudren, Kevin
    Kainz, Bernhard
    Malamateniou, Christina
    Rutherford, Mary
    Hajnal, Joseph V.
    Glocker, Ben
    Rueckert, Daniel
    [J]. MACHINE LEARNING MEETS MEDICAL IMAGING, 2015, 9487 : 13 - 22
  • [4] CNN-based transfer learning-BiLSTM network: A novel approach for COVID-19 infection detection
    Aslan, Muhammet Fatih
    Unlersen, Muhammed Fahri
    Sabanci, Kadir
    Durdu, Akif
    [J]. APPLIED SOFT COMPUTING, 2021, 98
  • [5] Fetal Brain Abnormality Classification from MRI Images of Different Gestational Age
    Attallah, Omneya
    Sharkas, Maha A.
    Gadelkarim, Heba
    [J]. BRAIN SCIENCES, 2019, 9 (09)
  • [6] Bergstra J, 2012, J MACH LEARN RES, V13, P281
  • [7] StackDPPIV: A novel computational approach for accurate prediction of dipeptidyl peptidase IV (DPP-IV) inhibitory peptides
    Charoenkwan, Phasit
    Nantasenamat, Chanin
    Hasan, Md Mehedi
    Moni, Mohammad Ali
    Lio, Pietro
    Manavalan, Balachandran
    Shoombuatong, Watshara
    [J]. METHODS, 2022, 204 : 189 - 198
  • [8] StackIL6: a stacking ensemble model for improving the prediction of IL-6 inducing peptides
    Charoenkwan, Phasit
    Chiangjong, Wararat
    Nantasenamat, Chanin
    Hasan, Md Mehedi
    Manavalan, Balachandran
    Shoombuatong, Watshara
    [J]. BRIEFINGS IN BIOINFORMATICS, 2021, 22 (06)
  • [9] Chen LC, 2018, ADV NEUR IN, V31
  • [10] Sequence-based prediction of protein interaction sites with an integrative method
    Chen, Xue-Wen
    Jeong, Jong Cheol
    [J]. BIOINFORMATICS, 2009, 25 (05) : 585 - 591