Brain MRI image classification based on transfer learning and support vector machine

Zimeng Li; Yan Zhao; Shigang Wang

doi:10.1117/12.2538076

18 November 2019 Brain MRI image classification based on transfer learning and support vector machine

Zimeng Li, Yan Zhao, Shigang Wang

Proceedings Volume 11187, Optoelectronic Imaging and Multimedia Technology VI; 111871P (2019) https://doi.org/10.1117/12.2538076
Event: SPIE/COS Photonics Asia, 2019, Hangzhou, China

Abstract

In order to improve the accuracy of brain magnetic resonance imaging (MRI) classification and to reduce classification time, this paper proposes a brain MRI image classification algorithm combined with transfer learning and support vector machine (SVM). Firstly, the three deep convolutional neural networks pre-trained on the ImageNet database, AlexNet, VGG16 and GoogleNet, are used as feature extractors to extract the features of brain MRI images in the Harvard Medical School Website database. The feature extraction process does not require to fine-tune the transferred networks. Then, the features extracted in each convolutional neural network are combined to form a feature vector of each brain MRI image and it is input to the SVM for classification. Finally the SVM classified the brain MRI images into healthy, Alzheimer's disease, and stroke. The experimental results show that the classification accuracy can achieve 100%, and the classification time is only 26 seconds. Compared with the brain MRI image classification algorithm proposed by the literature, the accuracy of the proposed method is increased by 8.67%, 1.09%, 0.55%, respectively. The proposed method can provide effective help for the diagnosis of brain diseases.

Citation Download Citation

Zimeng Li, Yan Zhao, and Shigang Wang "Brain MRI image classification based on transfer learning and support vector machine", Proc. SPIE 11187, Optoelectronic Imaging and Multimedia Technology VI, 111871P (18 November 2019); https://doi.org/10.1117/12.2538076

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available