대리 우도 추정기를 활용한 확장 가능한 추론 시점 어닐링

연구진은 분자 샘플링을 위한 확장 가능한 inference-time annealing 기법 SITA를 소개했다.

SITA는 에너지 기반 모델의 surrogate likelihood를 활용해 온도 스케줄 전반에서 flow-based 생성 모델을 재학습한다.

이 방법은 기존 접근법에서 필요했던 고비용 divergence 계산을 피한다.

Alanine Dipeptide와 Alanine Tripeptide에서 최고 수준의 성능을 보였으며, 저온 분자 샘플링 효율을 높일 가능성을 보여줬다.