VelocityFM: Short-Horizon Protein Trajectory Prediction via Flow Matching in Velocity Space

단백질을 실로 만든 작고 정교한 종이접기 조각상이라고 상상해 보세요. 오랫동안 과학자들은 이 조각상들이 어떻게 움직이는지 연구하기 위해 두 가지 주요 방법을 사용해 왔습니다.

1. 슬로 모션 카메라 (분자 역학): 이것은 종이접기가 움직이는 모습을 실시간으로 촬영하는 것과 같습니다. 믿을 수 없을 정도로 정확하지만, 카메라가 너무 무겁고 비싸서 단 몇 초 분량의 영상을 얻는 데에도 엄청난 시간이 걸립니다.
2. 정지 사진 (정적 예측 모델): 이것은 종이접기의 완벽한 순간을 포착한 단 한 장의 스냅샷과 같습니다. 지금 이 순간의 모양은 알려주지만, 실이 다음에 어떻게 흔들리거나 춤출지는 전혀 알려주지 않습니다.

VelocityFM은 이 간극을 메우려 노력하는 새로운 도구입니다. 이것은 값비싸고 느린 슬로 모션 카메라 없이도 단백질의 다음 몇 초간의 움직임을 예측할 수 있는 단백질을 위한 "영화 감독"이라고 생각하면 됩니다.

이것이 어떻게 작동하는지 일상적인 비유를 통해 설명하겠습니다.

"속도(Velocity)"라는 지름길

단백질의 모든 부분이 미래에 정확히 어디에 있을지 추측하는 대신(이는 무용수의 발이 10초 후에 정확히 어디에 있을지 맞히는 것과 같습니다), VelocityFM은 먼저 움직임의 속도와 방향(속도)을 추측합니다.

당신이 무용수를 보고 있다고 상상해 보세요. "그녀의 팔이 오른쪽 위로 움직이고 있다"라고 예측하는 것이, 1초 뒤에 그녀의 손이 정확히 어디에 있을지 예측하는 것보다 훨씬 쉽습니다. VelocityFM은 "속도 공간(velocity space)"에서 작동하며, 단백질 부위들의 새로운 위치를 계산하기 전에 그 움직임의 모멘텀을 먼저 파악합니다. 이 방식은 예측을 훨씬 빠르고 안정적으로 만듭니다.

"전문가 팀" 구조

이 예측을 수행하기 위해, 모델은 특별한 디지털 전문가 팀을 사용합니다:

• 기하학의 수호자 (IPA 블록): 이들은 단백질이 스스로의 규칙을 어기지 않도록 보장하는 건축가 팀과 같습니다. 이들은 단백질이 올바르게 접힌 상태를 유지하고, 존재할 수 없는 형태(존재해서는 안 되는 매듭 같은 형태)로 뒤틀리지 않도록 관리합니다.
• 시간 여행자 (시계열 셀프 어텐션): 이들은 타임라인을 살피는 역사가와 같습니다. 이들은 단백질이 지난 몇 프레임 동안 어떻게 움직였는지 살펴봄으로써 그 춤의 리듬을 이해하고, 다음 동작이 흐름에 적합하도록 보장합니다.

결과: 완벽한 춤

연구진은 이 새로운 감독을 본 적이 없는 72개의 서로 다른 단백질에 대해 테스트했습니다. 결과는 인상적이었습니다:

• "접힘(Fold)" 점수: 1.0이 실제와 완벽하게 일치함을 의미하는 척도에서, 이 모델은 0.929를 기록했습니다. 이는 예측된 영화가 실제 모습과 거의 똑같았음을 의미합니다.
• 충돌 없음: 100%의 경우에서 단백질이 스스로 충돌하지 않았습니다. 무용수가 회전하는 모습을 상상해 보세요. VelocityFM은 그들이 자신의 발에 걸려 넘어지거나 자신의 머리에 부딪히지 않도록 보장합니다.
• 형태 유지: 단백질의 내부 "뼈대"(백본)는 건강한 상태를 유지하며 자연스러운 움직임의 규칙을 91%의 확률로 따랐습니다.
• 보수적인 움직임: 모델은 신중했습니다. 움직임을 과장하지 않고, 격렬하고 불가능한 몸짓 대신 현실적이고 안전한 가동 범위를 예측했습니다.

결론

VelocityFM은 단백질이 짧은 시간 동안 어떻게 흔들리고 춤출지 예측할 수 있는 고속, 저비용 시뮬레이터와 같습니다. 이것은 느리고 비싼 과학용 카메라를 대체하는 것은 아니지만, 단백질이 스스로의 규칙을 어기지 않으면서도 본 적 없는 새로운 단백질의 움직임을 빠르고 신뢰성 있게 볼 수 있는 방법을 제공합니다.