超越软标签:通过正交梯度匹配进行数据集蒸馏
Beyond Soft Label: Dataset Distillation via Orthogonal Gradient Matching
OGM 通过正交化梯度并匹配奇异向量方向,为数据集蒸馏提供了一种无需反向传播的梯度匹配新思路,可复用于其他需要高效匹配梯度方向的视觉任务。
原文依据 Orthogonal Gradient Matching (OGM), which explicitly aligns the intrinsic direction of gradients, i.e., singular vectors.
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论将大规模高分辨率 ImageNet-1K 数据集进行压缩仍然是数据集蒸馏(DD)的一个挑战。现有方法通常匹配真实数据集与合成数据集之间的批归一化(BN)统计量,即均值和方差。尽管在软标签下有效,但在硬标签下性能显著下降。本文从理论上指出,BN 匹配主要对齐了真实梯度与合成梯度的尺度,而忽略了它们的方向。然而,实验证据表明,梯度方向而非尺度对模型训练至关重要,这阐明了先前方法的局限性。基于这一见解,我们引入了正交梯度匹配(OGM),它显式地对齐梯度的内在方向,即奇异向量。具体而言,OGM 首先通过将所有奇异值设为一来对真实梯度和合成梯度进行正交化,消除它们的尺度,然后最小化这些正交梯度之间的距离,使它们的奇异向量重合。为进一步减少计算,OGM 采用了一种最小二乘损失,其梯度可以在前向传播中获得,避免了反向传播。在 ImageNet-1K 上的大量实验验证了 OGM 的有效性。在每类仅十张图像(IPC = 10)的情况下,OGM 在软标签下达到 47.0% 的准确率,在硬标签下达到 16.7%,优于基于训练的 DD 方法和 RDED。
英文原文摘要 1437 字符
Condensing the large-scale, high-resolution ImageNet-1K dataset remains a challenge for dataset distillation (DD). Existing methods typically match batch normalization (BN) statistics, i.e., mean and variance, between real and synthetic datasets. Although effective with soft labels, their performance degrades substantially under hard labels. In this paper, we theoretically identify that BN matching mainly aligns the scales of real and synthetic gradients but overlooks their directions. However, experimental evidence demonstrates that gradient direction, rather than scale, is pivotal to model training, clarifying the limitations of prior methods. Building on this insight, we introduce Orthogonal Gradient Matching (OGM), which explicitly aligns the intrinsic direction of gradients, i.e., singular vectors. Specifically, OGM first orthogonalizes real and synthetic gradients by setting all singular values to one, eliminating their scales, and then minimizes the distance between these orthogonal gradients so that their singular vectors coincide. To further reduce computation, OGM employs a least-squares loss whose gradients can be obtained in the forward pass, avoiding back-propagation. Extensive experiments on ImageNet-1K validate the effectiveness of OGM. With only ten images per class (IPC = 10), OGM achieves 47.0% accuracy with soft labels and 16.7% with hard labels, outperforming training-based DD methods and RDED.