Sub-network Discovery and Soft-masking for Continual Learning of Mixed Tasks
Not on the map. Why?
Abstract (arXiv)
Continual learning (CL) has two main objectives: preventing catastrophic forgetting (CF) and encouraging knowledge transfer (KT). The existing literature mainly focused on overcoming CF. Some work has also been done on KT when the tasks are similar. To our knowledge, only one method has been proposed to learn a sequence of mixed tasks. However, these techniques still suffer from CF and/or limited KT. This paper proposes a new CL method to achieve both. It overcomes CF by isolating the knowledge of each task via discovering a subnetwork for it. A soft-masking mechanism is also proposed to preserve the previous knowledge and to enable the new task to leverage the past knowledge to achieve KT. Experiments using classification, generation, information extraction, and their mixture (i.e., heterogeneous tasks) show that the proposed method consistently outperforms strong baselines.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 51 of its 65 references that are in the map. First posted on arXiv 13 Oct 2023.
Builds on
51 of 65 references.
References · 51
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2016iCaRL: Incremental Classifier and Representation Learningcited by 2,313
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2017Gradient Episodic Memory for Continual Learningcited by 1,871
- 2016Progressive Neural Networkscited by 1,332
- 2017Continual Learning with Deep Generative Replaycited by 1,152
- 2018Efficient Lifelong Learning with A-GEMcited by 1,044
- 2021Learning to Prompt for Continual Learningcited by 938
- 2017PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruningcited by 882
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
38 more
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2018Overcoming catastrophic forgetting with hard attention to the taskcited by 672
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 434
- 2017PathNet: Evolution Channels Gradient Descent in Super Neural Networkscited by 362
- 2020The Lottery Ticket Hypothesis for Pre-trained BERT Networkscited by 275
- 2017Overcoming Catastrophic Forgetting by Incremental Moment Matchingcited by 243
- 2019What's Hidden in a Randomly Weighted Neural Network?cited by 236
- 2020Supermasks in Superpositioncited by 223
- 2020Mnemonics Training: Multi-Class Incremental Learning without Forgettingcited by 220
- 2023Continual Pre-training of Language Modelscited by 194
- 2019Episodic Memory in Lifelong Language Learningcited by 188
- 2019BERT Post-Training for Review Reading Comprehension and Aspect-based Sentiment Analysiscited by 176
- 2021An Empirical Investigation of the Role of Pre-training in Lifelong Learningcited by 134
- 2020On the Effect of Dropping Layers of Pre-trained Transformer Modelscited by 118
- 2021LFPT5: A Unified Framework for Lifelong Few-shot Language Learning Based on Prompt Tuning of T5cited by 103
- 2021Achieving Forgetting Prevention and Knowledge Transfer in Continual Learningcited by 91
- 2021Continual Learning for Text Classification with Information Disentanglement Based Regularizationcited by 86
- 2022Continual Learning of Natural Language Processing Tasks: A Surveycited by 83
- 2020Continual Learning in Task-Oriented Dialogue Systemscited by 82
- 2021Continual Learning of a Mixed Sequence of Similar and Dissimilar Taskscited by 80
- 2022TRGP: Trust Region Gradient Projection for Continual Learningcited by 79
- 2022A Theoretical Study on Solving Continual Learningcited by 78
- 2017Deep Generative Dual Memory Network for Continual Learningcited by 59
- 2022Beyond Not-Forgetting: Continual Learning with Backward Knowledge Transfercited by 59
- 2017Continual Learning in Generative Adversarial Netscited by 51
- 2019Complementary Learning for Overcoming Catastrophic Forgetting Using Experience Replaycited by 51
- 2021PARP: Prune, Adjust and Re-Prune for Self-Supervised Speech Recognitioncited by 50
- 2020Efficient Meta Lifelong-Learning with Limited Memorycited by 48
- 2021Differentiable Subset Pruning of Transformer Headscited by 46
- 2021Adapting BERT for Continual Learning of a Sequence of Aspect Sentiment Classification Taskscited by 46
- 2022Continual Prompt Tuning for Dialog State Trackingcited by 44
- 2023Parameter-Level Soft-Masking for Continual Learningcited by 39
- 2021ConvoSumm: Conversation Summarization Benchmark and Improved Abstractive Summarization with Argument Miningcited by 37
- 2022Continual Sequence Generation with Adaptive Compositional Modulescited by 31
- 2018Large-Scale Multi-Domain Belief Tracking with Knowledge Sharingcited by 29
- 2023Learnability and Algorithm for Continual Learningcited by 28
- 2022ConTinTin: Continual Learning from Task Instructionscited by 26
- 2020Pruning Redundant Mappings in Transformer Models via Spectral-Normalized Identity Priorcited by 23
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 3; 2026: 1.
7 papers.
- 2024Towards Lifelong Learning of Large Language Models: A Surveycited by 60
- 2024Disentangling and Mitigating the Impact of Task Similarity for Continual Learningcited by 19
- 2025Optimal Task Order for Continual Learning of Multiple Taskscited by 14
- 2025Recurrent Knowledge Identification and Fusion for Language Model Continual Learningcited by 14
- 2024MoFO: Momentum-Filtered Optimizer for Mitigating Forgetting in LLM Fine-Tuningcited by 13
- 2026FOREVER: Forgetting Curve-Inspired Memory Replay for Language Model Continual Learningcited by 7
- 2025AIMMerging: Adaptive Iterative Model Merging Using Training Trajectories for Language Model Continual Learningcited by 4
- 2026FOREVER: Forgetting Curve-Inspired Memory Replay for Language Model Continual Learning7 Jan 2026
- 2025AIMMerging: Adaptive Iterative Model Merging Using Training Trajectories for Language Model Continual Learning22 Sep 2025
- 2025Recurrent Knowledge Identification and Fusion for Language Model Continual Learning22 Feb 2025
- 2025Optimal Task Order for Continual Learning of Multiple Tasks5 Feb 2025
- 2024MoFO: Momentum-Filtered Optimizer for Mitigating Forgetting in LLM Fine-Tuning30 Jul 2024
- 2024Towards Lifelong Learning of Large Language Models: A Survey10 Jun 2024
- 2024Disentangling and Mitigating the Impact of Task Similarity for Continual Learning30 May 2024