Prompting without Panic: Attribute-aware, Zero-shot, Test-Time Calibration

TL;DR

This paper introduces a method for improving the calibration of vision-language models during test-time prompt tuning by using attribute-aware initialization and regularization, leading to more reliable confidence estimates.

Contribution

It proposes a novel attribute-aware prompt initialization and a regularization loss to enhance calibration during test-time prompt tuning in vision-language models.

Findings

Significantly reduces expected calibration error (ECE) from 11.7 to 4.11

Improves calibration across multiple datasets and architectures

Maintains high accuracy while enhancing confidence reliability

Abstract

Vision-language models (VLM) have demonstrated impressive performance in image recognition by leveraging self-supervised training on large datasets. Their performance can be further improved by adapting to the test sample using test-time prompt tuning (TPT). Unfortunately, the singular focus of TPT approaches on improving the accuracy suffers from tunnel vision, and leads to degradation in confidence calibration. This limits the applicability of TPT in critical applications. We make three contributions in this work. (1) We posit that random or naive initialization of prompts leads to overfitting on a particular test sample, and is the main reason for miscalibration of the VLM after TPT. To mitigate the problem, we propose careful initialization of test time prompt using prior knowledge about the target label attributes from a large language model (LLM); (2) To further maintain the quality of prompts during \tpt, we propose a novel regularization loss to reduce intraclass distance, and increase inter-class distance between the learnt Through extensive experiments on different CLIP architectures and 15 datasets, we show that our approach can effectively improve the calibration after TPT. We report an average expected calibration error (ECE) of 4.11 with our method, TCA, compared to 11.7 for vanilla TPT, 6.12 for C-TPT (ICLR'24), 6.78 for DiffTPT (CVPR'23), and 8.43 for PromptAlign (NeurIPS'23). The code is publicly accessible at: https://github.com/rhebbalaguppe/TCA_PromptWithoutPanic.