Scalar-pathway fidelity improves physical accuracy in short-range equivariant interatomic potentials

技術要約：短距離等変原子間ポテンシャルにおけるスカラー経路の忠実度（Scalar-Pathway Fidelity）

問題提起
MACE、Allegro、NequIPといった短距離等変原子間ポテンシャル（MLIP）は、$O(3)$等変テンソル特徴量を通じて局所的な原子幾何学を表現することで、分野を前進させてきた。これらのアーキテクチャは、高次ボディの幾何学的相関を捉えることに成功している。しかし、決定的なボトルネックが残っている。それは、最終的なデプロイメントに不可欠な予測（エネルギー、力、応力）が、これら豊かなテンソル表現を不変なスカラー（$\ell=0$）チャネルへと投影することによって導出されるという点である。

著者らは、この「スカラー経路」が、以下の2つの系統的な制限により、物理的な不正確さの原因となっていることが多いことを指摘している：

1. 一様集約（Uniform Aggregation）： スカラーメッセージは通常、重みのない置換不変な総和によって集約される。これは、局所的な配位数、密度、または幾何学的歪みに条件付けられた、適応的な振幅ゲーティングを欠いている。その結果、不均質な環境（表面、欠陥、歪んだ領域など）において解像度が不足する。
2. スペクトル・バイアス（Spectral Bias）： スカラー非線形性（例：SiLU）は、強い低周波スペクトル・バイアスを示す。これはポテンシャルエネルギー面（PES）における高周波構造を抑制し、短距離の反発力の軟化、光学フォノンのレッドシフト、および有限タイムステップのマイクロカノニカル積分中のエネルギー漂流といった物理的エラーを引き起こす。

従来のアプローチでは、スカラーの読み出し（readout）を、制御可能な設計次元としてではなく、固定された実装の詳細として扱うことが多い。本論文は、短距離MLIPにおける残留的な物理的エラーは、等変テンソル・バックボーンのみからではなく、このスカラー圧縮段階から大きく生じている可能性があると仮定している。

手法
これらの制限に対処するため、著者らは等変テンソル積バックボーンを変更することなく、$\ell=0$のスカラーチャネルに対してのみ作用する、2つの軽量で対称性を保持したモジュールを導入する。これらのモジュールは、スカラー・インターフェースの制御されたプローブとして機能する：

1. 物理学を考慮した近傍（PAN）プーリング：

   • メカニズム： 一様なスカラー集約を、シグモイド・ゲートを用いた振幅変調に置き換える。
   • 入力： ゲートは、原子番号、滑らかな配位数記述子（$\rho_i$）、およびカットオフ重み付き余弦角分散（$\eta_i$）を含む、不変なペア記述子から導出される。
   • 機能： これにより、置換不変性を破ることなく、スカラー集約を局所的な幾何学的歪み、配位子の喪失、および密度の変動に対して敏感にすることができる。

2. 物理学に基づいたスペクトル（PGS）ミキサー：

   • メカニズム： 短距離のスペクトル解像度を向上させるために、物理的に動機付けられたスペクトル基底を用いてスカラー特徴量を拡張する。
   • 実装：
     • エッジ・レベル： ペア距離に対するフーリエ・ベッセル展開を注入する。
     • 読み出し（Readout）レベル： 原子ごとの不変潜在変数に対して、テーパリングされた指数型半円（Exponential-of-Semicircle; ES）展開を注入する。
   • 機能： これらの基底は、標準的な非線形性が平滑化してしまう急峻な短距離曲率や高周波モードを、より良く表現するように設計されている。

実験設計
著者らは、これらの介入をテストするための高次ボディのメカニズム的足場として、主にMACEを使用している。実験プロトコルは以下を保証する：

• 等変バックボーン、相互作用の深さ、カットオフ、および学習予算は、ベースラインと修正モデル間で同一である。
• 比較は5つの独立した学習シードに対して行われる。
• モデルは、金属Ag、共有結合Si、短距離イオン性LiF/Li–Fサブセット、およびMD17/rMD17分子ベンチマークの多様なシステムでテストされる。
• 移植性テスト： PAN+PGSモジュールを、アーキテクチャ的に類似した位置にあるAllegroおよびNequIPに挿入し、その恩恵が異なる短距離等変バックボーン間で転移するかどうかを検証する。

主な結果

1. 精度の向上：

   • Ag、Si、およびLiF/Li–Fにわたって、結合されたMACE+PAN+PGSモデルは、ベースラインのMACEと比較して、力のMAEを22–27%、エネルギーMAEを**19–22%**減少させた。
   • 応力ラベルを持つシステムにおいて、応力誤差は**27–28%**減少した。
   • 分子ベンチマーク（rMD17）では、力の誤差が平均で**13.9%**減少し、最も顕著な利得は柔軟で非調和な分子で見られた。
   • 容量を一致させたコントロール（PAN/PGSなしでのバックボンドの拡張/深化）は、わずか4–7%の改善しか示さず、この利得が単なるパラメータ数の増加によるものではないことを証明した。

2. 移植性：

   • PAN+PGSをAllegroおよびNequIPに挿入したところ、力の誤差が**17–19%**減少し、この修正が異なる短距離等変バックボック間で移植可能であることを確認した（ただし、改善の大きさはアーキテクチャに依存し、MACEで最大であった）。

3. 物理的観測量：
   スカラー経路の修正は、著者らが予測した4つの明確な物理的失敗モードを特異的に緩和した：

   • 非調和ねじれ（Anharmonic Torsions）： アスピリンのねじれ分布におけるカルバック・ライブラリー（KL）ダイバージェンスを約3倍減少させ、「障壁漏れ（barrier leakage）」を修正した。
   • 振動モード： Siにおける高周波光学フォノンのレッドシフトを抑制し、振動状態密度（VDOS）のピーク位置誤差を約4倍改善した。
   • 積分漂流（Integration Drift）： バルクAgの500 ps NVEシミュレーションにおける永続的なエネルギー漂流を、因子として約7倍減少させた。
   • 液体構造： 液体Agの動径分布関数（RDF）における偽のサブシェルピークを減衰させ、非物理的な原子の接近を修正した。

4. 計算コスト：
   これらの改善は、約5%の追加推論FLOPsで達成され、コストと精度のフロンティアにおける低誤差・低オーバーヘッド領域に位置している。

意義と主張
本論文は、**スカラー経路の忠実度（scalar-pathway fidelity）**を、短距離等変原子間ポテンシャルのための実用的かつ直交する設計次元として特定したと主張している。著者らは以下のように述べている：

• テンソルの表現力とスカラーの忠実度は代替可能なものではない。表現力豊かな等変バックボンドであっても、ダウンストリームのスカラー操作が十分に適応的でない、あるいはスペクトル的に制限されている場合、物理的な詳細が失われる可能性がある。
• スラー経路は単なる実装の詳細ではなく、幾何学的情報がエネルギー地形へと変換される重要なインターフェースである。
• PANとPGSは、明示的な物理項（長距離静電相互作用など）を必要とせず、また等変バックボーンを置き換えることもなく、特定の、明確に区別された形態の情報損失（幾何学的解像度の不足およびスペクトル・バイアス）に対処する。
• 本手法は、既存の拡張（長距離相互作用や反応性に関するもの）と補完的な関係にあり、局所的な表現が構築された後に残る残留的な短距離エラーを特に対象としている。

著者らは、将来の短距離MLIPは、等変表現の物理的精度がデプロイ可能なエネルギー面として保存されることを確実にするために、スカラー集約とスペクトル混合を第一級の設計コンポーネントとして扱うべきであると結論付けている。