Litcius/Paper detail

ResNet-50 vs VGG-19 vs training from scratch: A comparative analysis of the segmentation and classification of Pneumonia from chest X-ray images

Agughasi Victor Ikechukwu, S. Murali, R Deepu, R. C. Shivamurthy

2021Global Transitions Proceedings240 citationsDOIOpen Access PDF

Abstract

In medical imaging, segmentation plays a vital role towards the interpretation of X-ray images where salient features are extracted with the help of image segmentation. Without undergoing surgery, clinicians employ various modalities ranging from X-rays and CT-Scans to ultrasonography, and other imaging techniques to visualise and examine interior human body organ and structures. To ensure appropriate convergence, training a deep convolutional neural network (CNN) from scratch is tough since it requires more computational time, a big amount of labelled training data and a considerable degree of experience. Fine-tuning a CNN that has been pre-trained using, for instance, a huge set of labelled medical datasets, is a viable alternative. In this paper, a comparative study was done using pre-trained models such as VGG-19 and ResNet-50 as against training from scratch. To reduce overfitting, data augmentation and dropout regularization was used. With a recall of 92.03%, our analysis showed that the pre-trained models with proper finetuning was comparable with Iyke-Net, a CNN trained from scratch.

Topics & Concepts

Convolutional neural networkSegmentationComputer scienceArtificial intelligenceOverfittingDeep learningScratchPattern recognition (psychology)Dropout (neural networks)Residual neural networkMedical imagingComputer visionArtificial neural networkMachine learningOperating systemCOVID-19 diagnosis using AILung Cancer Diagnosis and TreatmentRadiomics and Machine Learning in Medical Imaging