더 빠른 분자 동역학을 위한 투기적 샘플링 (Speculative Sampling)

분자 동역학 (Molecular dynamics, MD)은 원자 시스템의 동적 거동을 시뮬레이션하는 핵심 도구입니다. 그러나 MD는 본질적으로 직렬적(serial)이기 때문에, 병렬 컴퓨팅을 사용하더라도 단일 시스템의 처리량 (throughput)을 높이기가 어렵습니다. 이를 해결하기 위해, 우리는 상대적 오차를 추가하지 않으면서 MD를 가속화할 수 있는 분산형 및 모델 불가지론적 (model-agnostic) 투기적 샘플러인 Langevin Speculative Dynamics (LSD)를 소개합니다. 언어 및 확산 모델링 (diffusion modeling)에서의 투기적 방법론 (speculative methods)에서 영감을 얻은 LSD는 초안 모델 (draft model)을 사용하여 빠른 시뮬레이션 단계를 제안하고, 이를 더 느린 타겟 모델 (target model)과 병렬로 검증하며, 초안에서 타겟 분포로의 수송 맵 (transport map)을 적용합니다. 우리는 투기적 샘플링을 2차 Langevin 동역학 (second-order Langevin dynamics)으로 확장하고, 물리적 매개변수의 함수로서 달성 가능한 속도 향상 (speedup)을 도출했습니다. 또한 LSD가 다양한 시스템 및 초안-타겟 조합에 걸쳐 3~9배의 속도 향상을 보이며 일반화됨을 보여주고, LSD가 타겟 모델 분포로부터 궤적 (trajectories)을 샘플링한다는 것을 이론적 및 경험적으로 확인했습니다.