Uncertainty-guided Graph Attention Network for Parapneumonic Effusion Diagnosis

Jinkui Hao, Jiang Liu, Ella Pereira, Ri Liu, Jiong Zhang, Yangfan Zhang, Kun Yan, Yan Gong, Jianjun Zheng, Jingfeng Zhang, Yonghuai Liu, Yitian Zhao

Research output: Contribution to journalArticle (journal)peer-review

1 Citation (Scopus)

Abstract

Parapneumonic effusion (PPE) is a common condition that causes death in patients hospitalized with pneumonia. Rapid distinction of complicated PPE (CPPE) from uncomplicated PPE (UPPE) in Computed Tomography (CT) scans is of great importance for the management and medical treatment of PPE. However, UPPE and CPPE display similar appearances in CT scans, and it is challenging to distinguish CPPE from UPPE via a single 2D CT image, whether attempted by a human expert, or by any of the existing disease classification approaches. 3D convolutional neural networks (CNNs) can utilize the entire 3D volume for classification: however, they typically suer from the intrinsic defect of over-fitting. Therefore, it is important to develop a method that not only overcomes the heavy memory and computational requirements of 3D CNNs, but also leverages the 3D information. In this paper, we propose an uncertainty-guided
graph attention network (UG-GAT) that can automatically extract and integrate information from all CT slices in a 3D volume for classification into UPPE, CPPE, and normal control cases. Specifically, we frame the distinction of dierent cases as a graph classification problem. Each individual is represented as a directed graph with a topological structure, where vertices represent the image features of slices, and edges encode the spatial relationship between them. To estimate the contribution of each slice, we first extract the slice representations with uncertainty, using a Bayesian CNN: we then make use of the uncertainty information to weight each slice during the graph prediction phase in order to enable more reliable decision-making. We construct a dataset consisting of 302 chest CT volumetric data from dierent subjects (99 UPPE, 99 CPPE and 104 normal control cases) in this study, and to the best of our knowledge, this is the first attempt to classify UPPE, CPPE and normal cases using a deep learning
method. Extensive experiments show that our approach is lightweight in demands, and outperforms accepted state-of-the-art methods by a large margin. Code is available at https://github.com/iMED-Lab/UG-GAT.
Original languageEnglish
Pages (from-to)102217
JournalMedical Image Analysis
Volume75
Early online date3 Nov 2021
DOIs
Publication statusE-pub ahead of print - 3 Nov 2021

Keywords

  • Uncertainty
  • CT
  • Graph
  • Deep Learning
  • Parapneumonic effusion

Fingerprint

Dive into the research topics of 'Uncertainty-guided Graph Attention Network for Parapneumonic Effusion Diagnosis'. Together they form a unique fingerprint.

Cite this