1. 從“硬編碼”到“軟學(xué)習(xí)”為什么我們需要無集體變量的承諾函數(shù)在計(jì)算化學(xué)和分子模擬領(lǐng)域我們經(jīng)常面臨一個(gè)核心挑戰(zhàn)如何高效且準(zhǔn)確地描述一個(gè)復(fù)雜分子體系的“狀態(tài)”傳統(tǒng)上我們依賴“集體變量”。你可以把它想象成給一個(gè)復(fù)雜的、多自由度的分子系統(tǒng)比如一個(gè)正在折疊的蛋白質(zhì)安裝幾個(gè)“儀表盤”。這些儀表盤上的讀數(shù)比如某個(gè)關(guān)鍵化學(xué)鍵的長(zhǎng)度、兩個(gè)關(guān)鍵原子間的距離、或者一個(gè)二面角的角度就被定義為集體變量。模擬的目標(biāo)就是觀察這些“儀表盤”讀數(shù)如何隨時(shí)間變化從而理解蛋白質(zhì)是如何從一條松散的鏈折疊成特定三維結(jié)構(gòu)的。這個(gè)方法行之有效但它有個(gè)根本性的“硬傷”這些“儀表盤”是我們?cè)谀M開始前憑借化學(xué)直覺和先驗(yàn)知識(shí)“硬編碼”進(jìn)去的。這就帶來了兩個(gè)大問題。第一先驗(yàn)知識(shí)依賴過強(qiáng)。如果我們對(duì)一個(gè)新體系的理解不夠深入很可能選錯(cuò)或漏掉關(guān)鍵的“儀表盤”導(dǎo)致模擬無法捕捉到真實(shí)的反應(yīng)路徑或構(gòu)象變化。第二維度詛咒的隱憂。為了全面描述一個(gè)復(fù)雜過程我們可能會(huì)引入過多的集體變量這不僅增加了計(jì)算負(fù)擔(dān)更可能因?yàn)樽兞块g的冗余或非線性耦合讓后續(xù)的分析和采樣變得異常困難。那么有沒有可能讓機(jī)器自己從最原始的數(shù)據(jù)中“學(xué)會(huì)”一套更優(yōu)的描述體系狀態(tài)的方式呢這就是《Nat. Comput. Sci.》上這篇工作探討的核心基于原子坐標(biāo)的無集體變量承諾函數(shù)學(xué)習(xí)。這里的“原子坐標(biāo)”就是最原始、最底層的輸入數(shù)據(jù)即每個(gè)原子在三維空間中的位置?！盁o集體變量”意味著我們不預(yù)先定義任何具體的反應(yīng)坐標(biāo)。“承諾函數(shù)”則是一個(gè)數(shù)學(xué)上的關(guān)鍵概念它本質(zhì)上是一個(gè)將高維原子坐標(biāo)空間映射到一維“進(jìn)度”標(biāo)量的函數(shù)。這個(gè)“進(jìn)度”標(biāo)量在0到1之間變化0代表反應(yīng)的起始態(tài)如未折疊的蛋白質(zhì)1代表反應(yīng)的終態(tài)如折疊好的蛋白質(zhì)中間值則描述了反應(yīng)路徑上的任意一點(diǎn)。所以這篇工作的革命性在于它試圖用數(shù)據(jù)驅(qū)動(dòng)的方法繞開依賴人類經(jīng)驗(yàn)的“集體變量”設(shè)計(jì)環(huán)節(jié)直接從原子坐標(biāo)中學(xué)習(xí)出一個(gè)最優(yōu)的“承諾函數(shù)”。這個(gè)函數(shù)本身就可以被視為一個(gè)數(shù)據(jù)驅(qū)動(dòng)的、最優(yōu)的“反應(yīng)坐標(biāo)”。它不僅能更本質(zhì)地刻畫反應(yīng)過程還可能發(fā)現(xiàn)人類未曾預(yù)料到的、隱藏在原子運(yùn)動(dòng)背后的關(guān)鍵模式。2. 承諾函數(shù)連接微觀動(dòng)力學(xué)與宏觀命運(yùn)的數(shù)學(xué)橋梁要理解這個(gè)學(xué)習(xí)方法我們必須先吃透“承諾函數(shù)”到底是什么以及它在統(tǒng)計(jì)力學(xué)和分子動(dòng)力學(xué)中扮演的角色。這不是一個(gè)憑空創(chuàng)造的新詞而是有著堅(jiān)實(shí)理論根基的概念。想象一個(gè)復(fù)雜的能量景觀圖上面有無數(shù)個(gè)山谷穩(wěn)定態(tài)和山峰過渡態(tài)。一個(gè)分子體系就像在這個(gè)景觀圖上漫游的旅行者。承諾函數(shù)記為 ( C(\mathbf{x}) )其中 (\mathbf{x}) 代表所有原子的坐標(biāo)一個(gè)非常高維的向量。對(duì)于景觀圖上的任意一點(diǎn) (\mathbf{x})( C(\mathbf{x}) ) 的值給出了一個(gè)概率答案從當(dāng)前這個(gè)位置出發(fā)體系首次到達(dá)目標(biāo)態(tài)B的概率而不是先回到初始態(tài)A的概率。這是一個(gè)非常強(qiáng)大的定義。如果 ( C(\mathbf{x}) 0 )意味著從這個(gè)點(diǎn)出發(fā)幾乎注定會(huì)先回到A態(tài)如果 ( C(\mathbf{x}) 1 )則意味著幾乎注定會(huì)先到達(dá)B態(tài)。那么( C(\mathbf{x}) 0.5 ) 的點(diǎn)集合就構(gòu)成了一個(gè)非常特殊的曲面——等承諾曲面。這個(gè)曲面可以近似地理解為反應(yīng)坐標(biāo)上的“過渡態(tài)”或“分水嶺”因?yàn)閺倪@里出發(fā)體系“猶豫不決”前往A或B的概率各占一半。從數(shù)學(xué)上講承諾函數(shù)是向后Kolmogorov方程在特定邊界條件下的解。這個(gè)偏微分方程描述了在給定動(dòng)力學(xué)規(guī)則下某個(gè)“首次到達(dá)”事件的概率。對(duì)于滿足細(xì)致平衡條件的平衡態(tài)動(dòng)力學(xué)如常規(guī)分子動(dòng)力學(xué)模擬承諾函數(shù)還與體系的平衡概率分布和自由能景觀有著直接聯(lián)系。那么承諾函數(shù)好在哪里為什么說它可能比人工選擇的集體變量更優(yōu)最優(yōu)性在一定的數(shù)學(xué)準(zhǔn)則下如最大化弛豫速率或最小化過渡路徑的方差承諾函數(shù)本身被證明是描述A到B轉(zhuǎn)變的“最優(yōu)”反應(yīng)坐標(biāo)。它最大限度地利用了體系的動(dòng)力學(xué)信息。無偏性它的定義不依賴于任何先驗(yàn)的、人為定義的幾何量。只要給定了初始態(tài)A和目標(biāo)態(tài)B的微觀定義比如通過原子坐標(biāo)的某個(gè)范圍來界定承諾函數(shù)就可以從原理上被計(jì)算出來。概率解釋清晰它的值具有明確的物理意義——首次到達(dá)概率。這比一個(gè)單純的距離或角度值包含更豐富的動(dòng)力學(xué)信息。然而直接通過求解Kolmogorov方程來計(jì)算承諾函數(shù)對(duì)于高維分子體系是計(jì)算上不可行的。這就引出了核心問題我們?nèi)绾螐挠邢薜臄?shù)據(jù)分子動(dòng)力學(xué)模擬軌跡中“學(xué)習(xí)”出這個(gè)函數(shù)的近似形式這正是機(jī)器學(xué)習(xí)大顯身手的地方。3. 學(xué)習(xí)框架拆解如何教會(huì)神經(jīng)網(wǎng)絡(luò)“預(yù)測(cè)命運(yùn)”既然無法解析求解我們就用一個(gè)參數(shù)化的函數(shù)比如一個(gè)神經(jīng)網(wǎng)絡(luò)來近似承諾函數(shù) ( C_{\theta}(\mathbf{x}) )其中 (\theta) 代表神經(jīng)網(wǎng)絡(luò)的權(quán)重參數(shù)。學(xué)習(xí)的目標(biāo)是找到一組參數(shù) (\theta)使得 ( C_{\theta}(\mathbf{x}) ) 盡可能接近真實(shí)的承諾函數(shù)。關(guān)鍵就在于設(shè)計(jì)損失函數(shù)。損失函數(shù)就像給神經(jīng)網(wǎng)絡(luò)布置的“作業(yè)題目”題目出得好學(xué)生才能學(xué)到真本事?；诔兄Z函數(shù)的理論性質(zhì)我們可以構(gòu)造出以下幾種核心的損失項(xiàng)3.1 邊界條件損失明確起點(diǎn)與終點(diǎn)這是最直觀的約束。我們必須告訴網(wǎng)絡(luò)什么樣的構(gòu)型算是“反應(yīng)開始”A態(tài)什么樣的構(gòu)型算是“反應(yīng)完成”B態(tài)。通常我們可以從分子動(dòng)力學(xué)軌跡中根據(jù)某些簡(jiǎn)單的幾何準(zhǔn)則如RMSD挑選出一批明確的A態(tài)構(gòu)象 ({\mathbf{x}_A}) 和B態(tài)構(gòu)象 ({\mathbf{x}_B})。對(duì)應(yīng)的損失函數(shù)項(xiàng)為 [ \mathcal{L}{boundary} \frac{1}{N_A} \sum{\mathbf{x} \in A} [C_{\theta}(\mathbf{x}) - 0]^2 \frac{1}{N_B} \sum_{\mathbf{x} \in B} [C_{\theta}(\mathbf{x}) - 1]^2 ] 這一項(xiàng)強(qiáng)制網(wǎng)絡(luò)對(duì)于純A態(tài)輸出接近0對(duì)于純B態(tài)輸出接近1。3.2 動(dòng)力學(xué)一致性損失核心的“教學(xué)”環(huán)節(jié)這是整個(gè)方法的靈魂。承諾函數(shù)是一個(gè)“鞅”martingale這意味著在微觀動(dòng)力學(xué)的演化下承諾函數(shù)值的期望在未來時(shí)刻保持不變。用更直白的話說從當(dāng)前構(gòu)型 (\mathbf{x}t) 出發(fā)經(jīng)過一個(gè)很短的時(shí)間 (\tau) 后體系演化到 (\mathbf{x}{t\tau})那么 (C(\mathbf{x}t)) 應(yīng)該等于所有可能 (\mathbf{x}{t\tau}) 所對(duì)應(yīng)的 (C) 值的平均期望值。我們可以利用模擬軌跡中連續(xù)的時(shí)間幀來構(gòu)造這一約束。對(duì)于軌跡中的每一個(gè)時(shí)間點(diǎn) (t)我們有 ((\mathbf{x}t, \mathbf{x}{t\tau})) 這樣一個(gè)數(shù)據(jù)對(duì)。動(dòng)力學(xué)一致性損失要求 [ \mathcal{L}{dynamics} \frac{1}{N} \sum{t} [C_{\theta}(\mathbf{x}t) - C{\theta}(\mathbf{x}{t\tau})]^2 ]注意這里使用的是平方差但實(shí)際訓(xùn)練時(shí)更嚴(yán)謹(jǐn)?shù)淖龇ㄊ亲?(C{\theta}(\mathbf{x}t)) 去預(yù)測(cè) (C{\theta}(\mathbf{x}{t\tau})) 的期望。由于我們通常只有一條或有限條軌跡(C{\theta}(\mathbf{x}_{t\tau})) 本身是一個(gè)確定值而非分布所以直接最小化二者差異是一種可行的近似。更高級(jí)的方法會(huì)引入基于變分原理的損失函數(shù)。這個(gè)損失項(xiàng)是“無監(jiān)督”的它不依賴任何標(biāo)記數(shù)據(jù)只依賴軌跡本身的時(shí)序關(guān)系。它迫使神經(jīng)網(wǎng)絡(luò)去捕捉那些在短時(shí)間動(dòng)力學(xué)演化中保持不變的“慢變量”而這些慢變量恰恰是決定反應(yīng)進(jìn)程的關(guān)鍵。3.3 平滑性正則化損失避免過擬合與數(shù)值不穩(wěn)定神經(jīng)網(wǎng)絡(luò)容易對(duì)訓(xùn)練數(shù)據(jù)過擬合學(xué)到一些高頻的、無意義的噪聲。對(duì)于物理函數(shù)我們通常期望它是相對(duì)平滑的。因此加入對(duì)函數(shù)梯度相對(duì)于輸入坐標(biāo) (\mathbf{x})的懲罰項(xiàng)是必要的 [ \mathcal{L}{smooth} \lambda \cdot \frac{1}{N} \sum{\mathbf{x}} || \nabla_{\mathbf{x}} C_{\theta}(\mathbf{x}) ||^2 ] 其中 (\lambda) 是正則化強(qiáng)度系數(shù)。這一項(xiàng)懲罰函數(shù)變化劇烈的區(qū)域促使網(wǎng)絡(luò)學(xué)習(xí)到一個(gè)平滑變化的承諾函數(shù)這更符合物理直觀也能提升模型的泛化能力。最終總的損失函數(shù)是上述各項(xiàng)的加權(quán)和 [ \mathcal{L}{total} \alpha \mathcal{L}{boundary} \beta \mathcal{L}{dynamics} \gamma \mathcal{L}{smooth} ] 通過反向傳播優(yōu)化 (\theta)最小化總損失我們就得到了一個(gè)訓(xùn)練好的承諾函數(shù)神經(jīng)網(wǎng)絡(luò) (C_{\theta}^{*}(\mathbf{x}))。實(shí)操心得一軌跡數(shù)據(jù)準(zhǔn)備與時(shí)間延遲τ的選擇數(shù)據(jù)的質(zhì)量直接決定學(xué)習(xí)的上限。軌跡需要足夠長(zhǎng)能多次跨越A和B態(tài)之間的區(qū)域從而為動(dòng)力學(xué)一致性損失提供豐富的“轉(zhuǎn)變中”的構(gòu)象樣本。時(shí)間延遲 (\tau) 的選擇是個(gè)關(guān)鍵技巧它必須遠(yuǎn)小于體系在過渡態(tài)區(qū)域的停留時(shí)間但又需要長(zhǎng)于原子運(yùn)動(dòng)的振動(dòng)周期通常是幾個(gè)到幾百個(gè)飛秒。如果τ太小( \mathbf{x}t ) 和 ( \mathbf{x}{t\tau} ) 幾乎沒區(qū)別損失函數(shù)會(huì)趨于零學(xué)不到有用信息如果τ太大動(dòng)力學(xué)一致性假設(shè)本身可能不再成立。一個(gè)實(shí)用的方法是嘗試幾個(gè)不同的τ值觀察學(xué)習(xí)結(jié)果的穩(wěn)定性。4. 網(wǎng)絡(luò)架構(gòu)與輸入工程處理高維原子坐標(biāo)的實(shí)戰(zhàn)策略原子坐標(biāo) (\mathbf{x}) 是一個(gè)維度為 (3N) 的向量N是原子數(shù)。對(duì)于蛋白質(zhì)等大分子N可達(dá)數(shù)千甚至上萬直接將其拋入全連接網(wǎng)絡(luò)是低效且難以訓(xùn)練的因?yàn)樗雎粤嗽拥幕瘜W(xué)身份和三維空間結(jié)構(gòu)。4.1 對(duì)稱性處理旋轉(zhuǎn)平移不變性是鐵律分子的勢(shì)能及其動(dòng)力學(xué)與整個(gè)分子在空間中的絕對(duì)位置和朝向無關(guān)。因此我們學(xué)習(xí)的承諾函數(shù) (C_{\theta}(\mathbf{x})) 必須是旋轉(zhuǎn)和平移不變的。有兩種主流實(shí)現(xiàn)方式數(shù)據(jù)預(yù)處理對(duì)齊在將坐標(biāo)輸入網(wǎng)絡(luò)前先將每一幀的分子結(jié)構(gòu)通過最小二乘法擬合到一個(gè)參考結(jié)構(gòu)上通常選A態(tài)或B態(tài)的平均結(jié)構(gòu)消除整體的旋轉(zhuǎn)和平移。這是最直接的方法但需要注意參考結(jié)構(gòu)的選擇可能引入偏差。架構(gòu)內(nèi)置不變性設(shè)計(jì)本身就具有不變性的神經(jīng)網(wǎng)絡(luò)。這通常通過以下手段實(shí)現(xiàn)使用內(nèi)部坐標(biāo)輸入不是笛卡爾坐標(biāo)而是鍵長(zhǎng)、鍵角、二面角。這天然滿足不變性但可能會(huì)丟失一些全局信息。使用距離矩陣輸入所有原子對(duì)之間的距離 (r_{ij})。距離是旋轉(zhuǎn)平移不變的。但距離矩陣是 (N \times N) 的存在冗余且維度隨原子數(shù)平方增長(zhǎng)。使用等變網(wǎng)絡(luò)如SchNet、SE(3)-Transformer等這些網(wǎng)絡(luò)專門設(shè)計(jì)用于處理3D點(diǎn)云數(shù)據(jù)其層間特征變換滿足等變性而最終的標(biāo)量輸出可以通過不變性池化如求和、平均來獲得。這是目前最前沿且優(yōu)雅的方法但實(shí)現(xiàn)和訓(xùn)練復(fù)雜度較高。對(duì)于大多數(shù)初次嘗試的應(yīng)用采用預(yù)處理對(duì)齊 標(biāo)準(zhǔn)全連接網(wǎng)絡(luò)是一個(gè)穩(wěn)妥的起點(diǎn)。4.2 特征構(gòu)建從原始坐標(biāo)到信息豐富的描述符即使對(duì)齊后直接將 (3N) 維坐標(biāo)送入網(wǎng)絡(luò)仍非上策。更好的做法是構(gòu)建一些對(duì)分子狀態(tài)更敏感的特征描述符。這不同于預(yù)定義集體變量這里的特征是更基礎(chǔ)、更通用的數(shù)學(xué)變換旨在降低維度并保留關(guān)鍵信息。例如接觸圖計(jì)算所有重原子對(duì)之間的距離然后通過一個(gè)平滑的截?cái)嗪瘮?shù)如 (1/(1(r/r_0)^6))將其映射為0到1之間的值得到一個(gè)稠密的接觸特征向量。主慣性矩分子的三個(gè)主慣性矩反映了分子的整體形狀和延展度。二級(jí)結(jié)構(gòu)含量通過DSSP等算法實(shí)時(shí)計(jì)算軌跡每一幀中α螺旋、β折疊等二級(jí)結(jié)構(gòu)所占的比例。這些特征可以作為原始坐標(biāo)的補(bǔ)充與對(duì)齊后的坐標(biāo)一起拼接成輸入向量。關(guān)鍵在于這些特征仍然是通用、自動(dòng)計(jì)算的不針對(duì)特定反應(yīng)路徑。4.3 網(wǎng)絡(luò)結(jié)構(gòu)選擇一個(gè)典型的結(jié)構(gòu)可以是輸入層接收處理后的坐標(biāo)和/或特征描述符。若干隱藏層使用全連接層激活函數(shù)常用ReLU或Swish。層數(shù)和寬度需要根據(jù)問題復(fù)雜度調(diào)整可以從3層128神經(jīng)元開始嘗試。輸出層單個(gè)神經(jīng)元使用Sigmoid激活函數(shù)將輸出約束在(0,1)區(qū)間直觀對(duì)應(yīng)承諾概率。實(shí)操心得二處理原子序與輸入順序分子模擬軌跡中每個(gè)原子的索引通常是固定的。這意味著你的輸入向量中第i個(gè)位置永遠(yuǎn)對(duì)應(yīng)同一個(gè)原子。這允許網(wǎng)絡(luò)學(xué)習(xí)特定原子的行為模式。但這也要求你的所有訓(xùn)練和預(yù)測(cè)數(shù)據(jù)必須具有完全相同的原子數(shù)和順序。在數(shù)據(jù)預(yù)處理流水線中務(wù)必加入一致性檢查。如果使用基于距離或接觸圖的特征則可以部分緩解對(duì)原子順序的嚴(yán)格依賴。5. 訓(xùn)練流程、驗(yàn)證與應(yīng)用場(chǎng)景分析5.1 訓(xùn)練流程與技巧數(shù)據(jù)分割將分子動(dòng)力學(xué)軌跡按時(shí)間序列分割成訓(xùn)練集、驗(yàn)證集和測(cè)試集。切忌隨機(jī)打亂因?yàn)閯?dòng)力學(xué)一致性損失依賴于時(shí)間相鄰的幀。通?？梢园磿r(shí)間順序取前70%為訓(xùn)練集中間15%為驗(yàn)證集最后15%為測(cè)試集。損失權(quán)重調(diào)參(\alpha, \beta, \gamma) 的平衡至關(guān)重要。初期可以設(shè)置 (\alpha) 較大確保網(wǎng)絡(luò)先學(xué)會(huì)識(shí)別A/B態(tài)。然后逐步增加 (\beta) 的權(quán)重讓網(wǎng)絡(luò)專注于學(xué)習(xí)動(dòng)力學(xué)約束。平滑項(xiàng)權(quán)重 (\gamma) 通常設(shè)為一個(gè)較小的值如1e-4防止函數(shù)出現(xiàn)尖峰。監(jiān)控指標(biāo)損失曲線觀察總損失以及各分項(xiàng)損失在訓(xùn)練集和驗(yàn)證集上的下降情況防止過擬合。承諾值分布在驗(yàn)證集上繪制A態(tài)、B態(tài)以及中間態(tài)構(gòu)象的承諾函數(shù)值直方圖。理想情況下A態(tài)應(yīng)集中在0附近B態(tài)集中在1附近中間態(tài)呈0到1的分布。軌跡著色將學(xué)習(xí)到的 (C_{\theta}(\mathbf{x})) 值作為顏色繪制在原始的或降維后的構(gòu)象空間中直觀觀察是否形成了從A到B的平滑漸變。5.2 學(xué)成之后承諾函數(shù)的用武之地訓(xùn)練好的承諾函數(shù)神經(jīng)網(wǎng)絡(luò) (C_{\theta}^{*}(\mathbf{x})) 本身就是一個(gè)強(qiáng)大的分析工具和加速引擎。作為反應(yīng)坐標(biāo)進(jìn)行可視化與分析將模擬軌跡中的每一幀通過 (C_{\theta}^{*}) 映射到一個(gè)標(biāo)量值我們可以用這個(gè)值作為橫坐標(biāo)繪制自由能剖面圖。這個(gè)剖面圖是基于“學(xué)習(xí)到的最優(yōu)坐標(biāo)”的可能比基于人工CV的剖面圖更能清晰地揭示過渡態(tài)和中間態(tài)。指導(dǎo)增強(qiáng)采樣這是其最重要的應(yīng)用之一。承諾函數(shù)的值天然定義了“進(jìn)度”。我們可以基于此進(jìn)度設(shè)計(jì)偏置勢(shì)能進(jìn)行增強(qiáng)采樣。例如在元?jiǎng)恿W(xué)中可以將 (C_{\theta}^{}(\mathbf{x})) 直接作為集體變量在其上添加高斯偏置勢(shì)引導(dǎo)模擬快速跨越能壘。由于 (C_{\theta}^{}) 是數(shù)據(jù)驅(qū)動(dòng)得到的最優(yōu)坐標(biāo)在其上進(jìn)行偏置往往效率更高能更快地探索反應(yīng)路徑。構(gòu)象聚類與路徑分析可以對(duì) (C_{\theta}^{*}(\mathbf{x})) 值進(jìn)行分段將構(gòu)象空間劃分為從A到B的不同“承諾區(qū)間”從而識(shí)別反應(yīng)路徑上的關(guān)鍵中間態(tài)。轉(zhuǎn)移速率估計(jì)結(jié)合過渡態(tài)理論承諾函數(shù)可用于更精確地估計(jì)A態(tài)到B態(tài)的轉(zhuǎn)變速率常數(shù)。5.3 方法局限性討論盡管前景廣闊該方法仍有其局限和挑戰(zhàn)數(shù)據(jù)饑渴性學(xué)習(xí)一個(gè)可靠的承諾函數(shù)需要足夠多的、覆蓋了A到B之間主要過渡區(qū)域的軌跡數(shù)據(jù)。如果初始模擬完全無法跨越能壘即停留在A態(tài)則方法失效。通常需要先通過一些粗粒度的或基于簡(jiǎn)單CV的增強(qiáng)采樣獲得“種子”路徑。對(duì)動(dòng)力學(xué)的假設(shè)動(dòng)力學(xué)一致性損失建立在馬爾可夫性和時(shí)間可逆性等假設(shè)之上。對(duì)于非常復(fù)雜的非馬爾可夫動(dòng)力學(xué)或遠(yuǎn)離平衡的體系其適用性需要驗(yàn)證。神經(jīng)網(wǎng)絡(luò)的黑箱性學(xué)到的承諾函數(shù)是一個(gè)神經(jīng)網(wǎng)絡(luò)其內(nèi)部邏輯不易解釋。我們雖然得到了一個(gè)優(yōu)秀的反應(yīng)坐標(biāo)但可能難以直接將其分解為人類可理解的物理量如某個(gè)具體的二面角。計(jì)算成本訓(xùn)練神經(jīng)網(wǎng)絡(luò)需要額外的計(jì)算開銷對(duì)于非常長(zhǎng)的軌跡或非常大的體系特征計(jì)算和前向傳播可能成為瓶頸。6. 與相關(guān)機(jī)器學(xué)習(xí)方法的對(duì)比與定位在分子模擬領(lǐng)域利用機(jī)器學(xué)習(xí)學(xué)習(xí)反應(yīng)坐標(biāo)或慢變量的方法不止一種。理解本文方法與它們的區(qū)別能更好地定位其價(jià)值。方法名稱核心思想是否需要標(biāo)記數(shù)據(jù)與承諾函數(shù)學(xué)習(xí)的關(guān)鍵區(qū)別時(shí)間滯后獨(dú)立成分分析 (TICA)尋找在給定時(shí)間滯后下自相關(guān)函數(shù)最大的線性組合。否無監(jiān)督線性方法只能捕捉線性慢變量。承諾函數(shù)學(xué)習(xí)是非線性的表達(dá)能力更強(qiáng)。變分動(dòng)力學(xué)數(shù)據(jù)嵌入 (VDE)變分法學(xué)習(xí)一個(gè)編碼器最大化編碼狀態(tài)的時(shí)間相關(guān)性。否無監(jiān)督與承諾函數(shù)學(xué)習(xí)在理論上緊密相關(guān)很多VDE的損失函數(shù)形式與承諾函數(shù)的變分原理等價(jià)。可視為同一思想下的不同實(shí)現(xiàn)。深度自動(dòng)編碼器 動(dòng)力學(xué)正則化用自編碼器降維并在潛空間施加動(dòng)力學(xué)平滑約束。否無監(jiān)督主要目標(biāo)是降維和特征提取得到的潛變量不一定具有明確的“首次到達(dá)概率”物理意義。承諾函數(shù)直接輸出物理意義明確的標(biāo)量概率。監(jiān)督學(xué)習(xí)分類A/B態(tài)用A態(tài)和B態(tài)構(gòu)象作為正負(fù)樣本訓(xùn)練分類器。是需要明確標(biāo)簽分類器輸出的是“屬于某類的概率”而非“首次到達(dá)某類的概率”。對(duì)于過渡區(qū)域的構(gòu)象分類器可能給出模糊的~0.5概率但這與承諾概率的~0.5物理含義不同。承諾函數(shù)包含了動(dòng)力學(xué)的時(shí)序信息。本文的“基于原子坐標(biāo)的無集體變量承諾函數(shù)學(xué)習(xí)”方法可以看作是非線性、基于深度學(xué)習(xí)的、以首次到達(dá)概率為物理目標(biāo)的、無監(jiān)督動(dòng)力學(xué)模態(tài)提取方法。它站在TICA等線性方法的肩膀上引入了神經(jīng)網(wǎng)絡(luò)的強(qiáng)大非線性擬合能力并直接瞄準(zhǔn)了“承諾概率”這一具有清晰理論解釋的學(xué)習(xí)目標(biāo)。7. 復(fù)現(xiàn)指南與排坑要點(diǎn)如果你想在自己的體系上嘗試復(fù)現(xiàn)或應(yīng)用此方法以下是一個(gè)可操作的路線圖及可能遇到的“坑”。7.1 復(fù)現(xiàn)步驟概覽數(shù)據(jù)準(zhǔn)備運(yùn)行常規(guī)MD或增強(qiáng)采樣MD獲得一條或多條連接A和B態(tài)的軌跡。軌跡格式轉(zhuǎn)換至易處理的格式如.npy或.h5。定義A態(tài)和B態(tài)。通常使用RMSD閾值RMSD_to_A δA 的幀標(biāo)記為A態(tài)RMSD_to_B δB 的幀標(biāo)記為B態(tài)。δ的選擇需確保構(gòu)象確實(shí)穩(wěn)定在相應(yīng)態(tài)內(nèi)。對(duì)每一幀進(jìn)行旋轉(zhuǎn)平移對(duì)齊消除整體運(yùn)動(dòng)?？蛇x計(jì)算額外的通用特征描述符如接觸圖。按時(shí)間順序分割數(shù)據(jù)集。模型構(gòu)建使用PyTorch或TensorFlow定義神經(jīng)網(wǎng)絡(luò)。輸入維度等于處理后的特征維度。實(shí)現(xiàn)包含邊界損失、動(dòng)力學(xué)一致性損失和平滑損失的損失函數(shù)。設(shè)置優(yōu)化器如Adam和學(xué)習(xí)率調(diào)度器。訓(xùn)練與調(diào)優(yōu)在訓(xùn)練集上訓(xùn)練在驗(yàn)證集上監(jiān)控。調(diào)整損失權(quán)重α, β, γ、網(wǎng)絡(luò)深度/寬度、學(xué)習(xí)率、時(shí)間延遲τ等超參數(shù)。關(guān)鍵驗(yàn)證觀察驗(yàn)證集上A/B態(tài)承諾值的分離度以及承諾值沿軌跡的平滑變化情況。分析與應(yīng)用在測(cè)試集或新軌跡上應(yīng)用訓(xùn)練好的模型計(jì)算每幀的承諾值。繪制承諾值隨時(shí)間變化曲線、承諾值分布直方圖。將承諾值作為反應(yīng)坐標(biāo)通過WHAM等方法計(jì)算自由能剖面。將承諾函數(shù)作為集體變量集成到PLUMED等增強(qiáng)采樣插件中進(jìn)行元?jiǎng)恿W(xué)模擬。7.2 常見問題與排查思路問題1訓(xùn)練后承諾函數(shù)對(duì)所有構(gòu)象的輸出都趨近于0.5。排查這是最常見的問題。首先檢查動(dòng)力學(xué)一致性損失項(xiàng)是否有效。計(jì)算訓(xùn)練數(shù)據(jù)中連續(xù)幀的原子位移確保時(shí)間延遲τ設(shè)置合理使得 (\mathbf{x}t) 和 (\mathbf{x}{t\tau}) 既有區(qū)別又有相關(guān)性?？梢試L試增大τ。其次檢查邊界損失項(xiàng)確認(rèn)A/B態(tài)標(biāo)簽是否正確以及這些樣本在訓(xùn)練中是否被充分使用檢查損失貢獻(xiàn)?？赡苁沁吔鐡p失權(quán)重α太小被動(dòng)力學(xué)損失淹沒。問題2模型對(duì)訓(xùn)練集過擬合在驗(yàn)證集上A/B態(tài)承諾值分離很差。排查增加平滑性正則化權(quán)重γ。在神經(jīng)網(wǎng)絡(luò)中加入Dropout層。減小網(wǎng)絡(luò)規(guī)模減少層數(shù)或神經(jīng)元數(shù)。獲取更多樣化的訓(xùn)練數(shù)據(jù)更長(zhǎng)的軌跡或更多獨(dú)立的短軌跡。問題3承諾值變化不單調(diào)在軌跡中頻繁劇烈跳動(dòng)。排查強(qiáng)烈表明平滑性約束不足增大γ。同時(shí)檢查輸入特征是否包含高頻噪聲如某些快速振動(dòng)的鍵長(zhǎng)考慮對(duì)輸入特征進(jìn)行平滑處理或使用更整體的特征。也可能是τ太小動(dòng)力學(xué)損失沒有捕捉到有意義的慢模式。問題4將學(xué)到的承諾函數(shù)用于元?jiǎng)恿W(xué)采樣效率沒有提升甚至下降。排查承諾函數(shù)的質(zhì)量高度依賴訓(xùn)練數(shù)據(jù)。如果訓(xùn)練數(shù)據(jù)沒有充分覆蓋過渡態(tài)區(qū)域?qū)W到的函數(shù)在該區(qū)域可能不準(zhǔn)確甚至給出誤導(dǎo)性的梯度。確保用于訓(xùn)練的軌跡至少有一些成功跨越能壘的片段。此外在元?jiǎng)恿W(xué)中偏置勢(shì)的寬度和高度需要根據(jù)承諾函數(shù)值域的變化速度重新調(diào)整不能沿用針對(duì)幾何CV的經(jīng)驗(yàn)參數(shù)。這個(gè)領(lǐng)域正在快速發(fā)展將深度學(xué)習(xí)與分子動(dòng)力學(xué)深度融合代表了從“人工設(shè)計(jì)描述符”到“機(jī)器發(fā)現(xiàn)描述符”的范式轉(zhuǎn)變。雖然在實(shí)際應(yīng)用中仍有諸多細(xì)節(jié)需要打磨但它無疑為我們理解復(fù)雜分子體系的稀有事件動(dòng)力學(xué)打開了一扇新的大門。