Task-Anchored Representation Shaping for Pre-Trained Model-Based Continual Learning
Authors: Zhiming Xu, Huiyu Yi, Zhen-Hao Xie, Baile Xu, Furao Shen, Jian Zhao, Suorong Yang
Organizations: National Key Laboratory for Novel Software Technology, Nanjing University, Nanjing, China · School of Artificial Intelligence, Nanjing University, Nanjing, China · School of Electronic Science and Engineering, Nanjing University, China · School of Computer Science, Nanjing University, Nanjing, China
Pre-trained models (PTMs) provide a strong foundation for continual learning by offering stable representations that facilitate lightweight adaptation to new tasks. However, adapting well to each task does not ensure reliable inference over all learned tasks. Since task boundaries are often artificial and semantically entangled, an input from an unknown task can remain ambiguous even with strong PTM features, making cross-task prediction a key bottleneck. We propose Task-Anchored Inference Latent Shaping (TAILS), a lightweight post-PTM module that can be integrated into diverse continual learners and optimized through a decoupled step. TAILS uses fixed task anchors as persistent references to accumulated knowledge. It interprets each sample's feature representation relative to these references, then composes relevant evidence across tasks into latent recall. Rather than selecting a task-specific path or adjusting classifier outputs, TAILS uses latent recall to directly correct the feature representation before prediction. It therefore resolves cross-task ambiguity at the representation level, while leaving the original PTM, method-specific modules, and classifier unchanged. Extensive experiments across multiple PTM-based continual learning paradigms show that TAILS can improve classification and task-inference performance with modest parameter overhead and negligible inference cost.
Figures & tables
Figure 1: Motivation of TAILS. Left : Accuracy under the original and task-ID-oracle settings throughout continual learning. Right : For an ambiguous input, TAILS interprets its representation against accumulated task structures, composes sample-specific latent recall, and uses it to condition a feature correction before prediction by the original classifier.
Figure 2: Illustration of TAILS. After each task is learned by the base learner, its class prototypes and features are retained as task anchors. During continual optimization, the accumulated anchor pool maintains a coherent latent-recall space. At inference, a query derived from the current representation addresses the accumulated task structures and composes sample-specific latent recall, which conditions feature shaping before prediction by the original classifier.
Figure 3: Accuracy trajectories of different methods on ImageNet-A and ImageNet-R.
Lcls
Ltask
Lstab
SimpleCIL-TAILS
TUNA-TAILS
×
×
×
48.67
64.18
×
✓
✓
50.51
65.17
✓
×
✓
54.12
64.91
✓
✓
×
54.78
65.02
✓
✓
✓
54.86
65.30
Table 2: Ablation study of the training objectives in TAILS. All results are averaged over five seeds.
Variant
CARS
ImageNet-R
SimpleCIL
TUNA
MOS
CL-LoRA
SimpleCIL
TUNA
MOS
CL-LoRA
Anchor-MLP
58.81
77.32
71.34
65.92
62.43
75.91
72.15
73.57
Anchor Retrieval
56.66
77.04
71.03
66.02
60.47
75.51
71.37
73.48
Single-Center
62.23
77.72
70.54
67.52
65.95
76.23
72.55
73.91
Logit Calibration
45.22
76.63
70.81
64.61
56.82
75.33
70.42
73.31
TAILS
64.39
78.09
72.75
69.41
67.94
76.74
73.32
74.01
Table 3: Mechanism comparison on CARS and ImageNet-R. All results are averaged over five seeds. All variants use the same anchor and optimization budgets.
Method
ImageNet-A
ImageNet-R
AT
tˉinf (ms)
AT
tˉinf (ms)
SimpleCIL
48.67
4.73
54.59
4.82
SimpleCIL-TAILS
54.86
4.92
67.94
5.10
ACMap
54.49
4.93
69.33
5.05
TUNA
64.18
62.78
75.32
145.04
TUNA-TAILS
65.30
66.09
76.74
150.42
Table 4: Comparison of average final accuracy and the per-sample inference time measured over the test set.
Methods
Model Params.
Auxiliary Storage
Total Storage
ACMap
1.19
47.59
48.78
MOS
12.17
130.29
142.46
TUNA
12.48
118.12
130.60
CL-LoRA
7.56
0.00
7.56
TAILS
1.14
3.07
4.21
Table 5: Comparison of additional storage overhead on the ImageNet-R dataset, measured in millions of scalar values (M), excluding the PTM.
Figure 4: Impact of support anchor and task slot counts on the results across different PTM-based CL methods.
General continual learning (GCL) aims to learn from evolving data streams without task identities, explicit boundaries, or repeated access to previous data, making it a realistic yet challenging setting for continual intelligence. Although pretrained models (PTMs) provide rich prior knowledge for addressing the limited supervision and non-stationary nature of GCL, existing PTM-based methods often directly adapt pretrained representations and overlook two critical gaps: the misalignment between upstream pretraining and downstream continual adaptation, and the unreliability of conventional output alignment under blurry streams. Here we propose MePo++, a unified post-training framework that bridges pretrained knowledge and downstream GCL through representation refinement and reconciliation. MePo++ introduces two complementary components: MetaPrep, which improves representation plasticity for continual adaptation through unsupervised meta-refinement over pseudo continual sequences; and StreamAlign, which reinforces representation stability by reconciling evolving online features with a stable pretrained geometry. By improving representation learnability before adaptation and preserving alignment during continual learning, MePo++ enables PTMs to remain both plastic for new concepts and stable over evolving streams. Experiments across diverse PTMs, datasets, and continual learning baselines demonstrate the consistent effectiveness and generality of MePo++ for PTM-based GCL. Our code is available at https://github.com/SunGL001/MePo_Plus.
Guanglong Sun, Kanglei Zhou, Liyuan Wang +6
School of Life Sciences, IDG/McGovern Institute for Brain Research, Tsinghua University, Beijing, China · Department of Psychological and Cognitive Sciences, Tsinghua University, Beijing, China · Institute of Software, Chinese Academy of Sciences, Beijing, China +1
Class-incremental learning (CIL) requires a model to incrementally learn tasks that contain new classes without accessing earlier training data while preserving the ability to recognize all seen classes. Recently, pretrained-model-based approaches have become prevalent by adapting a frozen backbone with additional lightweight trainable modules. Existing methods, however, exhibit limitations: task-specific adapters learn explicit per-task representations but are parameter- and computation-inefficient, while LoRA-based merging methods combine per-task LoRA parameters into a single model whose static aggregated weights cause representation interference during inference. To address these problems, we present FACET: task-conditioned FeAture transformation with ConditionEd feature consisTency, achieving excellent parameter efficiency while producing highly discriminative features during inference. When continually trained on a task sequence, FACET learns a single shared adapter that employs a dynamic task-conditioned feature transformation, shaping the overall feature distribution of the adapter into a mixture of overlap-reduced task-specific components. On the other hand, we propose an efficient replay-free task-conditioned feature consistency loss, aiming to mitigate catastrophic forgetting of the learned mixture distribution in the adapter's feature space. Even when maintaining only a single adapter, FACET demonstrates robust scalability. On both very long task sequences (e.g., 200 tasks) and standard short task sequences (e.g., 20 tasks), our method achieves superior performance while using significantly fewer trainable parameters and GFLOPs. The code will be made open source upon acceptance.
Yunxiang Fu, Meng Lou, Yizhou Yu
School of Computing and Data Science The University of Hong Kong
The world is inherently dynamic, and continual learning aims to enable models to adapt to ever-evolving data streams. While pre-trained models have shown powerful performance in continual learning, they still require finetuning to adapt effectively to downstream tasks. However, prevailing Parameter-Efficient Fine-Tuning (PEFT) methods operate through empirical, black-box optimization at the weight level. These approaches lack explicit control over representation drift, leading to sensitivity to domain shifts and catastrophic forgetting in continual learning scenarios. In this work, we introduce Continual Representation Learning (CoRe), a novel framework that for the first time shifts the finetuning paradigm from weight space to representation space. Unlike conventional methods, CoRe performs task-specific interventions within a low-rank linear subspace of hidden representations, adopting a learning process with explicit objectives, which ensures stability for past tasks while maintaining plasticity for new ones. By constraining updates to a low-rank subspace, CoRe achieves exceptional parameter efficiency. Extensive experiments across multiple continual learning benchmarks demonstrate that CoRe not only preserves parameter efficiency but also significantly outperforms existing state-of-the-art methods. Our work introduces representation finetuning as a new, more effective and interpretable paradigm for continual learning.
Haihua Luo, Xuming Ran, Tommi Kärkkäinen +4
Faculty of Information Technology, University of Jyväskylä, Jyväskylä, Finland · National University of Singapore, Singapore, Singapore · School of Computer Science and Technology, Dalian University of Technology, Dalian, China