Application of Deep Transfer Learning in Medical Imaging for Thyroid Lesion Diagnostic Assistance

Abstract

This academic work evaluates and compares the performance of various deep convolutional neural network (DCNN) architectures in classifying thyroid nodules into two categories, malignant and benign, using ultrasound images. The dataset comprises 269 cases of benign lesions and 526 cases of malignant lesions. Given the limited dataset size, we employ a progressive learning approach with three established CNN models: VGG-16, ResNet-50, and EfficientNet. Initially pretrained on ImageNet, these models undergo further fine-tuning using a radiographic image dataset related to a different medical condition but similar to our domain. Different levels and fine-tuning strategies are applied to these models. A supervised softmax classifier is used for classifying lesions as malignant or benign, with the exception of the VGG-16 model. For the VGG-16 model, two additional classifiers, Support Vector Machine (SVM) and Random Forest (RF), are evaluated. The results obtained demonstrate the possibility of easily transitioning from the classification of one disease to another, even with a limited number of images, by leveraging the knowledge already acquired from another extensive database.