提出新方法让视觉语言模型在未知数据中更安全地适应已知类。
ProtoDCS: Towards Robust and Efficient Open-Set Test-Time Adaptation for Vision-Language Models
- 用概率混合模型动态区分已知与未知样本,取代固定阈值。
- 通过原型级更新和不确定性感知损失,提升准确率并降低计算开销。
- 适合需要稳定部署的开放集场景,尤其适用于大模型实时适应。
大规模视觉语言模型(VLMs)具备强大的零样本识别能力,但其实际部署常受分布偏移影响。现有测试时自适应(TTA)方法基于封闭集假设,在包含协变量偏移的已知类(csID)和未知类(csOOD)数据的开放集场景下失效。关键挑战在于:模型需精准识别未知样本以避免干扰,同时对已知类进行有效适应。现有开放集TTA(OSTTA)方法依赖硬阈值分离和熵最小化适应,易误判模糊样本并产生过自信预测,且参数更新计算成本高。为此,本文提出原型双检分离(ProtoDCS),通过概率高斯混合模型(GMM)验证实现鲁棒分离,并采用不确定性感知损失与原型级高效更新策略,缓解过自信问题并减少计算开销。在CIFAR-10/100-C和Tiny-ImageNet-C上的大量实验表明,ProtoDCS显著提升已知类准确率与OOD检测性能,达到当前最优水平。代码将公开于https://github.com/O-YangF/ProtoDCS。
原文摘要 · Abstract (English)
Large-scale Vision-Language Models (VLMs) exhibit strong zero-shot recognition, yet their real-world deployment is challenged by distribution shifts. While Test-Time Adaptation (TTA) can mitigate this, existing VLM-based TTA methods operate under a closed-set assumption, failing in open-set scenarios where test streams contain both covariate-shifted in-distribution (csID) and out-of-distribution (csOOD) data. This leads to a critical difficulty: the model must discriminate unknown csOOD samples to avoid interference while simultaneously adapting to known csID classes for accuracy. Current open-set TTA (OSTTA) methods rely on hard thresholds for separation and entropy minimization for adaptation. These strategies are brittle, often misclassifying ambiguous csOOD samples and inducing overconfident predictions, and their parameter-update mechanism is computationally prohibitive for VLMs. To address these limitations, we propose Prototype-based Double-Check Separation (ProtoDCS), a robust framework for OSTTA that effectively separates csID and csOOD samples, enabling safe and efficient adaptation of VLMs to csID data. Our main contributions are: (1) a novel double-check separation mechanism employing probabilistic Gaussian Mixture Model (GMM) verification to replace brittle thresholding; and (2) an evidence-driven adaptation strategy utilizing uncertainty-aware loss and efficient prototype-level updates, mitigating overconfidence and reducing computational overhead. Extensive experiments on CIFAR-10/100-C and Tiny-ImageNet-C demonstrate that ProtoDCS achieves state-of-the-art performance, significantly boosting both known-class accuracy and OOD detection metrics. Code will be available at https://github.com/O-YangF/ProtoDCS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。