Graph Autoencoders for Embedding Learning in Brain Networks and Major Depressive Disorder Identification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Noman, Fuad, Ting, Chee-Ming, Kang, Hakmook, Phan, Raphael C. -W., Boyd, Brian D., Taylor, Warren D., Ombao, Hernando
Format: Preprint
Published: 2021
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911767011000320
author Noman, Fuad
Ting, Chee-Ming
Kang, Hakmook
Phan, Raphael C. -W.
Boyd, Brian D.
Taylor, Warren D.
Ombao, Hernando
author_facet Noman, Fuad
Ting, Chee-Ming
Kang, Hakmook
Phan, Raphael C. -W.
Boyd, Brian D.
Taylor, Warren D.
Ombao, Hernando
contents Brain functional connectivity (FC) reveals biomarkers for identification of various neuropsychiatric disorders. Recent application of deep neural networks (DNNs) to connectome-based classification mostly relies on traditional convolutional neural networks using input connectivity matrices on a regular Euclidean grid. We propose a graph deep learning framework to incorporate the non-Euclidean information about graph structure for classifying functional magnetic resonance imaging (fMRI)-derived brain networks in major depressive disorder (MDD). We design a novel graph autoencoder (GAE) architecture based on the graph convolutional networks (GCNs) to embed the topological structure and node content of large-sized fMRI networks into low-dimensional latent representations. In network construction, we employ the Ledoit-Wolf (LDW) shrinkage method to estimate the high-dimensional FC metrics efficiently from fMRI data. We consider both supervised and unsupervised approaches for the graph embedding learning. The learned embeddings are then used as feature inputs for a deep fully-connected neural network (FCNN) to discriminate MDD from healthy controls. Evaluated on two resting-state fMRI (rs-fMRI) MDD datasets, results show that the proposed GAE-FCNN model significantly outperforms several state-of-the-art methods for brain connectome classification, achieving the best accuracy using the LDW-FC edges as node features. The graph embeddings of fMRI FC networks learned by the GAE also reveal apparent group differences between MDD and HC. Our new framework demonstrates feasibility of learning graph embeddings on brain networks to provide discriminative information for diagnosis of brain disorders.
format Preprint
id arxiv_https___arxiv_org_abs_2107_12838
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Graph Autoencoders for Embedding Learning in Brain Networks and Major Depressive Disorder Identification
Noman, Fuad
Ting, Chee-Ming
Kang, Hakmook
Phan, Raphael C. -W.
Boyd, Brian D.
Taylor, Warren D.
Ombao, Hernando
Neurons and Cognition
Artificial Intelligence
Machine Learning
Brain functional connectivity (FC) reveals biomarkers for identification of various neuropsychiatric disorders. Recent application of deep neural networks (DNNs) to connectome-based classification mostly relies on traditional convolutional neural networks using input connectivity matrices on a regular Euclidean grid. We propose a graph deep learning framework to incorporate the non-Euclidean information about graph structure for classifying functional magnetic resonance imaging (fMRI)-derived brain networks in major depressive disorder (MDD). We design a novel graph autoencoder (GAE) architecture based on the graph convolutional networks (GCNs) to embed the topological structure and node content of large-sized fMRI networks into low-dimensional latent representations. In network construction, we employ the Ledoit-Wolf (LDW) shrinkage method to estimate the high-dimensional FC metrics efficiently from fMRI data. We consider both supervised and unsupervised approaches for the graph embedding learning. The learned embeddings are then used as feature inputs for a deep fully-connected neural network (FCNN) to discriminate MDD from healthy controls. Evaluated on two resting-state fMRI (rs-fMRI) MDD datasets, results show that the proposed GAE-FCNN model significantly outperforms several state-of-the-art methods for brain connectome classification, achieving the best accuracy using the LDW-FC edges as node features. The graph embeddings of fMRI FC networks learned by the GAE also reveal apparent group differences between MDD and HC. Our new framework demonstrates feasibility of learning graph embeddings on brain networks to provide discriminative information for diagnosis of brain disorders.
title Graph Autoencoders for Embedding Learning in Brain Networks and Major Depressive Disorder Identification
topic Neurons and Cognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2107.12838