MAEM: Mitigating Recency Bias via a Replay-Free, Capacity-Controlled Dynamic Mixture of Experts for Class-Incremental Graph Learning
Authors: Junlin Chen, Wenhao Wang, Junchao Xu, Qingsong Yao, Fei Xue, Li Sun, Xiao Wang, Ruijie Wang
Organizations: Beihang University, Beijing, China · Renmin University of China, Beijing, China · Trine University, Indiana, USA · Beijing University of Posts and Telecommunications, Beijing, China
Graph Neural Networks (GNNs) excel on graph-structured data but suffer catastrophic forgetting when classes arrive sequentially. In class-incremental graph learning, predictions are often dominated by recently learned classes, causing severe recency bias. Dynamic expansion can isolate knowledge across sessions, but many methods assume task-incremental settings and require task identities at test time to select branches or heads---information unavailable in class-incremental learning. Learnable routers offer an alternative, yet continual fine-tuning the router on new data can bias routing toward recent components. We propose MAEM, a replay-free, capacity-controlled dynamic mixture-of-experts framework for class-incremental graph learning. MAEM builds representations with Simplified Graph Convolution (SGC) and trains a dedicated expert for each new session, then freezes it to protect prior knowledge. For routing without task IDs or a shared trainable router, each expert uses a masked autoencoder (MAE); test nodes are assigned to the expert with the lowest reconstruction error, reducing routing-level recency bias. When the expert pool reaches a fixed capacity, MAEM merges the least-used experts via Git Re-Basin alignment and further distills them into a fused expert using topology-aware pseudo data synthesized from MAE distributions (DeepInversion-style), without storing or revisiting historical graph data. On four class-incremental graph benchmarks, MAEM mitigates recency bias, outperforms strong continual graph learning baselines, and keeps model capacity bounded under strict replay-free learning.
Figures & tables
Figure 1: Session-wise accuracy matrix of LwF on cora-full. The entry in row i and column j denotes the accuracy on session j after training through session i .
Figure 2: Overall framework of MAEM.
OGBN-Arxiv
Cora-Full
Method
Replay
Micro
AP
AF
Micro
AP
AF
Bare
×
0.52±0.24
6.99±0.46
−83.71±0.87
1.60±0.16
6.86±0.48
−90.99±0.69
EWC
×
0.39±0.01
6.90±0.17
−75.86±1.59
2.02±0.31
7.38±0.62
−90.59±0.64
MAS
×
0.37±0.01
6.61±0.20
−75.58±1.16
1.18±0.03
5.53±0.13
−91.26±0.30
LwF
×
0.72±0.22
6.78±0.35
−81.97±1.84
2.03±0.17
7.99±0.55
−87.92±0.96
TWP
×
0.37±0.02
6.54±0.30
−76.68±1.41
1.10±0.03
4.89±0.11
−87.44±0.56
Table 1: Main results on four class-incremental graph benchmarks. Results are percentages reported as mean ± standard deviation over five runs. ✓ and × indicate whether historical information is replayed. MAEM N and MAEM M denote MAEM without and with expert merging, respectively. Bold and underlined values indicate the best and second-best results.
Figure 3: Final Micro accuracy, AP, and routing accuracy under different maximum numbers of experts on OGBN-Arxiv and Cora-Full.
Figure 4: Session-wise Micro accuracy and AP on Cora-Full under different expert-capacity limits.
Variant
Micro
AP
Full model
67.40±0.28
63.04±0.25
Random initialization
61.13±0.30
45.87±0.21
w/o distillation
60.21±0.39
45.20±0.19
w/o topology-aware loss
60.77±0.30
45.71±0.21
Table 2: Ablation results (%) on Cora-Full.
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
Dataset
Nodes
Edges
Feature Dimension
Classes
Class per Session
Sessions
ogbn-arxiv
169,343
1,166,243
128
40
4
10
Cora-Full
19,793
65,311
512 †
70
4
17
Coauthor-CS
18,333
81,894
256 †
15
2
7
WikiCS
11,701
216,123
300
10
2
5
Appendix
Table 3: Dataset statistics. “Feat.” is the feature dimension after preprocessing; † denotes Truncated-SVD reduction (original dimensions: 8,710 for Cora-Full and 6,805 for Coauthor-CS ). “ C /Sess.” is the number of classes per session. For Coauthor-CS , the final incomplete session is discarded, so 14 of 15 classes are used.
Figure 5: Hyperparameter sensitivity on Cora-Full. Left: MAE mask ratio. Right: number of pseudo samples per class used in expert fusion. Curves report Micro and AP (mean ± std over five seeds).
OGBN-Arxiv
Cora-Full
Backbone
Micro
AP
AF
Micro
AP
AF
Identity
33.18±1.65
35.12±0.20
−19.06±1.39
43.26±0.25
26.89±0.30
−1.63±0.27
SGC / GCN
50.34±1.41
46.35±0.27
−15.04±0.68
68.22±0.08
65.54±0.25
−7.85±0.36
GraphSAGE
43.24±2.03
45.64±0.36
−18.43±1.14
69.72±0.11
65.25±0.25
−7.04±0.34
Random Walk
48.45±1.60
47.04±0.43
−15.75±1.15
67.42±0.20
65.86±0.17
−9.33±0.44
PPR
49.52±1.56
46.77±0.24
−15.63±0.50
69.69±0.17
65.16±0.16
−6.02±0.21
Appendix
Table 4: Backbone ablation of MAEM (%). All operators are parameter-free; only the propagation rule changes. Mean ± std over five seeds. Best results are in bold and second-best are underlined.
Expert
Classes
MMD2
Accpseudo
Accreal
Ratio
0
0 – 3
0.239±0.019
0.734±0.087
0.903±0.001
0.813±0.097
1
4 – 7
0.170±0.034
0.835±0.017
0.909±0.002
0.918±0.018
2
8 – 11
0.147±0.029
0.850±0.027
0.955±0.002
0.891±0.029
3
12 – 15
0.187±0.032
0.837±0.017
0.941±0.003
0.890±0.019
4
16 – 19
0.262±0.060
0.858±0.077
0.932±0.008
0.921±0.083
5
20 – 23
0.162±0.023
0.803±0.022
0.904±0.011
0.888±0.022
Appendix
Table 5: Per-expert pseudo-embedding quality on ogbn-arxiv (no-fusion, K=10 ). MMD2 uses a Gaussian kernel with median-heuristic bandwidth and an unbiased estimator. Ratio=Accpseudo/Accreal under an identical classifier. Mean ± std over five seeds.