원문 캡처 · arxiv.org
원문 캡처 · arxiv.org

arXiv에 공개된 연구는 생성 모델이 학습한 데이터의 저차원 다양체 위에서 최적화하는 ‘미분동형 최적화’를 제안했다. 연구진은 단백질 설계 실험에서 기존 유도 방식보다 2차 구조 표적 비율이 높았다고 보고했다.

확산 모델과 플로 모델이 데이터 다양체를 단순한 기저 공간으로 옮기는 지도를 제공한다는 점을 이용한다. 이 공간에서 경사 하강을 수행하면 미분기하학적으로 데이터 다양체 위의 리만 경사 하강과 같아지며, 차이는 O(λ²) 항까지 남는다. 연구진은 이 방식이 최적화 경로를 다양체 위에 유지하고 더 매끄러운 최적화 표면을 제공한다고 설명한다.

단백질 설계에는 행렬 리 군 SO(3)와 SE(3)로 방법을 확장했다. 논문은 자동미분과 호환되는 SO(3) 기울기와 리 군 상미분방정식 해법을 통한 역전파를 위한 일반화된 수반 상태 방법을 제시한다.

FrameFlow를 이용한 2차 구조 표적화에서 목표 라마찬드란 영역에 들어간 잔기 비율은 91.3%로, 조정된 기존 유도 방식의 63.3%보다 높았다. 펩타이드 결합 친화도에서는 OC-Flow보다 높은 성능을 보였으며 속도는 2배였다. PDB 테스트 세트에서는 수백 개 잔기로 이뤄진 구조의 Rosetta 에너지를 수천 단위 낮췄다고 논문은 보고했다.

논문 초록 · PDF · HTML