Papers·4일 전
SITA: 분자 샘플링을 위한 확장 가능한 추론-시간 어닐링 — 알라닌 2·3펩타이드 SOTA
SITA는 에너지 기반 모델을 이용해 빠른 대용량 우도를 제공함으로써, 기존 방법의 비용이 큰 divergence 항을 피하면서 분자의 볼츠만 분포 샘플링을 개선했습니다. 알라닌 디펩타이드와 트리펩타이드에서 SOTA 성능을 달성했으며, 코드도 공개되었습니다.
SITA는 분자 샘플링에서 비용이 큰 divergence 항을 피하면서도 SOTA 성능을 달성한 추론-시간 어닐링 기법입니다.
핵심 결론
- 벤치 — 알라닌 디펩타이드와 트리펩타이드에서 기존 방법 대비 SOTA 성능.
- 효율 — 중요도 가중치를 위한 divergence 계산이 필요 없어 대규모 시스템으로 확장 가능.
방법
- 아이디어 — flow 기반 모델을 온도 사다리를 따라 점진적으로 fine-tuning, 에너지 기반 모델로 빠른 대용량 우도를 제공.
- 차별점 — 기존 방법은 score field divergence를 계산해야 했으나 SITA는 이를 회피하여 확장성을 확보.
한계·조건
- 벤치마크 — 소형 분자(알라닌 2·3펩타이드)에서만 검증, 더 큰 분자계에서의 성능은 추가 연구 필요.
- 코드 — GitHub에 공개되어 재현 가능.
편집자 한 줄
분자 샘플링에서 확장성 문제를 해결한 점이 인상적이지만, 실제 약물 설계 규모의 시스템에서도 통할지는 지켜봐야겠네요.
- #molecular-sampling
- #diffusion-models
- #flow-models
- #computational-chemistry
Daniel Peñaherrera