군(Group) 대칭성을 고려한 분수 공간 확산 모델 (Quotient-Space Diffusion Models)

요약

본 논문은 특정 시스템에 내재된 군(group) 대칭성(symmetry)을 활용하여 생성 모델의 효율성과 정확도를 높인 '분수 공간 확산 모델' 프레임워크를 제안합니다. 특히, 분자 구조 생성과 같이 $ ext{SE}(3)$와 같은 특수 유클리드 그룹(Special Euclidean Group)에 의해 대칭성을 갖는 분야에 적용 가능합니다. 이 방법은 기존의 군 등변(group-equivariant) 확산 모델들이 복잡하게 학습해야 했던 그룹 작용 관련 요소를 제거하여 학습 난이도를 낮추고, 샘플러가 목표 분포를 정확히 재현함을 보

핵심 포인트

• 제안된 프레임워크는 일반적인 분수 공간(quotient space)에서의 확산 모델링을 위한 공식적 틀을 제시합니다.
• 분자 구조 생성에 적용하여 $ ext{SE}(3)$ 대칭성을 가진 시스템의 효율적인 모델링이 가능함을 입증했습니다.
• 기존 방식 대비 그룹 작용 관련 요소를 학습할 필요가 없어 모델의 복잡성과 학습 난이도가 크게 감소했습니다.
• 샘플러가 목표 분포를 보장하는 반면, 기존 휴리스틱 정렬 전략은 적절한 샘플러를 갖지 못한다는 점을 명확히 했습니다.