arXiv:2607.15851cs.CVcs.LG2026-07中稿 · ICML被引 2

解决视觉语言模型测试时适应中的类别不平衡问题

Von Mises-Fisher Mixture Model with Dynamic Shrinkage for Realistic Test-Time Transduction

论文配图:Von Mises-Fisher Mixture Model with Dynamic Shrinkage for Realistic Test-Time Transduction
图 1 · 摘自论文原文
  • 用动态收缩机制建模单位超球面上的特征分布
  • 在9个数据集上提升准确率,最高达14.3%(如ImageNet-1K)
  • 无需训练、不依赖特定任务参数,适合快速部署

现有方法旨在提升视觉语言模型在测试时的性能,其中测试时适应(transduction)因其高兼容性和效率成为有前景的范式。然而,真实评估常面临严重类别不平衡问题,导致性能下降甚至崩溃。本文从惩罚似然估计(PLE)视角重新审视适应过程,发现带KL散度锚定项的PLE天然具备自适应收缩行为,介于先验锚点与经验估计之间。由此揭示:当前方法脆弱性的根源在于缺乏锚定机制和静态收缩强度建模。为此提出混合冯·米塞斯-费舍尔模型与动态收缩(MOON)。MOON基于单位超球面上的冯·米塞斯-费舍尔混合分布建模特征表示,通过零样本先验在实例与类别层面动态调整收缩强度,抑制不可靠分配并防止异常类别的有害更新,从而缓解负迁移。MOON具有模型无关性、无需训练、无需任务特异性超参数调优。大量实验验证其在性能与效率上的优势,在9个数据集上平均提升准确率,最高达14.3%(ImageNet-1K),代码已开源。

原文摘要 · Abstract (English)

A range of methods aim to enhance the performance of vision-language models (VLMs) at test time. Among them, transduction has emerged as a promising paradigm due to its strong compatibility and efficiency. However, realistic evaluations often involve highly imbalanced class distributions, which cause performance degradation or even collapse. In this work, we systematically revisit transduction from the perspective of penalized likelihood estimation (PLE), showing that PLE with a KL-divergence anchor term naturally yields an adaptive shrinkage behavior between prior anchors and empirical estimates. From this viewpoint, the brittleness of transductive methods can be attributed to the absence of anchoring mechanism and static modeling of the shrinkage strength. Therefore, we propose Mixture of Von Mises-Fisher Models with Dynamic Shrinkage (MOON). MOON is built upon a mixture of von Mises-Fisher distributions to model feature representations on the unit hypersphere. To handle imbalance, MOON dynamically adjusts the shrinkage strength using zero-shot priors at both instance and class levels. Thus, it suppresses unreliable assignments and prevents harmful updates from outlier classes, thereby mitigating negative transfer. MOON is model-agnostic, training-free, and requires no task-specific hyperparameter tuning. Extensive experiments further validate the advantage of MOON in both performance and efficiency. Our code is available at https://github.com/walawalagoose/MOON

测试时适应动态收缩类别不平衡无监督

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