Multi-site Hyper-graph Convolutional Neural Networks and Application

Abstract

Recently,the exploitation of graph neural networks for neurological brain disorder diagnosis has attracted much attention.However,the graphs used in the existing studies are usually based on the pairwise connections of different nodes,and thus cannot reflect the complex correlation of three or more subjects,especially in the multi-site dataset,i.e.,the dataset collected from different medical institutions with the problem of data heterogeneity resulted from various scanning parameters or subject population.To address this issue,a multi-site hypergraph data structure is proposed to describe the relationship between multi-site data.This hypergraph consists of two types hyper-edge,one is intra-site hyper-edge that describes the relationship within the site,and the other is inter-site hyper-edge that describes relationship between different sites.Also,a hypergraph convolutional network is proposed to learn the feature representation of each node.The hypergraph convolution consists of two parts:the first part is the hypergraph node convolution,the second part is the super edge convolution.Experimental results on two multi-site datasets can also validate the effectiveness of the proposed method.

Keywords

• multi-site dataset|data heterogeneity|brain diseases diagnosis|graph convolutional networks|hyper-graph convolutional networks