研究筆記

Neural ODE 可以符合物理,卻仍然沒有校準

一個合成基因迴路基準在未見介入下比較無約束與符號結構向量場,發現可行性不等於誠實不確定性。

2026年7月22日 · 約 20 分鐘閱讀

學習所得微分方程可以完美插值觀察軌跡,卻誤解機制。問題常在介入後才出現:輸入比訓練更強、抑制一個元件,或由不常見初態開始。第二個問題是,即使多個模型 fit 之間有差異,ensemble uncertainty 也未必在陌生介入處擴闊得足夠。

本研究問:

結構符號約束與 ensemble uncertainty,能否令學習向量場在預測雙穩態基因迴路的未見介入時更誠實?

答案是混合的。結構改善物理可行性並稍為降低軌跡誤差,但在留出介入下,不確定區間校準很差。

合成 toggle-switch 世界

資料生成器是兩狀態互相抑制開關:

x˙=αx1+ynydxx+ux(t),y˙=αy1+xnxdyy+uy(t).\dot x=\frac{\alpha_x}{1+y^{n_y}}-d_xx+u_x(t), \qquad \dot y=\frac{\alpha_y}{1+x^{n_x}}-d_yy+u_y(t).

產生項為正、自衰減為負、兩狀態互相抑制,介入 ux,uyu_x,u_y 有已聲明方向。30 條訓練軌跡提供 1,350 個狀態—介入—導數樣本。

兩個向量場競爭:

  1. 無約束 multilayer perceptron;
  2. 把 cross-repression、輸入、衰減與產生項限制到指定符號的結構 production–decay 網絡。

每個 family 有六個 bootstrap 成員。校準只使用 in-domain 資料,更強脈衝與 knockout-like 介入留作測試。

留出介入下真實與學習 toggle-switch 軌跡及不確定帶。
圖 1。 介入 rollout。不確定帶在強留出擾動前校準;結構模型保持非負,但真值常落在帶外。

結構改變 hypothesis class

無約束網絡可表示靈活局部導數,包括生物上不合理符號。結構模型分開產生與衰減,並以正值變換限制權重。這不證明找到真生成器,只排除部分已知違反定性機制的向量場。

留出介入下,結構模型平均 RMSE 為 0.2370.237,無約束為 0.2590.259。結構模型最小預測保持 0.01460.0146;無約束最低達 1.178-1.178,在模擬器中是不可能濃度。

基因迴路相平面上的真實、無約束及符號結構向量場。
圖 2。 相場比較。相似訓練 loss 可隱藏不同全局幾何;符號結構排除部分不可能流,但未正確恢復所有 separatrix。

誤差改善不大,較強結果是可行性:結構 architecture 防止負值 rollout。但可行性不足;一條正值但錯誤的軌跡仍然錯誤。

校準帶來負面發現

Ensemble standard deviation 乘上一個由 in-domain 校準軌跡選出的尺度。無約束 ensemble 需要 3.523.52 倍,留出 coverage 達 0.9830.983;結構 ensemble 需要更大的 39.5739.57 倍,coverage 仍只有 0.7210.721

兩個 neural ODE family 的 coverage、誤差、不確定性關聯與 abstention 曲線。
圖 3。 校準診斷。較低 RMSE 與較好物理符號,不代表介入轉移下有有效預測區間。

為何結構 ensemble 會過度自信?所有成員共享同一限制 architecture 及相似訓練支持,可以彼此同意但共同帶有相同外推偏差。Bootstrap spread 量度重抽樣引起的變化,不會自動表示遺漏機制或介入轉移的不確定性。

結構模型的誤差—不確定性相關為 0.5140.514,無約束為 0.9530.953。移除高不確定案例後,結構 RMSE 由 0.2370.237 降至約 0.1980.198,所以不確定性並非無用,只是未校準到可當作有效區間。

分開檢查求解器誤差

Neural ODE 預測混合向量場誤差與數值積分誤差。Tolerance 由 10310^{-3} 收緊至 2×1062\times10^{-6} 時,相對軌跡差由約 9.55×1049.55\times10^{-4} 降至 3.09×1063.09\times10^{-6},遠低於模型 RMSE。

改變 ODE solver 不能修補錯誤外推向量場;相反,好的向量場若積分太粗也會看似很差。兩類誤差必須分開。

「介入校準」應要求甚麼

有科學用途的介入區間應在目標介入 family 評估,而非只看觀察軌跡。理想協議會預先聲明介入類別、分開弱校準與強測試介入、報告邊際及整條軌跡 coverage、檢查 interval width 與失敗位置、加入機制錯設模擬,並在支持不足時容許 abstention。

標題描述研究目標,結果則顯示目標尚未完全達成。這個負面發現比只報告 RMSE 更有資訊。

聲稱界線

所有軌跡均為合成,學習權重不辨認真實基因網絡。可支持結論是:

在此 toggle-switch 基準,符號結構降低平均留出 RMSE 並防止負預測,但 bootstrap ensemble 在強介入下仍嚴重 under-cover。

結構與校準是兩個不同設計問題。後續可結合符號約束、機制 ensembles、跨介入 family 的 conformal calibration、顯式 epistemic prior 或 reject option。

參考文獻

  1. R. T. Q. Chen , “Neural ordinary differential equations,” NeurIPS, 2018. Proceedings.
  2. P. Kidger , “Neural controlled differential equations for irregular time series,” NeurIPS, 2020. Proceedings.
  3. Y. Gal and Z. Ghahramani, “Dropout as a Bayesian approximation,” ICML, 2016. Proceedings.
  4. A. Guo , “On calibration of modern neural networks,” ICML, 2017. Proceedings.