A Comparative Analysis of Transfer Learning Methods for Medical Super-Resolution
DOI:
https://doi.org/10.54097/6pvy8418Keywords:
Transfer learning; image super-resolution; deep learning.Abstract
Precisely and rapidly acquiring high-resolution medical images with a limited radiation dose is of great importance in clinical diagnosis. However, due to the challenges in gathering medical images and the high costs of developing a deep learning model from scratch, the super-resolution of medical images is still a tough problem. To deal with above, transfer learning is a favorable technique that leverages a previously trained model for a different but related task, significantly cutting down on the amount of training data and time needed for the downstream task. In this study, a pre-trained U-shaped Network++ (UNet++) model is initially trained using the ImageNet dataset to learning the general information. Subsequently, multiple finetuning methods are adopted to transfer the natural images super-resolution to the super-resolution of skin lesion images. 1279 manually annotated skin lesion images are selected for training and testing. The results indicate that transfer learning not only improve the convergence speed of deep learning models but also generate clearer high-resolution skin lesion images. Compared to the normal training group, the Mean Square Error (MSE) value was dropped by 24% to 70%. Notably, BitFit might lead to worse performance.
Downloads
References
[1] Mingfeng Jiang, Minghao Zhi, Liying Wei, et al. FA-GAN: Fused attentive generative adversarial networks for MRI image super-resolution. Computerized Medical Imaging and Graphics, 2021, 92: 101969.
[2] Zhisheng Lu, Juncheng Li, Hong Liu, et al. Transformer for single image super-resolution. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, 2022: 457-466.
[3] Lepcha Dawa Chyophel, Goyal Bhawna, Dogra Ayush, et al. Image super-resolution: A comprehensive review, recent trends, challenges and applications. Information Fusion, 2023, 91: 230-260.
[4] Jiang Mingfeng, Zhi Minghao, Wei Liying, et al. FA-GAN: Fused attentive generative adversarial networks for MRI image super-resolution. Computerized Medical Imaging and Graphics, 2021, 92: 101969.
[5] Li Yufei, Bruno Sixou, and Francois Peyrin. A review of the deep learning methods for medical images super resolution problems. Irbm, 2021, 42(2): 120-133.
[6] Park Junyoung, Hwang Donghwi, Kim Kyeong Yun, et al. Computed tomography super-resolution using deep convolutional neural network. Physics in Medicine & Biology, 2018, 63(14): 145011.
[7] Gu Yuchong, Zeng Zitao, Chen Haibin, et al. MedSRGAN: medical images super-resolution using generative adversarial networks. Multimedia Tools and Applications, 2020, 79: 21815-21840.
[8] Iman Mohammadreza, Hamid Reza Arabnia, Khaled Rasheed. A review of deep transfer learning and recent advancements. Technologies, 2023, 11(2): 40.
[9] Voghoei Sahar, Tonekaboni Navid Hashemi, Wallace Jason, et al. Deep learning at the edge. International Conference on Computational Science and Computational Intelligence, 2018: 895-901.
[10] Gutman David, Codella Noel, Celebi Emre, et al. Skin lesion analysis toward melanoma detection: A challenge at the international symposium on biomedical imaging (ISBI) 2016, hosted by the international skin imaging collaboration (ISIC). arXiv preprint arXiv, 2016: 1605.01397.
[11] Deng Jia, Dong Wei, Socher Richard, et al. Imagenet: A large-scale hierarchical image database. IEEE conference on computer vision and pattern recognition. 2009: 248-255.
[12] Zhou Zongwei, Rahman Siddiquee Md Mahfuzur, Tajbakhsh Nima, et al. Unet++: A nested u-net architecture for medical image segmentation. Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, 2018: 3-11.
[13] Zaken Elad Ben, Shauli Ravfogel, Yoav Goldberg. Bitfit: Simple parameter-efficient fine-tuning for transformer-based masked language-models. arXiv preprint arXiv, 2021: 2106.10199.
[14] Chen Shoufa, Ge Chongjian, Tong Zhan, et al. Adaptformer: Adapting vision transformers for scalable visual recognition. Advances in Neural Information Processing Systems, 2022, 35: 16664-16678.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Highlights in Science, Engineering and Technology

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.







