An Experiment on Lung Disease Classification using YOLOv8

Abstract

Lung diseases are a leading cause of illness and mortality worldwide. Accurate and timely diagnosis is crucial for improving patient outcomes, but manual interpretation of chest X-rays by radiologists may consume a significant amount of time and is susceptible to errors. This study leverages the state-of-the-art YOLOv8 deep learning model to develop an automated system for classifying lung diseases from chest X-ray images. The proposed approach utilizes a large, diverse dataset of 21,165 chest X-ray images categorized into four classes: COVID-19, viral pneumonia, lung opacity, and normal. The YOLOv8-cls model is fine-tuned using transfer learning and advanced data augmentation techniques. The system achieves a high accuracy of approximately 95% across all disease classes while maintaining real-time performance. Confusion matrix analysis demonstrates strong performance in correctly identifying each condition, with COVID-19, normal, viral pneumonia and lung opacity cases correctly identified 97%, 97%, 99% and 93% of the time respectively. The outcomes validate the adaptability and reliability of deep learning, specifically YOLOv8, for automated lung disease detection, offering the potential to improve clinical workflows and patient care by providing efficient screening tools for practitioners. The study addresses limitations of previous works by utilizing a large consolidated dataset, reporting comprehensive computational efficiency metrics, and extending classification complexity without substantially impacting accuracy or speed. Future research should focus on further optimizations and extending the curated image repositories to include under-represented patient groups to enhance the model's inclusiveness and robustness.