Classification of MR Brain Images for Detection of Tumor with Transfer Learning from Pre-trained CNN Models
Autor: | R. Shantha Selva Kumari, R. Meena Prakash |
---|---|
Rok vydání: | 2019 |
Předmět: |
medicine.diagnostic_test
Computer science business.industry Brain tumor Pattern recognition Magnetic resonance imaging 02 engineering and technology medicine.disease Convolutional neural network Residual neural network 030218 nuclear medicine & medical imaging 03 medical and health sciences 0302 clinical medicine Softmax function 0202 electrical engineering electronic engineering information engineering medicine 020201 artificial intelligence & image processing Artificial intelligence Transfer of learning business |
Zdroj: | 2019 International Conference on Wireless Communications Signal Processing and Networking (WiSPNET). |
Popis: | Magnetic resonance imaging is one of the non-invasive imaging techniques widely employed to diagnose brain diseases. Early diagnosis and treatment of brain tumors is essential. It is a time consuming process for the radiologists to manually classify MR brain images into normal and images with tumors. In this paper, an automated method based on Convolutional Neural Network (CNN) is proposed for detection of tumor in brain images. The CNN model pre-trained on the huge image database of ImageNet, is used to train the input brain images. The high level features extracted are given as input to the fully connected layer followed by softmax activation. The method is tested on MR brain images from database of Harvard medical school. An analysis is done with utilization of three pre-trained models – VGG16, ResNet and Inception. It is able to achieve an accuracy of 100% on the experimented database. Also, from the results, it is inferred that data augmentation improves classification accuracy. |
Databáze: | OpenAIRE |
Externí odkaz: |