arXiv:2605.07194cs.CVcs.AI2026-05被引 1

用闭式解压缩图像数据集,加速预训练模型微调。

Closed-Form Linear-Probe Dataset Distillation for Pre-trained Vision Models

论文配图:Closed-Form Linear-Probe Dataset Distillation for Pre-trained Vision Models
图 1 · 摘自论文原文
  • 直接利用预训练特征计算最优线性探针,无需迭代或无限宽近似。
  • 在ImageNet-100上性能接近现有最佳方法,计算成本降低超90%。
  • 适合需要高效微调的科研与工业场景,尤其关注显存与速度者。

数据蒸馏将大规模训练集压缩为少量合成样本,保留下游训练效用。现有方法多针对从零训练网络,而现代视觉迁移学习常使用冻结的预训练编码器配合轻量线性探针。当前蒸馏方法要么通过轨迹梯度匹配模拟迭代更新,要么依赖基于神经正切核(NTK)近似的闭式公式,均未充分利用冻结特征线性探针可直接由预训练特征解析求解的特性。本文提出闭式线性探针数据蒸馏(CLP-DD),采用双层优化框架:内层用样本空间核岭回归求解合成集诱导的线性探针;外层通过温度缩放Softmax交叉熵评估该分类器在真实特征上的表现,以类锚点形式更新合成图像。实验表明,外层目标选择至关重要:标准MSE损失性能远低于轨迹法,而判别性损失可显著缩小差距。在ImageNet-100上,四款预训练骨干网络下,CLP-DD优于无DSA的LGM,逼近含DSA的LGM,计算成本仅为几分之一。在ImageNet-1K上,三款骨干网络性能匹配或超越含DSA的LGM,推理速度提升约14倍,显存占用不足1/8。

原文摘要 · Abstract (English)

Dataset distillation compresses a large training set into a small synthetic set that preserves downstream training utility. While most existing methods target training networks from scratch, modern visual transfer learning often uses frozen pre-trained encoders followed by lightweight linear probing. Existing distillation methods for this setting either unroll iterative linear-probe updates with trajectory-based gradient matching, or rely on closed-form formulations originally designed for from-scratch training with neural-tangent-kernel (NTK) approximations. Neither route exploits the fact that frozen-feature linear probing admits a closed-form solution determined directly by the pre-trained features themselves, with no infinite-width approximation and no inner-loop trajectory. We propose Closed-Form Linear-Probe Dataset Distillation (CLP-DD), a bilevel formulation that computes the linear probe induced by the synthetic set with a sample-space kernel ridge solver. The synthetic images are then updated by evaluating this induced classifier on real features through a temperature-scaled softmax cross-entropy, where the classifier columns act as learned class anchors in feature space. We further show that the choice of outer objective is decisive: pairing the closed-form inner solver with a standard MSE outer loss substantially underperforms trajectory-based methods, while the discriminative outer loss closes most of the gap. On ImageNet-100 with four pre-trained backbones, CLP-DD substantially improves over LGM without DSA and approaches LGM with DSA at a fraction of the computational cost. On ImageNet-1K, CLP-DD matches or surpasses LGM with DSA on three of four backbones while running roughly $14\times$ faster and using less than one-eighth of the GPU memory.

数据蒸馏线性探针预训练模型高效微调

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