CAGAN: Classifier‐augmented generative adversarial networks for weakly‐supervised COVID‐19 lung lesion localisation

• Published in: IET computer vision Vol. 18; no. 1; pp. 1 - 14
• Main Authors: Li, Xiaojie, Fei, Xin, Yan, Zhe, Ren, Hongping, Shi, Canghong, Zhang, Xian, Mumtaz, Imran, Luo, Yong, Wu, Xi
• Format: Journal Article
• Language: English
• Published: Stevenage John Wiley & Sons, Inc 01.02.2024; Wiley
• Subjects: Abnormalities; Algorithms; Artificial intelligence; biomedical MRI; Computed tomography; computer graphics; COVID-19; Deep learning; Discriminators; Disease; Feature maps; Generative adversarial networks; Health care; Lesions; Localization; Lungs; Machine learning; Medical imaging; Methods; patient diagnosis; Patients; Pneumonia; Public health; Severe acute respiratory syndrome coronavirus 2; China
• ISSN: 1751-9632; 1751-9640
• DOI: 10.1049/cvi2.12216

The Coronavirus Disease 2019 (COVID‐19) epidemic has constituted a Public Health Emergency of International Concern. Chest computed tomography (CT) can help early reveal abnormalities indicative of lung disease. Thus, accurate and automatic localisation of lung lesions is particularly important to assist physicians in rapid diagnosis of COVID‐19 patients. The authors propose a classifier‐augmented generative adversarial network framework for weakly supervised COVID‐19 lung lesion localisation. It consists of an abnormality map generator, discriminator and classifier. The generator aims to produce the abnormality feature map M to locate lesion regions and then constructs images of the pseudo‐healthy subjects by adding M to the input patient images. Besides constraining the generated images of healthy subjects with real distribution by the discriminator, a pre‐trained classifier is introduced to enhance the generated images of healthy subjects to possess similar feature representations with real healthy people in terms of high‐level semantic features. Moreover, an attention gate is employed in the generator to reduce the noise effect in the irrelevant regions of M. Experimental results on the COVID‐19 CT dataset show that the method is effective in capturing more lesion areas and generating less noise in unrelated areas, and it has significant advantages in terms of quantitative and qualitative results over existing methods. (1) The authors propose an effective classifier‐augmented generative adversarial network framework for COVID‐19 lung lesion localisation, which provides a more accurate feature map indicating the lesion regions. The proposed framework incorporating the pre‐trained classifier enforces the output of the generator to have similar intermediate feature representations (M) with normal people and thus leads to improved precise lesion localisation. (2) The authors construct an L1 norm reconstruction loss and regularisation loss on M, which keep the patient’s lung structure unchanged when lesion location maps are generated.