Atomic Trajectory Modeling with State Space Models for Biomolecular Dynamics

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shi, Liang, Lu, Jiarui, Liu, Junqi, Shi, Chence, Yang, Zhi, Tang, Jian
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914405971656704
author Shi, Liang
Lu, Jiarui
Liu, Junqi
Shi, Chence
Yang, Zhi
Tang, Jian
author_facet Shi, Liang
Lu, Jiarui
Liu, Junqi
Shi, Chence
Yang, Zhi
Tang, Jian
contents Understanding the dynamic behavior of biomolecules is fundamental to elucidating biological function and facilitating drug discovery. While Molecular Dynamics (MD) simulations provide a rigorous physical basis for studying these dynamics, they remain computationally expensive for long timescales. Conversely, recent deep generative models accelerate conformation generation but are typically either failing to model temporal relationship or built only for monomeric proteins. To bridge this gap, we introduce ATMOS, a novel generative framework based on State Space Models (SSM) designed to generate atom-level MD trajectories for biomolecular systems. ATMOS integrates a Pairformer-based state transition mechanism to capture long-range temporal dependencies, with a diffusion-based module to decode trajectory frames in an autoregressive manner. ATMOS is trained across crystal structures from PDB and conformation trajectory from large-scale MD simulation datasets including mdCATH and MISATO. We demonstrate that ATMOS achieves state-of-the-art performance in generating conformation trajectories for both protein monomers and complex protein-ligand systems. By enabling efficient inference of atomic trajectory of motions, this work establishes a promising foundation for modeling biomolecular dynamics.
format Preprint
id arxiv_https___arxiv_org_abs_2603_17633
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Atomic Trajectory Modeling with State Space Models for Biomolecular Dynamics
Shi, Liang
Lu, Jiarui
Liu, Junqi
Shi, Chence
Yang, Zhi
Tang, Jian
Biomolecules
Machine Learning
Understanding the dynamic behavior of biomolecules is fundamental to elucidating biological function and facilitating drug discovery. While Molecular Dynamics (MD) simulations provide a rigorous physical basis for studying these dynamics, they remain computationally expensive for long timescales. Conversely, recent deep generative models accelerate conformation generation but are typically either failing to model temporal relationship or built only for monomeric proteins. To bridge this gap, we introduce ATMOS, a novel generative framework based on State Space Models (SSM) designed to generate atom-level MD trajectories for biomolecular systems. ATMOS integrates a Pairformer-based state transition mechanism to capture long-range temporal dependencies, with a diffusion-based module to decode trajectory frames in an autoregressive manner. ATMOS is trained across crystal structures from PDB and conformation trajectory from large-scale MD simulation datasets including mdCATH and MISATO. We demonstrate that ATMOS achieves state-of-the-art performance in generating conformation trajectories for both protein monomers and complex protein-ligand systems. By enabling efficient inference of atomic trajectory of motions, this work establishes a promising foundation for modeling biomolecular dynamics.
title Atomic Trajectory Modeling with State Space Models for Biomolecular Dynamics
topic Biomolecules
Machine Learning
url https://arxiv.org/abs/2603.17633