arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1809.07653physics.chem-ph
收藏

化学机器学习中快速且准确的不确定性估计

Fast and Accurate Uncertainty Estimation in Chemical Machine Learning

Felix Musil, Michael J. Willatt, Mikhail A. Langovoy, Michele Ceriotti

AI总结:

本研究提出一种基于重采样的低成本化学机器学习不确定性估计方案,可通过最大似然估计基准测试与校正,在稀疏高斯过程回归模型上实现可忽略的计算成本,验证了分子能量与核化学屏蔽估计的可靠性,可推广至其他任务与模型。

摘要:

我们提出了一种方案,可获取原子与分子性质机器学习模型预测相关的低成本且可靠的不确定性估计值。该方案基于重采样技术,通过对同一训练数据进行子采样生成多个模型。不确定性预测的准确性可通过最大似然估计进行基准测试,该方法还可用于校正重采样模型间的相关性,并通过交叉验证流程提升不确定性估计的性能。对于稀疏高斯过程回归(Gaussian Process Regression)模型,这种重采样估计器的计算成本可忽略不计。我们在分子能量预测以及分子晶体中核化学屏蔽的估计任务上验证了这些估计值的可靠性。将其扩展用于估计能量差、力或其他相关预测的不确定性也十分简便。该方法可轻松应用于其他机器学习方案,有助于提升数据驱动预测的可靠性,并推动训练集优化与主动学习策略的发展。

英文摘要:

We present a scheme to obtain an inexpensive and reliable estimate of the uncertainty associated with the predictions of a machine-learning model of atomic and molecular properties. The scheme is based on resampling, with multiple models being generated based on sub-sampling of the same training data. The accuracy of the uncertainty prediction can be benchmarked by maximum likelihood estimation, which can also be used to correct for correlations between resampled models, and to improve the performance of the uncertainty estimation by a cross-validation procedure. In the case of sparse Gaussian Process Regression models, this resampled estimator can be evaluated at negligible cost. We demonstrate the reliability of these estimates for the prediction of molecular energetics, and for the estimation of nuclear chemical shieldings in molecular crystals. Extension to estimate the uncertainty in energy differences, forces, or other correlated predictions is straightforward. This method can be easily applied to other machine learning schemes, and will be beneficial to make data-driven predictions more reliable, and to facilitate training-set optimization and active-learning strategies.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑