FETCH: A Memory-Efficient Replay Approach for Continual Learning in Image Classification

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Weißflog, Markus, Protzel, Peter, Neubert, Peer
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910531071246336
author Weißflog, Markus
Protzel, Peter
Neubert, Peer
author_facet Weißflog, Markus
Protzel, Peter
Neubert, Peer
contents Class-incremental continual learning is an important area of research, as static deep learning methods fail to adapt to changing tasks and data distributions. In previous works, promising results were achieved using replay and compressed replay techniques. In the field of regular replay, GDumb achieved outstanding results but requires a large amount of memory. This problem can be addressed by compressed replay techniques. The goal of this work is to evaluate compressed replay in the pipeline of GDumb. We propose FETCH, a two-stage compression approach. First, the samples from the continual datastream are encoded by the early layers of a pre-trained neural network. Second, the samples are compressed before being stored in the episodic memory. Following GDumb, the remaining classification head is trained from scratch using only the decompressed samples from the reply memory. We evaluate FETCH in different scenarios and show that this approach can increase accuracy on CIFAR10 and CIFAR100. In our experiments, simple compression methods (e.g., quantization of tensors) outperform deep autoencoders. In the future, FETCH could serve as a baseline for benchmarking compressed replay learning in constrained memory scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2407_12375
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FETCH: A Memory-Efficient Replay Approach for Continual Learning in Image Classification
Weißflog, Markus
Protzel, Peter
Neubert, Peer
Computer Vision and Pattern Recognition
Machine Learning
Class-incremental continual learning is an important area of research, as static deep learning methods fail to adapt to changing tasks and data distributions. In previous works, promising results were achieved using replay and compressed replay techniques. In the field of regular replay, GDumb achieved outstanding results but requires a large amount of memory. This problem can be addressed by compressed replay techniques. The goal of this work is to evaluate compressed replay in the pipeline of GDumb. We propose FETCH, a two-stage compression approach. First, the samples from the continual datastream are encoded by the early layers of a pre-trained neural network. Second, the samples are compressed before being stored in the episodic memory. Following GDumb, the remaining classification head is trained from scratch using only the decompressed samples from the reply memory. We evaluate FETCH in different scenarios and show that this approach can increase accuracy on CIFAR10 and CIFAR100. In our experiments, simple compression methods (e.g., quantization of tensors) outperform deep autoencoders. In the future, FETCH could serve as a baseline for benchmarking compressed replay learning in constrained memory scenarios.
title FETCH: A Memory-Efficient Replay Approach for Continual Learning in Image Classification
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2407.12375