Enhanced performance of Dark-Nets for brain tumor classification and segmentation using colormap-based superpixel techniques

被引:35
作者
Ahuja, Sakshi [1 ]
Panigrahi, Bijaya Ketan [1 ]
Gandhi, Tapan Kumar [1 ]
机构
[1] Indian Inst Technol Delhi, Elect Engn Dept, New Delhi 110016, India
来源
MACHINE LEARNING WITH APPLICATIONS | 2022年 / 7卷
关键词
Brain tumor; MRI; Deep learning; Superpixel segmentation; FEATURES; TEXTURE; MACHINE;
D O I
10.1016/j.mlwa.2021.100212
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The brain tumor is the deadliest disease in adults as it arises due to an abnormal mass of cells that grows rapidly and it alters the proper functioning of the organs. In clinical practice, radiographic images of different modalities are used to diagnose types of brain tumors, their size, and location. The proposed work aims to automatically classify, localize, and segment brain tumors from T1W-CE Magnetic Resonance Image (MRI) datasets. The T1W-CE MRI dataset is divided into 8:1:1, i.e., 80% training set, 10% of each validation, and testing set. To address the overfitting issues, the training data set is augmented using 2-levels wavelet decomposition and geometrical operations (scaling, rotation, translation). Performance of pre-trained DarkNet model (DarkNet-19 and DarkNet-53) is evaluated for the multi-class classification and localization of brain tumors. The best performing pre-trained DarkNet model achieved 99.60% of training accuracy and 98.81% of validation accuracy. The performance evaluation parameters confirm the superiority of the proposed methodology in comparison to the state -of -the -art on the T1W-CE MRI dataset. On 1070 T1W-CE testing images, the best-performing pre-trained DarkNet-53 model obtained a testing accuracy of 98.54% and Area Under Curve (AUC) of 0.99. The tumor is segmented using a 2-D superpixel segmentation technique with an average dice index of 0.94 +/- 2.6% on the 793 brain tumor testing data. To prove the superiority of the proposed technique, it is implemented on MRI images from the BraTS2018 dataset. The comparative analysis of performance evaluation parameters of the proposed methodology with the state -of -the -art technique proves its robustness and clinical significance.
引用
收藏
页数:13
相关论文
共 75 条
[1]   Two-phase multi-model automatic brain tumour diagnosis system from magnetic resonance images using convolutional neural networks [J].
Abd-Ellah, Mahmoud Khaled ;
Awad, Ali Ismail ;
Khalaf, Ashraf A. M. ;
Hamed, Hesham F. A. .
EURASIP JOURNAL ON IMAGE AND VIDEO PROCESSING, 2018,
[2]   Brain Tumor Classification Using Convolutional Neural Network [J].
Abiwinanda, Nyoman ;
Hanif, Muhammad ;
Hesaputra, S. Tafwida ;
Handayani, Astri ;
Mengko, Tati Rajab .
WORLD CONGRESS ON MEDICAL PHYSICS AND BIOMEDICAL ENGINEERING 2018, VOL 1, 2019, 68 (01) :183-189
[3]   SLIC Superpixels Compared to State-of-the-Art Superpixel Methods [J].
Achanta, Radhakrishna ;
Shaji, Appu ;
Smith, Kevin ;
Lucchi, Aurelien ;
Fua, Pascal ;
Suesstrunk, Sabine .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2012, 34 (11) :2274-2281
[4]   A clustering fusion technique for MR brain tissue segmentation [J].
Al-Dmour, Hayat ;
Al-Ani, Ahmed .
NEUROCOMPUTING, 2018, 275 :546-559
[5]   Big data analysis for brain tumor detection: Deep convolutional neural networks [J].
Amin, Javeria ;
Sharif, Muhammad ;
Yasmin, Mussarat ;
Fernandes, Steven Lawrence .
FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2018, 87 :290-297
[6]   Tree-based Ensemble Classifier Learning for Automatic Brain Glioma Segmentation [J].
Amiri, Samya ;
Mahjoub, Mohamed Ali ;
Rekik, Islem .
NEUROCOMPUTING, 2018, 313 :135-142
[7]   Brain tumour segmentation from MRI using superpixels based spectral clustering [J].
Angulakshmi, Maruthamuthu ;
Priya, Gnanapandithan G. Lakshmi .
JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2020, 32 (10) :1182-1193
[8]   Deep CNN for Brain Tumor Classification [J].
Ayadi, Wadhah ;
Elhamzi, Wajdi ;
Charfi, Imen ;
Atri, Mohamed .
NEURAL PROCESSING LETTERS, 2021, 53 (01) :671-700
[9]   Detection of white matter lesion regions in MRI using SLICO and convolutional neural network [J].
Bandeira Diniz, Pedro Henrique ;
Azevedo Valente, Thales Levi ;
Bandeira Diniz, Joao Otavio ;
Silva, Aristofanes Correa ;
Gattass, Marcelo ;
Ventura, Nina ;
Muniz, Bernardo Carvalho ;
Gasparetto, Emerson Leandro .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2018, 167 :49-63
[10]  
Board CE, 2019, Brain tumor: statistics