MLIP Arena：超越误差指标机器学习原子间势能基准平台

NeurIPS 2025 Spotlight项目MLIP Arena提供了一个统一、透明的机器学习原子间势能基准测试框架，突破传统基于DFT误差指标的局限，专注于评估模型的物理合理性和实际应用价值。

• 原作者/维护者： Yuan Chiang, Atomind AI团队
• 来源平台： GitHub
• 原始标题： mlip-arena: Fair and transparent benchmark of machine learning interatomic potentials
• 原始链接： https://github.com/atomind-ai/mlip-arena
• 发表时间： 2025年（NeurIPS 2025 Spotlight接收）

---

近年来，基于海量密度泛函理论（DFT）计算数据训练的机器学习原子间势能（MLIPs）彻底改变了分子和材料建模领域。这些基础模型通过在包含数百万DFT计算的数据集上进行训练，能够以接近DFT的精度预测原子间相互作用，同时计算速度提升数个数量级。

然而，现有的基准测试方法存在三个根本性缺陷：

数据泄漏问题：许多基准测试的训练集和测试集之间存在重叠，导致模型性能被高估，无法真实反映泛化能力。

迁移性局限：传统基准往往局限于特定材料体系或单一任务，缺乏跨体系、跨任务的全面评估。

过度依赖误差指标：现有方法主要关注与特定DFT参考值之间的数值误差，忽视了模型学习到的物理规律是否合理、能否在实际应用中保持稳定性。

---

MLIP Arena是由Atomind AI团队开发的开源基准平台，该项目荣获NeurIPS 2025 Spotlight（录取率约3.5%）和ICLR AI4Mat Spotlight双重认可。其核心设计理念是超越传统的误差指标，从物理合理性和实际应用价值两个维度全面评估MLIP模型。

架构无关性：MLIP Arena不依赖于特定的模型架构或训练数据集，支持公平比较各种MLIP方法，包括等变图神经网络、消息传递网络等不同技术路线。

物理一致性评估：平台不仅关注预测数值的准确性，更注重模型是否学习到了符合物理规律的原子间相互作用，例如能量守恒、力的旋转不变性等。

实际应用场景测试：通过分子动力学模拟、结构优化、声子计算等真实任务，评估模型在长时间模拟中的稳定性和可靠性。

---

MLIP Arena基于现代化的Python工作流编排框架Prefect构建，支持任务链式调用、分布式扩展和智能缓存。平台提供统一的ASE（Atomic Simulation Environment）兼容接口，支持多种MLIP模型的无缝集成。

结构优化（OPT）：评估模型在弛豫原子结构至能量最低状态时的收敛性和精度。

状态方程（EOS）：通过能量-体积扫描测试模型对不同晶格常数下材料能量的预测能力。

分子动力学（MD）：支持NVE、NVT、NPT等多种系综，可配置温度/压力调度策略（如退火、剪切等），测试模型在长时间模拟中的能量漂移和结构稳定性。

声子计算（PHONON）：基于phonopy驱动，评估模型预测晶格振动特性的能力。

弹性张量（ELASTICITY）：计算材料的弹性常数，测试模型对力学性质的预测准确性。

过渡态搜索（NEB）：实现Nudged Elastic Band方法，评估模型预测反应路径和能垒的能力，支持线性插值和IDPP（Image Dependent Pair Potential）两种图像插值策略。

---

MLIP Arena提供简洁的Python API，用户可以通过枚举类型遍历所有支持的模型，快速进行对比实验：

from mlip_arena.models import MLIPEnum
from mlip_arena.tasks import MD
from mlip_arena.tasks.utils import get_calculator
from ase import units
from ase.build import bulk

# 构建铜超晶胞
atoms = bulk("Cu", "fcc", a=3.6) * (5, 5, 5)

results = []

# 遍历所有支持的MLIP模型
for model in MLIPEnum:
    result = MD(
        atoms=atoms,
        calculator=get_calculator(
            model,
            dispersion=True,  # 启用色散校正
            dispersion_kwargs=dict(
                damping='bj', xc='pbe', cutoff=40.0 * units.Bohr
            ),
        ),
        ensemble="nve",           # NVE系综
        dynamics="velocityverlet",  # Velocity Verlet积分器
        total_time=1e3,             # 1皮秒模拟
        time_step=2,                # 2飞秒步长
    )
    results.append(result)

对于大规模并行计算，平台支持Prefect的任务提交机制：

from prefect import flow

@flow
def run_all_tasks():
    futures = []
    for model in MLIPEnum:
        future = MD.submit(atoms=atoms, ...)
        futures.append(future)
    return [f.result(raise_on_failure=False) for f in futures]

---