Lune

ICML2026Top-tier venue

USE: A Unified Self-Ensembling Framework for Test-Time Prompt Tuning

Siru Jiang, Jian Liang, Ran He, Tieniu Tan

2026Year

Abstract

Test-time adaptation (TTA) has emerged as a popular paradigm for improving the performance of vision–language models ( e.g. , CLIP) on downstream tasks. Among existing CLIP-based TTA methods, Test-Time Prompt Tuning (TPT) is a pioneering work that optimizes textual prompts using multiple test-time augmentations and remains a strong baseline to date. In this work, we revisit TPT and reveal that its optimization can be interpreted as implicitly learning from self-generated pseudo labels. Building on this perspective, we propose a unified self-ensembling framework ( USE ) that ensures consistency between the optimization and inference stages. During optimization, we introduce a simple yet effective self-ensembling ( SE ) strategy that emphasizes the test image itself over its augmented views adaptively to obtain more reliable pseudo labels. To fully exploit the potential of augmentations, we further apply the same strategy at inference time, unifying the objectives of both stages. Notably, SE can also act as a lightweight optimization-free TTA method. Extensive experiments across multiple datasets demonstrate that SE and USE outperform their counterparts, respectively. Furthermore, SE yields consistent performance gains when integrated with existing TTA methods. The code is available at https://github.com/sirujiang/USE .

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext f455ef21-cff0-4eab-9468-17671ef01b38

Builds on38

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines