arXiv:2511.16674cs.CVcs.AI2025-11NeurIPS被引 4

为预训练自监督模型设计高效数据蒸馏方法,用少量合成图像提升线性探测器性能。

Dataset Distillation for Pre-Trained Self-Supervised Vision Models

  • 通过匹配真实数据梯度优化合成图像,使线性分类器在预训练特征上表现更优。
  • 合成数据在多个模型间通用,用DINO蒸馏数据可训练出媲美CLIP的线性探针。
  • 适用于细粒度分类与模型可解释性分析,能预测模型嵌入空间相似性及对抗敏感性。

数据蒸馏旨在生成一组少量合成图像,使模型在这些图像上训练的效果接近在大量真实图像上训练的表现。现有方法主要针对从零开始训练的随机初始化模型,而当前最先进的视觉模型多基于大规模预训练的自监督模型。本文研究如何为这类预训练模型蒸馏数据,以最优地训练线性探测器。我们提出一种新方法——线性梯度匹配(Linear Gradient Matching),通过优化合成图像,使其经过预训练特征提取器后,在线性分类器上产生的梯度与真实数据相似。该方法生成的合成数据超越所有真实图像基线,并惊人地具备跨模型泛化能力:例如,使用DINO主干蒸馏的数据可训练出性能媲美的线性CLIP探测器。此外,我们的蒸馏数据在细粒度分类中表现出色,且成为模型可解释性的有力工具,可预测不同模型嵌入空间的相似性(基于柏拉图表征假设),或判断模型是否对对抗数据中的伪相关性敏感。

原文摘要 · Abstract (English)

The task of dataset distillation aims to find a small set of synthetic images such that training a model on them reproduces the performance of the same model trained on a much larger dataset of real samples. Existing distillation methods focus on synthesizing datasets that enable training randomly initialized models. In contrast, state-of-the-art vision approaches are increasingly building on large, pre-trained self-supervised models rather than training from scratch. In this paper, we investigate the problem of distilling datasets that enable us to optimally train linear probes on top of such large, pre-trained vision models. We introduce a method of dataset distillation for this task called Linear Gradient Matching that optimizes the synthetic images such that, when passed through a pre-trained feature extractor, they induce gradients in the linear classifier similar to those produced by the real data. Our method yields synthetic data that outperform all real-image baselines and, remarkably, generalize across pre-trained vision models, enabling us, for instance, to train a linear CLIP probe that performs competitively using a dataset distilled via a DINO backbone. Further, we show that our distilled datasets are exceptionally effective for fine-grained classification and provide a valuable tool for model interpretability, predicting, among other things, how similar two models' embedding spaces are under the platonic representation hypothesis or whether a model is sensitive to spurious correlations in adversarial datasets.

数据蒸馏自监督学习线性探测模型可解释性

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