BrainNet-ML-ToolBox

Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.

Regression Graph Neural Network (regGNN) for cognitive score prediction.

Awesome graph neural networks for brain network learning. Collections of related research papers with implementations, commonly used datasets and tools. We also invite researchers interested in brain graph learning with GNNs to join the project.

The Brain Activity Flow ("Actflow") Toolbox. Tools to quantify the relationship between connectivity and task activity through network simulations and machine learning prediction. Helps determine how connections contribute to specific brain functions.

Implementations for CHIL 2023 paper ''PTGB: Pre-Train Graph Neural Networks for Brain Network Analysis'' and KDD 2022 paper ''Data-Efficient Brain Connectome Analysis via Multi-Task Meta-Learning''

An Explainable Geometric-Weighted Graph Attention Network (xGW-GAT) for Identifying Functional Networks Associated with Gait Impairment

Federating temporally-varying graph timeseries

Federated time-dependent graph evolution prediction with missing timepoints.

Federated multigraph integration with application to connectional brain template estimation.

StairwayGraphNet for inter- and inter-modality graph superresolution.