Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2023
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2307.02245 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913231580168192 |
|---|---|
| author | Muttenthaler, Lukas Vandermeulen, Robert A. Zhang, Qiuyi Unterthiner, Thomas Müller, Klaus-Robert |
| author_facet | Muttenthaler, Lukas Vandermeulen, Robert A. Zhang, Qiuyi Unterthiner, Thomas Müller, Klaus-Robert |
| contents | Model overconfidence and poor calibration are common in machine learning and difficult to account for when applying standard empirical risk minimization. In this work, we propose a novel method to alleviate these problems that we call odd-$k$-out learning (OKO), which minimizes the cross-entropy error for sets rather than for single examples. This naturally allows the model to capture correlations across data examples and achieves both better accuracy and calibration, especially in limited training data and class-imbalanced regimes. Perhaps surprisingly, OKO often yields better calibration even when training with hard labels and dropping any additional calibration parameter tuning, such as temperature scaling. We demonstrate this in extensive experimental analyses and provide a mathematical theory to interpret our findings. We emphasize that OKO is a general framework that can be easily adapted to many settings and a trained model can be applied to single examples at inference time, without significant run-time overhead or architecture changes. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2307_02245 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Set Learning for Accurate and Calibrated Models Muttenthaler, Lukas Vandermeulen, Robert A. Zhang, Qiuyi Unterthiner, Thomas Müller, Klaus-Robert Machine Learning Computer Vision and Pattern Recognition Information Theory Model overconfidence and poor calibration are common in machine learning and difficult to account for when applying standard empirical risk minimization. In this work, we propose a novel method to alleviate these problems that we call odd-$k$-out learning (OKO), which minimizes the cross-entropy error for sets rather than for single examples. This naturally allows the model to capture correlations across data examples and achieves both better accuracy and calibration, especially in limited training data and class-imbalanced regimes. Perhaps surprisingly, OKO often yields better calibration even when training with hard labels and dropping any additional calibration parameter tuning, such as temperature scaling. We demonstrate this in extensive experimental analyses and provide a mathematical theory to interpret our findings. We emphasize that OKO is a general framework that can be easily adapted to many settings and a trained model can be applied to single examples at inference time, without significant run-time overhead or architecture changes. |
| title | Set Learning for Accurate and Calibrated Models |
| topic | Machine Learning Computer Vision and Pattern Recognition Information Theory |
| url | https://arxiv.org/abs/2307.02245 |