Iso-CapsNet: Isomorphic Capsule Network for Brain Graph Representation Learning

by   Jiawei Zhang, et al.

Brain graph representation learning serves as the fundamental technique for brain diseases diagnosis. Great efforts from both the academic and industrial communities have been devoted to brain graph representation learning in recent years. The isomorphic neural network (IsoNN) introduced recently can automatically learn the existence of sub-graph patterns in brain graphs, which is also the state-of-the-art brain graph representation learning method by this context so far. However, IsoNN fails to capture the orientations of sub-graph patterns, which may render the learned representations to be useless for many cases. In this paper, we propose a new Iso-CapsNet (Isomorphic Capsule Net) model by introducing the graph isomorphic capsules for effective brain graph representation learning. Based on the capsule dynamic routing, besides the subgraph pattern existence confidence scores, Iso-CapsNet can also learn other sub-graph rich properties, including position, size and orientation, for calculating the class-wise digit capsules. We have compared Iso-CapsNet with both classic and state-of-the-art brain graph representation approaches with extensive experiments on four brain graph benchmark datasets. The experimental results also demonstrate the effectiveness of Iso-CapsNet, which can out-perform the baseline methods with significant improvements.


Contrastive Brain Network Learning via Hierarchical Signed Graph Pooling Model

Recently brain networks have been widely adopted to study brain dynamics...

CommPOOL: An Interpretable Graph Pooling Framework for Hierarchical Graph Representation Learning

Recent years have witnessed the emergence and flourishing of hierarchica...

Robust Causal Graph Representation Learning against Confounding Effects

The prevailing graph neural network models have achieved significant pro...

Split-Brain Autoencoders: Unsupervised Learning by Cross-Channel Prediction

We propose split-brain autoencoders, a straightforward modification of t...

Node-Centric Graph Learning from Data for Brain State Identification

Data-driven graph learning models a network by determining the strength ...

Periodic Graph Transformers for Crystal Material Property Prediction

We consider representation learning on periodic graphs encoding crystal ...

Please sign up or login with your details

Forgot password? Click here to reset