Effectiveness of Vision Language Models for
Open-world Single Image Test Time Adaptation

Indian Institute of Science, Bengaluru, India
Transactions in Machine Learning Research (TMLR), 2025    

ROSITA Framework

PromptAlign design
Overview of ROSITA framework: The test samples with Desired and Undesired classes arrive one at a time. The image features are matched with the text based classifier, the confidence scores of which are used to distinguish between desired and undesired class samples through a simple LDA based class identifier. Based on this classification and if a sample is identified to be reliable, the respective feature banks are updated and the proposed ReDUCe loss is optimized to update the LayerNorm parameters of the Vision Encoder.

Abstract

Adapting models to dynamic, real-world environments characterized by shifting data distributions and unseen test scenarios is a critical challenge in deep learning. In this paper, we consider a realistic and challenging Test-Time Adaptation setting, where a model must continuously adapt to test samples that arrive sequentially, one at a time, while distinguishing between known and unknown classes. Current Test-Time Adaptation methods operate under closed-set assumptions or batch processing, differing from the real-worls open-set scenarios. We address this limitation by establishing a comprehensive benchmark for Open-set Single-image Test-Time Adaptation using Vision-Language Models. Furthermore, we propose ROSITA, a novel framework that leverages dynamically updated feature banks to identify reliable test samples and employs a contrastive learning objective to improve the separation between known and unknown classes. Our approach effectively adapts models to domain shifts for known classes while rejecting unfamiliar samples. Extensive experiments across diverse real-world benchmarks demonstrate that ROSITA sets a new state-of-the-art in open-set TTA, achieving both strong performance and computational efficiency for real-time deployment.

BibTeX


        @article{sreenivas2025rosita,
          title={Efficient Open Set Single Image Test Time Adaptation of Vision Language Models},
          author={Sreenivas, Manogna and Biswas, Soma},
          journal={Transactions on Machine Learning Research},
            year={2025}
        }