关于现代神经网络的校准

郭川, Geoff Pleiss, 孙宇, Kilian Q. Weinberger
第34届国际机器学习会议论文集, PMLR 70:1321-1330, 2017.

摘要

置信度校准——预测概率估计值以代表真实的正确性可能性这一问题——对于许多应用中的分类模型至关重要。我们发现,与十年前的神经网络不同,现代神经网络的校准效果较差。通过大量的实验,我们观察到深度、宽度、权重衰减和批量归一化是影响校准的重要因素。我们评估了各种后处理校准方法在最先进的架构上,使用图像和文档分类数据集。我们的分析和实验不仅提供了对神经网络学习的见解,还为实际设置提供了一个简单直接的方案:在大多数数据集上,温度缩放——Platt Scaling的一个单参数变体——在校准预测方面出奇地有效。

引用本文


BibTeX
@InProceedings{pmlr-v70-guo17a, title = {关于现代神经网络的校准}, author = {郭川 and Geoff Pleiss and 孙宇 and Kilian Q. Weinberger}, booktitle = {第34届国际机器学习会议论文集}, pages = {1321--1330}, year = {2017}, editor = {Precup, Doina and Teh, Yee Whye}, volume = {70}, series = {机器学习研究论文集}, month = {06--11 Aug}, publisher = {PMLR}, pdf = {https://pmlr.com.cn/v70/guo17a/guo17a.pdf}, url = {https://pmlr.com.cn/v70/guo17a.html}, abstract = {置信度校准——预测概率估计值以代表真实的正确性可能性这一问题——对于许多应用中的分类模型至关重要。我们发现,与十年前的神经网络不同,现代神经网络的校准效果较差。通过大量的实验,我们观察到深度、宽度、权重衰减和批量归一化是影响校准的重要因素。我们评估了各种后处理校准方法在最先进的架构上,使用图像和文档分类数据集。我们的分析和实验不仅提供了对神经网络学习的见解,还为实际设置提供了一个简单直接的方案:在大多数数据集上,温度缩放——Platt Scaling的一个单参数变体——在校准预测方面出奇地有效。} }
Endnote
%0 会议论文 %T 关于现代神经网络的校准 %A 郭川 %A Geoff Pleiss %A 孙宇 %A Kilian Q. Weinberger %B 第34届国际机器学习会议论文集 %C 机器学习研究论文集 %D 2017 %E Doina Precup %E Yee Whye Teh %F pmlr-v70-guo17a %I PMLR %P 1321--1330 %U https://pmlr.com.cn/v70/guo17a.html %V 70 %X 置信度校准——预测概率估计值以代表真实的正确性可能性这一问题——对于许多应用中的分类模型至关重要。我们发现,与十年前的神经网络不同,现代神经网络的校准效果较差。通过大量的实验,我们观察到深度、宽度、权重衰减和批量归一化是影响校准的重要因素。我们评估了各种后处理校准方法在最先进的架构上,使用图像和文档分类数据集。我们的分析和实验不仅提供了对神经网络学习的见解,还为实际设置提供了一个简单直接的方案:在大多数数据集上,温度缩放——Platt Scaling的一个单参数变体——在校准预测方面出奇地有效。
APA
郭川,Pleiss, G., 孙宇, & Weinberger, K.Q. (2017). 关于现代神经网络的校准. 第34届国际机器学习会议论文集, in 机器学习研究论文集 70:1321-1330 可从 https://pmlr.com.cn/v70/guo17a.html 获取.

相关材料