Saved in:
| Main Authors: | , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2509.05485 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866910001518346240 |
|---|---|
| author | Kazanskii, Maksim Kasianov, Artem |
| author_facet | Kazanskii, Maksim Kasianov, Artem |
| contents | We study how the training data distribution affects confidence and performance in image classification models. We introduce Embedding Density, a model-agnostic framework that estimates prediction confidence by measuring the distance of test samples from the training distribution in embedding space, without requiring retraining. By filtering low-density (low-confidence) predictions, our method significantly improves classification accuracy. We evaluate Embedding Density across multiple architectures and compare it with state-of-the-art out-of-distribution (OOD) detection methods. The proposed approach is potentially generalizable beyond computer vision. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2509_05485 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Prior Distribution and Model Confidence Kazanskii, Maksim Kasianov, Artem Machine Learning We study how the training data distribution affects confidence and performance in image classification models. We introduce Embedding Density, a model-agnostic framework that estimates prediction confidence by measuring the distance of test samples from the training distribution in embedding space, without requiring retraining. By filtering low-density (low-confidence) predictions, our method significantly improves classification accuracy. We evaluate Embedding Density across multiple architectures and compare it with state-of-the-art out-of-distribution (OOD) detection methods. The proposed approach is potentially generalizable beyond computer vision. |
| title | Prior Distribution and Model Confidence |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2509.05485 |