PFP/MM: A Hybrid Approach Combining a Universal Neural Network Potential with Classical Force Fields for Large-Scale Reactive Simulations

Motivation

• Known: QM/MM 방법은 반응 영역에 QM, 주변 환경에 MM을 적용하여 비용을 줄이지만, QM 부분의 계산 비용이 여전히 높고 긴 시간 샘플링이 제한된다. 최근 ML/MM 접근법이 등장했으나 대부분의 기존 MLIP는 제한된 원소(H, C, N, O, F, S, Cl)만 지원한다.
• Gap: Universal MLIP 기반 ML/MM 방법이 금속 중심 반응성과 96개 원소 커버리지를 갖추고도 대규모 응축상 시스템에 실제로 적용된 사례가 부족하다. 특히 효소 반응과 같은 현실적인 생물분자 환경에서의 검증이 필요하다.
• Why: 대규모 반응성 시뮬레이션의 효율적 수행은 신약 설계, 촉매 개발, 효소 공학 등에서 메커니즘 규명과 설계 지원에 필수적이다. 광범위한 원소 커버리지를 가진 uMLIP는 금속 효소와 같은 복합 화학 환경을 다룰 수 있어 실용적 가치가 높다.
• Approach: PFP uMLIP와 classical MM을 region-partitioned 하이브리드 방식으로 결합하되, 화학적으로 활성인 영역만 PFP로 계산하고 나머지는 MM으로 처리하여 계산 효율을 극대화했다. OpenMM 기반 GPU 가속 구현과 MN-Core 2 AI 가속기 지원을 통해 확장성을 확보했다.

Achievement

• 계산 속도 향상: PFP-only 대비 V100에서 17.8배, MN-Core 2에서 56.5배 빠른 성능을 달성하여 1,000,002원자 시스템에서 1.07-1.51 ns/day 시뮬레이션 가능
• 향상된 샘플링: 아라닌 디펩타이드 시스템에서 하루 단위 enhanced sampling을 통해 다중 나노초 규모 Ramachandran plot 수렴 달성, 기존 알려진 conformational basin 재현
• 용매 효과 정확한 모사: 분자 내 친핵성 첨가 반응에서 극성 용매의 stabilization 효과를 자유에너지 프로파일로 정확히 재현
• 금속 효소 반응 적용: 96개 원소 커버리지의 PFP를 활용하여 cytochrome P450 Compound I 수산화 반응의 자유에너지 경관을 계산하고 수용된 반응 메커니즘과 일치하는 결과 획득
• 확장성: 1백만 개 이상 원자를 처리 가능한 하이브리드 시스템 구현으로 현실적인 대규모 생물분자 환경 시뮬레이션 가능성 입증

How

• Additive hybrid scheme 적용: UML/MM = UML(rML) + UMM(rMM) + UML−MM(rML,rMM) 형태의 총 에너지 계산
• Mechanical embedding 사용: PFP-MM 경계의 상호작용을 classical nonbonded interaction(Coulomb + Lennard-Jones)으로 처리
• Link atom 방식: PFP/MM 경계를 넘는 공유 결합에 virtual hydrogen atom 부착하여 dangling bond 제거
• FIRES (Flexible Inner Region Ensemble Separator) 적용: 용매 효과가 중요한 반응의 경우 용질 근처 용매 분자를 PFP 영역에 포함하면서 flexible restraint로 앙상블 평균 보존
• OpenMM 기반 GPU 가속 구현: MM 영역은 GPU 병렬 계산, PFP 추론은 V100 또는 MN-Core 2 가속기에서 실행
• PFP v6.0.0 두 가지 모드 활용: CRYSTAL_U0_PLUS_D3(PBE functional, 96원소, D3 분산 보정), MOLECULE(ωB97X-D functional, 소형 유기분자)

Originality

• 96개 원소를 지원하는 uMLIP(PFP)를 ML/MM에 처음 적용하여 금속 중심 반응성과 광범위한 화학 공간 커버리지 확보
• Universal MLIP 기반 프레임워크로 Compound I 같은 금속 효소 반응을 대규모 환경에서 시뮬레이션한 최초의 사례
• OpenMM 통합을 통해 GPU 가속 MM 계산과 uMLIP 추론의 효율적 결합, MN-Core 2 같은 전문 AI 가속기 지원으로 추가 성능 향상 실현
• FIRES와 결합하여 용매 극성, 수소 결합 등 정교한 용매 효과를 포함하면서도 앙상블 통계 유지

Limitation & Further Study

• Mechanical embedding 사용으로 PFP 영역이 MM 영역의 고전적 정전기 상호작용만 받음: 용매 극성화(polarization)나 전자 재배치가 중요한 반응은 추가적으로 용매 분자를 PFP 영역에 포함 필요
• Link atom force redistribution 절차의 정확도와 다양한 계 결합 유형(특히 금속-배위 결합)에 대한 검증 부족
• PFP v6.0.0의 학습 데이터셋 특성(PBE 함수형)에 따른 정확도 편향, 다른 DFT 함수형(B3LYP, CAM-B3LYP 등)과의 차이 미평가
• 현실적 효소 시스템(전체 단백질 + 용매)에 대한 수렴성과 안정성, 장시간 시뮬레이션의 누적 오차에 대한 심층 검토 필요
• 후속 연구: 전자 분극 효과를 포함한 polarizable embedding 개발, 금속-리간드 상호작용에 특화된 uMLIP 세부 검증, 실험치와 직접 비교하는 대규모 단백질 시스템 적용

Evaluation

총평: Universal neural network potential을 ML/MM 프레임워크에 처음으로 대규모, 장시간 시뮬레이션에 적용하여 금속 효소 반응까지 다룰 수 있음을 입증했다. GPU 가속과 전문 AI 가속기 지원으로 실용적 계산 효율을 확보한 점이 매우 강점이지만, mechanical embedding의 근본적 한계와 다양한 화학 환경에서의 정확도 검증이 추가로 필요하다.