Software

Research software and analysis code accompanying our statistical and biomedical AI methods. Explore by research area below, or visit the Zhao-team GitHub collection ↗. Entries include reusable packages, paper implementations, and code supplied through journal supplements.

Brain networks & cognition

BSTD

Bayesian deep learning for dynamic connectivity

Link stationary and time-varying functional connectivity to cognition, with posterior summaries of predictive edges, brain regions, and temporal patterns.

Python · Biometrics, 82(4), ujag174 (2026).

SF Gradient Coupling

Multimodal brain organization

Construct and analyze structural–functional gradient coupling, including behavioral and genetic associations.

SBP

Supervised brain parcellation

Construct brain nodes and networks using voxel-level imaging and a behavioral outcome.

Original implementation by Wanwan Xu and collaborators.

ABCModel

Attributes-informed brain connectivity

Estimate group-level structural or functional connectivity jointly with regional attributes through a latent space model.

R · Alzheimer’s & Dementia.

HyBRiD

Higher-order brain relationships

Learn relationships among groups of brain regions and use the resulting representations for cognitive outcome prediction.

Python · ICML 2024.

GenCPM

Generalized connectome-based predictive modeling

Predict continuous, binary, categorical, and time-to-event outcomes from brain connectivity, with optional non-imaging covariates and regularized regression.

R · Frontiers in Neuroscience.

COSR

Connectivity-on-shape regression

Study spatially varying associations between brain shape and functional connectivity.

Group fork of Naomi-Ding/COSR.

CoCR

Covariance-on-covariance regression

Relate paired covariance structures through regression, with an R implementation and example script.

R · Biometrics.

Multimodal learning & subtyping

scMODAL

Single-cell multi-omics alignment

Align single-cell datasets across modalities with deep generative learning and feature links, supporting integrated representations and downstream analyses.

Developed by Gefei Wang and collaborators.

MMBeans

Multi-state brain network subtyping

Identify subtypes and state-specific network modules by combining stochastic block models with Bayesian nonparametric clustering.

R · Biostatistics.

Nebula

Network-based multimodal clustering

Nonparametric Bayesian clustering of multi-view and multimodal data, using network structure to learn groupings.

Disease pathways & event histories

BHPI

Bayesian hypergraph inference

Estimate overlapping disease groups and risk-factor associations, with posterior uncertainty.

Group fork of Naomi-Ding/BHPI.

BNMM

Bayesian Network Mediation Model

Perform mediation analysis with a network-valued mediator.

Network mediation for survival

Brain network pathways to disease onset

Estimate exposure–network and network–survival relationships jointly, with pathway summaries at the subnetwork and node levels.

R · Biometrics.

BMZ-DP

Recurrent and terminal event modeling

Jointly model clustered recurrent events with zero inflation and a terminal event using Bayesian semiparametric inference.

R · Journal of the Royal Statistical Society: Series C.

CoSA

Collaborative survival analysis

Integrate high-dimensional imaging and genetic features to predict time-to-event outcomes and identify discriminative features.

R · Statistics in Biosciences.

Genomics & imaging genetics

GIANT

Genetically informed brain atlas

Construct heritability-aware brain parcellations for imaging genetics; the repository provides atlas files, analysis code, and summary statistics.

Developed by Jingxuan Bao and collaborators · Nature Communications.

NRSS

Network response structural shrinkage

Estimate genetic associations with brain structural networks using structured shrinkage to identify affected connectivity patterns.

R / MATLAB utilities · Journal of the American Statistical Association.

BNME

Genetic association with related samples

Model genetic associations with brain network phenotypes while accounting for relatedness and population structure through Bayesian mixed effects.

R · Biostatistics.

Code in journal supplements

Biomarker trajectories before disease onset

Semiparametric joint modeling

Estimate biomarker trajectories on age and disease-onset time scales while accounting for delayed entry and right censoring.

Biometrics · Code and implementation details are supplied with the journal supplements.

BSSS

Bayesian interaction selection

Jointly select imaging predictors and interactions across modalities using a prior that incorporates brain topology and the relationship between main and interaction effects.

R · Biometrics · Code and implementation details are supplied with the journal supplements.

BCHA

Bayesian connectome-based heterogeneity analysis

Identify subgroups with distinct brain–cognition relationships by jointly modeling multi-state functional connectomes, network features, and cognitive outcomes.

R · Journal of the American Statistical Association · Code and implementation details are supplied with the journal supplements.