Search papers, labs, and topics across Lattice.
This paper introduces C-Flat Turbo, an optimized variant of the C-Flat continual learning algorithm that reduces computational overhead by identifying and skipping redundant gradient computations in the flatness-promoting steps. The method exploits the observation that flatness gradients contain direction-invariant components and stabilize across tasks, enabling a linear scheduling strategy for adaptive step allocation. Experiments demonstrate that C-Flat Turbo achieves comparable or improved accuracy while being 1.0x to 1.25x faster than the original C-Flat across various continual learning benchmarks.
Continual learning just got a turbo boost: C-Flat Turbo cuts training time by up to 25% without sacrificing accuracy, thanks to a clever gradient-skipping trick.
Continual Learning (CL) aims to train neural networks on a dynamic stream of tasks without forgetting previously learned knowledge. Among optimization-based approaches, C-Flat has emerged as a promising solution due to its plug-and-play nature and its ability to encourage uniformly low-loss regions for both new and old tasks. However, C-Flat requires three additional gradient computations per iteration, imposing substantial overhead on the optimization process. In this work, we propose C-Flat Turbo, a faster yet stronger optimizer that significantly reduces the training cost. We show that the gradients associated with first-order flatness contain direction-invariant components relative to the proxy-model gradients, enabling us to skip redundant gradient computations in the perturbed ascent steps. Moreover, we observe that these flatness-promoting gradients progressively stabilize across tasks, which motivates a linear scheduling strategy with an adaptive trigger to allocate larger turbo steps for later tasks. Experiments show that C-Flat Turbo is 1.0$\times$ to 1.25$\times$ faster than C-Flat across a wide range of CL methods, while achieving comparable or even improved accuracy.