[논문 리뷰] Improving the Efficiency of Variationally Enhanced Sampling with Wavelet-Based Bias Potentials

Benjamin Pampel, Ómar Valsson|arXiv (Cornell University)|2022. 02. 27.

Protein Structure and Dynamics참고 문헌 101인용 수 9

한 줄 요약

이 논문은 변분적 강화 샘플링(VES)의 기저 함수로 다우베치스 웨이브릿을 도입하여, 기존의 전역 기저 함수인 체비셰프 다항식과 메타다이내믹스와 비교해 수렴성 향상과 편향의 변동 감소를 입증한다. 모델 시스템과 수용액 내 칼슘탄산염의 상호작용에서 웨이브릿은 자유에너지 표면 추정에 있어 뛰어난 강건성과 정확도를 보인다.

ABSTRACT

Collective variable-based enhanced sampling methods are routinely used on systems with metastable states, where high free energy barriers impede proper sampling of the free energy landscapes when using conventional molecular dynamics simulations. One such method is variationally enhanced sampling (VES), which is based on a variational principle where a bias potential in the space of some chosen slow degrees of freedom, or collective variables, is constructed by minimizing a convex functional. In practice, the bias potential is taken as a linear expansion in some basis function set. So far, primarily basis functions delocalized in the collective variable space, like plane waves, Chebyshev, or Legendre polynomials, have been used. However, there has not been an extensive study of how the convergence behavior is affected by the choice of the basis functions. In particular, it remains an open question if localized basis functions might perform better. In this work, we implement, tune, and validate Daubechies wavelets as basis functions for VES\@. The wavelets construct orthogonal and localized bases that exhibit an attractive multiresolution property. We evaluate the performance of wavelet and other basis functions on various systems, going from model potentials to the calcium carbonate association process in water. We observe that wavelets exhibit excellent performance and much more robust convergence behavior than all other basis functions, as well as better performance than metadynamics. In particular, using wavelet bases yields far smaller fluctuations of the bias potential within individual runs and smaller differences between independent runs. Based on our overall results, we can recommend wavelets as basis functions for VES.

연구 동기 및 목표

• 지역화된 기저 함수, 특히 웨이브릿이 변분적 강화 샘플링(VES)에서 전역 기저 함수보다 우월한 성능을 보이는지 조사하기.
• 웨이브릿 기반의 편향 잠재에너지가 메타다이내믹스와 다항식 기반 방법과 비교해 수렴성 행동과 강건성 측면에서 어떻게 성능을 발휘하는지 평가하기.
• 분자의 시뮬레이션 환경에서 PLUMED 2 소프트웨어 프레임워크 내에서 다우베치스 웨이브릿을 구현하고 최적화하기.
• 성능과 안정성 측면에서 가우시안과 3차 B스플라인과 같은 다른 지역화된 기저 함수와의 비교를 통해 웨이브릿의 성능 평가하기.
• 다양한 시스템에서의 경험적 성능 기반으로 VES에 적합한 최적의 기저 함수 선택에 대한 권고 제시하기.

제안 방법

• PLUMED 2 내 VES 프레임워크에 다우베치스 웨이브릿을 정규직교이며 국소화된 기저 함수로 구현하기.
• 웨이브릿 기저 함수의 선형 확장으로 편향 잠재에너지 구축하기. 이때 다중해상도 성질을 활용하기.
• 모델 잠재에너지와 칼슘탄산염 시스템을 대상으로 체계적 최적화를 통해 웨이브릿의 파arameter(스케일링 및 이동) 조정하기.
• 상대 엔트로피와 쿨백-라이블러 발산과 관련된 볼록 함수를 최소화하기 위해 변분 원리를 적용하여 편향 잠재에너지 도출하기.
• 재가중 기법을 사용해 편향된 시뮬레이션으로부터 자유에너지 표면 계산하고, 다양한 기저 함수 간 결과 비교하기.
• 각 기저 유형별로 독립적인 시뮬레이션을 수행하여 통계적 변동성과 수렴 강건성 평가하기.

실험 결과

연구 질문

• RQ1 다우베치스 웨이브릿과 같은 지역화된 기저 함수가 체비셰프 다항식과 같은 전역 기저 함수보다 VES에서 더 나은 수렴 성질을 보이는가?
• RQ2 웨이브릿의 자유에너지 표면 추정 성능이 메타다이내믹스 및 기타 지역화된 기저 함수(Gaussians, B-splines 등)와 비교해 어떻게 되는가?
• RQ3 웨이브릿은 개별 시뮬레이션 내 편향 잠재에너지의 변동성을 줄이고, 독립적인 런 간의 차이를 감소시킬 수 있는가?
• RQ4 웨이브릿의 다중해상도 성질은 시뮬레이션 중에 적응형 해상도 정밀도 향상이 가능하게 하여 정확도 향상에 기여하는가?
• RQ5 특정 유형의 시스템에서 웨이브릿이 수렴 속도와 정확도 측면에서 다른 기저 함수보다 뚜렷이 뛰어난 성능을 보이는가?

주요 결과

• 다우베치스 웨이브릿은 체비셰프 다항식과 메타다이내믹스보다 개별 VES 런 내에서 편향 잠재에너지의 변동성이 현저히 작다.
• 독립적인 웨이브릿 기반 시뮬레이션은 자유에너지 표면 추정치의 차이가 작아지며, 더 뛰어난 강건성과 재현 가능성 나타낸다.
• 웨이브릿 기반 VES는 체비셰프 다항식과 메타다이내믹스보다 더 나은 수렴 성질을 보이며, 런 간 자유에너지 차이가 1 kJ/mol 이하로 유지된다.
• 모든 테스트된 모델 시스템에서 웨이브릿은 가우시안과 3차 B스플라인보다 뛰어난 성능을 보이며, 특히 일부 시스템에선 B스플라인 기반 결과가 사용 불가능한 수준이었다.
• 웨이브릿의 다중해상도 성질은 향후 실시간으로 해상도를 정밀하게 조정할 수 있는 적응형 해상도 편향 잠재에너지의 구현 가능성을 열어준다.
• 종합적인 평가를 바탕으로 저자들은 다우베치스 웨이브릿이 수렴성과 안정성 측면에서 뛰어나므로 VES에 가장 선호되는 기저 함수로 권장한다.