TorchSISSO:基于PyTorch的独立筛选与稀疏化操作的高效可解释模型发现实现
💡
原文中文,约600字,阅读约需2分钟。
📝
内容提要
Matbench Discovery使用机器学习模型模拟无机晶体的稳定性,解决了热力学稳定性与形成能的差异。通过Python包和排行榜分析,CHGNet、M3GNet和MACE在预测热力学稳定性方面表现最佳,F1得分约为0.6,发现加速因子最高达5倍。强调分类指标比全局回归指标更相关,接近决策边界时易出现高假阳性率。
🎯
关键要点
-
Matbench Discovery使用机器学习模型模拟无机晶体的稳定性。
-
解决了热力学稳定性与形成能之间的脱节问题。
-
通过Python包和排行榜分析,探索了各个性能指标之间的权衡。
-
在热力学稳定性预测中,CHGNet、M3GNet和MACE表现最佳,F1得分约为0.6。
-
发现加速因子最高可达5倍。
-
强调分类指标比全局回归指标更相关。
-
接近决策边界时,准确的回归器易出现高假阳性率。
-
结果突显了关注与提高稳定性命中率实际相关的分类指标的必要性。
➡️