Anglia Ruskin Research Online (ARRO)
Browse
frai-06-1235204.pdf (3.81 MB)

COVID-19 lateral flow test image classification using deep CNN and StyleGAN2

Download (3.81 MB)
journal contribution
posted on 2024-02-02, 16:00 authored by Vishnu Pannipulath Venugopal, Lakshmi Babu Saheer, Mahdi Maktabdar Oghaz

Artificial intelligence (AI) in healthcare can enhance clinical workflows and diagnoses, particularly in large-scale operations like COVID-19 mass testing. This study presents a deep Convolutional Neural Network (CNN) model for automated COVID-19 RATD image classification.MethodsTo address the absence of a RATD image dataset, we crowdsourced 900 real-world images focusing on positive and negative cases. Rigorous data augmentation and StyleGAN2-ADA generated simulated images to overcome dataset limitations and class imbalances.ResultsThe best CNN model achieved a 93% validation accuracy. Test accuracies were 88% for simulated datasets and 82% for real datasets. Augmenting simulated images during training did not significantly improve real-world test image performance but enhanced simulated test image performance.DiscussionThe findings of this study highlight the potential of the developed model in expediting COVID-19 testing processes and facilitating large-scale testing and tracking systems. The study also underscores the challenges in designing and developing such models, emphasizing the importance of addressing dataset limitations and class imbalances.ConclusionThis research contributes to the deployment of large-scale testing and tracking systems, offering insights into the potential applications of AI in mitigating outbreaks similar to COVID-19. Future work could focus on refining the model and exploring its adaptability to other healthcare scenarios.

History

Refereed

  • Yes

Volume

6

Publication title

Frontiers in Artificial Intelligence

ISSN

2624-8212

Publisher

Frontiers Media SA

File version

  • Published version

Affiliated with

  • School of Computing and Information Science Outputs

Usage metrics

    ARU Outputs

    Licence

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC