Abstract
Emergence of convolutional neural networks (CNNs) have offered better predictive performance and the possibility to replace traditional workflows with single network architecture. Recently developed MRNet CNN for the Knee MRI dataset has used AlexNet for their transfer learning implementation. This paper explores the effect of structural variations, data augmentation and various transfer learning implementations on the performance of a deep neural network in the classification task of knee MRI. Modifications of MRNet were generated by freezing the layers of the AlexNet backbone, replacing the backbone network AlexNet to other and applying the valid data augmentation techniques used on the dataset prior to input to the network. AlexNet based CNNs with layer-freezing achieved AUC for Abnormal, ACL lesion and Meniscal lesion classification of 0.913, 0.859, 0.792, an improvement over no layer freezing which had AUCs of 0.896, 0.842 and 0.773. although the result is less than that reported by Stanford’s AlexNet based classifiers of 0.937, 0.965 and 0.847 AUC. ResNet18 based classifier achieved AUCs of 0.843, 0.774, 0.671. VGG16 based classifier achieved AUCs of 0.728 0.690 0.711. Using color jitter for data augmentation resulted 0.938 AUC in abnormal classification.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Hosny, K.M, Kassem, M.A., Foaud, M.M.: Classification of skin lesions using transfer learning and augmentation with alex-net. PLOS ONE 14, e0217293 (2019)
Maqsood, M., et al.: Transfer learning assisted classification and detection of alzheimer’s disease stages using 3D MRI scans. Sensors 19, 2645 (2019)
Choi, J.Y., Yoo, T.K., Seo, J.G., Kwak, J., Um, T.T., Rim, T.H.: Multicategorical deep learning neural network to classify retinal images: a pilot study employing small database. PLOS ONE 12, e0187336 (2017)
Pardamean, B., Cenggoro, T.W., Rahutomo, R., Budiarto, A., Karuppiah, E.K.: Transfer learning from chest x-ray pre-trained convolutional neural network for learning mammogram data, Procedia Comput. Sci. 135, 400–407, 2018. The 3rd International Conference on Computer Science and Computational Intelligence (ICCSCI 2018): Empowering Smart Technology in Digital Era for a Better Life (2018)
Shallu, Mehra, R.: Breast cancer histology images classification: training from scratch or transfer learning. ICT Express 4(4), 247–254 (2018)
Kim, M., Zuallaert, J., De Neve W.: Towards novel methods for effective transfer learning and unsupervised deep learning for medical image analysis, pp. 32–39 (2017)
Bien, N., et al.: Deep-learning-assisted diagnosis for knee magnetic resonance imaging: development and retrospective validation of MRNet. PLOS Med. 15, e1002699 (2018)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition (2014)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. CoRR, abs/1512.03385 (2015)
Szegedy, C., Ioffe, S., Vanhoucke, V.: Inception-v4, Inception-ResNet and the impact of residual connections on learning, CoRR, abs/1602.07261 (2016)
Ioffe, S. Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift (2015)
Irmakci, I., Anwar, S.M., Torigian, D.A., Bagci, U.: Deep learning for musculoskeletal image analysis (2020)
Talo, M., Baloglu, U.B., Aydin, G., Acharya, U.R.: Convolutional neural networks for multi-class brain disease detection using MRI images. Comput. Med. Imag. Graph. 78, 101673 (2019)
Li, H., et al.: Fully convolutional network ensembles for white matter hyperintensities segmentation in MR images. Neuroimage 183, 650–665 (2018). https://doi.org/10.1016/j.neuroimage.2018.07.005. Epub PMID 30125711
Author information
Authors and Affiliations
Corresponding authors
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
ďż˝ 2022 Springer Nature Switzerland AG
About this paper
Cite this paper
Haddadian, J., Balamurali, M. (2022). Transfer Learning and Data Augmentation in the Diagnosis of Knee MRI. In: Long, G., Yu, X., Wang, S. (eds) AI 2021: Advances in Artificial Intelligence. AI 2022. Lecture Notes in Computer Science(), vol 13151. Springer, Cham. https://doi.org/10.1007/978-3-030-97546-3_37
Download citation
DOI: https://doi.org/10.1007/978-3-030-97546-3_37
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-97545-6
Online ISBN: 978-3-030-97546-3
eBook Packages: Computer ScienceComputer Science (R0)