VelocityFM: Short-Horizon Protein Trajectory Prediction via Flow Matching in Velocity Space (opens in new tab)

Protein dynamics is fundamentally a trajectory prediction problem, but molecular dynamics (MD) simulation remains expensive and static structure predictors do not model time-ordered motion. We present VelocityFM, a short-horizon protein trajectory predictor that applies rectified flow matching in velocity space over residue frames and torsions. The model combines six Invariant Point Attention (IPA) blocks with a two-layer per-residue temporal self-attention encoder, and is trained on 710 ATLA...