arXiv · 2209.11604
Neural Clamping: Joint Input Perturbation and Temperature Scaling for Neural Network Calibration
Abstract
Neural network calibration is an essential task in deep learning to ensure consistency between the confidence of model prediction and the true correctness likelihood. In this paper, we propose a new post-processing calibration method called Neural Clamping, which employs a simple joint input-output transformation on a pre-trained classifier via a learnable universal input perturbation and an output temperature scaling parameter. Moreover, we provide theoretical explanations on why Neural Clamping is provably better than temperature scaling. Evaluated on BloodMNIST, CIFAR-100, and ImageNet image recognition datasets and a variety of deep neural network models, our empirical results show that Neural Clamping significantly outperforms state-of-the-art post-processing calibration methods. The code is available at github.com/yungchentang/NCToolkit, and the demo is available at huggingface.co/spaces/TrustSafeAI/NCTV.
Explore related subjects
Keep this discovery
Yung-Chen Tang, Pin-Yu Chen, Tsung-Yi Ho. 2022-09-23. Neural Clamping: Joint Input Perturbation and Temperature Scaling for Neural Network Calibration. https://arxiv.org/abs/2209.11604
Cite the original work for its findings. Save a collection to share your selection of sources.