概念引导微调:引导 ViT 远离虚假相关以提升鲁棒性
Concept-Guided Fine-Tuning: Steering ViTs away from Spurious Correlations to Improve Robustness
用 LLM 无标签提出类相关概念、VLM 分割成概念掩码,再以微调目标把 ViT 的内部相关性图对齐到这些概念区域并抑制背景,是一种可直接复用的、无需人工标注的鲁棒性微调手段,也可用来提升模型可解释性。
原文依据 optimizes the model's internal relevance maps to align with spatially grounded concept masks
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论视觉 Transformer(ViT)在分布偏移下往往性能下降,因为它们依赖虚假相关(如背景线索),而非语义上有意义的特征。现有的正则化方法通常依赖简单的前景-背景掩码,无法捕捉定义物体的细粒度语义概念(例如“鸟”的“长喙”和“翅膀”)。因此,这些方法对分布偏移提供的鲁棒性有限。为解决这一局限,我们提出一种新的微调框架,将模型推理引导至概念级语义。我们的方法优化模型内部的相关性图,使其与空间落地的概念掩码对齐。这些掩码自动生成,无需人工标注:首先使用基于 LLM 的无标签方法提出类相关概念,然后用 VLM 进行分割。微调目标在将这些概念区域与相关性对齐的同时,抑制对虚假背景区域的关注。值得注意的是,该过程仅需少量图像,并使用数据集类别的一半。在五个分布外基准上的大量实验表明,我们的方法提升了多种基于 ViT 的模型的鲁棒性。此外,我们表明所得相关性图与语义物体部件具有更强的对齐,为更鲁棒、更可解释的视觉模型提供了一条可扩展的路径。最后,我们确认概念引导掩码比传统分割图能为模型鲁棒性提供更有效的监督,支持了我们的核心假设。
英文原文摘要 1633 字符
Vision Transformers (ViTs) often degrade under distribution shifts because they rely on spurious correlations, such as background cues, rather than semantically meaningful features. Existing regularization methods, typically relying on simple foreground-background masks, which fail to capture the fine-grained semantic concepts that define an object (e.g., "long beak" and "wings" for a "bird"). As a result, these methods provide limited robustness to distribution shifts. To address this limitation, we introduce a novel finetuning framework that steers model reasoning toward concept-level semantics. Our approach optimizes the model's internal relevance maps to align with spatially grounded concept masks. These masks are generated automatically, without manual annotation: class-relevant concepts are first proposed using an LLM-based, label-free method, and then segmented using a VLM. The finetuning objective aligns relevance with these concept regions while simultaneously suppressing focus on spurious background areas. Notably, this process requires only a minimal set of images and uses half of the dataset classes. Extensive experiments on five out-of-distribution benchmarks demonstrate that our method improves robustness across multiple ViT-based models. Furthermore, we show that the resulting relevance maps exhibit stronger alignment with semantic object parts, offering a scalable path toward more robust and interpretable vision models. Finally, we confirm that concept-guided masks provide more effective supervision for model robustness than conventional segmentation maps, supporting our central hypothesis.