[论文解读] Iterative trajectory reweighting for estimation of equilibrium and non-equilibrium observables

John D. Russo, Jeremy Copperman+1 more|arXiv (Cornell University)|Jun 16, 2020

Spectroscopy and Quantum Chemical Studies被引用 4

一句话总结

该论文提出了一种迭代轨迹重加权算法，可在不假设马尔可夫性且无需构建转移矩阵的前提下，估计平衡态与非平衡态可观测量（如稳态分布、击穿概率和平均首达时间）。该方法通过利用连续轨迹上概率分布和击穿值的平稳性，基于时间平均的箱概率或下游击穿值迭代优化轨迹权重，从而收敛到无偏、自洽的估计结果。

ABSTRACT

We present two algorithms by which a set of short, unbiased trajectories can be iteratively reweighted to obtain various observables. The first algorithm estimates the stationary (steady state) distribution of a system by iteratively reweighting the trajectories based on the average probability in each state. The algorithm applies to equilibrium or non-equilibrium steady states, exploiting the `left' stationarity of the distribution under dynamics -- i.e., in a discrete setting, when the column vector of probabilities is multiplied by the transition matrix expressed as a left stochastic matrix. The second procedure relies on the `right' stationarity of the committor (splitting probability) expressed as a row vector. The algorithms are unbiased, do not rely on computing transition matrices, and make no Markov assumption about discretized states. Here, we apply the procedures to a one-dimensional double-well potential, and to a 208$\\mu$s atomistic Trp-cage folding trajectory from D.E. Shaw Research.

研究动机与目标

• 解决分子动力学模拟中采样时间尺度有限的问题，特别是针对罕见事件与长时间动力学。
• 克服依赖马尔可夫假设与滞后时间选择的马尔可夫状态模型（MSMs）的局限性。
• 开发一种方法，从短时、无偏轨迹中估计平衡态与非平衡态可观测量（如稳态分布、击穿概率和平均首达时间）。
• 实现在不构建转移矩阵或不将动力学离散化为马尔可夫链的前提下估计可观测量。
• 提供一种基于连续轨迹数据中概率分布与击穿值平稳性的自洽、迭代框架。

提出的方法

• 基于轨迹在离散相空间箱中的时间平均概率，对轨迹进行迭代重加权，以估计稳态（平衡态或非平衡稳态）分布。
• 应用概率向量在动力学下的左平稳性：当概率向量与转移矩阵相乘时保持不变，从而实现自洽的权重更新。
• 在击穿概率估计中，利用击穿向量的右平稳性：轨迹在某箱下游的平均击穿概率收敛至该箱的击穿值。
• 将轨迹权重初始化为均匀分布，并基于前一轮迭代的时间平均概率，迭代更新权重，确保收敛至平稳解。
• 应用边界条件：当轨迹达到目标宏观态时，击穿值设为1；当达到非目标态时，击穿值设为0。
• 仅以连续轨迹作为输入，保留所有动力学细节，不假设离散化状态之间的马尔可夫转移。

实验结果

研究问题

• RQ1 我们能否在不假设马尔可夫动力学或构建转移矩阵的前提下，仅从短时无偏轨迹中估计系统的平衡态分布？
• RQ2 当在轨迹准备过程中正确考虑外部或边界条件时，该方法能否可靠估计非平衡稳态（NESS）分布？
• RQ3 与暴力参考计算相比，该迭代重加权方法在复杂系统（如Trp-cage蛋白）中估计击穿概率（分裂概率）的准确性如何？
• RQ4 该迭代过程是否能仅通过轨迹数据，利用Hill关系收敛至平均首达时间（MFPT）的无偏估计？
• RQ5 尽管缺乏马尔可夫假设，当相空间被离散化为箱时，该方法在多大程度上仍保持鲁棒性与准确性？

主要发现

• 在1D双势阱系统中，迭代平衡态分布估计器收敛至精确的玻尔兹曼分布，即使在第一轮迭代后也未观察到偏差。
• 在Trp-cage折叠轨迹中，迭代方法生成的平衡态分布与简单箱计数结果吻合良好，但最右侧箱存在显著偏差，需进一步研究。
• 在1D系统中，迭代击穿概率估计器的结果与暴力参考值高度一致，其S形分布轮廓符合理论预期。
• 在Trp-cage系统中，迭代击穿概率估计值与暴力估计值高度一致，尤其在宏观态附近，独立实验中相关系数接近1。
• 该方法在不依赖转移矩阵或马尔可夫假设的情况下实现收敛，仅依赖连续轨迹与迭代自洽性。
• 该方法在形式上等价于对一种非标准转移矩阵应用幂法，该矩阵可捕捉所有时间尺度的完整动力学，此结论经合作者的理论分析证实。