Deep-Stacked Convolutional Neural Networks for Brain Abnormality Classification Based on MRI Images

Dewinda Julianensi Rumala, Peter van Ooijen, Reza Fuad Rachmadi, Anggraini Dwi Sensusiati, I. Ketut Eddy Purnama

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Anautomated diagnosis system is crucial for helping radiologists identify brain abnormalities efficiently. The convolutional neural network (CNN) algorithm of deep learning has the advantage of automated feature extraction beneficial for an automated diagnosis system. However, several challenges in the CNN-based classifiers of medical images, such as a lack of labeled data and class imbalance problems, can significantly hinder the performance. Meanwhile, the expertise of multiple clinicians may be required to achieve accurate diagnoses, which can be reflected in the use of multiple algorithms. In this paper, we present Deep-Stacked CNN, a deep heterogeneous model based on stacked generalization to harness the advantages of different CNN-based classifiers. The model aims to improve robustness in the task of multi-class brain disease classification when we have no opportunity to train single CNNs on sufficient data. We propose two levels of learning processes to obtain the desired model. At the first level, different pre-trained CNNs fine-tuned via transfer learning will be selected as the base classifiers through several procedures. Each base classifier has a unique expert-like character, which provides diversity to the diagnosis outcomes. At the second level, the base classifiers are stacked together through neural network, representing the meta-learner that best combines their outputs and generates the final prediction. The proposed Deep-Stacked CNN obtained an accuracy of 99.14% when evaluated on the untouched dataset. This model shows its superiority over existing methods in the same domain. It also requires fewer parameters and computations while maintaining outstanding performance.

Original languageEnglish
Pages (from-to)1460-1479
Number of pages20
JournalJournal of Digital Imaging
Volume36
Issue number4
DOIs
Publication statusPublished - Aug 2023

Keywords

  • Brain disease
  • Convolutional neural network
  • Deep transfer learning
  • Ensemble classifier
  • Magnetic resonance images
  • Stacking

Fingerprint

Dive into the research topics of 'Deep-Stacked Convolutional Neural Networks for Brain Abnormality Classification Based on MRI Images'. Together they form a unique fingerprint.

Cite this