Feb, 2024

超扩散:利用单一模型估计认知和随机不确定性

TL;DR通过训练和样本采集大量扩散模型的集合来准确估计表观不确定性(可以通过更多的训练数据来减少的不确定性)和内在不确定性(与手头任务相关的不确定性)在将机器学习(ML)应用于医学影像和天气预报等高风险应用的过程中至关重要。而在模型架构复杂性增加时,训练这样的集合变得计算上难以处理。本论文介绍了一种新的集合方法,称为超扩散,它允许使用单个模型准确估计表观和内在不确定性。不同于现有的基于蒙特卡洛 dropout 的单模型集合方法,超扩散提供了与多模型集合相同的预测准确性。我们在两个不同的任务上验证了我们的方法:X 射线计算机断层扫描(CT)重建和天气温度预测。