Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Expectation consistency for calibration of neural networks
 
conference paper

Expectation consistency for calibration of neural networks

Clarte, Lucas Andry  
•
Loureiro, Bruno  
•
Krzakala, Florent  
Show more
2023
Uncertainty in Artificial Intelligence
Proceedings of the 39th Conference on Uncertainty in Artificial Intelligence (UAI 2023), PMLR 216:443–453.

Despite their incredible performance, it is well reported that deep neural networks tend to be overoptimistic about their prediction confidence. Finding effective and efficient calibration methods for neural networks is therefore an important endeavour towards better uncertainty quantification in deep learning. In this manuscript, we introduce a novel calibration technique named expectation consistency (EC), consisting of a post-training rescaling of the last layer weights by enforcing that the average validation confidence coincides with the average proportion of correct labels. First, we show that the EC method achieves similar calibration performance to temperature scaling (TS) across different neural network architectures and data sets, all while requiring similar validation samples and computational resources. However, we argue that EC provides a principled method grounded on a Bayesian optimality principle known as the Nishimori identity. Next, we provide an asymptotic characterization of both TS and EC in a synthetic setting and show that their performance crucially depends on the target function. In particular, we discuss examples where EC significantly outperforms TS.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

clarte23a.pdf

Type

N/a

Access type

openaccess

License Condition

n/a

Size

403.47 KB

Format

Adobe PDF

Checksum (MD5)

2d3052e076ae7df17b8326e72f3ff42e

Loading...
Thumbnail Image
Name

clarte23a-supp.pdf

Type

N/a

Access type

openaccess

License Condition

n/a

Size

232.33 KB

Format

Adobe PDF

Checksum (MD5)

50b813eab320bca3055ba22d1304e3d7

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés