TorchSISSO:基于PyTorch的独立筛选与稀疏化操作的高效可解释模型发现实现

💡 原文中文,约600字,阅读约需2分钟。
📝

内容提要

Matbench Discovery使用机器学习模型模拟无机晶体的稳定性,解决了热力学稳定性与形成能的差异。通过Python包和排行榜分析,CHGNet、M3GNet和MACE在预测热力学稳定性方面表现最佳,F1得分约为0.6,发现加速因子最高达5倍。强调分类指标比全局回归指标更相关,接近决策边界时易出现高假阳性率。

🎯

关键要点

  • Matbench Discovery使用机器学习模型模拟无机晶体的稳定性。

  • 解决了热力学稳定性与形成能之间的脱节问题。

  • 通过Python包和排行榜分析,探索了各个性能指标之间的权衡。

  • 在热力学稳定性预测中,CHGNet、M3GNet和MACE表现最佳,F1得分约为0.6。

  • 发现加速因子最高可达5倍。

  • 强调分类指标比全局回归指标更相关。

  • 接近决策边界时,准确的回归器易出现高假阳性率。

  • 结果突显了关注与提高稳定性命中率实际相关的分类指标的必要性。

➡️

继续阅读