BAM! Born-Again Multi-Task Networks for Natural Language Understanding
Not on the map. Why?
Abstract (arXiv)
It can be challenging to train multi-task neural networks that outperform or even match their single-task counterparts. To help address this, we propose using knowledge distillation where single-task models teach a multi-task model. We enhance this training with teacher annealing, a novel method that gradually transitions the model from distillation to supervised learning, helping the multi-task model surpass its single-task teachers. We evaluate our approach by multi-task fine-tuning BERT on the GLUE benchmark. Our method consistently improves over standard single-task and multi-task training.
Key facts. Cited by 85 papers in the map (data through 14 Oct 2026). First posted on arXiv 10 Jul 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 36; 2022: 16; 2023: 17; 2024: 8; 2025: 6; 2026: 2.
85 papers.
Sort:
By area:
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,780
- 2021Parameter-efficient Multi-task Fine-tuning for Transformers via Shared Hypernetworkscited by 284
- 2021UniT: Multimodal Multitask Learning with a Unified Transformercited by 195
- 2023Multitask Prompt Tuning Enables Parameter-Efficient Transfer Learningcited by 115
- 2024Knowledge Fusion of Large Language Modelscited by 115
- 2021Beyond Distillation: Task-level Mixture-of-Experts for Efficient Inferencecited by 96
- 2021Multi-Task Self-Training for Learning General Representationscited by 85
- 2021Multi-Task Learning in Natural Language Processing: An Overviewcited by 62
- 2021Annealing Knowledge Distillationcited by 61
- 2021Object DGCNN: 3D Object Detection using Dynamic Graphscited by 58
- 2022Uni-Perceiver-MoE: Learning Sparse Generalist Models with Conditional MoEscited by 58
- 2021Universal Representation Learning from Multiple Domains for Few-shot Classificationcited by 56
38 more
- 2021PAIR: Leveraging Passage-Centric Similarity Relation for Improving Dense Passage Retrievalcited by 50
- 2022Model Cascading: Towards Jointly Improving Efficiency and Accuracy of NLP Systemscited by 44
- 2023FAME-ViL: Multi-Tasking Vision-Language Model for Heterogeneous Fashion Taskscited by 38
- 2021The Stem Cell Hypothesis: Dilemma behind Multi-Task Learning with Transformer Encoderscited by 32
- 2021MATE-KD: Masked Adversarial TExt, a Companion to Knowledge Distillationcited by 27
- 2024UNIC: Universal Classification Models via Multi-teacher Distillationcited by 27
- 2022Universal Representations: A Unified Look at Multiple Task and Domain Learningcited by 23
- 2021Deep Multi-Task Learning for Joint Localization, Perception, and Predictioncited by 22
- 2021Specializing Multilingual Language Models: An Empirical Studycited by 21
- 2021Student Surpasses Teacher: Imitation Attack for Black-Box NLP APIscited by 18
- 2021Pro-KD: Progressive Distillation by Following the Footsteps of the Teachercited by 17
- 2023Towards a Unified Transformer-based Framework for Scene Graph Generation and Human-object Interaction Detectioncited by 15
- 2021Towards a Unified Foundation Model: Jointly Pre-Training Transformers on Unpaired Images and Textcited by 14
- 2021MergeDistill: Merging Pre-trained Language Models using Distillationcited by 13
- 2023The Expressive Power of Tuning Only the Normalization Layerscited by 13
- 2021De-identification of Privacy-related Entities in Job Postingscited by 12
- 2021Not Far Away, Not So Close: Sample Efficient Nearest Neighbour Data Augmentation via MiniMaxcited by 10
- 2021Improving Tree-Structured Decoder Training for Code Generation via Mutual Learningcited by 10
- 2022Nearest Neighbor Knowledge Distillation for Neural Machine Translationcited by 10
- 2022Can Open-Domain QA Reader Utilize External Knowledge Efficiently like Humans?cited by 10
- 2023Preventing Catastrophic Forgetting in Continual Learning of New Natural Language Taskscited by 10
- 2021What's in your Head? Emergent Behaviour in Multi-Task Transformer Modelscited by 9
- 2022FS-BAN: Born-Again Networks for Domain Generalization Few-Shot Classificationcited by 9
- 2025Clinical QA 2.0: Multi-Task Learning for Answer Extraction and Categorizationcited by 9
- 2021Multitasking Inhibits Semantic Driftcited by 8
- 2022Let the Model Decide its Curriculum for Multitask Learningcited by 8
- 2023Visual Exemplar Driven Task-Prompting for Unified Perception in Autonomous Drivingcited by 8
- 2021Dual-View Distilled BERT for Sentence Embeddingcited by 7
- 2021Language Modelling via Learning to Rankcited by 7
- 2022Confidence Based Bidirectional Global Context Aware Training Framework for Neural Machine Translationcited by 7
- 2022Mask More and Mask Later: Efficient Pre-training of Masked Language Models by Disentangling the [MASK] Tokencited by 7
- 2023Adversarial Multi-task Learning for End-to-end Metaphor Detectioncited by 7
- 2024PEMT: Multi-Task Correlation Guided Mixture-of-Experts Enables Parameter-Efficient Transfer Learningcited by 7
- 2024ToXCL: A Unified Framework for Toxic Speech Detection and Explanationcited by 7
- 2021Latent-Optimized Adversarial Neural Transfer for Sarcasm Detectioncited by 6
- 2021Improving Question Answering Performance Using Knowledge Distillation and Active Learningcited by 6
- 2022Sentiment-Aware Word and Sentence Level Pre-training for Sentiment Analysiscited by 6
- 2022Transformers Go for the LOLs: Generating (Humourous) Titles from Scientific Abstracts End-to-Endcited by 6
50 most cited of 85 All 85 (JSON)
- 2026Progressive2: A Teacher-Student Progressive Co-Evolving Knowledge Distillation Method for Substantial Model Compression31 Jul 2026
- 2026Trade-offs in Ensembling, Merging and Routing Among Parameter-Efficient Experts3 Mar 2026
- 2025Revisit the Imbalance Optimization in Multi-task Learning: An Experimental Analysis28 Sep 2025
- 2025Unraveling the cognitive patterns of Large Language Models through module communities25 Aug 2025
- 2025Multi-task Learning for Heterogeneous Multi-source Block-Wise Missing Data30 May 2025
- 2025Distil-xLSTM: Learning Attention Mechanisms through Recurrent Structures24 Mar 2025
- 2025Clinical QA 2.0: Multi-Task Learning for Answer Extraction and Categorization18 Feb 2025
- 2025The Effect of Optimal Self-Distillation in Noisy Gaussian Mixture Model27 Jan 2025
- 2024Foldable SuperNets: Scalable Merging of Transformers with Different Initializations and Tasks2 Oct 2024
- 2024Enhancing Romanian Offensive Language Detection through Knowledge Distillation, Multi-Task Learning, and Data Augmentation30 Sep 2024
- 2024UNIC: Universal Classification Models via Multi-teacher Distillation9 Aug 2024
- 2024Topological Persistence Guided Knowledge Distillation for Wearable Sensor Data7 Jul 2024
38 more
- 2024ToXCL: A Unified Framework for Toxic Speech Detection and Explanation25 Mar 2024
- 2024PEMT: Multi-Task Correlation Guided Mixture-of-Experts Enables Parameter-Efficient Transfer Learning23 Feb 2024
- 2024DFML: Decentralized Federated Mutual Learning2 Feb 2024
- 2024Knowledge Fusion of Large Language Models19 Jan 2024
- 2023Less or More From Teacher: Exploiting Trilateral Geometry For Knowledge Distillation22 Dec 2023
- 2023Heterogeneous Transfer Learning for Building High-Dimensional Generalized Linear Models with Disparate Datasets20 Dec 2023
- 2023Data exploitation: multi-task learning of object detection and semantic segmentation on partially annotated data7 Nov 2023
- 2023Towards a Unified Transformer-based Framework for Scene Graph Generation and Human-object Interaction Detection3 Nov 2023
- 2023Talking Models: Distill Pre-trained Knowledge to Downstream Models via Interactive Communication4 Oct 2023
- 2023Challenges and Opportunities of Using Transformer-Based Multi-Task Learning in NLP Through ML Lifecycle: A Survey16 Aug 2023
- 2023EM-Network: Oracle Guided Self-distillation for Sequence Learning14 Jun 2023
- 2023Adversarial Multi-task Learning for End-to-end Metaphor Detection26 May 2023
- 2023Label Agnostic Pre-training for Zero-shot Text Classification25 May 2023
- 2023Multitask Prompt Tuning Enables Parameter-Efficient Transfer Learning6 Mar 2023
- 2023FAME-ViL: Multi-Tasking Vision-Language Model for Heterogeneous Fashion Tasks4 Mar 2023
- 2023Visual Exemplar Driven Task-Prompting for Unified Perception in Autonomous Driving3 Mar 2023
- 2023Preventing Catastrophic Forgetting in Continual Learning of New Natural Language Tasks22 Feb 2023
- 2023The Expressive Power of Tuning Only the Normalization Layers15 Feb 2023
- 2023Improving Text-based Early Prediction by Distillation from Privileged Time-Series Text26 Jan 2023
- 2023ERNIE 3.0 Tiny: Frustratingly Simple Method to Improve Task-Agnostic Distillation Generalization9 Jan 2023
- 2023Multi-Task Learning with Prior Information4 Jan 2023
- 2022Transformers Go for the LOLs: Generating (Humourous) Titles from Scientific Abstracts End-to-End20 Dec 2022
- 2022Continuation KD: Improved Knowledge Distillation through the Lens of Continuation Optimization12 Dec 2022
- 2022Can Open-Domain QA Reader Utilize External Knowledge Efficiently like Humans?23 Nov 2022
- 2022Mask More and Mask Later: Efficient Pre-training of Masked Language Models by Disentangling the [MASK] Token9 Nov 2022
- 2022Sentiment-Aware Word and Sentence Level Pre-training for Sentiment Analysis18 Oct 2022
- 2022Model Cascading: Towards Jointly Improving Efficiency and Accuracy of NLP Systems11 Oct 2022
- 2022FS-BAN: Born-Again Networks for Domain Generalization Few-Shot Classification23 Aug 2022
- 2022Coarse-to-Fine: Hierarchical Multi-task Learning for Natural Language Understanding19 Aug 2022
- 2022Uni-Perceiver-MoE: Learning Sparse Generalist Models with Conditional MoEs9 Jun 2022
- 2022All Birds with One Stone: Multi-task Text Classification for Efficient Inference with One Forward Pass22 May 2022
- 2022Let the Model Decide its Curriculum for Multitask Learning19 May 2022
- 2022Nearest Neighbor Knowledge Distillation for Neural Machine Translation1 May 2022
- 2022Universal Representations: A Unified Look at Multiple Task and Domain Learning6 Apr 2022
- 2022Look Backward and Forward: Self-Knowledge Distillation with Bidirectional Decoder for Neural Machine Translation10 Mar 2022
- 2022Confidence Based Bidirectional Global Context Aware Training Framework for Neural Machine Translation28 Feb 2022
- 2022ASC me to Do Anything: Multi-task Training for Embodied AI14 Feb 2022
- 2021Towards a Unified Foundation Model: Jointly Pre-Training Transformers on Unpaired Images and Text14 Dec 2021
50 newest of 85 All 85 (JSON)
- 2021Annealing Knowledge Distillationcited by 61
- 2021MATE-KD: Masked Adversarial TExt, a Companion to Knowledge Distillationcited by 27
- 2024UNIC: Universal Classification Models via Multi-teacher Distillationcited by 27
- 2021Towards a Unified Foundation Model: Jointly Pre-Training Transformers on Unpaired Images and Textcited by 14
- 2022Nearest Neighbor Knowledge Distillation for Neural Machine Translationcited by 10
- 2022Confidence Based Bidirectional Global Context Aware Training Framework for Neural Machine Translationcited by 7
- 2021Improving Question Answering Performance Using Knowledge Distillation and Active Learningcited by 6
- 2022Continuation KD: Improved Knowledge Distillation through the Lens of Continuation Optimizationcited by 3
- 2023ERNIE 3.0 Tiny: Frustratingly Simple Method to Improve Task-Agnostic Distillation Generalizationcited by 3
- 2023Data exploitation: multi-task learning of object detection and semantic segmentation on partially annotated datacited by 3
- 2023Less or More From Teacher: Exploiting Trilateral Geometry For Knowledge Distillationcited by 2
- 2025The Effect of Optimal Self-Distillation in Noisy Gaussian Mixture Modelcited by 2
12 most cited of 17 All 17 (JSON)
- 2026Progressive2: A Teacher-Student Progressive Co-Evolving Knowledge Distillation Method for Substantial Model Compression31 Jul 2026
- 2025The Effect of Optimal Self-Distillation in Noisy Gaussian Mixture Model27 Jan 2025
- 2024Enhancing Romanian Offensive Language Detection through Knowledge Distillation, Multi-Task Learning, and Data Augmentation30 Sep 2024
- 2024UNIC: Universal Classification Models via Multi-teacher Distillation9 Aug 2024
- 2024Topological Persistence Guided Knowledge Distillation for Wearable Sensor Data7 Jul 2024
- 2023Less or More From Teacher: Exploiting Trilateral Geometry For Knowledge Distillation22 Dec 2023
- 2023Data exploitation: multi-task learning of object detection and semantic segmentation on partially annotated data7 Nov 2023
- 2023EM-Network: Oracle Guided Self-distillation for Sequence Learning14 Jun 2023
- 2023ERNIE 3.0 Tiny: Frustratingly Simple Method to Improve Task-Agnostic Distillation Generalization9 Jan 2023
- 2022Continuation KD: Improved Knowledge Distillation through the Lens of Continuation Optimization12 Dec 2022
- 2022Nearest Neighbor Knowledge Distillation for Neural Machine Translation1 May 2022
- 2022Look Backward and Forward: Self-Knowledge Distillation with Bidirectional Decoder for Neural Machine Translation10 Mar 2022
12 newest of 17 All 17 (JSON)
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,780
- 2021Parameter-efficient Multi-task Fine-tuning for Transformers via Shared Hypernetworkscited by 284
- 2021UniT: Multimodal Multitask Learning with a Unified Transformercited by 195
- 2023Multitask Prompt Tuning Enables Parameter-Efficient Transfer Learningcited by 115
- 2024Knowledge Fusion of Large Language Modelscited by 115
- 2021Multi-Task Self-Training for Learning General Representationscited by 85
- 2021Multi-Task Learning in Natural Language Processing: An Overviewcited by 62
- 2021Object DGCNN: 3D Object Detection using Dynamic Graphscited by 58
- 2021Universal Representation Learning from Multiple Domains for Few-shot Classificationcited by 56
- 2021PAIR: Leveraging Passage-Centric Similarity Relation for Improving Dense Passage Retrievalcited by 50
- 2022Model Cascading: Towards Jointly Improving Efficiency and Accuracy of NLP Systemscited by 44
- 2021The Stem Cell Hypothesis: Dilemma behind Multi-Task Learning with Transformer Encoderscited by 32
12 most cited of 64 All 64 (JSON)
- 2026Trade-offs in Ensembling, Merging and Routing Among Parameter-Efficient Experts3 Mar 2026
- 2025Revisit the Imbalance Optimization in Multi-task Learning: An Experimental Analysis28 Sep 2025
- 2025Unraveling the cognitive patterns of Large Language Models through module communities25 Aug 2025
- 2025Multi-task Learning for Heterogeneous Multi-source Block-Wise Missing Data30 May 2025
- 2025Distil-xLSTM: Learning Attention Mechanisms through Recurrent Structures24 Mar 2025
- 2025Clinical QA 2.0: Multi-Task Learning for Answer Extraction and Categorization18 Feb 2025
- 2024Foldable SuperNets: Scalable Merging of Transformers with Different Initializations and Tasks2 Oct 2024
- 2024ToXCL: A Unified Framework for Toxic Speech Detection and Explanation25 Mar 2024
- 2024DFML: Decentralized Federated Mutual Learning2 Feb 2024
- 2024Knowledge Fusion of Large Language Models19 Jan 2024
- 2023Heterogeneous Transfer Learning for Building High-Dimensional Generalized Linear Models with Disparate Datasets20 Dec 2023
- 2023Towards a Unified Transformer-based Framework for Scene Graph Generation and Human-object Interaction Detection3 Nov 2023
12 newest of 64 All 64 (JSON)