用扩散模型提升蒸馏数据语义表达,让小数据集在不同模型上表现更稳。
DIVER:Diving Deeper into Distilled Data via Expressive Semantic Recovery

- 分两阶段蒸馏:先提取通用语义,再融合指导重建
- 在ImageNet上仅用4GB显存,性能超越传统方法且跨架构泛化强
- 适合关注隐私保护与高效训练的科研人员和工程师
数据蒸馏旨在生成紧凑的代理数据集,以保护隐私并实现高效学习。然而,现有方法多采用单阶段蒸馏,易受预设架构限制,导致语义表达不足,跨架构性能下降。为此,本文提出双阶段蒸馏框架DIVER,利用预训练扩散模型深入挖掘蒸馏数据中的丰富语义。通过语义继承,将抽象蒸馏图像的高层语义注入潜在空间,过滤掉架构特异性噪声;语义引导则在逆过程期间增强原始语义保留;语义融合仅在逆过程具体阶段提供引导,避免模糊与伪影。大量实验表明,DIVER显著提升经典蒸馏技术效果,在ImageNet(256×256)上处理时间接近原始DiT,仅需4 GB GPU内存,跨架构泛化能力大幅提升。
原文摘要 · Abstract (English)
Dataset distillation aims to synthesize a compact proxy dataset that is unreadable or non-raw from the original dataset for privacy protection and highly efficient learning. However, previous approaches typically adopt a single-stage distillation paradigm, which suffers from learning specific patterns that overfit on a prior architecture, consequently suppressing the expression of semantics and leading to performance degradation across heterogeneous architectures. To address this issue, we propose a novel dual-stage distillation framework called ${\textbf{DIVER}}$, which leverages the pre-trained diffusion model to dive deeper into $\textbf{DI}$stilled data $\textbf{V}$ia $\textbf{E}$xpressive semantic $\textbf{R}$ecovery, an entire process of semantic inheritance, guidance, and fusion. Semantic inheritance distills high-level semantics of abstract distilled images into the latent space to filter out architecture-specific ``noise" and retain the intrinsic semantics. Furthermore, semantic guidance improves the preservation of the original semantics by directing the reverse procedure. Finally, semantic fusion is designed to provide semantic guidance only during the concrete phase of the reverse process, preventing semantic ambiguity and artifacts while maintaining the guidance information. Extensive experiments validate the effectiveness and efficiency of DIVER in improving classical distillation techniques and significantly improving cross-architecture generalization, requiring processing time comparable to raw DiT on ImageNet (256$\times$256) with only 4 GB of GPU memory usage.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。