arXiv:2608.10723cs.CV2026-08

让ViT在数据少时表现更好,保留卷积网络的空间结构优势

Grid-Preserving Knowledge Distillation: Transferring Convolutional Inductive Bias to Vision Transformers under Data Scarcity

论文配图:Grid-Preserving Knowledge Distillation: Transferring Convolutional Inductive Bias to Vision Transformers under Data Scarcity
图 1 · 摘自论文原文
  • 用网格保持机制将卷积网络的空间先验知识传给ViT
  • 在小数据下提升多个ViT模型性能,数据越少效果越明显
  • 训练时加模块不改部署模型,推理无额外开销

视觉变换器虽具强大全局建模能力,但在数据稀缺场景下表现不佳。通过从卷积神经网络教师中蒸馏卷积归纳偏置可有效改善,但通用特征蒸馏在此情境下效果有限。在CNN到CNN的蒸馏中,池化、展平和对数空间投影会破坏编码局部性和平移等变性的空间网格。与卷积学生不同,ViT无法自主重建该结构。本文提出iBKD框架,全程保留空间网格。其核心模块诱导偏置注意力模块,使用学习权重将学生各层特征聚合到教师网格,并通过通道和可变形空间注意力增强结构线索,再通过直接作用于空间网格的卷积交叉注意力注入。该模块仅用于训练,部署模型仍为未修改的ViT,无推理开销。在七个Transformer骨干网络和六个数据稀缺基准上,iBKD持续优于局部性引导方法和通用知识蒸馏基线,且数据越少优势越显著。

原文摘要 · Abstract (English)

Vision Transformers demonstrate remarkable global modeling capacity but often underperform in data-scarce regimes. Distilling convolutional inductive biases from a CNN teacher provides an effective remedy while leaving the deployed model unchanged. However, general-purpose feature distillation transfers little in this setting. In CNN-to-CNN distillation, pooling, flattening, and logit-space projections remove the spatial grid that encodes locality and translation equivariance. Unlike a convolutional student, a ViT cannot readily reconstruct this structure on its own. In this paper, we propose iBKD, a distillation framework that preserves the spatial grid throughout the entire transfer process. Its core module, the Inductive Bias Attention Module, aggregates features from all student layers onto the teacher's grid using learned weights. It then enhances structural cues through channel and deformable spatial attention and injects them via convolutional cross-attention operating directly between spatial grids rather than token sets. The module is used only during training, leaving the deployed model as an unmodified ViT with no inference overhead. Across seven Transformer backbones and six data-scarce benchmarks, iBKD consistently outperforms both locality-guidance methods and general knowledge distillation baselines, with its advantage increasing as the amount of training data decreases.

知识蒸馏视觉Transformer小样本学习结构保持

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