arXiv:2507.04619cs.LGcs.AI2025-07被引 10

用信息论指导扩散采样,让小样本数据集更保真。

Information-Guided Diffusion Sampling for Dataset Distillation

  • 从信息论出发,同时保留类别特征和类内差异。
  • 在低样本每类(IPC)下,准确率提升显著。
  • 适合追求高保真小样本数据集的研究者。

数据集蒸馏旨在生成紧凑数据集以保留关键信息并维持模型性能。扩散模型(DMs)虽有潜力,但在低图像每类(IPC)设置下生成样本多样性不足。本文从信息论视角识别两类必要信息:原型信息 $\mathrm{I}(X;Y)$(捕捉标签相关特征)和上下文信息 $\mathrm{H}(X | Y)$(保留类内变异性)。观察到上下文信息随 IPC 增长,提出在扩散模型采样中最大化 $\mathrm{I}(X;Y) + β\mathrm{H}(X | Y)$,其中 $β$ 与 IPC 相关。由于直接计算不可行,采用数据驱动的变分估计方法紧下界这两项。所提方法信息引导扩散采样(IGDS)可无缝集成于扩散模型,在所有 IPC 设置下均优于现有方法。在 Tiny ImageNet 与 ImageNet 子集上的实验表明,尤其在低 IPC 情况下显著领先。代码将在接受后发布。

原文摘要 · Abstract (English)

Dataset distillation aims to create a compact dataset that retains essential information while maintaining model performance. Diffusion models (DMs) have shown promise for this task but struggle in low images-per-class (IPC) settings, where generated samples lack diversity. In this paper, we address this issue from an information-theoretic perspective by identifying two key types of information that a distilled dataset must preserve: ($i$) prototype information $\mathrm{I}(X;Y)$, which captures label-relevant features; and ($ii$) contextual information $\mathrm{H}(X | Y)$, which preserves intra-class variability. Here, $(X,Y)$ represents the pair of random variables corresponding to the input data and its ground truth label, respectively. Observing that the required contextual information scales with IPC, we propose maximizing $\mathrm{I}(X;Y) + β\mathrm{H}(X | Y)$ during the DM sampling process, where $β$ is IPC-dependent. Since directly computing $\mathrm{I}(X;Y)$ and $\mathrm{H}(X | Y)$ is intractable, we develop variational estimations to tightly lower-bound these quantities via a data-driven approach. Our approach, information-guided diffusion sampling (IGDS), seamlessly integrates with diffusion models and improves dataset distillation across all IPC settings. Experiments on Tiny ImageNet and ImageNet subsets show that IGDS significantly outperforms existing methods, particularly in low-IPC regimes. The code will be released upon acceptance.

数据集蒸馏扩散模型信息论

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