Neural Spectral Decomposition for Dataset Distillation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yang, Shaolei, Cheng, Shen, Hong, Mingbo, Fan, Haoqiang, Wei, Xing, Liu, Shuaicheng
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914928468688896
author Yang, Shaolei
Cheng, Shen
Hong, Mingbo
Fan, Haoqiang
Wei, Xing
Liu, Shuaicheng
author_facet Yang, Shaolei
Cheng, Shen
Hong, Mingbo
Fan, Haoqiang
Wei, Xing
Liu, Shuaicheng
contents In this paper, we propose Neural Spectrum Decomposition, a generic decomposition framework for dataset distillation. Unlike previous methods, we consider the entire dataset as a high-dimensional observation that is low-rank across all dimensions. We aim to discover the low-rank representation of the entire dataset and perform distillation efficiently. Toward this end, we learn a set of spectrum tensors and transformation matrices, which, through simple matrix multiplication, reconstruct the data distribution. Specifically, a spectrum tensor can be mapped back to the image space by a transformation matrix, and efficient information sharing during the distillation learning process is achieved through pairwise combinations of different spectrum vectors and transformation matrices. Furthermore, we integrate a trajectory matching optimization method guided by a real distribution. Our experimental results demonstrate that our approach achieves state-of-the-art performance on benchmarks, including CIFAR10, CIFAR100, Tiny Imagenet, and ImageNet Subset. Our code are available at \url{https://github.com/slyang2021/NSD}.
format Preprint
id arxiv_https___arxiv_org_abs_2408_16236
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Neural Spectral Decomposition for Dataset Distillation
Yang, Shaolei
Cheng, Shen
Hong, Mingbo
Fan, Haoqiang
Wei, Xing
Liu, Shuaicheng
Computer Vision and Pattern Recognition
In this paper, we propose Neural Spectrum Decomposition, a generic decomposition framework for dataset distillation. Unlike previous methods, we consider the entire dataset as a high-dimensional observation that is low-rank across all dimensions. We aim to discover the low-rank representation of the entire dataset and perform distillation efficiently. Toward this end, we learn a set of spectrum tensors and transformation matrices, which, through simple matrix multiplication, reconstruct the data distribution. Specifically, a spectrum tensor can be mapped back to the image space by a transformation matrix, and efficient information sharing during the distillation learning process is achieved through pairwise combinations of different spectrum vectors and transformation matrices. Furthermore, we integrate a trajectory matching optimization method guided by a real distribution. Our experimental results demonstrate that our approach achieves state-of-the-art performance on benchmarks, including CIFAR10, CIFAR100, Tiny Imagenet, and ImageNet Subset. Our code are available at \url{https://github.com/slyang2021/NSD}.
title Neural Spectral Decomposition for Dataset Distillation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.16236