Harnessing ResNet50 and DenseNet201 for Enhanced Lymphoma Diagnosis via Feature Extraction

Abstract

Introduction: A form of malignant tumour known as lymphoma originated in lymphoid hematopoietic organs. Because the physical characteristics of the many lymphoma classes are similar, accurately diagnosing lymphomas is one of the most difficult tasks. Hence, an efficient classification of lymphoma plays a very important role in order to provide patients with prompt care. The purpose of this work is to evaluate the performance of pre-trained Convolutional Neural Networks (CNNs) in the multiclass categorization of lymphomas.

Objectives: Classification of Non-Hodgkin lymphomas by adopting pre-trained CNN architectures like ResNet50, VGG16, InceptionV3 and DenseNet201 are adopted. Utilize several pre-processing techniques for denoising, rescaling, and enriching the input images, including gaussian filter, min-max normalisation, and data augmentation. Perform a detailed performance analysis of the proposed work with existing models.

Methods: This research uses the different CNN architectures such as VGG16, DenseNet201, InceptionV3 to classify the lymphoma. In pre-processing, the gaussian filter is used to denoise and smoothen the images, min-max normalization is used to rescale the images and the data augmentation is used for solving the data imbalance issue. Transfer Learning and Fine-Tuning is done which improves the overall performance of the model.

Results: This study makes use of the multi cancer dataset from Kaggle. The performance of these pre-trained CNN models is evaluated using accuracy, precision, recall, and the F-measure. Based on simulation findings, DenseNet201 outperforms VGG16 and InceptionV3 with an accuracy of 99.90%. Furthermore, FFNN-ResNet50 and HPC are two current studies that are used to compare ResNet50 and DenseNet201. ResNet50-DenseNet201 has a high accuracy of 99.90% compared to FFNN-ResNet50 and HPC.

Conclusions: Several CNN architectures, including VGG16, InceptionV3 and DenseNet201 are employed in this study to categorize lymphomas. Several NHL classifications, including FL, CLL, and MCL, are classified using the pre-trained CNN architecture. The gaussian filter, which aids in smoothing the pictures, is used to eliminate noise from the histopathology images. The pixel limits are then scaled using min-max normalization to increase pixel intensity, and data augmentation is employed to prevent data imbalance problems. Improved categorization is achieved by the ResNet50 by extracting multi-scale characteristics from the images. Based on the simulation findings, it is evident that DenseNet201, which incorporates ResNet50 features, outperforms VGG16 and InceptionV3 due to the intricate interactions between data that dense connectivity enables. Furthermore, ResNet50-DenseNet201 performs better than FFNN-ResNet50 and HPC. In comparison to FFNN-ResNet50 and HPC, ResNet50-DenseNet201 has a high accuracy of 99.90%.