Auxiliary Learning by Implicit Differentiation
Not on the map. Why?
Abstract (arXiv)
Training neural networks with auxiliary tasks is a common practice for improving the performance on a main task of interest. Two main challenges arise in this multi-task learning setting: (i) designing useful auxiliary tasks; and (ii) combining auxiliary tasks into a single coherent loss. Here, we propose a novel framework, AuxiLearn, that targets both challenges based on implicit differentiation. First, when useful auxiliaries are known, we propose learning a network that combines all losses into a single coherent objective function. This network can learn non-linear interactions between tasks. Second, when no useful auxiliary task is known, we describe how to learn a network that generates a meaningful, novel auxiliary task. We evaluate AuxiLearn in a series of tasks and domains, including image segmentation and learning with attributes in the low data regime, and find that it consistently outperforms competing methods.
Key facts. Cited by 45 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 10; 2023: 9; 2024: 9; 2025: 9; 2026: 3.
45 papers.
Sort:
By area:
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2022AdaTask: A Task-aware Adaptive Learning Rate Approach to Multi-task Learningcited by 31
- 2023ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learningcited by 28
- 2023Prismer: A Vision-Language Model with Multi-Task Expertscited by 28
- 2021Meta-Learning to Improve Pre-Trainingcited by 27
- 2021Should We Be Pre-training? An Argument for End-task Aware Training as an Alternativecited by 26
- 2021Pareto Domain Adaptationcited by 20
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2023MELTR: Meta Loss Transformer for Learning to Fine-tune Video Foundation Modelscited by 15
- 2023Improving Few-Shot Generalization by Exploring and Exploiting Auxiliary Datacited by 13
- 2022AANG: Automating Auxiliary Learningcited by 12
33 more
- 2022Optimization with Access to Auxiliary Informationcited by 12
- 2023Auxiliary Learning as an Asymmetric Bargaining Gamecited by 12
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methodscited by 10
- 2021Linear Speedup in Personalized Collaborative Learningcited by 8
- 2022Towards Optimization and Model Selection for Domain Generalization: A Mixup-guided Solutioncited by 6
- 2022Adaptive Mixing of Auxiliary Losses in Supervised Learningcited by 5
- 2022Image Classification with Small Datasets: Overview and Benchmarkcited by 4
- 2021Auxiliary Learning for Self-Supervised Video Representation via Similarity-based Knowledge Distillationcited by 3
- 2024Meta-Auxiliary Learning for Micro-Expression Recognitioncited by 3
- 2022Benchmarking Visual-Inertial Deep Multimodal Fusion for Relative Pose Regression and Odometry-aided Absolute Pose Regressioncited by 2
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspectivecited by 2
- 2026FGTBT: Frequency-Guided Task-Balancing Transformer for Unified Facial Landmark Detectioncited by 2
- 2026Non-Stationary Functional Bilevel Optimizationcited by 2
- 2022Context Label Learning: Improving Background Class Representations in Semantic Segmentationcited by 1
- 2023No Data Augmentation? Alternative Regularizations for Effective Training on Small Datasetscited by 1
- 2024Enhancing Molecular Property Prediction with Auxiliary Learning and Task-Specific Adaptationcited by 1
- 2024Robust Analysis of Multi-Task Learning Efficiency: New Benchmarks on Light-Weighed Backbones and Effective Measurement of Multi-Task Learning Challenges by Feature Disentanglementcited by 1
- 2024Data-Driven Preference Sampling for Pareto Front Learningcited by 1
- 2024Aux-NAS: Exploiting Auxiliary Labels with Negligibly Extra Inference Costcited by 1
- 2024Learning Representation for Multitask learning through Self Supervised Auxiliary learningcited by 1
- 2025Adapting Foundation Models for Few-Shot Medical Image Segmentation: Actively and Sequentiallycited by 1
- 2025Behavior Importance-Aware Graph Neural Architecture Search for Cross-Domain Recommendationcited by 1
- 2025Auxiliary Gene Learning: Spatial Gene Expression Estimation by Auxiliary Gene Selectioncited by 1
- 2023FedL2P: Federated Learning to Personalizecited by 0
- 2023Disentangled Latent Spaces Facilitate Data-Driven Auxiliary Learningcited by 0
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2024Selecting the Best Sequential Transfer Path for Medical Image Segmentation with Limited Labeled Datacited by 0
- 2025Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learningcited by 0
- 2025Towards Fair and Robust Face Parsing for Generative AI: A Multi-Objective Approachcited by 0
- 2025Adversarial Attacks in Weight-Space Classifierscited by 0
- 2025Generating Auxiliary Tasks with Reinforcement Learningcited by 0
- 2025Learning Visual Affordance from Audiocited by 0
- 2026Feed m Birds with One Scone: Accelerating Multi-task Gradient Balancing via Bi-level Optimizationcited by 0
- 2026Feed m Birds with One Scone: Accelerating Multi-task Gradient Balancing via Bi-level Optimization8 Mar 2026
- 2026Non-Stationary Functional Bilevel Optimization21 Jan 2026
- 2026FGTBT: Frequency-Guided Task-Balancing Transformer for Unified Facial Landmark Detection19 Jan 2026
- 2025Learning Visual Affordance from Audio1 Dec 2025
- 2025Auxiliary Gene Learning: Spatial Gene Expression Estimation by Auxiliary Gene Selection23 Nov 2025
- 2025Generating Auxiliary Tasks with Reinforcement Learning27 Oct 2025
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspective21 Oct 2025
- 2025Behavior Importance-Aware Graph Neural Architecture Search for Cross-Domain Recommendation11 Mar 2025
- 2025Adversarial Attacks in Weight-Space Classifiers27 Feb 2025
- 2025Towards Fair and Robust Face Parsing for Generative AI: A Multi-Objective Approach6 Feb 2025
- 2025Adapting Foundation Models for Few-Shot Medical Image Segmentation: Actively and Sequentially3 Feb 2025
- 2025Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learning12 Jan 2025
33 more
- 2024Selecting the Best Sequential Transfer Path for Medical Image Segmentation with Limited Labeled Data9 Oct 2024
- 2024Learning Representation for Multitask learning through Self Supervised Auxiliary learning25 Sep 2024
- 2024Aux-NAS: Exploiting Auxiliary Labels with Negligibly Extra Inference Cost9 May 2024
- 2024Meta-Auxiliary Learning for Micro-Expression Recognition18 Apr 2024
- 2024Data-Driven Preference Sampling for Pareto Front Learning12 Apr 2024
- 2024Functional Bilevel Optimization for Machine Learning29 Mar 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
- 2024Robust Analysis of Multi-Task Learning Efficiency: New Benchmarks on Light-Weighed Backbones and Effective Measurement of Multi-Task Learning Challenges by Feature Disentanglement5 Feb 2024
- 2024Enhancing Molecular Property Prediction with Auxiliary Learning and Task-Specific Adaptation29 Jan 2024
- 2023Disentangled Latent Spaces Facilitate Data-Driven Auxiliary Learning13 Oct 2023
- 2023FedL2P: Federated Learning to Personalize3 Oct 2023
- 2023No Data Augmentation? Alternative Regularizations for Effective Training on Small Datasets4 Sep 2023
- 2023MELTR: Meta Loss Transformer for Learning to Fine-tune Video Foundation Models23 Mar 2023
- 2023Prismer: A Vision-Language Model with Multi-Task Experts4 Mar 2023
- 2023Unified Convergence Theory of Stochastic and Variance-Reduced Cubic Newton Methods23 Feb 2023
- 2023Improving Few-Shot Generalization by Exploring and Exploiting Auxiliary Data1 Feb 2023
- 2023Auxiliary Learning as an Asymmetric Bargaining Game31 Jan 2023
- 2023ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learning30 Jan 2023
- 2022Image Classification with Small Datasets: Overview and Benchmark23 Dec 2022
- 2022Context Label Learning: Improving Background Class Representations in Semantic Segmentation16 Dec 2022
- 2022AdaTask: A Task-aware Adaptive Learning Rate Approach to Multi-task Learning28 Nov 2022
- 2022Towards Optimization and Model Selection for Domain Generalization: A Mixup-guided Solution1 Sep 2022
- 2022Benchmarking Visual-Inertial Deep Multimodal Fusion for Relative Pose Regression and Odometry-aided Absolute Pose Regression1 Aug 2022
- 2022Optimization with Access to Auxiliary Information1 Jun 2022
- 2022AANG: Automating Auxiliary Learning27 May 2022
- 2022Adaptive Mixing of Auxiliary Losses in Supervised Learning7 Feb 2022
- 2022Auto-Lambda: Disentangling Dynamic Task Relationships7 Feb 2022
- 2022Multi-Task Learning as a Bargaining Game2 Feb 2022
- 2021Pareto Domain Adaptation8 Dec 2021
- 2021Auxiliary Learning for Self-Supervised Video Representation via Similarity-based Knowledge Distillation7 Dec 2021
- 2021Linear Speedup in Personalized Collaborative Learning10 Nov 2021
- 2021Meta-Learning to Improve Pre-Training2 Nov 2021
- 2021Should We Be Pre-training? An Argument for End-task Aware Training as an Alternative15 Sep 2021
- 2023Prismer: A Vision-Language Model with Multi-Task Expertscited by 28
- 2023Prismer: A Vision-Language Model with Multi-Task Experts4 Mar 2023
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2022AdaTask: A Task-aware Adaptive Learning Rate Approach to Multi-task Learningcited by 31
- 2023ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learningcited by 28
- 2021Meta-Learning to Improve Pre-Trainingcited by 27
- 2021Should We Be Pre-training? An Argument for End-task Aware Training as an Alternativecited by 26
- 2021Pareto Domain Adaptationcited by 20
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2023MELTR: Meta Loss Transformer for Learning to Fine-tune Video Foundation Modelscited by 15
- 2023Improving Few-Shot Generalization by Exploring and Exploiting Auxiliary Datacited by 13
- 2022AANG: Automating Auxiliary Learningcited by 12
- 2022Optimization with Access to Auxiliary Informationcited by 12
12 most cited of 42 All 42 (JSON)
- 2026Feed m Birds with One Scone: Accelerating Multi-task Gradient Balancing via Bi-level Optimization8 Mar 2026
- 2026Non-Stationary Functional Bilevel Optimization21 Jan 2026
- 2026FGTBT: Frequency-Guided Task-Balancing Transformer for Unified Facial Landmark Detection19 Jan 2026
- 2025Learning Visual Affordance from Audio1 Dec 2025
- 2025Auxiliary Gene Learning: Spatial Gene Expression Estimation by Auxiliary Gene Selection23 Nov 2025
- 2025Generating Auxiliary Tasks with Reinforcement Learning27 Oct 2025
- 2025NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel Perspective21 Oct 2025
- 2025Behavior Importance-Aware Graph Neural Architecture Search for Cross-Domain Recommendation11 Mar 2025
- 2025Adversarial Attacks in Weight-Space Classifiers27 Feb 2025
- 2025Towards Fair and Robust Face Parsing for Generative AI: A Multi-Objective Approach6 Feb 2025
- 2025Adapting Foundation Models for Few-Shot Medical Image Segmentation: Actively and Sequentially3 Feb 2025
- 2024Selecting the Best Sequential Transfer Path for Medical Image Segmentation with Limited Labeled Data9 Oct 2024
12 newest of 42 All 42 (JSON)