Evaluating Custom and Transfer Learning Convolutional Neural Networks for Pneumonia Classification Under Data Scarcity and Low-Compute Conditions for Under-Resourced Environments

在数据稀缺和计算资源不足的环境下,评估用于肺炎分类的定制和迁移学习卷积神经网络

阅读:1

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.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。