Log In Sign Up

Edge-variational Graph Convolutional Networks for Uncertainty-aware Disease Prediction

by   Yongxiang Huang, et al.

There is a rising need for computational models that can complementarily leverage data of different modalities while investigating associations between subjects for population-based disease analysis. Despite the success of convolutional neural networks in representation learning for imaging data, it is still a very challenging task. In this paper, we propose a generalizable framework that can automatically integrate imaging data with non-imaging data in populations for uncertainty-aware disease prediction. At its core is a learnable adaptive population graph with variational edges, which we mathematically prove that it is optimizable in conjunction with graph convolutional neural networks. To estimate the predictive uncertainty related to the graph topology, we propose the novel concept of Monte-Carlo edge dropout. Experimental results on four databases show that our method can consistently and significantly improve the diagnostic accuracy for Autism spectrum disorder, Alzheimer's disease, and ocular diseases, indicating its generalizability in leveraging multimodal data for computer-aided diagnosis.


page 1

page 2

page 3

page 4


AMA-GCN: Adaptive Multi-layer Aggregation Graph Convolutional Network for Disease Prediction

Recently, Graph Convolutional Networks (GCNs) have proven to be a powerf...

Spectral Graph Convolutions for Population-based Disease Prediction

Exploiting the wealth of imaging and non-imaging information for disease...

InceptionGCN: Receptive Field Aware Graph Convolutional Network for Disease Prediction

Geometric deep learning provides a principled and versatile manner for t...

Online Disease Self-diagnosis with Inductive Heterogeneous Graph Convolutional Networks

We propose a Healthcare Graph Convolutional Network (HealGCN) to offer d...

Bootstrapping Graph Convolutional Neural Networks for Autism Spectrum Disorder Classification

Using predictive models to identify patterns that can act as biomarkers ...

Multi Layered-Parallel Graph Convolutional Network (ML-PGCN) for Disease Prediction

Structural data from Electronic Health Records as complementary informat...