Machine Learning-Aided Tool Generates High-Quality Chest X-Ray Images to Diagnose COVID-19 More Accurately

The need for rapid and accurate COVID-19 testing is high, including testing that can determine if COVID-19 is impacting a patient's respiratory system. Many clinicians use X-ray technology to classify images of possible cases of COVID-19, but the limited data available makes it more challenging to classify those images accurately.

The UMBC researchers developed their tool as an extension of generative adversarial networks (GANs) - machine learning frameworks that can quickly generate new data based on statistics from a training set. The team's more advanced method uses what they call Mean Teacher + Transfer Generative Adversarial Networks (MTT-GAN). The MTT-GANs are superior to GANs because the images they generate are much more similar to authentic images generated by x-ray machines. The MTT-GAN classification system has the potential to help improve the accuracy of COVID-19 classifiers, making it an important diagnostic tool for physicians who are still working to understand the range of ways this complex disease presents in patients.

"The availability of data is one of the most important aspects of machine learning and our research has taken an incremental theoretical step towards generating data using the MTT-GAN," said Sumeet Menon, a Ph.D. student in computer science at UMBC who led the research team. "This paper mainly focuses on generating more COVID-19 X-rays using the MTT-GAN, which could be widely used to train machine learning models and could have many applications, including classification of CT-scans and segmentation."

Related Links:
University of Maryland, Baltimore County