[논문 리뷰] Equivariant Energy-Guided SDE for Inverse Molecular Design

Fan Bao, Min Zhao외 4명|arXiv (Cornell University)|2022. 09. 30.

Spectroscopy and Quantum Chemical Studies인용 수 15

한 줄 요약

이 논문은 3차원 분자의 기하 대칭성을 활용하기 위해 직교 변환에 대해 불변인 에너지 함수를 활용하는, 제어 가능한 3차원 분자 설계를 위한 확산 기반 생성 모델인 등변 에너지 유도 SDE(EEGSDE)를 제안한다. EEGSDE는 QM9 및 GEOM-Drug 데이터셋에서 기저 모델인 조건부 확산 모델보다 우수한 성능을 보이며, 분자의 전기 이중극 모멘트와 같은 양자 성질의 평균 절대 오차를 30퍼센트 이상 향상시킨다. 또한 선형 조합을 통한 에너지 함수의 조합을 통해 다중 성질 동시 생성이 가능하다.

ABSTRACT

Inverse molecular design is critical in material science and drug discovery, where the generated molecules should satisfy certain desirable properties. In this paper, we propose equivariant energy-guided stochastic differential equations (EEGSDE), a flexible framework for controllable 3D molecule generation under the guidance of an energy function in diffusion models. Formally, we show that EEGSDE naturally exploits the geometric symmetry in 3D molecular conformation, as long as the energy function is invariant to orthogonal transformations. Empirically, under the guidance of designed energy functions, EEGSDE significantly improves the baseline on QM9, in inverse molecular design targeted to quantum properties and molecular structures. Furthermore, EEGSDE is able to generate molecules with multiple target properties by combining the corresponding energy functions linearly.

연구 동기 및 목표

• 정확한 목표 성질을 갖는 분자를 생성하는 데 있어 조건부 확산 모델의 한계를 해결하기 위해.
• 직교 변환에 대한 불변성을 통해 3차원 분자 구조에서 기하 대칭성을 통합하기 위해.
• 에너지 함수를 지침으로 사용하여 역분자 설계를 위한 융통성 있고 제어 가능한 프레임워크를 개발하기 위해.
• 다중 에너지 함수의 선형 조합을 통해 동시에 여러 목표 성질을 만족하는 분자를 생성하기 위해.
• 분자 설계에서 양자적 및 구조적 성질에 대한 생성 품질과 안정성을 향상시키기 위해.

제안 방법

• 3차원 기하 대칭성을 존중하는 등변 확률 미분 방정식(SDE)으로 분자 생성을 수식화한다.
• 직각 변환(회전 및 반사)에 대해 불변인 에너지 함수를 도입하여 등변 지침을 보장한다.
• 역방향 SDE 과정에서 에너지 함수의 기울기를 지침으로 사용하여 목표 성질 쪽으로 생성을 이끌어낸다.
• 에너지 함수가 드리프트 항을 수정하는 스코어 기반 노이즈 제거 프레임워크를 활용하여 역방향 과정을 지도한다.
• 에너지 지침의 강도를 제어하기 위해 스케일링 인자를 적용하여 충실도와 다양성 사이의 트레이드오프를 가능하게 한다.
• 다중 에너지 함수를 선형 조합하여 동시에 여러 목표 성질을 만족하는 분자를 생성한다.

실험 결과

연구 질문

• RQ1 기하 대칭성을 활용하는 에너지 유도 확산 모델은 3차원 분자 생성에서 더 높은 제어 가능성을 달성할 수 있는가?
• RQ2 에너지 유도 SDE는 특정 양자 성질을 갖는 분자를 생성할 때 표준 조건부 확산 모델보다 우수한가?
• RQ3 에너지 함수의 조합을 통해 EEGSDE는 다중 목표 성질을 갖는 분자를 효과적으로 생성할 수 있는가?
• RQ4 에너지 함수가 직각 변환에 대해 불변일 경우, 더 안정적이고 대칭적인 분자 구조가 유도되는가?
• RQ5 에너지 지침의 스케일링 인자가 생성된 분자의 품질과 목표 구조와의 일치도에 얼마나 큰 영향을 미치는가?

주요 결과

• EEGSDE는 QM9에서 조건부 EDM에 비해 전기 이중극 모멘트 성질의 평균 절대 오차를 30퍼센트 이상 감소시켰다.
• 구조 목표 설정 생성에 있어서 EEGSDE는 GEOM-Drug에서 기준 모델 대비 구조 유사도를 10퍼센트 이상 향상시켰다.
• 에너지 함수의 선형 조합을 통해 $\alpha$, $\mu$, 및 $\Delta\varepsilon$와 같은 다중 목표 성질을 만족하는 분자를 효과적으로 생성할 수 있었다.
• 다중 성질 목표에서 평균적으로 높은 신규성(85.6%), 원자 안정성(97.7%), 분자 안정성(74.6%)을 유지하였다.
• 에너지 지침의 스케일링 인자가 구조 일치도를 크게 향상시키며, 높은 값일수록 목표 구조와의 일치도가 뛰어났다.
• 생성된 분자 성질의 분포가 훈련 세트의 분포와 밀접하게 일치하여 성질 모델링의 정확성이 높음을 시사했다.