概念引导的微调:引导视觉Transformer远离虚假相关性以提升鲁棒性 Concept-Guided Fine-Tuning: Steering ViTs away from Spurious Correlations to Improve Robustness
用LLM发现类别概念+VLM定位,微调ViT相关性图对齐语义区域,提升分布偏移鲁棒性
前置知识
Vision Transformer (ViT)
Vision Transformer是将Transformer架构从NLP迁移到计算机视觉领域的模型。它将输入图像分割为固定大小的patch(如16×16像素),将每个patch线性映射为token嵌入,加上位置编码后送入标准Transformer编码器处理。ViT通过自注意力机制(self-attention)在patch之间建立全局依赖关系,能够捕获长距离视觉关系。典型代表包括ViT-B(Base规模)、DeiT-III(数据高效的ImageNet训练变体)和DINOv2(自监督预训练的ViT)。尽管在标准benchmark上表现优异,ViT在分布偏移下容易依赖虚假线索而非语义特征进行分类。
本文的核心目标就是改善ViT在分布偏移下的鲁棒性,理解ViT的工作机制和已知缺陷是理解本文动机的前提。
Out-of-Distribution (OOD) 鲁棒性
OOD鲁棒性指模型在面对与训练数据分布不同的输入时仍能正确工作的能力。常见的分布偏移包括:自然对抗样本(ImageNet-A中自然存在的、导致模型失败的图像)、视角/背景变化(ObjectNet中控制了物体姿态和背景的数据集)、艺术风格变化(ImageNet-R中的卡通/素描/雕塑风格)、几何变换(SI-Score中系统地改变物体位置、大小和旋转)。模型如果依赖训练集中的虚假相关性(如背景纹理)而非语义特征,在这些OOD数据上性能会急剧下降。
提升OOD鲁棒性是本文的直接目标,五个OOD基准测试是评估CFT效果的核心实验,理解这些分布偏移类型才能判断方法的实际价值。
注意力感知逐层相关性传播 (AttnLRP)
AttnLRP是一种专为Transformer架构设计的可解释性方法,源自逐层相关性传播(LRP)原理。它将模型的输出预测信号通过注意力权重和MLP模块反向传播到输入token,生成相关性图(relevance map)。对于每一层ℓ,token i在前一层的相关性通过注意力权重和后一层token j的相关性计算得到。与纯梯度方法相比,AttnLRP满足守恒性(conservation property)——传播的总相关性等于模型输出分数,保证了解释的忠实性。
CFT的核心创新之一就是用AttnLRP计算的相关性图作为优化目标,将其对齐到语义概念区域。AttnLRP的忠实性保证了这个优化目标的合理性。
视觉-语言模型 (VLM) 零样本分割
视觉-语言模型结合了视觉理解和语言理解能力,能够根据文本描述在图像中定位和分割目标。本文使用的GroundedSAM是GroundingDINO(开放集目标检测器)和SAM(Segment Anything Model,通用分割模型)的组合。给定文本提示(如"long beak"、"wings"),GroundingDINO首先在图像中定位相关区域,然后SAM生成精确的像素级分割掩码。这个过程是零样本的,无需针对特定概念进行训练。对于本文来说,这意味着可以自动为每个类别生成语义概念的空间定位掩码,完全不需要人工标注。
CFT能够自动、无需标注地生成概念级别的空间指导掩码,正是依赖GroundedSAM的零样本分割能力。这解决了传统方法需要人工标注前景掩码的瓶颈。
虚假相关性 (Spurious Correlations)
虚假相关性指训练数据中偶然存在但与目标类别无因果关系的统计关联。例如,如果训练集中大部分鸟的图片背景是树枝,模型可能学会用"树枝"来识别"鸟",而非鸟的实际特征(如翅膀、喙)。当测试数据中鸟出现在不同背景(如水面上)时,模型就会失败。这种依赖虚假线索的行为被称为"捷径学习"(shortcut learning),是现代深度学习模型泛化能力差的主要原因之一。对于ViT来说,研究表明其自注意力机制特别容易被背景纹理等虚假线索吸引。
消除虚假相关性的影响是CFT的出发点和最终目标。理解这个问题才能理解为什么需要将模型的推理引导到语义概念上。
研究动机
现代Vision Transformer虽然在ImageNet等标准基准上表现优异,但在面对分布偏移时鲁棒性严重不足。大量证据表明,这些模型依赖虚假相关性(spurious correlations)——如背景纹理、上下文线索——而非目标物体的语义内容来进行分类。例如,在ImageNet-A(自然对抗样本)上,原本在ImageNet验证集上准确率超过80%的ViT-B模型,Top-1准确率骤降至约13%。这种依赖在实际部署中是危险的,因为真实环境中的背景、视角、光照条件都不可控。现有试图解决这个问题的方法存在两个关键缺陷:一是依赖人工标注的前景-背景二元分割掩码,这种方法过于粗糙——识别一只"鸟"应该关注"翅膀"和"长喙"这样的判别性部件,而不是把整个前景当作一个均匀区域;二是大多数方法需要完整的训练或重新训练过程,对于大规模预训练模型来说计算代价过高。此外,一些方法(如RRR、GradMask)依赖输入梯度作为解释的代理,这对于ViT来说尤其成问题,因为输入梯度在ViT上可能不稳定或不忠实。
本文的目标是本文的具体目标是提出一种数据高效的后处理微调框架(CFT),能够在仅使用1,500张图像(ImageNet-1K一半类别,每类3张)的情况下,引导ViT模型的内部推理远离虚假线索、转向语义上有意义的概念级别特征,从而在五个OOD基准测试上一致地提升鲁棒性,同时基本维持甚至改善分布内准确率。此外,CFT产生的相关性图应与物体前景区域有更好的对齐,使模型的决策过程更加可解释。
与已有工作不同的是,本文的独特切入角度在于将"概念级别"的语义监督引入模型微调,而不仅仅是粗粒度的前景-背景分离。现有方法(如GradMask、RRR、RRDA)都将前景视为一个统一的区域,忽略了其内部的语义结构。CFT认识到:真正的鲁棒识别依赖于有结构的语义线索层次——例如识别"鹦鹉"需要关注特定的"长喙"和"翅膀",而"树枝"可以作为"鹦鹉"的上下文证据,"水面"可以支持"鸭子"的识别。通过用LLM自动发现每个类别的语义概念、用VLM进行空间定位,CFT首次实现了概念级别的空间指导,这是现有工作从未探索过的。此外,CFT采用后处理微调而非完整重新训练,并使用AttnLRP(而非不稳定的输入梯度)计算相关性图,使得方法对大规模预训练模型既实用又可靠。
核心方法
CFT的整体思路可以用一个简单的类比来理解:假设你是一个老师,发现学生在考试中总是用错误的线索答题(比如通过纸张颜色判断这是数学卷子)。传统方法是粗暴地告诉学生"看题目内容,别看纸张",而CFT则是更精细地指出"这道题的关键是看翅膀形状和喙的长度"——即引导模型关注具体的语义概念而非笼统的前景。技术路线上,CFT分三个阶段运行:首先,利用LLM(GPT-4o-mini)自动发现每个类别的语义概念(如"bird"类的"long beak"、"wings"、"feathered body"等),并通过视觉验证过滤掉不可靠的概念;其次,使用GroundedSAM在训练图像中定位这些概念的空间位置,生成二值语义掩码;最后,微调模型使其内部的相关性图(通过AttnLRP计算)与这些语义掩码对齐——鼓励模型在概念区域内分配高相关性,同时抑制对非概念区域的关注。整个过程只需要1,500张图像和50个epoch的微调。
CFT的核心创新在于用"概念级别的语义掩码"替代传统的"前景-背景二元掩码"作为微调的监督信号。这个区别是本质性的:传统方法将整个前景视为均匀区域,模型学到的是"看前景而非背景";而CFT提供了粒度更细的指导——告诉模型"对于这个类别,哪些具体的视觉概念是判别性的,它们在图像中的什么位置"。例如,对于"鸟"这个类别,CFT会生成"翅膀"、"喙"、"羽毛"等概念的独立掩码,模型需要学会在这些概念区域上分配高相关性。这种概念级别的指导不仅能区分前景的不同部分,还能包含有用的上下文信息(如"树枝"之于"鹦鹉"),因为概念集是由LLM根据类别语义自动发现的,不受前景-背景二分法的限制。实验表明,概念引导的掩码比标准分割掩码提供了更有效的监督信号。
方法步骤详情
CFT的完整流程包含三个主要步骤。第一步是概念集创建与验证。对于数据集中的每个类别c,使用LLM(GPT-4o-mini)提出一组语义属性ξ_c(如"bird"类的"long beak"、"wings")。然后使用P=30张样本来验证每个概念:通过GroundedSAM检测概念在图像中的出现率(occurrence rate,至少15%)和空间覆盖率(spatial coverage,至少20%),过滤掉不满足条件的概念,最终得到1,852个验证后的概念。第二步是语义掩码生成。对于每张训练图像I,使用GroundedSAM对所有验证后的概念生成二值分割掩码M_k(I),然后取所有概念掩码的最大值得到最终的语义指导掩码S(I)。第三步是微调优化。损失函数由两部分组成:对齐损失L_align和分类一致性损失L_cls。对齐损失又分为两部分——概念损失L_concept鼓励概念区域内的高相关性(通过最小化-log(Φ_p)),非概念损失L_non-concept抑制非概念区域的相关性(通过最小化-log(1-Φ_p))。分类一致性损失L_cls用模型自己的预测作为伪标签,确保微调不会破坏分类能力。最终损失为L = λ_align × L_align + λ_cls × L_cls,其中λ_align=0.8、λ_cls=0.2、λ_non-concept=1.2、λ_concept=0.5。所有模型使用AdamW优化器,batch size 8,训练50个epoch。
技术新颖性
CFT的技术新颖性体现在多个方面。首先,与所有已有工作不同,CFT首次将"概念级别"的语义信息引入模型微调的监督信号中。现有的saliency引导方法(GradMask、RRR、RRDA)都依赖前景-背景二元掩码,无法捕获物体内部的语义结构。CFT通过LLM+VLM的自动化流程实现了概念级别的空间指导,无需任何人工标注。其次,CFT采用后处理微调(post-hoc fine-tuning)而非完整训练/重新训练,仅需1,500张图像和50个epoch,这对于大规模预训练模型更加实用。第三,CFT使用AttnLRP(而非输入梯度)计算相关性图作为优化目标,AttnLRP满足守恒性且在ViT上更加稳定忠实,这解决了输入梯度方法在ViT上的已知问题。第四,分类一致性损失使用模型自己的预测作为目标,而非ground-truth标签,这避免了微调过程中分类能力的退化。消融实验证实,这种自监督的分类损失比标准的ground-truth交叉熵损失在鲁棒性提升上更有效。
实验结果
CFT在五个OOD基准测试和四个现代视觉模型上进行了全面评估,实验结果令人信服地证明了该方法的有效性。在真实的分布偏移数据集上,CFT取得了最显著的改进。以ViT-B模型为例,在最具挑战性的ImageNet-A(自然对抗样本)上,CFT将Top-1准确率从原始的13.26%提升至27.76%,绝对提升14.5个百分点,相比最佳基线RRDA的25.12%也有2.64个百分点的优势。在ObjectNet(控制背景和视角变化)上,CFT从30.26%提升至54.28%,绝对提升24个百分点。在分布内数据上,CFT基本维持甚至小幅改善了准确率——在ImageNet验证集上,ViT-B从80.41%提升至81.35%,CNv2从86.49%提升至87.27%。关键的是,CFT的改进不仅限于训练中见过的类别。Table 4显示,在未参与微调的类别上,CFT同样带来了显著的鲁棒性提升(如ViT-B在IN-A上,训练类提升7.31个百分点,非训练类提升7.98个百分点),这表明CFT确实改善了模型的底层推理机制,而非仅仅记忆了特定类别的线索。在相关性图与物体区域的对齐方面,CFT也取得了明显改进:在ImageNet-Segmentation数据集上,ViT-B的mIoU从62.91%提升至68.23%,mAP从78.67%提升至84.26%,像素准确率从72.23%提升至80.34%。在SI-Score合成基准测试上(测试几何变换鲁棒性),CFT的改进更为突出,表明概念聚焦的推理天然具有更好的几何不变性。消融实验证实,概念级别的指导信号显著优于标准分割掩码的指导(Table 5),以及每个损失组件都不可或缺(Table 6),AttnLRP是最佳的相关性计算方法(Table 7)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| ImageNet-A (自然对抗样本) | Top-1 Accuracy (%) | 27.93 (CNv2) | 25.89 (RRDA, CNv2) | +2.04 绝对提升 (CNv2); ViT-B上13.26→27.76 (+14.50) |
| ObjectNet (视角/背景变化) | Top-1 Accuracy (%) | 54.28 (ViT-B) | 52.54 (RRDA, ViT-B) | +1.74 绝对提升 (ViT-B); 原始30.26→54.28 (+24.02) |
| ImageNet-R (艺术风格变体) | Top-1 Accuracy (%) | 48.53 (DINOv2) | 47.93 (RRDA, DINOv2) | +0.60 绝对提升 (DINOv2) |
| ImageNet-Sketch (素描风格) | Top-1 Accuracy (%) | 46.14 (CNv2) | 45.87 (RRDA, CNv2) | +0.27 绝对提升 (CNv2) |
| SI-Score (几何变换合成基准) | Top-1 Accuracy (%) | 63.15 (ViT-B, SI-location) | 62.72 (RRDA, ViT-B) | +0.43 绝对提升; 原始55.26→63.15 (+7.89) |
| 相关性图对齐 (mIoU) | mIoU (%) | 68.23 (ViT-B) | 62.91 (原始 ViT-B) | +5.32 绝对提升 |
| 相关性图对齐 (mAP) | mAP (%) | 84.26 (ViT-B) | 78.67 (原始 ViT-B) | +5.59 绝对提升 |
局限与改进
尽管CFT整体表现优异,但论文承认了几个重要的局限性。首先,CFT依赖GroundedSAM的分割质量,当概念是抽象或非视觉的(如"aggressive behaviour"对于狮子),或物体部件非常小(占图像面积不足2%),分割质量会下降。其次,LLM生成的概念可能包含"幻觉"——语义上合理但实际不存在的属性,或基于LLM训练数据中的统计共现而非真正的类别判别特征。第三,在医学影像、卫星图像等专业领域,GPT-4o-mini的概念词汇可能不精确或不完整,需要领域特定的LLM或专家人工构建概念集。第四,虽然微调本身是轻量的(1,500张图像、50个epoch),但初始的概念创建和验证阶段需要对每个类别的30张样本运行GroundedSAM,对于有数千个类别的数据集,这个预处理步骤可能成为实际瓶颈。第五,CFT在ImageNet-R和ImageNet-Sketch等艺术/抽象风格数据集上的改进不如在ImageNet-A和ObjectNet上显著,因为这些数据集本身背景偏差较小。从独立观察来看,论文使用的是ImageNet子集(500类),每类仅3张图像,这在真实场景中是否足够还需要进一步验证;此外,损失权重参数在所有模型和数据集上使用固定值,未进行自适应调整。
独立分析的弱点
从独立分析的角度,CFT存在几个值得注意的弱点。首先,概念集的质量高度依赖于LLM(GPT-4o-mini)的能力边界。在实际应用中,如果目标领域与LLM的训练数据分布差异较大(如工业缺陷检测、农业病虫害识别),LLM可能生成的概念既不完整也不准确,导致指导掩码质量下降。改进方向是引入多轮交互验证机制——让LLM生成概念后,通过在目标数据上的实际分割效果反馈来迭代优化概念集,或允许人类专家介入审核关键概念。其次,当前方法将所有验证后的概念平等对待,但实际上不同概念对鲁棒性的贡献是不均等的——"翅膀"之于"鸟"显然比"羽毛纹理"更重要。改进方向是开发自适应概念加权机制,根据每个概念在OOD数据上的判别力动态调整其在损失函数中的权重。第三,CFT的相关性图优化目标是二值的(概念区域=1,非概念区域=0),但实际的最佳相关性分布可能是连续的——某些概念区域应该比其他区域获得更多关注。改进方向是引入软化的、基于概念重要性的连续指导信号。第四,分类一致性损失使用模型自己的预测作为目标,但如果原始模型的预测本身就是错误的(这在OOD数据上很常见),这个损失可能强化错误的推理模式。改进方向是引入置信度阈值或不确定性估计来选择性地应用这个损失。
未来方向
论文作者提出了几个有前景的未来方向。首先是自适应概念加权(Adaptive Concept Weighting),根据不同概念对特定分布偏移的判别力动态调整权重,而非平等对待所有概念。其次是层次化和组合概念(Hierarchical and Compositional Concepts),当前框架将概念视为独立实体,但真实世界的物体具有层次结构和组合语义——将复杂概念由简单原语组合构建,可能进一步增强可解释性和鲁棒性。第三是扩展到其他领域——CFT的核心原理(用语义概念引导模型推理)可以自然地扩展到目标检测、语义分割、视频理解等计算机视觉任务,甚至可能适用于结构化可解释表示有价值的非视觉领域。基于CFT的成果,还可以看到几个延伸方向:一是将CFT与数据增强结合——用概念掩码指导语义保持的数据增强策略;二是探索概念级别的迁移学习——如果两个类别共享某些概念(如"毛茸茸"),在一个类别上学到的概念关注模式可以迁移到相关类别;三是将CFT的思路应用于多模态模型(如CLIP),用文本来引导视觉编码器的概念关注。
复现评估
从复现角度来看,CFT具有较好的可复现性。论文已开源代码(https://github.com/yonisGit/cft),使用的所有模型(DINOv2、ViT-B、DeiT-III、ConvNeXt-V2)都来自公开的timm库,数据集(ImageNet-1K及其五个OOD变体)都是广泛使用的公开基准。训练设置相对简单:50个epoch、AdamW优化器、batch size 8,在NVIDIA A100 GPU上进行。然而,复现也面临几个实际挑战:首先,需要调用GPT-4o-mini API进行概念集创建,这涉及API成本和网络访问;其次,GroundedSAM的运行需要相当的GPU内存和计算时间,特别是处理每个类别30张样本的概念验证阶段;第三,论文使用的loss权重在所有模型上固定(λ_align=0.8、λ_cls=0.2、λ_non-concept=1.2、λ_concept=0.5),但学习率需要通过网格搜索选择(5e-7到5e-6),这增加了调参成本。总体而言,对于有经验的深度学习研究者,复现CFT的完整pipeline估计需要几天到一周的时间,主要瓶颈在于概念创建和验证的预处理步骤。
论文图表
该图展示了CFT方法的动机,使用ViT-B在ImageNet-A和ImageNet-R上的例子。左列为原始图像,中列为CFT微调前的相关性图,右列为微调后的相关性图。微调前,相关性图分散在背景纹理等误导性区域;微调后,相关性图紧密集中在物体的核心语义概念上,如鸟的喙和翅膀、鱼的鳍和嘴部。值得注意的是,这些数据集从未参与微调过程。
这是全文最核心的动机图,直观展示了ViT依赖虚假相关性的问题以及CFT如何解决这个问题。没有这张图,读者很难理解为什么需要概念级别的指导。