arXiv:2604.27715cs.CV2026-04被引 2

通过初始化更平坦的提示,提升视觉语言模型测试时调优的预测可靠性。

Improving Calibration in Test-Time Prompt Tuning for Vision-Language Models via Data-Free Flatness-Aware Prompt Pretraining

论文配图:Improving Calibration in Test-Time Prompt Tuning for Vision-Language Models via Data-Free Flatness-Aware Prompt Pretraining
图 1 · 摘自论文原文
  • 用无标签数据预训练提示,使其初始状态位于损失函数平坦区域。
  • 在不增加计算开销的前提下,同时改善模型校准性和准确率。
  • 适合需高可靠性预测的部署场景,如医疗或金融领域应用。

测试时提示调优(TPT)通过利用未标注测试数据优化文本提示,提升了视觉语言模型的适应能力。然而,已有研究发现TPT常导致模型校准性差,影响预测可信度。现有方法引入正则化项约束输出以改进校准,但往往损害性能。本文揭示这些策略隐式促使优化趋向更平坦的极小值,且提示附近的损失曲面平坦度是决定校准质量的关键因素。为此,我们提出平坦度感知提示预训练(FPP),一种简单高效的TPT预训练框架,在适应前将提示初始化于损失景观的平坦区域。实验表明,仅替换现有TPT流程中的初始化方式,无需修改其他组件,即可同时提升校准性和性能。值得注意的是,FPP无需标签数据,且测试时调优无额外计算成本,极具实际部署价值。代码已公开于:https://github.com/YonseiML/fpp。

原文摘要 · Abstract (English)

Test-time prompt tuning (TPT) has emerged as a promising technique for enhancing the adaptability of vision-language models by optimizing textual prompts using unlabeled test data. However, prior studies have observed that TPT often produces poorly calibrated models, raising concerns about the reliability of their predictions. Recent works address this issue by incorporating additional regularization terms that constrain model outputs, which improve calibration but often degrade performance. In this work, we reveal that these regularization strategies implicitly encourage optimization toward flatter minima, and that the sharpness of the loss landscape around adapted prompts is a key factor governing calibration quality. Motivated by this observation, we introduce Flatness-aware Prompt Pretraining (FPP), a simple yet effective pretraining framework for TPT that initializes prompts within flatter regions of the loss landscape prior to adaptation. We show that simply replacing the initialization in existing TPT pipelines--without modifying any other components--is sufficient to improve both calibration and performance. Notably, FPP requires no labeled data and incurs no additional computational costs during test-time tuning, making it highly practical for real-world deployment. The code is available at: https://github.com/YonseiML/fpp.

提示调优模型校准视觉语言模型无监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。