[论文解读] 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

一句话总结

本文将Daubechies小波引入变分增强采样（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软件框架中实现并调优Daubechies小波用于分子模拟。
• 从性能与稳定性角度，比较小波与其他局域化基函数（如高斯函数和三次B样条）的差异。
• 基于在多样化体系中的实证表现，为VES中基函数的最优选择提供推荐。

提出的方法

• 在PLUMED 2中实现Daubechies小波作为正交、局域化的基函数，应用于VES框架。
• 通过小波基函数的线性展开构建偏差势能，利用其多分辨率特性。
• 通过系统调优（如缩放与平移参数）在模型势能与碳酸钙体系中优化小波参数。
• 应用变分原理，最小化与相对熵及Kullback-Leibler散度相关的凸泛函，以获得偏差势能。
• 采用重加权技术从偏置模拟中计算自由能面，并比较不同基函数的结果。
• 对每种基函数类型独立运行模拟，以评估统计波动与收敛鲁棒性。

实验结果

研究问题

• RQ1 局域化基函数（如Daubechies小波）是否在VES中表现出优于非局域化全局基函数（如切比雪夫多项式）的收敛行为？
• RQ2 小波在估算自由能面方面与元动力学及其他局域基函数（如高斯函数、B样条）相比表现如何？
• RQ3 小波能否减少单次模拟中偏差势能的波动以及不同独立运行之间的差异？
• RQ4 小波的多分辨率特性是否可实现模拟过程中自适应分辨率的偏差势能优化，从而提升精度？
• RQ5 是否存在某些体系类型，使得小波在收敛速度与精度方面显著优于其他基函数？

主要发现

• 在单次VES运行中，Daubechies小波表现出远小于切比雪夫多项式与元动力学的偏差势能波动。
• 基于小波的独立模拟在自由能面估算上差异更小，表明其具有更优的鲁棒性与可重复性。
• 基于小波的VES在收敛行为上优于切比雪夫多项式与元动力学，各次运行间的自由能差值低于1 kJ/mol。
• 在所有测试的模型体系中，小波优于高斯函数与三次B样条，后者在某些体系中产生不可用结果。
• 小波的多分辨率特性为未来实现可动态调整分辨率的偏差势能提供了可能。
• 基于全面评估，作者推荐将Daubechies小波作为VES的首选基函数，因其在收敛性与稳定性方面表现更优。