Evaluating Custom and Transfer Learning Convolutional Neural Networks for Pneumonia Classification Under Data Scarcity and Low-Compute Conditions for Under-Resourced Environments
Year: 2025
Type: article
Abstract: Pneumonia is an infectious disease that causes many deaths across the world. By accurately and efficiently recognizing pneumonia from images using machine learning, we can facilitate faster treatment. Yet, in under-resourced environments, there is often a lack of training data, high-quality images for training, as well as a lack of high computing power. As such, we aim to investigate and address these issues. In this paper, we deployed and tested different convolutional neural networks (CNNs), optimizing them to use fewer resources. We simulated the low training image count, low image quality, and low compute power, using a Raspberry Pi and the Pneumonia Modified National Institute of Standards and Technology (MNIST) dataset with 64x64 images. We were able to achieve accuracies of 95% for our laptop-trained, quantized model and 94% for our P100 quantized model. The F1 scores, as well as the recall for the pneumonia class, were also evaluated to provide additional insights into the performance of the models. Furthermore, we were able to obtain a rounded average latency of 1.46 milliseconds (ms) for our P100 quantized model, with a rounded model size of 1.45 megabytes (MB). Our work allows for an accurate diagnosis of pneumonia with limited resources.
Source: Cureus
DOI: 10.7759/cureus.97959
Authors: Andy L Chen, Eric Cheek, Morteza Samradi
Institutions: University of Michigan, Massachusetts Institute of Technology
Language: English
Topic: COVID-19 diagnosis using AI
Subfield: Radiology, Nuclear Medicine and Imaging
Field: Medicine
Domain: Health Sciences
Open Access status: diamond
OpenAlex ID: W4416756341