TopoFlow: Evolutionarily Conditioned Flow Matching for Protein Conformational Ensemble Generation

Abstract

Proteins function through dynamic conformational ensembles rather than a single static structure, yet accurately modeling these ensembles remains challenging. Here, we introduce TopoFlow, an evolutionarily conditioned flow-matching framework for protein conformational ensemble generation. TopoFlow uses evolutionary representations from multiple sequence alignments (MSAs) to encode global conformational heterogeneity. It combines this with structural variables from a variational autoencoder (VAE) to capture local flexibility. These complementary representations are then adaptively fused through a conditional modulation module to construct unified conditioning features that guide protein conformational ensemble generation. On the ATLAS molecular dynamics (MD) benchmark, TopoFlow improved the Jensen Shannon metrics for pairwise distances (JS-PwD), radius of gyration (JS-Rg), and time-lagged independent components (JS-TIC) by 10.73%, 5.78% and 7.75%, respectively, relative to BioEmu. Notably, this performance generalizes to experiment-based evaluation, where TopoFlow exhibits strong agreement with experimental observables, including chemical shifts (CS) and small-angle X-ray scattering (SAXS). Ablation analyses further indicated that the evolutionary representations and structural latent variables contributed complementary information to ensemble generation. Our results suggest that TopoFlow, through its integration of evolutionary and structural latent information, offers a promising strategy for generating structurally plausible and conformationally informative protein ensembles.