Utilising VGG-16 of Convolutional Neural Network for Medical Image Classification

Abstract

Medical image classification, which involves accurately classifying anomalies or abnormalities within images, is an important area of attention in healthcare domain. It requires a fast and exact classification to ensure appropriate and timely treatment to the patients. This paper introduces a model based on Convolutional Neural Network (CNN) that utilises the VGG16 architecture for medical image classification, specifically in brain tumour and Alzheimer dataset. The VGG16 architecture, is known for its remarkable ability to extract important features, that is crucial in medical image classification. To enhance the precision of diagnosis, a detailed experimental setup is conducted, which includes the careful selection and organisation of a collection of medical images that cover different illnesses and anomalies to the dataset. The architecture of the model is then adjusted to achieve optimal performance in for image classification. The results show the model's efficiency in identifying anomalies in medical images especially for brain tumour dataset. The sensitivity, specificity, and F1-score evaluation metrics are presented, emphasising the model's ability to accurately differentiate between various medical image diseases.