Semantic-Powered Explainable Model-Free Few-Shot Learning Scheme of Diagnosing COVID-19 on Chest X-Ray

Chest X-ray (CXR) is commonly performed as an initial investigation in COVID-19, whose fast and accurate diagnosis is critical. Recently, deep learning has a great potential in detecting people who are suspected to be infected with COVID-19. However, deep learning resulting with black-box models, which often breaks down when forced to make predictions about data for which limited supervised information is available and lack inter-pretability, still is a major barrier for clinical integration. In this work, we hereby propose a semantic-powered explainable model-free few-shot learning scheme to quickly and precisely diagnose COVID-19 with higher reliability and transparency. Specifically, we design a Report Image Explanation Cell (RIEC) to exploit clinically indicators derived from radiology reports as interpretable driver to introduce prior knowledge at training. Meanwhile, multi-task collaborative diagnosis strategy (MCDS) is developed to construct ${\boldsymbol{N}}$-way ${\boldsymbol{K}}$-shot tasks, which adopts a cyclic and collaborative training approach for producing better generalization performance on new tasks. Extensive experiments demonstrate that the proposed scheme achieves competitive results (accuracy of 98.91%, precision of 98.95%, recall of 97.94% and F1-score of 98.57%) to diagnose COVID-19 and other pneumonia infected categories, even with only 200 paired CXR images and radiology reports for training. Furthermore, statistical results of comparative experiments show that our scheme provides an interpretable window into the COVID-19 diagnosis to improve the performance of the small sample size, the reliability and transparency of black-box deep learning models. Our source codes will be released on https://github.com/AI-medical-diagnosis-team-of-JNU/SPEMFSL-Diagnosis-COVID-19.

[1]  K. He,et al.  An interpretable machine learning model based on a quick pre-screening system enables accurate deterioration risk prediction for COVID-19 , 2021, Scientific Reports.

[2]  S. Satapathy,et al.  SARS-Net: COVID-19 detection from chest x-rays by combining graph convolutional network and convolutional neural network , 2021, Pattern Recognition.

[3]  Qing Cai,et al.  COVID-19 Screening in Chest X-Ray Images Using Lung Region Priors , 2021, IEEE Journal of Biomedical and Health Informatics.

[4]  Yizhou Yu,et al.  Cross-Modal Self-Attention with Multi-Task Pre-Training for Medical Visual Question Answering , 2021, ICMR.

[5]  May D. Wang,et al.  COVID-19 Automatic Diagnosis With Radiographic Imaging: Explainable Attention Transfer Deep Neural Networks , 2021, IEEE Journal of Biomedical and Health Informatics.

[6]  G. Alexopoulos,et al.  Frequency and profile of objective cognitive deficits in hospitalized patients recovering from COVID-19 , 2020, Neuropsychopharmacology.

[7]  Joseph D. Janizek,et al.  AI for radiographic COVID-19 detection selects shortcuts over signal , 2020, Nature Machine Intelligence.

[8]  Angelica I. Avilés-Rivero,et al.  Common pitfalls and recommendations for using machine learning to detect and prognosticate for COVID-19 using chest radiographs and CT scans , 2020, Nature Machine Intelligence.

[9]  Yu Zhang,et al.  Momentum contrastive learning for few-shot COVID-19 diagnosis from chest CT images , 2020, Pattern Recognition.

[10]  Rishav Singh,et al.  MetaMed: Few-shot medical image classification using gradient-based meta-learning , 2021, Pattern Recognit..

[11]  Till Döhmen,et al.  DeepCOVIDExplainer: Explainable COVID-19 Diagnosis from Chest X-ray Images , 2020, 2020 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[12]  Yoshua Bengio,et al.  Cross-Modal Information Maximization for Medical Imaging: CMIM , 2020, ArXiv.

[13]  M. Shamim Hossain,et al.  MetaCOVID: A Siamese neural network framework with contrastive loss for n-shot diagnosis of COVID-19 patients , 2020, Pattern Recognition.

[14]  Jun Zheng,et al.  Few-Shot Learning for Medical Image Classification , 2020, ICANN.

[15]  Hang Joon Jo,et al.  Classification of femur fracture in pelvic X-ray images using meta-learned deep neural network , 2020, Scientific Reports.

[16]  G. Kicska,et al.  Review of Chest Radiograph Findings of COVID-19 Pneumonia and Suggested Reporting Language , 2020, Journal of thoracic imaging.

[17]  Mizuho Nishio,et al.  Automatic classification between COVID-19 pneumonia, non-COVID-19 pneumonia, and the healthy on chest X-ray image: combination of data augmentation methods , 2020, Scientific Reports.

[18]  Z. Fayad,et al.  Artificial intelligence–enabled rapid diagnosis of patients with COVID-19 , 2020, Nature Medicine.

[19]  L. Mombaerts,et al.  An interpretable mortality prediction model for COVID-19 patients , 2020, Nature Machine Intelligence.

[20]  Rita Orji,et al.  Deep Sentiment Classification and Topic Discovery on Novel Coronavirus or COVID-19 Online Discussions: NLP Using LSTM Recurrent Neural Network Approach , 2020, bioRxiv.

[21]  Milan Sonka,et al.  Deep-COVID: Predicting COVID-19 from chest X-ray images using deep transfer learning , 2020, Medical Image Analysis.

[22]  Jong Chul Ye,et al.  Deep Learning COVID-19 Features on CXR Using Limited Training Data Sets , 2020, IEEE Transactions on Medical Imaging.

[23]  Joseph Paul Cohen,et al.  COVID-19 Image Data Collection , 2020, ArXiv.

[24]  Ioannis D. Apostolopoulos,et al.  Covid-19: automatic detection from X-ray images utilizing transfer learning with convolutional neural networks , 2020, Physical and Engineering Sciences in Medicine.

[25]  A. Wong,et al.  COVID-Net: a tailored deep convolutional neural network design for detection of COVID-19 cases from chest X-ray images , 2020, Scientific Reports.

[26]  Fang Liu,et al.  Clinical and CT imaging features of the COVID-19 pneumonia: Focus on pregnant women and children , 2020, Journal of Infection.

[27]  Q. Tao,et al.  Correlation of Chest CT and RT-PCR Testing in Coronavirus Disease 2019 (COVID-19) in China: A Report of 1014 Cases , 2020, Radiology.

[28]  Jun Liu,et al.  Chest CT for Typical 2019-nCoV Pneumonia: Relationship to Negative RT-PCR Testing , 2020, Radiology.

[29]  K. Yuen,et al.  Imaging Profile of the COVID-19 Infection: Radiologic Findings and Literature Review , 2020, Radiology. Cardiothoracic imaging.

[30]  Chi-Wing Fu,et al.  Difficulty-aware Meta-Learning for Rare Disease Diagnosis , 2019, MICCAI.

[31]  Liang Chen,et al.  Self-supervised learning for medical image analysis using image context restoration , 2019, Medical Image Anal..

[32]  Li Fei-Fei,et al.  Composing Text and Image for Image Retrieval - an Empirical Odyssey , 2018, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[33]  Joaquin Vanschoren,et al.  Meta-Learning: A Survey , 2018, Automated Machine Learning.

[34]  Kristen Grauman,et al.  Attributes as Operators , 2018, ECCV.

[35]  Wesley De Neve,et al.  Few-shot Learning Using a Small-Sized Dataset of High-Resolution FUNDUS Images for Glaucoma Diagnosis , 2017, MMHealth@MM.

[36]  Yuan Luo,et al.  Recurrent Neural Networks for Classifying Relations in Clinical Notes , 2017, AMIA.

[37]  Massimo Piccardi,et al.  Recurrent neural networks with specialized word embeddings for health-domain named-entity recognition , 2017, J. Biomed. Informatics.

[38]  Sergey Levine,et al.  Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks , 2017, ICML.

[39]  Oriol Vinyals,et al.  Matching Networks for One Shot Learning , 2016, NIPS.

[40]  Bohyung Han,et al.  Image Question Answering Using Convolutional Neural Network with Dynamic Parameter Prediction , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[41]  Mario Fritz,et al.  Ask Your Neurons: A Neural-Based Approach to Answering Questions about Images , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[42]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.