언어모델이 분자 최적화 알고리즘을 자동 설계
원제 Optimizing the Optimizer: Language Models Discover Faster Molecular Relaxation
추천 20댓글 2
Key Point
언어모델이 과학 계산의 핵심 병목인 분자 최적화 알고리즘을 자동으로 개선해 계산 비용을 40% 이상 줄인 사례로, AI가 직접 과학 연구 방법론을 최적화할 수 있음을 보여준다.
핵심 요약
- 양자화학 계산에서 기하 최적화는 각 단계마다 힘 계산이 필요해 가장 비용이 큰 부분이다.
- 연구팀은 언어모델 에이전트를 이용해 최적화 알고리즘 자체를 개선할 수 있는지 시도했다.
- 에이전트가 기존 최고 성능의 오픈소스 옵티마이저 Sella를 기반으로 자동으로 재작성하면서 힘 계산 횟수를 최소화한다.
- 조기 종료 거부와 미확인 분자에 대한 일반화 실패를 걸러내는 두 개의 검증 게이트로 품질을 보장했다.
- 결과적으로 AutoSella 계열의 두 옵티마이저를 발견했으며, 학습에 사용하지 않은 분자와 포텐셜에서도 Sella 대비 일관된 개선을 보였다.
- 특히 r2SCAN-3c DFT 수준에서 최고 성능 변형은 동일한 에너지 감소를 달성하면서 Sella의 40.2~77.2%만큼의 힘 계산만 필요했으며, 에이전트는 DFT 그래디언트를 사용하지 않았다.