Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Activation Function Design Sustains Plasticity in Continual Learning

About

In independent, identically distributed (i.i.d.) training regimes, activation functions have been benchmarked extensively, and their differences often shrink once model size and optimization are tuned. In continual learning, however, the picture is different: beyond catastrophic forgetting, models can progressively lose the ability to adapt (referred to as loss of plasticity) and the role of the non-linearity in this failure mode remains underexplored. We show that activation choice is a primary, architecture-agnostic lever for mitigating plasticity loss. Building on a property-level analysis of negative-branch shape and saturation behavior, we introduce two drop-in nonlinearities (Smooth-Leaky and Randomized Smooth-Leaky) and evaluate them in two complementary settings: (i) supervised class-incremental benchmarks and (ii) reinforcement learning with non-stationary MuJoCo environments designed to induce controlled distribution and dynamics shifts. We also provide a simple stress protocol and diagnostics that link the shape of the activation to the adaptation under change. The takeaway is straightforward: thoughtful activation design offers a lightweight, domain-general way to sustain plasticity in continual learning without extra capacity or task-specific tuning.

Lute Lillo, Nick Cheney• 2025

Related benchmarks

TaskDatasetResultRank
Continual LearningCIFAR100 Split
Average Per-Task Accuracy25
117
Continual Supervised LearningCIFAR Random Label
Total Average Online Task Accuracy98.42
49
Continual Supervised LearningContinual ImageNet
Total Average Online Task Accuracy86.23
49
Continual Supervised LearningCIFAR 5+1
Total Average Online Task Accuracy67.9
49
Continual LearningPermuted MNIST
Average Accuracy81.3
32
Continual LearningMNIST Random-Label
Average Accuracy85.9
32
Continual LearningMNIST Shuffled Labels
Accuracy (ACC)93.33
22
Plasticity MeasurementLocomotion Tasks Aggregate (Ant, HalfCheetah, Humanoid) (train)
Plasticity Score (IQM)38.53
17
Continual Reinforcement LearningMuJoCo Locomotion Sequence HalfCheetah, Hopper, Walker2d, Ant v4
IQM38.75
5
Showing 9 of 9 rows

Other info

Follow for update