2025-11-14T19:19:11.421355

GO-Diff: Data-free and amortized global structure optimization

RÃ¸nne, Vegge, Bhowmik

We introduce GO-Diff, a diffusion-based method for global structure optimization that learns to directly sample low-energy atomic configurations without requiring prior data or explicit relaxation. GO-Diff is trained from scratch using a Boltzmann-weighted score-matching loss, leveraging only the known energy function to guide generation toward thermodynamically favorable regions. The method operates in a two-stage loop of self-sampling and model refinement, progressively improving its ability to target low-energy structures. Compared to traditional optimization pipelines, GO-Diff achieves competitive results with significantly fewer energy evaluations. Moreover, by reusing pretrained models across related systems, GO-Diff supports amortized optimization - enabling faster convergence on new tasks without retraining from scratch.

academic

GO-Diff: 데이터 무관 및 분할 상환 전역 구조 최적화

기본 정보

논문 ID: 2510.13448
제목: GO-Diff: Data-free and amortized global structure optimization
저자: Nikolaj Rønne, Tejs Vegge, Arghya Bhowmik (덴마크 기술대학교)
분류: physics.comp-ph cond-mat.dis-nn cond-mat.mtrl-sci cs.CE
발표 시간: 2025년 10월 15일 (사전 인쇄본)
논문 링크: https://arxiv.org/abs/2510.13448

초록

본 논문은 확산 모델 기반의 전역 구조 최적화 방법인 GO-Diff를 소개한다. 이 방법은 사전 데이터나 명시적 완화 없이 저에너지 원자 구성을 직접 샘플링할 수 있다. GO-Diff는 볼츠만 가중 점수 매칭 손실을 사용하여 처음부터 훈련되며, 알려진 에너지 함수만을 활용하여 생성 과정을 열역학적으로 유리한 영역으로 유도한다. 이 방법은 자체 샘플링과 모델 정제의 2단계 순환을 채택하여 저에너지 구조 발견 능력을 점진적으로 향상시킨다. 기존 최적화 절차와 비교하여, GO-Diff는 훨씬 적은 에너지 평가로 경쟁력 있는 결과를 달성한다. 더욱이, 관련 시스템 간에 사전 훈련된 모델을 재사용함으로써 GO-Diff는 분할 상환 최적화를 지원한다. 즉, 처음부터 재훈련할 필요 없이 새로운 작업에서 더 빠르게 수렴할 수 있다.

연구 배경 및 동기

해결하려는 문제

본 연구는 원자 시스템의 전역 구조 최적화 문제, 즉 퍼텐셜 에너지 표면(PES)에서 저에너지의 안정적인 원자 구성을 찾는 문제를 다룬다. 퍼텐셜 에너지 표면은 원자 위치를 해당 퍼텐셜 에너지에 매핑하는 고차원 비볼록 함수이며, 이 표면을 탐색하여 저에너지 구조를 식별하는 것은 계산 재료 과학, 화학 및 촉매 분야의 기본적인 과제이다.

문제의 중요성

전역 구조 최적화는 촉매 표면 발견부터 기능성 재료 설계에 이르는 응용의 기초이며, 다음을 위해 중요하다:

새로운 촉매 표면의 발견
기능성 재료의 설계
안정적인 원자 구성의 예측
재료 특성의 이해

기존 방법의 한계

전통적인 전역 최적화 방법은 다음과 같은 문제점을 가진다:

높은 계산 비용: 무작위 구조 탐색(RSS), 분지 점프, 유전 알고리즘, 모의 담금질 등의 방법은 국소 완화 및 기울기 기반 최적화기에 의존하며, 많은 에너지 및 힘 평가가 필요하다
국소 최적화에 국한: 국소 최솟값에 쉽게 빠지며, 복잡한 에너지 경관의 탐색을 제한한다
데이터 의존성: 기계학습 원자간 퍼텐셜은 관련 최솟값을 포착하기 위해 신중하게 선택된 훈련 데이터가 필요하며, 그렇지 않으면 자기강화 국소 최솟값에 빠질 수 있다
이전 가능성 부족: 기존 방법은 관련 시스템 간에 학습된 지식을 재사용하기 어렵다

연구 동기

확산 모델은 분자 및 재료 과학의 구조 생성에서 전망을 보여주었지만, 이를 전역 최적화 작업에 적용하는 것은 도전적이다. 왜냐하면 목표는 PES의 전역 최솟값에 해당하는 희귀한 저에너지 구성을 샘플링하는 것이지만, 이러한 구조의 데이터 분포는 일반적으로 미지수이거나 접근 불가능하기 때문이다.

핵심 기여

데이터 무관 생성 최적화 방법 제안: 사전 데이터나 명시적 완화 없이 퍼텐셜 에너지 표면의 최솟값을 직접 샘플링
볼츠만 가중 손실 함수 개발: 어닐링 전략과 결합하여 저에너지 영역으로의 샘플링을 유도하면서 탐색성 유지
분할 상환 최적화 구현: 관련 시스템 간 사전 훈련된 모델 전이를 통한 지식 재사용
우수한 샘플 효율성 검증: 고전적 탐색 방법과 비교하여 높은 샘플 효율성

방법론 상세 설명

작업 정의

입력: 원자 시스템의 에너지 함수 E(x), 여기서 x는 원자 구성 출력: 저에너지의 안정적인 원자 구성 목표: 볼츠만 분포에서 샘플링: $\pi_T(x) = \frac{\exp(-E(x)/T)}{Z_T}$

모델 아키텍처

훈련 순환

GO-Diff는 자체 샘플링 순환 작동을 채택한다:

모델이 역확산을 통해 원자 구조 생성
생성된 구조의 에너지 평가
결과 샘플을 사용하여 모델 정제

재생 버퍼 $B = \{(x_0^{(i)}, E^{(i)})\}$ 를 유지하여 생성된 구성 및 해당 에너지 저장

볼츠만 가중 점수 매칭

핵심 혁신은 볼츠만 가중 점수 매칭 손실이다:

$L_{\theta}^{Boltzmann} = E_{t\sim U(0,1)}\left[\lambda(t)E_{x_0\sim q, x_t\sim p_{t|0}(x_t|x_0)} w(E) \|s_\theta(x_t,t) - \nabla_{x_t}\log p_{t|0}(x_t|x_0)\|_2^2\right]$