Transferability of coVariance Neural Networks and Application to Interpretable Brain Age Prediction using Anatomical Features

05/02/2023
by   Saurabh Sihag, et al.
0

Graph convolutional networks (GCN) leverage topology-driven graph convolutional operations to combine information across the graph for inference tasks. In our recent work, we have studied GCNs with covariance matrices as graphs in the form of coVariance neural networks (VNNs) that draw similarities with traditional PCA-driven data analysis approaches while offering significant advantages over them. In this paper, we first focus on theoretically characterizing the transferability of VNNs. The notion of transferability is motivated from the intuitive expectation that learning models could generalize to "compatible" datasets (possibly of different dimensionalities) with minimal effort. VNNs inherit the scale-free data processing architecture from GCNs and here, we show that VNNs exhibit transferability of performance over datasets whose covariance matrices converge to a limit object. Multi-scale neuroimaging datasets enable the study of the brain at multiple scales and hence, can validate the theoretical results on the transferability of VNNs. To gauge the advantages offered by VNNs in neuroimaging data analysis, we focus on the task of "brain age" prediction using cortical thickness features. In clinical neuroscience, there has been an increased interest in machine learning algorithms which provide estimates of "brain age" that deviate from chronological age. We leverage the architecture of VNNs to extend beyond the coarse metric of brain age gap in Alzheimer's disease (AD) and make two important observations: (i) VNNs can assign anatomical interpretability to elevated brain age gap in AD, and (ii) the interpretability offered by VNNs is contingent on their ability to exploit specific principal components of the anatomical covariance matrix. We further leverage the transferability of VNNs to cross validate the above observations across different datasets.

READ FULL TEXT

page 19

page 21

research
05/27/2023

Explainable Brain Age Prediction using coVariance Neural Networks

In computational neuroscience, there has been an increased interest in d...
research
10/28/2022

Predicting Brain Age using Transferable coVariance Neural Networks

The deviation between chronological age and biological age is a well-rec...
research
05/31/2022

coVariance Neural Networks

Graph neural networks (GNN) are an effective framework that exploit inte...
research
03/29/2021

IA-GCN: Interpretable Attention based Graph Convolutional Network for Disease prediction

Interpretability in Graph Convolutional Networks (GCNs) has been explore...
research
06/06/2021

Brain Age Estimation From MRI Using Cascade Networks with Ranking Loss

Chronological age of healthy people is able to be predicted accurately u...
research
07/11/2021

BrainNNExplainer: An Interpretable Graph Neural Network Framework for Brain Network based Disease Analysis

Interpretable brain network models for disease prediction are of great v...

Please sign up or login with your details

Forgot password? Click here to reset