GT-Mamba: a Topology-Aware Graph-State space model for robust and interpretable epigenetic age prediction

Current epigenetic clocks face a trade-off between predictive accuracy and biological interpretability, often relying on dataset-specific correction to generalize across cohorts. We propose GT-Mamba, a novel architecture that integrates a Structure-Aware Graph Transformer with the Mamba state space model. This design captures CpG topological correlations and genome-wide long-range dependencies.

GT-Mamba demonstrates strong out-of-the-box robustness across heterogeneous independent validation cohorts, achieving a weighted average MAE of 4.43 years. Notably, it effectively generalizes to EPIC 850k arrays despite partial feature missingness, and maintains consistent performance across homologous age distribution shifts (MAE 2.94 years in a young cohort). Ablation studies confirm that graph topology contributes to improved robustness against noise. Mechanistic analysis suggests that the model captures methylation patterns associated with both developmental and functional processes.

Source code and pre-trained models are freely available at https://github.com/NENUBioCompute/GT-Mamba and archived on Zenodo (DOI: 10.5281/zenodo.19703155).