Cited time in webofscience Cited time in scopus

Dual attention multiple instance learning with unsupervised complementary loss for COVID-19 screening

Title
Dual attention multiple instance learning with unsupervised complementary loss for COVID-19 screening
Author(s)
Chikontwe, PhilipLuna, Acevedo Miguel AndresKang, MyeongkyunHong, Kyung SooAhn, June HongPark, Sang Hyun
Issued Date
2021-08
Citation
Medical Image Analysis, v.72, pp.102105
Type
Article
Author Keywords
COVID-19CT imagesDeep learningMultiple instance learningUnsupervised complementary loss
Keywords
Overall accuraciesReverse transcription-polymerase chain reactionSpatial attentionSpatial contextDiagnosisComputerized tomographyLarge datasetLearning systemsPolymerase chain reactionTranscriptionClinical diagnosisFalse positiveLarge datasetsMultiple-instance learning
ISSN
1361-8415
Abstract
Chest computed tomography (CT) based analysis and diagnosis of the Coronavirus Disease 2019 (COVID-19) plays a key role in combating the outbreak of the pandemic that has rapidly spread worldwide. To date, the disease has infected more than 18 million people with over 690k deaths reported. Reverse transcription polymerase chain reaction (RT-PCR) is the current gold standard for clinical diagnosis but may produce false positives; thus, chest CT based diagnosis is considered more viable. However, accurate screening is challenging due to the difficulty in annotation of infected areas, curation of large datasets, and the slight discrepancies between COVID-19 and other viral pneumonia. In this study, we propose an attention-based end-to-end weakly supervised framework for the rapid diagnosis of COVID-19 and bacterial pneumonia based on multiple instance learning (MIL). We further incorporate unsupervised contrastive learning for improved accuracy with attention applied both in spatial and latent contexts, herein we propose Dual Attention Contrastive based MIL (DA-CMIL). DA-CMIL takes as input several patient CT slices (considered as bag of instances) and outputs a single label. Attention based pooling is applied to implicitly select key slices in the latent space, whereas spatial attention learns slice spatial context for interpretable diagnosis. A contrastive loss is applied at the instance level to encode similarity of features from the same patient against representative pooled patient features. Empirical results show that our algorithm achieves an overall accuracy of 98.6% and an AUC of 98.4%. Moreover, ablation studies show the benefit of contrastive learning with MIL. © 2021
URI
http://hdl.handle.net/20.500.11750/15514
DOI
10.1016/j.media.2021.102105
Publisher
Elsevier BV
Related Researcher
  • 박상현 Park, Sang Hyun
  • Research Interests 컴퓨터비전; 인공지능; 의료영상처리
Files in This Item:

There are no files associated with this item.

Appears in Collections:
Department of Robotics and Mechatronics Engineering Medical Image & Signal Processing Lab 1. Journal Articles

qrcode

  • twitter
  • facebook
  • mendeley

Items in Repository are protected by copyright, with all rights reserved, unless otherwise indicated.

BROWSE