SD-GAN: A Style Distribution Transfer Generative Adversarial Network for Covid-19 Detection Through X-Ray Images

Abstract

The Covid-19 pandemic is a prevalent health concern around the world in recent times. Therefore, it is essential to screen the infected patients at the primary stage to prevent secondary infections from person to person. The reverse transcription polymerase chain reaction (RT-PCR) test is commonly performed for Covid-19 diagnosis, while it requires significant effort from health professionals. Automated Covid-19 diagnosis using chest X-ray images is one of the promising directions to screen infected patients quickly and effectively. Automatic diagnostic approaches are used with the assumption that data originating from different sources have the same feature distributions. However, the X-ray images generated in different laboratories using different devices experience style variations e.g., intensity and contrast which contradict the above assumption. The prediction performance of deep models trained on such heterogeneous images of different distributions with different noises is affected. To address this issue, we have designed an automatic end-to-end adaptive normalization-based model called style distribution transfer generative adversarial network (SD-GAN). The designed model is equipped with the generative adversarial network (GAN) and task-specific classifier to transform the style distribution of images between different datasets belonging to different race people and carried out Covid-19 detection effectively. Evaluated results on four different X-ray datasets show the superiority of the proposed model to state-of-the-art methods in terms of the visual quality of style transferred images and the accuracy of Covid-19 infected patient detection. SD-GAN is publicly available at: https://github.com/tasleem-hello/SD-GAN/tree/SD-GAN.