Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment
Created by
Haebom
저자
Yuhui Ding, Thomas Hofmann
개요
본 논문은 3D 분자 생성에서 우수한 성능을 보이는 등변(equivariant) 확산 모델의 확장성 및 효율성 한계를 해결하는 새로운 방법을 제시합니다. 기존 등변 확산 모델은 SE(3)-등변 디노이징 네트워크를 사용하여 3D 분자의 유클리드 대칭성을 고려하지만, 특수한 등변 아키텍처로 인해 확장성과 효율성이 제한됩니다. 본 논문에서는 각 분자에 대해 표본 의존적인 SO(3) 변환을 학습하여 정렬된 잠재 공간을 구성하는 접근 방식을 제안합니다. 그런 다음 정렬된 표현에 대해 비등변 확산 모델을 훈련합니다. 실험 결과에 따르면, 제안된 방법은 기존의 비등변 모델보다 훨씬 우수한 성능을 보이며, 최첨단 등변 확산 모델과 비교할 만한 샘플 품질을 제공하면서 훈련 및 샘플링 효율성을 향상시킵니다. 코드는 https://github.com/skeletondyh/RADM 에서 이용 가능합니다.