arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2003.06820cs.LGcs.CVstat.ML
收藏

Intra Order-preserving Functions for Calibration of Multi-Class Neural Networks

ANU(澳大利亚国立大学);ACRV(澳大利亚机器人视觉中心);Georgia Tech(佐治亚理工学院);Microsoft Research(微软研究院);Google Research(谷歌研究院);University of Washington(华盛顿大学)

Amir Rahimi, Amirreza Shaban, Ching-An Cheng, Richard Hartley, Byron Boots

英文摘要:

Predicting calibrated confidence scores for multi-class deep networks is important for avoiding rare but costly mistakes. A common approach is to learn a post-hoc calibration function that transforms the output of the original network into calibrated confidence scores while maintaining the network's accuracy. However, previous post-hoc calibration techniques work only with simple calibration functions, potentially lacking sufficient representation to calibrate the complex function landscape of deep networks. In this work, we aim to learn general post-hoc calibration functions that can preserve the top-k predictions of any deep network. We call this family of functions intra order-preserving functions. We propose a new neural network architecture that represents a class of intra order-preserving functions by combining common neural network components. Additionally, we introduce order-invariant and diagonal sub-families, which can act as regularization for better generalization when the training data size is small. We show the effectiveness of the proposed method across a wide range of datasets and classifiers. Our method outperforms state-of-the-art post-hoc calibration methods, namely temperature scaling and Dirichlet calibration, in several evaluation metrics for the task.

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

↑