提出无需训练的几何引导扩散数据蒸馏方法,提升小规模数据集的代表性与多样性。
ManifoldGD: Training-Free Hierarchical Manifold Guidance for Diffusion-Based Dataset Distillation
- 通过分层聚类生成多尺度原型点,构建每步去噪的局部流形结构。
- 在去噪过程中投影对齐向量至流形切空间,保持语义一致性与图像质量。
- 无需重训练,显著优于现有训练自由与需训练的方法,适合高效模型训练场景。
近年来,大规模数据集阻碍了高效模型训练,且包含冗余信息。数据蒸馏旨在合成紧凑数据集,在大幅降低存储与计算成本的同时保留大规模训练集的知识。近期扩散模型的发展使基于预训练生成先验的免训练蒸馏成为可能;然而现有引导策略仍有限。当前基于得分的方法或进行无引导去噪,或依赖简单的实例原型中心(IPC)模式引导,常过于粗略且次优。本文提出曼德拉引导蒸馏(ManifoldGD),一种免训练的扩散基框架,可在每一步去噪中整合流形一致性引导。方法通过分层分裂聚类VAE隐空间特征生成多尺度的实例原型点(IPCs)共核,捕捉粗粒度语义模式与细粒度类内变化。利用提取的IPC中心邻域构建每个去噪步的隐空间流形。在每一步去噪中,将模态对齐向量投影至估计的隐空间流形局部切空间,从而约束生成轨迹保持流形忠实性,同时维持语义一致性。该设计在不需任何模型重训练的情况下,显著提升表示能力、多样性和图像保真度。实验表明,其在FID、真实与合成数据嵌入间的l2距离及分类准确率上均持续优于现有免训练与需训练基线,确立了首个几何感知的免训练数据蒸馏框架。
原文摘要 · Abstract (English)
In recent times, large datasets hinder efficient model training while also containing redundant concepts. Dataset distillation aims to synthesize compact datasets that preserve the knowledge of large-scale training sets while drastically reducing storage and computation. Recent advances in diffusion models have enabled training-free distillation by leveraging pre-trained generative priors; however, existing guidance strategies remain limited. Current score-based methods either perform unguided denoising or rely on simple mode-based guidance toward instance prototype centroids (IPC centroids), which often are rudimentary and suboptimal. We propose Manifold-Guided Distillation (ManifoldGD), a training-free diffusion-based framework that integrates manifold consistent guidance at every denoising timestep. Our method employs IPCs computed via a hierarchical, divisive clustering of VAE latent features, yielding a multi-scale coreset of IPCs that captures both coarse semantic modes and fine intra-class variability. Using a local neighborhood of the extracted IPC centroids, we create the latent manifold for each diffusion denoising timestep. At each denoising step, we project the mode-alignment vector onto the local tangent space of the estimated latent manifold, thus constraining the generation trajectory to remain manifold-faithful while preserving semantic consistency. This formulation improves representativeness, diversity, and image fidelity without requiring any model retraining. Empirical results demonstrate consistent gains over existing training-free and training-based baselines in terms of FID, l2 distance among real and synthetic dataset embeddings, and classification accuracy, establishing ManifoldGD as the first geometry-aware training-free data distillation framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。