MatterSim-MT: A multi-task foundation model for in silico materials characterization

35M 構造・89 元素・0-5000K・0-1000GPa を覆う multi-task uMLIP（Microsoft Research AI for Science）

🎯 PES だけでなく Bader 電荷・Born effective charge・誘電率まで同一モデルで出力し、ferroelectric ヒステリシスや高圧相安定化など PES では届かない物性を統一シミュレーションする。

① 背景と課題

universal MLIP (uMLIP) は materials discovery を加速するが、既存モデルは PES（E/F/σ）のみを学習対象とし、双極子・電子分極を要する物性（ferroelectric hysteresis、LO-TO phonon splitting、redox transition 等）が射程外にある。multi-task の試みは限定的で、ほぼ expert model に留まる。

PES だけでは BaTiO3 強誘電体ヒステリシスや Li-rich cathode の redox 遷移を扱えない。

既存データセットは温度・圧力レンジが狭く、極限条件下の relevant materials 探索に届かない。

→ 単一 foundation model で PES + atomic property を同時学習し、極限条件（5000 K / 1000 GPa）まで届かせる。

② 手法 (a) GeoMFormer + multi-task heads

transformer backbone に invariant + equivariant 特徴量を併用。energy conservation・PBC・SO(3) 不変性を physical prior として強制。共通 atomic embedding から PES / Bader / magmom / Z* / ε∞ を分岐予測。

atomic structure → GeoMFormer embedding → {PES head, Bader head, Z* head, ε∞ head, magmom head}

10M parameter が MD throughput と精度のスイートスポット。1.3B では free-energy MAE が 18.50 → 9.62 meV/atom に改善。

② 手法 (b) Active learning + 35M / 89 元素

materials explorer による active learning で未踏領域構造を選別 → first-principles ラベル付け → iterative 再学習。MPF2021 / MPtraj / Alexandria / OMat24 と組み合わせて学習データを構築。

35M / 89 元素

構造数 / 力ベクトル 450M、T 0–5000 K、P 0–1000 GPa

未踏系統への extension は from-scratch 比 6.6% のデータで同等精度（Li2B12H12 で実証）。

③ 本研究で示したこと（要点）

PES + Bader / Z* / ε∞ / magmom を単一モデルで出力する初の foundation 規模 multi-task NN-IP
LaH10 超伝導相 (150 GPa)・MgO B1-B2 相境界 (10000 K) など極限条件で既存 uMLIP を上回る
BaTiO3 ferroelectric hysteresis、SiC LO-TO splitting、Li-rich cathode redox transition を再現
rev-PBE0-D3 fine-tune は 60 構造で from-scratch (900) 同等 → データ生成コスト 92% 削減

④ 主な結果 (a) スケーリング則

model / data size に対する validation loss（自由エネルギー予測 MAE）

2 桁スケールアップで MAE が約半減。foundation paradigm が materials science でも成立。

④ 主な結果 (b) uMLIP 比較 (stress / phonon)

stress MAE 8.5 GPa, phonon max freq MAE ~1 THz, group velocity MAE 23 km/s

④ 主な結果 (c) PES 以外の demonstrations

現象	システム	必要なヘッド
LO-TO phonon splitting	SiC (高圧)	Z* + ε∞
Ferroelectric hysteresis	BaTiO3	Z* + 電場 coupling
Cation→anion redox	Li-rich cathode	Bader 電荷
Phase boundary B1-B2	MgO (10000K)	PES (高 T,P)
Superconducting hydride	LaH10 (150 GPa)	PES (高圧)

④ 主な結果 (d) Active learning 効率

系統	fine-tune データ	精度
Li2B12H12	6.6%	from-scratch 同等
molten P / B	低データ	同等精度
water (rev-PBE0-D3)	60 構造	RDF / ADF / D 同等 (vs 900 構造)

data generation cost 92% 削減。高精度関数への昇格コスト劇的低下。

⑤ テイクホームメッセージ

🧲 Multi-task ヘッド
PES + Bader / Z* / ε∞ / magmom を単一モデルで。ferroelectric や redox を扱える。

🌡️ 極限条件
0-5000 K / 0-1000 GPa を学習データで連続被覆。LaH10 / MgO で既存超え。

📈 スケーリング則
data・model size の両軸で精度が単調改善。foundation paradigm が成立。

🎯 少量 fine-tune
active learning + 60-構造 fine-tune で高精度関数へ昇格、コスト 92% 削減。

他 uMLIP との対比

モデル	Multi-task	T,P 範囲
MACE-MP-0	—	限定
CHGNet	—	限定
Orb-v3 / SevenNet	—	限定
OMat24	—	中
MatterSim-MT	○	0-5000K / 0-1000GPa

本研究のインパクト

materials science の foundation model paradigm を 35M scale で実証
PES 超えの物性予測（双極子・redox）を NN-IP で扱う設計テンプレートを提示
chemexp 文脈: 直接スコープ外だが、bio-organic NN-IP (FeNNix-Bio1 等) の将来設計の参照点