확장 가능한 시공간 SE(3) 확산 모델을 이용한 장기 단백질 동역학 연구

Scalable Spatio-Temporal SE(3) Diffusion for Long-Horizon Protein Dynamics

분자 동역학(MD) 시뮬레이션은 단백질 동역학 연구의 표준으로 남아 있지만, 계산 비용으로 인해 생물학적으로 관련된 시간 척도까지 접근하기 어렵습니다. 최근 생성 모델은 시뮬레이션 속도를 향상시키는 데 유망한 결과를 보였지만, 구조적 제약, 오류 누적, 그리고 부적절한 시공간 동역학 모델링으로 인해 장기 예측에 어려움을 겪습니다. 본 연구에서는 STAR-MD(Spatio-Temporal Autoregressive Rollout for Molecular Dynamics)라는 확장 가능한 SE(3) 등변 확산 모델을 제시합니다. 이 모델은 물리적으로 타당한 단백질 궤적을 마이크로초 단위의 시간 척도로 생성합니다. 핵심적인 혁신은 인과 관계 기반 확산 트랜스포머로, 이는 공동 시공간 어텐션을 통해 복잡한 시공간 의존성을 효율적으로 포착하면서 기존 방법의 메모리 병목 현상을 피합니다. 표준 ATLAS 벤치마크에서 STAR-MD는 모든 지표에서 최첨단 성능을 달성했으며, 기존 방법과 비교하여 구조적 타당성, 동적 충실도, 그리고 전체적인 구조 다양성을 크게 향상시켰습니다. STAR-MD는 기존 방법이 실패하는 경우에도 안정적인 마이크로초 단위의 궤적을 생성하며, 확장된 예측 과정에서도 높은 구조 품질을 유지합니다. 종합적인 평가 결과, 현재 모델은 장기 예측에 심각한 한계를 가지고 있으며, STAR-MD의 공동 시공간 모델링은 생물학적으로 관련된 시간 척위에서 강력한 동역학 시뮬레이션을 가능하게 한다는 것을 보여줍니다. 이는 단백질 기능 연구의 가속화를 위한 중요한 발판이 될 것입니다.

Molecular dynamics (MD) simulations remain the gold standard for studying protein dynamics, but their computational cost limits access to biologically relevant timescales. Recent generative models have shown promise in accelerating simulations, yet they struggle with long-horizon generation due to architectural constraints, error accumulation, and inadequate modeling of spatio-temporal dynamics. We present STAR-MD (Spatio-Temporal Autoregressive Rollout for Molecular Dynamics), a scalable SE(3)-equivariant diffusion model that generates physically plausible protein trajectories over microsecond timescales. Our key innovation is a causal diffusion transformer with joint spatio-temporal attention that efficiently captures complex space-time dependencies while avoiding the memory bottlenecks of existing methods. On the standard ATLAS benchmark, STAR-MD achieves state-of-the-art performance across all metrics--substantially improving conformational coverage, structural validity, and dynamic fidelity compared to previous methods. STAR-MD successfully extrapolates to generate stable microsecond-scale trajectories where baseline methods fail catastrophically, maintaining high structural quality throughout the extended rollout. Our comprehensive evaluation reveals severe limitations in current models for long-horizon generation, while demonstrating that STAR-MD's joint spatio-temporal modeling enables robust dynamics simulation at biologically relevant timescales, paving the way for accelerated exploration of protein function.