Online continual learning aims to continuously train neural networks on a non-stationary data stream in a single pass. Rehearsal-based methods alleviate catastrophic forgetting by replaying stored samples, but the imbalance between current and previous data often leads to predictions biased toward current classes. Many approaches have been proposed to overcome the forgetting problem by correcting the bias; however, they still need to be improved in an online fashion. By decomposing the dot-product logits into an angle factor and a norm factor, our analysis reveals that the angle factor efficiently captures novel knowledge but is particularly susceptible to prediction bias, whereas the norm factor helps preserve historical knowledge. Based on these observations, we propose Dual-Logit Experience Replay (DLER), which exploits dual-logits for both model optimization and memory management. Specifically, DLER learns current samples using cosine logits and incorporates additional dot-product logits when replaying previous samples to mitigate prediction bias. To strengthen dual-logit learning under limited replay memory, an auxiliary supervised contrastive objective promotes angular compactness within each class, extending corrections learned from limited replay samples to other samples of the same class. Meanwhile, we introduce a memory update strategy that preferentially retains samples for which dot-product prediction assigns a higher probability to the correct class than cosine prediction, aligning sample selection with dual-logit training. Extensive experiments on three datasets demonstrate the effectiveness of DLER, particularly under limited memory budgets.
Figures & tables
Fig. 1: Illustration of our work. (a) The dot-product logits can be decomposed into a norm factor and an angle factor; (b) Existing studies, which learn all samples by calculating logits in the same way; (c) Different from existing studies, the model in our work learns current samples only by the angle factor, while replays previous samples by both the angle factor and the norm factor. (d) By leveraging the norm factor, the model can correctly classify samples.
Fig. 2: Average accuracy rate on observed learning stages on three datasets when the memory buffer size is 1000.
Components
Buffer size
Method
Dual-logits
α
SC
BM
Advantage
400
1000
2000
4000
ER
5.2 ± 0.6
7.6 ± 0.5
10.1 ± 0.9
12.7 ± 1.1
Dual-logit training (Fixed)
✓
8.0 ± 0.6
10.9 ± 0.7
14.2 ± 0.8
16.1 ± 0.8
Dual-logit training
✓
✓
9.7 ± 0.4
13.4 ± 0.5
16.2 ± 0.7
18.9 ± 0.8
Dual-logit training + SC
✓
✓
✓
10.0 ± 0.5
13.9 ± 0.3
17.2 ± 0.5
19.6 ± 0.7
Dual-logit training + SC + CB
✓
✓
✓
✓
10.2 ± 0.5
14.5 ± 0.3
17.4 ± 0.8
19.7 ± 0.5
TABLE II: Ablation study on Split TinyImageNet with different memory buffer sizes.
Fig. 3: Average accuracy rate of previous classes, current classes, and their average on Split MiniImageNet with a 1000 memory buffer size.
Fig. 4: Improvement of existing methods with dual-logits training on three datasets while the buffer size is 1000.
Fig. 5: The performance of dual-logits training on Split CIFAR100 (buffer size=1000) with different values of α .
Method Setting
All Classes
Current Classes
Previous Classes
ER
15.6±2.6
50.5
11.7
ER-ACE
18.9±0.9
17.3
19.1
OBC
18.3±0.7
17.5
18.4
Dual-logit training (feature)
18.6±1.0
30.6
17.2
Dual-logit training (predictor)
19.6±1.5
34.7
18.0
Dual-logit training (both)
20.1±1.2
31.8
18.8
TABLE III: Final accuracy (%) on Split CIFAR100 with a buffer of 1,000 samples after adding a linear layer to the feature extractor.
Fig. 6: Conditional coverage and previous-class accuracy of dual-logits training and its enhancement with SC on Split TinyImageNet.
Metric
High-score
Low-score
Spatial JS divergence
0.2256
0.0463
Feature norm
15.7856
12.8475
Ground-truth cosine probability
0.1173
0.0260
Gradient norm (all)
7.3114
52.7980
Gradient norm (backbone)
7.0487
51.4370
Gradient conflict rate
41.5%
44.6%
TABLE IV: Comparison of the highest- and lowest-scoring training samples within the 96 selected classes on Split TinyImageNet.
Logit configuration
Accuracy (%)
Weight norm
Weight statistics
Bias
Configuration
Learning
Replaying
Testing
Aall
Ap
Ac
∥w∥p
∥w∥c
μWp±σWp
μWc±σWc
μbp
μbc
1 (ER)
Dot
Dot
Dot
27.3
17.7
36.9
2.08
1.86
0.0044±0.0274
−0.0038±0.0221
−0.22
0.22
2
Dot
Dot
Cos
24.1
12.7
35.6
2.08
1.86
0.0044±0.0274
−0.0038±0.0221
−0.22
0.22
3 (LUCIR)
Cos
Cos
Cos
26.9
14.9
38.9
2.00
1.64
0.0398±0.0235
0.0471±0.0146
−0.00
0.00
4
Cos
Cos
Dot
28.2
28.8
27.6
2.00
1.64
0.0398±0.0235
0.0471±0.0146
−0.00
0.00
5
Dot
Cos
Dot
19.6
0.0
39.1
1.37
2.09
0.0197±0.0115
0.0219±0.0270
−1.03
1.03
TABLE VI: Comparison of logit configurations for training and inference on Split CIFAR100 with a memory buffer of 5,000 samples. Aall , Ap , and Ac denote the final accuracies (%) on all, previous, and current classes, respectively. ∥w∥ denotes the mean class-weight vector norm; μW and σW denote the mean and standard deviation of weight entries; and μb denotes the mean bias. Subscripts p and c indicate previous and current classes. Dot and Cos denote dot-product and cosine logits, respectively.
Fig. 7: The analysis results of different ways during the learning process on Split CIFAR100 when the memory buffer size is 5000.
Fig. 8: Examples of high-score and low-score samples and their class activation maps obtained using dot-product and cosine logits.