arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.07911cs.CV

超越启发式提示:一种基于概念的贝叶斯框架用于零样本图像识别

Beyond Heuristic Prompting: A Concept-Guided Bayesian Framework for Zero-Shot Image Recognition

City University of Hong Kong(香港城市大学);Harbin Institute of Technology(哈尔滨工业大学);Nanyang Technological University(南洋理工大学)

Hui Liu, Kecheng Chen, Jialiang Wang, Xianming Liu, Wenya Wang, Haoliang Li

AI总结:

本文提出一种基于概念的贝叶斯框架,通过整合类特定概念提升零样本图像识别性能,采用多阶段概念合成和自适应软修剪似然,实现更鲁棒和高效的分类效果。

摘要:

视觉-语言模型(VLMs),如CLIP,已显著推动了零样本图像识别的发展。然而,其性能仍然受到次优提示工程和对目标类别的适应性差的限制。尽管最近的方法试图通过多样类描述来改进提示,但它们往往依赖于启发式设计,缺乏通用性,并且容易受到异常提示的影响。本文通过整合类特定的概念来增强提示。通过将概念视为潜在变量,我们从贝叶斯视角重新思考零样本图像分类,将预测视为对概念空间的边缘化,其中每个概念被权重为先验和测试图像条件似然。这种公式强调了良好的结构化概念提案分布和概念先验的优化的重要性。为了构建一个表达力强且高效的提案分布,我们引入了一个由大语言模型驱动的多阶段概念合成管道来生成判别性和组合性概念,随后通过决定性点过程来强制多样性。为了减轻异常概念的影响,我们提出了一种无训练、自适应的软修剪似然,它在单次前向传递中削弱其影响。我们进一步提供了鲁棒性保证,并为我们的框架推导了多类超额风险界。广泛的实验表明,我们的方法在零样本图像分类中始终优于最先进的方法,验证了其有效性。我们的代码可在https://github.com/less-and-less-bugs/CGBC上获得。

英文摘要:

Vision-Language Models (VLMs), such as CLIP, have significantly advanced zero-shot image recognition. However, their performance remains limited by suboptimal prompt engineering and poor adaptability to target classes. While recent methods attempt to improve prompts through diverse class descriptions, they often rely on heuristic designs, lack versatility, and are vulnerable to outlier prompts. This paper enhances prompt by incorporating class-specific concepts. By treating concepts as latent variables, we rethink zero-shot image classification from a Bayesian perspective, casting prediction as marginalization over the concept space, where each concept is weighted by a prior and a test-image conditioned likelihood. This formulation underscores the importance of both a well-structured concept proposal distribution and the refinement of concept priors. To construct an expressive and efficient proposal distribution, we introduce a multi-stage concept synthesis pipeline driven by LLMs to generate discriminative and compositional concepts, followed by a Determinantal Point Process to enforce diversity. To mitigate the influence of outlier concepts, we propose a training-free, adaptive soft-trim likelihood, which attenuates their impact in a single forward pass. We further provide robustness guarantees and derive multi-class excess risk bounds for our framework. Extensive experiments demonstrate that our method consistently outperforms state-of-the-art approaches, validating its effectiveness in zero-shot image classification. Our code is available at https://github.com/less-and-less-bugs/CGBC.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑