[論文レビュー] Improving the Efficiency of Variationally Enhanced Sampling with Wavelet-Based Bias Potentials

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

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フレームワーク内に、直交的かつ局所化された基底関数としてダウベチーズウェーブレットを実装すること。
• マルチスケール性を活用して、ダウベチーズウェーブレット基底関数の線形展開としてバイアスポテンシャルを構築すること。
• モデルポテンシャルおよび炭酸カルシウム系を用いた体系的なチューニングにより、ウェーブレットのパラメータ（スケーリングおよびトランスレーション）を最適化すること。
• 相対エントロピーおよびカルバック・ライブラーダイバージェンスに関連する凸汎関数を最小化するための変分原理を適用してバイアスポテンシャルを取得すること。
• リウェートリング技術を用いて、バイアス付きシミュレーションから自由エネルギー表面を計算し、基底関数間で結果を比較すること。
• 各基底関数タイプごとに独立したシミュレーションを実施し、統計的フラクチュエーションおよび収束のロバスト性を評価すること。

実験結果

リサーチクエスチョン

• RQ1 局所化基底関数（例えばダウベチーズウェーブレット）は、非局所的グローバル基底関数（例えばチェビシェフ多項式）と比較して、VESにおける収束性が優れているか？
• RQ2 自由エネルギー表面推定において、ウェーブレットの性能はメタダイナミクスや他の局所化基底（例えばガウス関数、Bスプライン）と比較してどうか？
• RQ3 個々のシミュレーション内でのバイアスポテンシャルのフラクチュエーションが、ウェーブレットによって低減されるか？
• RQ4 ウェーブレットのマルチスケール性により、シミュレーション中に適応的解像度を向上させるバイアスポテンシャルの実装が可能になるか？
• RQ5 特定の系タイプにおいて、ウェーブレットは収束速度および正確性の観点で、他の基底関数よりも顕著に優れるか？

主な発見

• ダウベチーズウェーブレットは、チェビシェフ多項式やメタダイナミクスと比較して、個々のVES実行内でのバイアスポテンシャルのフラクチュエーションが著しく小さいことが示された。
• 独立したウェーブレットベースのシミュレーションでは、自由エネルギー表面推定の差が小さく、より優れたロバスト性と再現性を示している。
• ウェーブレットベースのVESは、チェビシェフ多項式およびメタダイナミクスの両方と比較して、より良好な収束性を示し、実行間の自由エネルギー差が1 kJ/mol未満に抑えられた。
• すべてのテストされたモデル系において、ウェーブレットはガウス関数および3次Bスプラインを上回ったが、後者は一部の系で使用不能な結果をもたらした。
• ウェーブレットのマルチスケール性により、将来的にシミュレーション中にリアルタイムで解像度を細分化するバイアスポテンシャルの実装が可能になる可能性がある。
• 包括的な評価の結果、著者らはダウベチーズウェーブレットを、収束性と安定性に優れるため、VESにおける推奨基底関数として推奨する。