English

Towards stable training of parallel continual learning

Machine Learning 2024-07-12 v1 Artificial Intelligence

Abstract

Parallel Continual Learning (PCL) tasks investigate the training methods for continual learning with multi-source input, where data from different tasks are learned as they arrive. PCL offers high training efficiency and is well-suited for complex multi-source data systems, such as autonomous vehicles equipped with multiple sensors. However, at any time, multiple tasks need to be trained simultaneously, leading to severe training instability in PCL. This instability manifests during both forward and backward propagation, where features are entangled and gradients are conflict. This paper introduces Stable Parallel Continual Learning (SPCL), a novel approach that enhances the training stability of PCL for both forward and backward propagation. For the forward propagation, we apply Doubly-block Toeplit (DBT) Matrix based orthogonality constraints to network parameters to ensure stable and consistent propagation. For the backward propagation, we employ orthogonal decomposition for gradient management stabilizes backpropagation and mitigates gradient conflicts across tasks. By optimizing gradients by ensuring orthogonality and minimizing the condition number, SPCL effectively stabilizing the gradient descent in complex optimization tasks. Experimental results demonstrate that SPCL outperforms state-of-the-art methjods and achieve better training stability.

Keywords

Cite

@article{arxiv.2407.08214,
  title  = {Towards stable training of parallel continual learning},
  author = {Li Yuepan and Fan Lyu and Yuyang Li and Wei Feng and Guangcan Liu and Fanhua Shang},
  journal= {arXiv preprint arXiv:2407.08214},
  year   = {2024}
}
R2 v1 2026-06-28T17:36:47.573Z