Budgeted Online Continual Learning by Adaptive Layer Freezing and Frequency-based Sampling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Seo, Minhyuk, Koh, Hyunseo, Choi, Jonghyun
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929760874004480
author Seo, Minhyuk
Koh, Hyunseo
Choi, Jonghyun
author_facet Seo, Minhyuk
Koh, Hyunseo
Choi, Jonghyun
contents The majority of online continual learning (CL) advocates single-epoch training and imposes restrictions on the size of replay memory. However, single-epoch training would incur a different amount of computations per CL algorithm, and the additional storage cost to store logit or model in addition to replay memory is largely ignored in calculating the storage budget. Arguing different computational and storage budgets hinder fair comparison among CL algorithms in practice, we propose to use floating point operations (FLOPs) and total memory size in Byte as a metric for computational and memory budgets, respectively, to compare and develop CL algorithms in the same 'total resource budget.' To improve a CL method in a limited total budget, we propose adaptive layer freezing that does not update the layers for less informative batches to reduce computational costs with a negligible loss of accuracy. In addition, we propose a memory retrieval method that allows the model to learn the same amount of knowledge as using random retrieval in fewer iterations. Empirical validations on the CIFAR-10/100, CLEAR-10/100, and ImageNet-1K datasets demonstrate that the proposed approach outperforms the state-of-the-art methods within the same total budget
format Preprint
id arxiv_https___arxiv_org_abs_2410_15143
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Budgeted Online Continual Learning by Adaptive Layer Freezing and Frequency-based Sampling
Seo, Minhyuk
Koh, Hyunseo
Choi, Jonghyun
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
The majority of online continual learning (CL) advocates single-epoch training and imposes restrictions on the size of replay memory. However, single-epoch training would incur a different amount of computations per CL algorithm, and the additional storage cost to store logit or model in addition to replay memory is largely ignored in calculating the storage budget. Arguing different computational and storage budgets hinder fair comparison among CL algorithms in practice, we propose to use floating point operations (FLOPs) and total memory size in Byte as a metric for computational and memory budgets, respectively, to compare and develop CL algorithms in the same 'total resource budget.' To improve a CL method in a limited total budget, we propose adaptive layer freezing that does not update the layers for less informative batches to reduce computational costs with a negligible loss of accuracy. In addition, we propose a memory retrieval method that allows the model to learn the same amount of knowledge as using random retrieval in fewer iterations. Empirical validations on the CIFAR-10/100, CLEAR-10/100, and ImageNet-1K datasets demonstrate that the proposed approach outperforms the state-of-the-art methods within the same total budget
title Budgeted Online Continual Learning by Adaptive Layer Freezing and Frequency-based Sampling
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.15143