|
This article is cited in 1 scientific paper (total in 1 paper)
On transfer learning methods in biomedical images classification tasks
E. Yu. Shchetinina, L. A. Sevastianovbc a Financial University under the Government of the Russian Federation, 49 Leningradsky Prospekt, Moscow 125993, Russian Federation
b Peoples' Friendship University of Russia (RUDN University), 6 Miklukho-Maklaya Str., Moscow 117198, Russian Federation
c Joint Institute for Nuclear Research, 6 Joliot-Curie Str., Dubna, Moscow Region 141980, Russian Federation
Abstract:
Computer studies of the effectiveness of deep transfer learning methods for solving the problem of human brain tumors recognition based on magtetic resonance imaging (MRI) are carried out. Various strategies of transfer learning and fine-tuning of the models are proposed and implemented. Deep convolutional networks VGG-16, ResNet-50, Xception, and MobileNetV2 were used as the baseline models, pre-trained on ImageNet. Also, a deep convolutional neural network 2D_CNN was built and trained from scratch. Computer analysis of their performance metrics showed that when using the strategy of fine-tuning models on augmented MRI-scans data set, Xception model demonstrated higher accuracy values compared to other deep learning models. For Xception model, the accuracy of classification of MRI-scans with brain tumors was 96%, precision 99.43%, recall 96.03%, f1-score 97.7%, and AUC 98.92%.
Keywords:
MRI scans, brain tumor, deep learning transfer, convolutional neural networks.
Received: 12.09.2021
Citation:
E. Yu. Shchetinin, L. A. Sevastianov, “On transfer learning methods in biomedical images classification tasks”, Inform. Primen., 15:4 (2021), 59–64
Linking options:
https://www.mathnet.ru/eng/ia757 https://www.mathnet.ru/eng/ia/v15/i4/p59
|
Statistics & downloads: |
Abstract page: | 159 | Full-text PDF : | 96 | References: | 33 |
|