Assessing Transfer Learning Models for Medical Image Classification: A Comparative Study on Alzheimer’s MRI, Chest CT-Scan, and Chest X-ray Images
Article Sidebar
Main Article Content
Abstract
Deep learning has revolutionized the field of neural network models, offering limitless applications in various do- mains. This study focuses on Transfer Learning (TL), a technique leveraging pre-trained deep learning models trained on large datasets for image classification tasks. Specifically, this research explores the effectiveness of various transfer learning models in three medical image datasets: Alzheimer’s MRI images, Chest CT-Scan images, and Chest X-ray images. The main objective of this study is to assess and compare the performance of various TL models, including MobileNetV2, ResNet50, Xception, and InceptionV3, on the three medical image datasets. Additionally, a customized Convolutional Neural Network (CNN) model is developed to compare its performance against the pre-trained TL models. Each model was trained and evaluated on the three medical image datasets. The performance of the TL models was compared in terms of accuracy and training time. The results of this study revealed that ResNet50 consistently outperforms other TL models, demonstrating accurate predictions at the expense of longer training times. MobileNetV2 and InceptionV3 exhibit the fastest training times across all datasets, but they demonstrate poorer performance in certain datasets. The developed CNN model performs poorly in terms of accuracy and tends to overfit, indicating that creating a CNN model for medical image classification is not feasible in this study. The findings of this study offer valuable insights into the performance of TL models in medical image datasets. Researchers can utilize this information to make informed decisions when selecting TL models for medical imaging applications. Understanding the strengths and weaknesses of different TL models enhances the potential for accurate and efficient medical image classification. The insights gained from this study contribute to researchers’ understanding of selecting transfer learning models for medical imaging applications, aiding in the advancement of medical image analysis and diagnosis.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
How to Cite
References
M. Raghu, C. Zhang, J. Kleinberg, and S. Bengio, “Transfusion: Understanding transfer learning for medical imaging,” Advances in neural information processing systems, vol. 32, 2019.
C. Hernandez-Matas, X. Zabulis, A. Triantafyllou, P. Anyfanti, S. Douma, and A. Argyros, “Fire: Fundus image registration dataset,” Journal for Modeling in Opthalmology, Jan. 2017. https://www.maio-journal.com/index.php/MAIO/article/view/42/90https://doi.org/10.35119/maio.v1i4.42
C. Chen, Y. Ren, and C.-C. Kuo, “Global-attributes assisted outdoor scene geometric labeling,” in Feb. 2016, pp. 93–120, ISBN: 978-981-10-0629-6. DOI: 10 . 1007 / 978-981-10-0631-9 5.
P. Rajpurkar, J. Irvin, R. L. Ball, et al., “Deep learning for chest radiograph diagnosis: A retrospective comparison of the chexnext algorithm to practicing radiologists,” PLoS medicine, vol. 15, no. 11, e1002686, 2018. https://doi.org/10.1371/journal.pmed.1002686
V. Cheplygina, M. de Bruijne, and J. P. Pluim, “Not-so-supervised: A survey of semi-supervised, multi-instance, and transfer learning in medical image analysis,” Medical image analysis, vol. 54, pp. 280–296, 2019. https://doi.org/10.1016/j.media.2019.03.009
P. Malhotra, S. Gupta, D. Koundal, A. Zaguia, and W. Enbeyle, “Deep neural networks for medical image segmentation,” Journal of Healthcare Engineering, vol. 2022, 2022. https://doi.org/10.1155/2022/9580991
Sk Imran Hossain, Jocelyn de Goër de Herve, Md Shahriar Hassan, Delphine Martineau, Evelina Petrosyan, Violaine Corbin, Jean Beytout, Isabelle Lebert, Jonas Durand, Irene Carravieri, Annick Brun-Jacob, Pascale Frey-Klett, Elisabeth Baux, Céline Cazorla, Carole Eldin, Yves Hansmann, Solene Patrat-Delon, Thierry Prazuck, Alice Raffetin, Pierre Tattevin, Gwenaël Vourc'h, Olivier Lesens, Engelbert Mephu Nguifo, Exploring convolutional neural networks with transfer learning for diagnosing Lyme disease from skin lesion images, Computer Methods and Programs in Biomedicine, Volume 215, 2022,106624, ISSN 01692607,https://doi.org/10.1016/j.cmpb.2022.106624. https://www.sciencedirect.com/science/article/pii/S016 260722000098) https://doi.org/10.1016/j.cmpb.2022.106624
Zhang L, Bian Y, Jiang P, Zhang F. A Transfer Residual Neural Network Based on ResNet-50 for Detection of Steel Surface Defects. Applied Sciences. 2023;13(9):5260. https://doi.org/10.3390/app13095260.
H. K. Kondaveeti and P. Edupuganti, "Skin Cancer Classification using Transfer Learning," 2020 IEEE International Conference on Advent Trends in Multidisciplinary Research and Innovation (ICATMRI), Buldhana, India, 2020, pp. 1-4, doi: 10.1109/ICATMRI51801.2020.9398388. https://doi.org/10.1109/ICATMRI51801.2020.9398388
A. Shah, "Monkeypox Skin Lesion Classification Using Transfer Learning Approach," 2022 IEEE Bombay Section Signature Conference (IBSSC), Mumbai, India, 2022, pp. 1-5, doi: 10.1109/IBSSC56953.2022.10037374. https://doi.org/10.1109/IBSSC56953.2022.10037374
Jaradat AS, Al Mamlook RE, Almakayeel N, Alharbe N, Almuflih AS, Nasayreh A, Gharaibeh H, Gharaibeh M, Gharaibeh A, Bzizi H. Automated Monkeypox Skin Lesion Detection Using Deep Learning and Transfer Learning Techniques. Int J Environ Res Public Health. 2023 Mar 1;20(5):4422. doi: 10.3390/ijerph20054422. PMID: 36901430; PMCID: PMC10001976. https://doi.org/10.3390/ijerph20054422
Qian Xiang, Xiaodan Wang, Rui Li, Guoling Zhang, Jie Lai, Qingshuang Hu, CSAE '19: Proceedings of the 3rd International Conference on Computer Science and Application EngineeringOctober 2019Article No.: 121Pages 1–7https://doi.org/10.1145/3331453.3361658. https://doi.org/10.1145/3331453.3361658
Qin, Y.; Tang, Q.; Xin, J.; Yang, C.; Zhang, Z.; Yang, X. A Rapid Identification Technique of Moving Loads Based on MobileNetV2 and Transfer Learning. Buildings 2023, 13, 572. https://doi.org/10.3390/buildings13020572.
Li T, Huang H, Peng Y, Zhou H, Hu H, Liu M. Quality Grading Algorithm of Oudemansiella raphanipes Based on Transfer Learning and MobileNetV2. Horticulturae. 2022; 8(12):1119. https://doi.org/10.3390/horticulturae8121119.
R. Patel and A. Chaware, "Transfer Learning with Fine-Tuned MobileNetV2 for Diabetic Retinopathy," 2020 International Conference for Emerging Technology (INCET), Belgaum, India, 2020, pp. 1-4, doi: 10.1109/INCET49848.2020.9154014. https://doi.org/10.1109/INCET49848.2020.9154014
S. Dubey, Alzheimer’s Dataset ( 4 class of Images), https: //www.kaggle.com/datasets/tourist55/alzheimers-dataset--4-class-of-images, 2020.
M. Hany, Chest CT-Scan images Dataset), https://www.kaggle.com/datasets/mohamedhanyyy/ches-ctscan-images9. P. Raikote, Covid-19 Image Dataset, https://www.kaggle.com / datasets / pranavraikokte / covid19image-dataset?datasetId=627146&sortBy=voteCount, 2020.