О методах переноса глубокого обучения в задачах классификации биомедицинских изображений

Проведены компьютерные исследования эффективности применения методов переноса глубокого обучения для решения задачи распознавания опухолей головного мозга человека на основе его МРТ-снимков. Предложены и реализованы различные стратегии глубокого обучения и тонкой настройки моделей. В качестве базовых моделей были использованы глубокие сверточные сети VGG-16, ResNet-50, Xception и MobileNetV2, предварительно обученные на наборе изображений ImageNet. Также разработана и обучена глубокая сверточная нейронная сеть 2D-CNN. Компьютерный анализ показателей их производительности показал, что стратегия тонкой настройки модели Xception на расширенном наборе данных продемонстрировала более высокие значения точности по сравнению с другими моделями глубокого обучения: точность классификации опухолей головного мозга по МРТ-снимкам составила 96%, precision - 99,9%, recall - 96,03%, f1-score - 98%, AUC - 98,92%.

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%.

Authors
Щетинин Е.Ю.1 , Севастьянов Л.А. 2, 3
Publisher
Федеральный исследовательский центр "Информатика и управление" РАН
Number of issue
4
Language
Russian
Pages
59-64
Status
Published
Volume
15
Year
2021
Organizations
  • 1 Финансовый университет при Правительстве РФ
  • 2 Российский университет дружбы народов
  • 3 Объединенный институт ядерных исследований
Keywords
MRI scans; brain tumor; deep learning transfer; convolutional neural networks; МРТ-снимки; опухоль головного мозга; перенос глубокого обучения; сверточные нейронные сети
Date of creation
19.07.2022
Date of change
19.07.2022
Short link
https://repository.rudn.ru/en/records/article/record/91115/
Share

Other records

Кочеткова И.А., Кущазли А.И., Харин П.А., Шоргин С.Я.
Informatika i ee Primeneniya. Федеральный исследовательский центр "Информатика и управление" РАН. Vol. 15. 2021. P. 87-92