A key challenge in lifelong imitation learning (LIL) is enabling agents to acquire new skills from expert demonstrations while retaining prior knowledge. This requires preserving the low-dimensional manifolds and geometric structures that underlie task representations across sequential learning. Existing distillation methods, which rely on L2-norm feature matching in raw feature space, are sensitive to noise and high-dimensional variability, often failing to preserve intrinsic task manifolds. To address this, we introduce SPREAD, a geometry-preserving framework that employs singular value decomposition (SVD) to align policy representations across tasks within low-rank subspaces. This alignment maintains the underlying geometry of multimodal features, facilitating stable transfer, robustness, and generalization. Additionally, we propose a confidence-guided distillation strategy that applies a Kullback-Leibler divergence loss restricted to the top-M most confident action samples, emphasizing reliable modes and improving optimization stability. Experiments on the LIBERO, lifelong imitation learning benchmark, show that SPREAD substantially improves knowledge transfer, mitigates catastrophic forgetting, and achieves state-of-the-art performance.
Figures & tables
Fig. 2 : Overview of our proposed SPREAD method. Subspace Representation Distillation aligns the latent representations from different input modality encoders ( e.g. , Task, AgentView, HandEye, Joint, and Gripper information), while confidence-guided policy distillation maps the action distribution of the GMM policy between incremental steps Tk−1 and Tk .
Method
LIBERO-OBJECT
LIBERO-GOAL
LIBERO-SPATIAL
FWT ( ↑ )
NBT ( ↓ )
AUC ( ↑ )
FWT ( ↑ )
NBT ( ↓ )
AUC ( ↑ )
FWT ( ↑ )
NBT ( ↓ )
AUC ( ↑ )
Sequential
62.0 ( ± 1.0)
63.0 ( ± 2.0)
30.0 ( ± 1.0)
55.0 ( ± 1.0)
70.0 ( ± 1.0)
23.0 ( ± 1.0)
72.0 ( ± 1.0)
81.0 ( ± 1.0)
20.0 ( ± 1.0)
EWC [ 6 ]
56.0 ( ± 3.0)
69.0 ( ± 2.0)
16.0 ( ± 2.0)
32.0 ( ± 2.0)
48.0 ( ± 3.0)
6.0 ( ± 1.0)
23.0 ( ± 1.0)
33.0 ( ± 1.0)
6.0 ( ± 1.0)
ER [ 8 ]
56.0 ( ± 1.0)
24.0 ( ± 1.0)
49.0 ( ± 1.0)
53.0 ( ± 1.0)
36.0 ( ± 1.0)
47.0 ( ± 2.0)
65.0 ( ± 3.0)
27.0 ( ± 3.0)
56.0 ( ± 1.0)
BUDS [ 26 ]
52.0 ( ± 2.0)
21.0 ( ± 1.0)
47.0 ( ± 1.0)
50.0 ( ± 1.0)
39.0 ( ± 1.0)
42.0 ( ± 1.0)
-
-
-
LOTUS [ 9 ]
74.0 ( ± 3.0)
11.0 ( ± 1.0)
65.0 ( ± 3.0)
61.0 ( ± 3.0)
30.0 ( ± 1.0)
56.0 ( ± 1.0)
-
-
-
TABLE I : Experimental results across three different LIBERO task suites. The reported values are averages from three seeds, including the mean and standard error. The best values are highlighted in bold, and the second-best values are underlined. The dash (-) indicates a failure to reproduce results. All metrics are measured based on success rates (%).
Fig. 3 : Average success rate across incremental tasks on (a) LIBERO-OBJECT and (b) LIBERO-GOAL (Higher numbers are better).
Fig. 4 : Forward Transfer (FWT) across incremental tasks on (a) LIBERO-OBJECT and (b) LIBERO-GOAL (Higher numbers are better).
Fig. 5 : Negative Backward Transfer (NBT) across incremental tasks on LIBERO-GOAL (lower is better).
Fig. 6 : Drift analysis across incremental tasks for different modalities on LIBERO-OBJECT. The radial axes (1-9) indicate the incremental steps in LIL. Lower representation drift indicates better preservation of previously learned knowledge across tasks in lifelong imitation learning.
Ltext
Limage
Laction
Lextra
LIBERO-GOAL
FWT ↑
NBT ↓
AUC ↑
✓
✓
✓
✓
0.78
0.09
0.72
✗
✓
✓
✓
0.78
0.11
0.69
✓
✗
✓
✓
0.67
0.20
0.57
✓
✓
✗
✓
0.70
0.02
0.70
✓
✓
✓
✗
0.74
0.03
0.71
TABLE II : Ablation studies on the contribution of each component in our method. Experiments were performed on the LIBERO-GOAL task suite. Reported results are the mean across three different seed values.
Fig. 7 : Ablation study on policy distillation strategies using LIBERO-GOAL task suite. Bar represents mean across three random seeds while standard error is represented with darker color. Our top- M confidence-weighted approach outperforms baselines in FWT, NBT and AUC.
Rank of Subspace
FWT ( ↑ )
BWT ( ↓ )
AUC ( ↑ )
SPREAD (Full-Rank)
76.0 ( ± 1.0)
16.0 ( ± 8.0)
66.0 ( ± 5.0)
SPREAD (Rank =48 )
78.0 ( ± 1.0)
9.0 ( ± 5.0)
72.0 ( ± 1.0)
SPREAD (Rank =32 )
74.0 ( ± 1.0)
6.0 ( ± 4.0)
71.0 ( ± 2.0)
SPREAD (Rank =16 )
76.0 ( ± 3.0)
9.0 ( ± 4.0)
69.0 ( ± 2.0)
TABLE III : Ablation study on the dimensionality/rank of subspace representation distillation strategies using LIBERO-GOAL task suite. Our proposed SPREAD loss with 75% -rank performs best in FWT, NBT and AUC.
We introduce a lifelong imitation learning framework that enables continual policy refinement across sequential tasks under realistic memory and data constraints. Our approach departs from conventional experience replay by operating entirely in a multimodal latent space, where compact representations of visual, linguistic, and robot's state information are stored and reused to support future learning. To further stabilize adaptation, we introduce an incremental feature adjustment mechanism that regularizes the evolution of task embeddings through an angular margin constraint, preserving inter-task distinctiveness. Our method establishes a new state of the art in the LIBERO benchmarks, achieving 10-17 point gains in AUC and up to 65% less forgetting compared to previous leading methods. Ablation studies confirm the effectiveness of each component, showing consistent gains over alternative strategies. The code is available at: https://github.com/yfqi/lifelong_mlr_ifa.
Fanqi Yu, Matteo Tiezzi, Tommaso Apicella +2
AI for Good (AIGO), Istituto Italiano di Tecnologia, Genoa, Italy · DITEN, University of Genoa, Genoa, Italy · PAVIS, Istituto Italiano di Tecnologia, Genoa, Italy +1
Continual learning in Large Language Models (LLMs) is hindered by the plasticity-stability dilemma, where acquiring new capabilities often leads to catastrophic forgetting of previous knowledge. Existing methods typically treat parameters uniformly, failing to distinguish between specific task knowledge and shared capabilities. We introduce Mixture of Sparse Experts for Task Agnostic Continual Learning (SETA), a framework that resolves the plasticity-stability conflict through adaptive sparse subspace decomposition into task-specific expert modules. Unlike standard updates, where tasks compete for the same parameters, SETA separates knowledge into unique experts, designed to isolate task-specific patterns, and shared experts, responsible for capturing common features. This structure is maintained through adaptive elastic anchoring and a routing-aware regularization that jointly protect shared knowledge at both the weight and routing levels and enable a unified gating network to automatically retrieve the correct expert combination during inference. Extensive experiments across diverse domain-specific benchmarks demonstrate that SETA achieves competitive or superior overall performance relative to state-of-the-art continual learning baselines, with particularly strong retention of early-task knowledge and improved backward transfer on LLaMA-2 7B and Qwen3-4B.
Fatema Siddika, Md Anwar Hossen, Tanwi Mallick +1
Iowa State University, Ames, USA · Argonne National Laboratory, USA
Catastrophic forgetting remains a primary hurdle in sequential task learning for artificial neural networks. We propose a silicon-native modular architecture that achieves structural parameter isolation using Task-Specific Experts and a distributed, outlier-based Gatekeeper. Moving beyond traditional sequential consolidation, our framework utilizes a Simultaneous Pipeline where Teacher learning, Student distillation, and Router manifold acquisition occur in parallel while raw data is present in a localized training session. This approach ensures computational efficiency and complies with privacy mandates like GDPR by deleting raw data as soon as a task is learned. We demonstrate that a Tight-Bottleneck Autoencoder (TB-AE) can effectively distinguish semantically crowded manifolds in high-dimensional latent spaces, overcoming the posterior collapse inherent to standard variational methods. By establishing strict topological boundaries, our TB-AE resolves latent space crowding in 4096-D LLM embeddings to provide a robust, unsupervised novelty signal. Furthermore, we validate an Autonomous Retrieval mechanism that confidently identifies returning manifolds, enabling stable lifelong learning without redundant module instantiation. Empirical results demonstrate that our ``Live Distillation'' approach acts as a natural regularizer, achieving strong retention across computer vision and natural language processing domains without suffering a student fidelity gap.