Data-Distill-Net: A Data Distillation Approach Tailored for Reply-based Continual Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liao, Wenyang, Wang, Quanziang, Wu, Yichen, Wang, Renzhen, Meng, Deyu
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909625849217024
author Liao, Wenyang
Wang, Quanziang
Wu, Yichen
Wang, Renzhen
Meng, Deyu
author_facet Liao, Wenyang
Wang, Quanziang
Wu, Yichen
Wang, Renzhen
Meng, Deyu
contents Replay-based continual learning (CL) methods assume that models trained on a small subset can also effectively minimize the empirical risk of the complete dataset. These methods maintain a memory buffer that stores a sampled subset of data from previous tasks to consolidate past knowledge. However, this assumption is not guaranteed in practice due to the limited capacity of the memory buffer and the heuristic criteria used for buffer data selection. To address this issue, we propose a new dataset distillation framework tailored for CL, which maintains a learnable memory buffer to distill the global information from the current task data and accumulated knowledge preserved in the previous memory buffer. Moreover, to avoid the computational overhead and overfitting risks associated with parameterizing the entire buffer during distillation, we introduce a lightweight distillation module that can achieve global information distillation solely by generating learnable soft labels for the memory buffer data. Extensive experiments show that, our method can achieve competitive results and effectively mitigates forgetting across various datasets. The source code will be publicly available.
format Preprint
id arxiv_https___arxiv_org_abs_2505_20135
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Data-Distill-Net: A Data Distillation Approach Tailored for Reply-based Continual Learning
Liao, Wenyang
Wang, Quanziang
Wu, Yichen
Wang, Renzhen
Meng, Deyu
Machine Learning
Replay-based continual learning (CL) methods assume that models trained on a small subset can also effectively minimize the empirical risk of the complete dataset. These methods maintain a memory buffer that stores a sampled subset of data from previous tasks to consolidate past knowledge. However, this assumption is not guaranteed in practice due to the limited capacity of the memory buffer and the heuristic criteria used for buffer data selection. To address this issue, we propose a new dataset distillation framework tailored for CL, which maintains a learnable memory buffer to distill the global information from the current task data and accumulated knowledge preserved in the previous memory buffer. Moreover, to avoid the computational overhead and overfitting risks associated with parameterizing the entire buffer during distillation, we introduce a lightweight distillation module that can achieve global information distillation solely by generating learnable soft labels for the memory buffer data. Extensive experiments show that, our method can achieve competitive results and effectively mitigates forgetting across various datasets. The source code will be publicly available.
title Data-Distill-Net: A Data Distillation Approach Tailored for Reply-based Continual Learning
topic Machine Learning
url https://arxiv.org/abs/2505.20135