Adopting transfer learning for neuroimaging: a comparative analysis with a custom 3D convolution neural network modelShow others and affiliations
2022 (English)In: BMC Medical Informatics and Decision Making, E-ISSN 1472-6947, Vol. 22, p. 1-15, article id 318Article in journal (Refereed) Published
Abstract [en]
Background: In recent years, neuroimaging with deep learning (DL) algorithms have made remarkable advances in the diagnosis of neurodegenerative disorders. However, applying DL in different medical domains is usually challenged by lack of labeled data. To address this challenge, transfer learning (TL) has been applied to use state-of-the-art convolution neural networks pre-trained on natural images. Yet, there are differences in characteristics between medical and natural images, also image classification and targeted medical diagnosis tasks. The purpose of this study is to investigate the performance of specialized and TL in the classification of neurodegenerative disorders using 3D volumes of 18F-FDG-PET brain scans. Results: Results show that TL models are suboptimal for classification of neurodegenerative disorders, especially when the objective is to separate more than two disorders. Additionally, specialized CNN model provides better interpretations of predicted diagnosis. Conclusions: TL can indeed lead to superior performance on binary classification in timely and data efficient manner, yet for detecting more than a single disorder, TL models do not perform well. Additionally, custom 3D model performs comparably to TL models for binary classification, and interestingly perform better for diagnosis of multiple disorders. The results confirm the superiority of the custom 3D-CNN in providing better explainable model compared to TL adopted ones. © 2022, The Author(s).
Place, publisher, year, edition, pages
London: BioMed Central (BMC), 2022. Vol. 22, p. 1-15, article id 318
Keywords [en]
Brain Neurodegenerative Disorders, Convolution Neural Networks, Medical Image Classification, Transfer Learning
National Category
Computer Vision and Robotics (Autonomous Systems)
Identifiers
URN: urn:nbn:se:hh:diva-49079DOI: 10.1186/s12911-022-02054-7ISI: 000904994900001PubMedID: 36476613Scopus ID: 2-s2.0-85143570393OAI: oai:DiVA.org:hh-49079DiVA, id: diva2:1725115
2023-01-102023-01-102023-08-21Bibliographically approved