arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.12314stat.MLcs.LG

神经网络中带噪声注入的不确定性量化:一种贝叶斯视角

Uncertainty Quantification With Noise Injection in Neural Networks: A Bayesian Perspective

Xueqiong Yuan, Jipeng Li, Ercan Engin Kuruoglu

AI总结:

本文从贝叶斯视角建立噪声注入与不确定性量化的联系,提出蒙特卡洛噪声注入方法,在训练时注入噪声、推断时多次前向传播,实验证明其性能优于基线模型。

摘要:

模型不确定性量化涉及测量和评估与模型预测相关的不确定性,有助于评估其可靠性和置信度。噪声注入是一种通过引入随机性来增强神经网络鲁棒性的技术。在本文中,我们从贝叶斯的角度建立了噪声注入与不确定性量化之间的联系。我们从理论上证明,向神经网络的权重注入噪声等价于在深度高斯过程上进行贝叶斯推断。因此,我们提出了一种蒙特卡洛噪声注入(MCNI)方法,该方法在训练期间向参数注入噪声,并在推断期间执行多次前向传播以估计预测的不确定性。通过在回归和分类任务上的模拟和实验,我们的方法表现出优于基线模型的性能。

英文摘要:

Model uncertainty quantification involves measuring and evaluating the uncertainty linked to a model's predictions, helping assess their reliability and confidence. Noise injection is a technique used to enhance the robustness of neural networks by introducing randomness. In this paper, we establish a connection between noise injection and uncertainty quantification from a Bayesian standpoint. We theoretically demonstrate that injecting noise into the weights of a neural network is equivalent to Bayesian inference on a deep Gaussian process. Consequently, we introduce a Monte Carlo Noise Injection (MCNI) method, which involves injecting noise into the parameters during training and performing multiple forward propagations during inference to estimate the uncertainty of the prediction. Through simulation and experiments on regression and classification tasks, our method demonstrates superior performance compared to the baseline model.

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

↑