EDBench: Large-Scale Electron Density Data for Molecular Modeling
電子密度 ρ(r) を軸に MLFFs を原子レベルから電子レベルへ · Xiang, Zeng ら (arXiv:2505.09262, 2025)
🎯 創薬様336万分子の ED を付与した最大規模データと ED 中心ベンチマークを構築し、電子レベル表現学習の可能性を初めて体系的に検証する。
① 背景と課題

Hohenberg–Kohn 定理が示すように電子密度 ρ(r) は基底状態のあらゆる物性を一意に決めるが、既存の機械学習力場(MLFFs)は原子・分子・単純な量子化学量（エネルギー・力）の学習に留まり、電子分布を明示的に扱ってこなかった。

ED の算出は高コストな第一原理 DFT に依存し、大規模 ED データが存在しない
ED を入力/出力/検索対象として評価する統一ベンチマークがない
GDL 系 MLFFs は1体〜5体間の原子多体相互作用に偏り、微視的電子分布を見落としている

→ ED を明示的に取り込むための大規模データと評価軸を提供し、MLFFs を電子レベルへ拡張する

② 手法の概要: EDBench + HGEGNN
EDBench: DFT → ED → 学習モデルPCQM4Mv23.36M molsPsi4 DFTB3LYP/6-31G**CUBE EDρ(r) gridsED encodersPointVector/X-3DHGEGNNgen ρ(r)

Psi4 で B3LYP/6-31G**(S含有は+G**) DFT を 336万分子に実行し CUBE 形式 ED を生成（計205,000コア時）。グリッド間隔 0.4 Bohr。ED 点群は密度閾値 ρτ で間引き、原子+電子の異種グラフに EGNN を拡張した HGEGNN で構造から ED を生成する。

③ 本研究で示したこと
  • 創薬様336万分子の ED を持つ既知最大規模データ EDBench を構築
  • 予測・検索・生成の3系統6タスクからなる ED 中心ベンチマークを設計
  • ED 単独で量子性質（エネルギー・軌道・多極子）を高精度推定できると実証
  • HGEGNN が構造から ED を DFT 比約4桁高速に高相関で予測
④ 主な結果 (a) ED生成速度 (Table 7)
0.013HGEGNN245.8DFTED生成時間 (秒/mol, 約4桁高速)
④ 主な結果 (b) ED生成精度 (Table 7)
56.4ρ=0.187.00.1591.00.2ED生成 Pearson(%) vs 閾値ρτ
④ 主な結果 (c) DFT vs 予測 ρ(r)
DFT ρ(r)予測 ρ(r)DFT ED vs HGEGNN予測 (Spearman 99.2%)
④ 主な結果 (d) 予測 MAE (Table 3)
243.49PV-E1190.77X3D-E117.54PV-E58.13X3D-E5エネルギー成分 MAE (低いほど良)
⑤ テイクホームメッセージ
電子レベルへの拡張
ED を明示的に扱うことで MLFFs を原子レベルからより物理的な電子レベル表現へ拡張できる。
DFT を置換する高速サロゲート
HGEGNN は 0.013秒/mol（DFT 245.8秒）で Pearson 0.91 の ED を生成。
ED ベース仮想スクリーニング
InfoNCE で MS↔ED を整列し、EquiformerV2 系が検索で最良。
生成 ED は下流に有利
生成 ED は滑らかで下流モデルに適合しやすく、DFT 起源より MAE が低下したケースも。
量子化学データセットとの比較
談データ規模ED 提供対象
QM9~134Kなし原子/値
QM9-VASP~134KCHGCAR原子/値
PubChemQC85Mなし構造/gap
EDBench (本研究)3.36MCUBE ρ(r)電子密度
本研究のインパクト
  • ED 駆動の創薬・材料科学への基盤を提供し、電子レベル表現学習を加速
  • 高速 ED サロゲートにより MD 規模で電子記述子を算出可能に
  • 公開予定のデータとコードが公平で再現可能な ED ベンチマークを確立
Hongxin Xiang, Ke Li, ..., Xin Jin, Xiangxiang Zeng. EDBench: Large-Scale Electron Density Data for Molecular Modeling. arXiv preprint. 2025. https://arxiv.org/abs/2505.09262