无需修改模型,仅用少量未标注图像就能提升开放词汇语义分割精度。
Test-Time Prototype Adaptation for Open-Vocabulary Semantic Segmentation

- 部署时通过自适应生成类别原型,不改动原有模型结构。
- 仅需约10%未标注数据即可构建有效原型库,显著提升分割准确率。
- 兼容多种主流方法,适配不同模型和场景的开放词汇分割任务。
开放词汇语义分割(OVSS)利用预训练的CLIP编码器进行密集预测,无需额外标注。现有方法通过重设计内部注意力或引入辅助视觉基础模型特征来改进空间表现,但均需访问主机内部计算且针对特定前向过程定制。本文提出测试时原型自适应(TPA),一种无需训练的插件式方法,仅在输出层操作,保持主机前向传播和权重不变。TPA通过轻量级归纳适应阶段,从主机自身在少量无标签部署域图像上的输出中识别置信锚点块,并将其冻结的DINO特征聚合为每类原型;推理时,通过与冻结原型库的余弦相似度查找,获得辅助得分并与主机逻辑值线性融合。TPA可与五种代表性OVSS主机兼容,覆盖三种CLIP主干、八个基准和多种内部视觉基础模型选择。使用单组超参数且无需针对各主机调优或参数更新,TPA始终提升分割精度,多数基准上仅需约10%的无标签部署域图像即可有效构建原型库。
原文摘要 · Abstract (English)
Open-vocabulary semantic segmentation (OVSS) repurposes a pretrained CLIP encoder for dense prediction without additional labeled supervision. Existing methods improve CLIP's spatial behavior either by redesigning its internal attention or by injecting features from auxiliary vision foundation models; both require access to the host's internal computation and are tailored to its specific forward pass. In this work, we propose Test-time Prototype Adaptation (TPA), a training-free plug-in that operates at the output level, leaving the host's forward pass and weights unmodified. By leveraging a lightweight transductive adaptation phase, TPA identifies confident anchor patches from the host's own output predictions on a small pool of unlabeled deployment-domain images, and aggregates their frozen DINO features into per-class prototypes; at inference, a single cosine similarity lookup against this frozen bank provides an auxiliary score fused linearly with the host's logits. TPA composes with five representative OVSS hosts spanning attention-redesign and VFM-injection designs, across three CLIP backbones, eight benchmarks, and multiple internal VFM choices. Under a single set of hyper-parameters and without per-host tuning or parameter updates, TPA consistently improves segmentation accuracy, with as few as approximately 10% of unlabeled deployment-domain images sufficing for effective bank construction on most benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。