딥러닝 기반 특징 앙상블을 사용한 MRI 영상의 뇌종양 분류
Deep Learning-Based Brain Tumor Classification in MRI images using Ensemble of Deep Features
1
조회
0
다운로드
1
인용
0
참고문헌
키워드
초록
뇌 MRI 영상의 자동 분류는 뇌종양의 조기 진단을 하는 데 있어 중요한 역할을 한다. 본 연구에서 우리는 심층 특징 앙상블을 사용한 MRI 영상에서의 딥 러닝 기반 뇌종양 분류 모델을 제안한다. 우선 사전 학습된 3개의 합성 곱 신경망을 사용하여 입력 MRI 영상에 대한 심층 특징들을추출한다. 그 이후 추출된 심층 특징들은 완전 연결 계층들로 구성된 분류 모듈의 입력 값으로들어간다. 분류 모듈에서는 우선 3개의 서로 다른 심층 특징들 각각에 대해 먼저 완전 연결 계층을 거쳐 특징 차원을 줄인다. 그 이후 3개의 차원이 준 특징들을 결합하여 하나의 특징 벡터를생성한 뒤 다시 완전 연결 계층의 입력값으로 들어가서 최종적인 분류 결과를 예측한다. 우리가제안한 모델을 평가하기 위해 웹상에 공개된 뇌 MRI 데이터 셋을 사용하였다. 실험 결과 우리가제안한 모델이 다른 기계학습 기반 모델보다 더 좋은 성능을 나타냄을 확인하였다.
Abstract
Automatic classification of brain MRI images play an important role in early diagnosis of brain tumors. In this work, we present a deep learning-based brain tumor classification model in MRI images using ensemble of deep features. In our proposed framework, three different deep features from brain MR image are extracted using three different pre-trained models. After that, the extracted deep features are fed to the classification module. In the classification module, the three different deep features are first fed into the fully-connected layers individually to reduce the dimension of the features. After that, the output features from the fully-connected layers are concatenated and fed into the fully-connected layer to predict the final output. To evaluate our proposed model, we use openly accessible brain MRI dataset from web. Experimental results show that our proposed model outperforms other machine learning-based models.
참고문헌 (20)
- H. Selvaraj, "Brain MRI Slices Classification Using Least Squares Support Vector Machine", International Journal of Intelligent Computing in Medical Sciences & Image Processing, 1(1), pp.21~33 (2007), Informa UK Limited[DOI]
- P. John, "Brain tumor classification using wavelet and texture based neural network", International Journal of Scientific & Engineering Research, 3(10), pp.1~7 (2012)
- Zahid Ullah, "A hybrid image enhancement based brain MRI images classification technique", Medical Hypotheses, 143, pp.109922~ (2020), Elsevier BV[DOI]
- A. Kharrat, "A hybrid approach for automatic classification of brain MRI using genetic algorithm and support vector machine", Leonardo journal of sciences, 17(1), pp.71~82 (2010)
- N. Varuna Shree, "Identification and classification of brain tumor MRI images with feature extraction using DWT and probabilistic neural network", Brain Informatics, 5(1), pp.23~30 (2018), Springer Science and Business Media LLC[DOI]
- Murugan Arunachalam, "An efficient and automatic glioblastoma brain tumor detection using shift-invariant shearlet transform and neural networks", International Journal of Imaging Systems and Technology, 27(3), pp.216~226 (2017), Wiley[DOI]
- P. G. Rajan, "Brain Tumor Detection and Segmentation by Intensity Adjustment", Journal of Medical Systems, 43(8) (2019), Springer Science and Business Media LLC[DOI]
- Om Prakash, "Human detection in complex real scenes based on combination of biorthogonal wavelet transform and Zernike moments", Optik, 157, pp.1267~1281 (2018), Elsevier BV[DOI]
- T.K.K. Ho, "Recent trends in communication, computing, and electronics", pp.431~440 (2019), Springer
- S. Deepak, "Brain tumor classification using deep CNN features via transfer learning", Computers in Biology and Medicine, 111, pp.103345~ (2019), Elsevier BV[DOI]
- Ahmet Çinar, "Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture", Medical Hypotheses, 139, pp.109684~ (2020), Elsevier BV[DOI]
- Saed Khawaldeh, "Noninvasive Grading of Glioma Tumor Using Magnetic Resonance Imaging with Convolutional Neural Networks", Applied Sciences, 8(1), pp.27~ (2018), MDPI AG[DOI]
- P. Saxena, "Innovations in Computational Intelligence and Computer Vision", pp.275~285 (2021), Springer
- P.M. Siva Raja, "Brain tumor classification using a hybrid deep autoencoder with Bayesian fuzzy clustering-based segmentation approach", Biocybernetics and Biomedical Engineering, 40(1), pp.440~453 (2020), Elsevier BV[DOI]
- Jaeyong Kang, "MRI-Based Brain Tumor Classification Using Ensemble of Deep Features and Machine Learning Classifiers", Sensors, 21(6), pp.2222~ (2021), MDPI AG[DOI]
- Alex Krizhevsky, "ImageNet classification with deep convolutional neural networks", Communications of the ACM, 60(6), pp.84~90 (2017), Association for Computing Machinery (ACM)[DOI]
- G. Huang, "Densely connected convolutional networks", Proc. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), pp.4700~4708 (2017)
- N. Ma, "Shufflenet v2 : Practical guidelines for efficient cnn architecture design", Proc. Eur. Conf. Comput. Vis. (ECCV), 2018, pp.116~131 (2018)
- M. Tan, "Mnasnet : Platform-aware neural architecture search for mobile", Proc. IEEE Conf. Comput. Vis. Pattern Recognit. (CVPR), 2019, pp.2820~2828 (2019)
- "Brain Tumor Classification (MRI) Dataset" (2020)[URL]