Technical report on label-informed logit redistribution for better domain generalization in low-shot classification with foundation models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Khan, Behraj, Syed, Tahir
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914054941966336
author Khan, Behraj
Syed, Tahir
author_facet Khan, Behraj
Syed, Tahir
contents Confidence calibration is an emerging challenge in real-world decision systems based on foundations models when used for downstream vision classification tasks. Due to various reasons exposed, logit scores on the CLIP head remain large irrespective of whether the image-language pairs reconcile. It is difficult to address in data space, given the few-shot regime. We propose a penalty incorporated into loss objective that penalizes incorrect classifications whenever one is made during finetuning, by moving an amount of log-likelihood to the true class commensurate to the relative amplitudes of the two likelihoods. We refer to it as \textit{confidence misalignment penalty (CMP)}. Extensive experiments on $12$ vision datasets and $5$ domain generalization datasets supports the calibration performance of our method against stat-of-the-art. CMP outperforms the benchmarked prompt learning methods, demonstrating average improvement in Expected Calibration Error (ECE) by average $6.01$\%, $4.01$ \% at minimum and $9.72$\% at maximum.
format Preprint
id arxiv_https___arxiv_org_abs_2501_17595
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Technical report on label-informed logit redistribution for better domain generalization in low-shot classification with foundation models
Khan, Behraj
Syed, Tahir
Computer Vision and Pattern Recognition
Confidence calibration is an emerging challenge in real-world decision systems based on foundations models when used for downstream vision classification tasks. Due to various reasons exposed, logit scores on the CLIP head remain large irrespective of whether the image-language pairs reconcile. It is difficult to address in data space, given the few-shot regime. We propose a penalty incorporated into loss objective that penalizes incorrect classifications whenever one is made during finetuning, by moving an amount of log-likelihood to the true class commensurate to the relative amplitudes of the two likelihoods. We refer to it as \textit{confidence misalignment penalty (CMP)}. Extensive experiments on $12$ vision datasets and $5$ domain generalization datasets supports the calibration performance of our method against stat-of-the-art. CMP outperforms the benchmarked prompt learning methods, demonstrating average improvement in Expected Calibration Error (ECE) by average $6.01$\%, $4.01$ \% at minimum and $9.72$\% at maximum.
title Technical report on label-informed logit redistribution for better domain generalization in low-shot classification with foundation models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2501.17595