Addressing the Loss-Metric Mismatch with Adaptive Loss Alignment
Not on the map. Why?
Abstract (arXiv)
In most machine learning training paradigms a fixed, often handcrafted, loss function is assumed to be a good proxy for an underlying evaluation metric. In this work we assess this assumption by meta-learning an adaptive loss function to directly optimize the evaluation metric. We propose a sample efficient reinforcement learning approach for adapting the loss dynamically during training. We empirically show how this formulation improves performance by simultaneously optimizing the evaluation metric and smoothing the loss landscape. We verify our method in metric learning and classification scenarios, showing considerable improvements over the state-of-the-art on a diverse set of tasks. Importantly, our method is applicable to a wide range of loss functions and evaluation metrics. Furthermore, the learned policies are transferable across tasks and data, demonstrating the versatility of the method.
Key facts. Cited by 33 papers in the map (data through 14 Oct 2026). First posted on arXiv 15 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 8; 2023: 7; 2024: 1; 2025: 4; 2026: 2.
33 papers.
Sort:
By area:
- 2021FASA: Feature Augmentation and Sampling Adaptation for Long-Tailed Instance Segmentationcited by 76
- 2021Meta-learning PINN loss functionscited by 51
- 2022Customizing ML Predictions for Online Algorithmscited by 21
- 2022The worst of both worlds: A comparative analysis of errors in learning from data in psychology and machine learningcited by 17
- 2021Searching for Robustness: Loss Learning for Noisy Classification Taskscited by 13
- 2022Scalable Decision-Focused Learning in Restless Multi-Armed Bandits with Application to Maternal and Child Healthcited by 12
- 2022FORML: Learning to Reweight Data for Fairnesscited by 12
- 2021Score-oriented loss (SOL) functionscited by 11
- 2021Training Over-parameterized Models with Non-decomposable Objectivescited by 11
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metricscited by 9
- 2021CCMN: A General Framework for Learning with Class-Conditional Multi-Label Noisecited by 9
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Taskscited by 9
21 more
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learningcited by 9
- 2022Reliable Decision from Multiple Subtasks through Threshold Optimization: Content Moderation in the Wildcited by 7
- 2021Optimizing Black-box Metrics with Iterative Example Weightingcited by 6
- 2023Cost-Sensitive Self-Training for Optimizing Non-Decomposable Metricscited by 4
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learningcited by 4
- 2021Learning MDPs from Features: Predict-Then-Optimize for Sequential Decision Problems by Reinforcement Learningcited by 3
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awarenesscited by 3
- 2023A comprehensive theoretical framework for the optimization of neural networks classification performance with respect to weighted metricscited by 3
- 2021Implicit Acceleration and Feature Learning in Infinitely Wide Neural Networks with Bottleneckscited by 2
- 2023OWAdapt: An adaptive loss function for deep learning using OWA operatorscited by 2
- 2023L2T-DLN: Learning to Teach with Dynamic Loss Networkcited by 2
- 2025DaMo: Data Mixing Optimizer in Fine-tuning Multimodal LLMs for Mobile Phone Agentscited by 2
- 2022Reinforcement Teachingcited by 1
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learningcited by 1
- 2025Aligning the Evaluation of Probabilistic Predictions with Downstream Valuecited by 1
- 2022Implicit Training of Energy Model for Structure Predictioncited by 0
- 2023Label Budget Allocation in Multi-Task Learningcited by 0
- 2025Enhancing Online Reinforcement Learning with Meta-Learned Objective from Offline Datacited by 0
- 2025An Efficient Plugin Method for Metric Optimization of Black-Box Modelscited by 0
- 2026Don't stop me now: How Validation Criteria Affect Checkpoint Selection and Early Stoppingcited by 0
- 2026Distribution-Free Pretraining of Classification Losses via Evolutionary Dynamicscited by 0
- 2026Distribution-Free Pretraining of Classification Losses via Evolutionary Dynamics5 May 2026
- 2026Don't stop me now: How Validation Criteria Affect Checkpoint Selection and Early Stopping25 Feb 2026
- 2025DaMo: Data Mixing Optimizer in Fine-tuning Multimodal LLMs for Mobile Phone Agents22 Oct 2025
- 2025Aligning the Evaluation of Probabilistic Predictions with Downstream Value25 Aug 2025
- 2025An Efficient Plugin Method for Metric Optimization of Black-Box Models3 Mar 2025
- 2025Enhancing Online Reinforcement Learning with Meta-Learned Objective from Offline Data13 Jan 2025
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learning1 Mar 2024
- 2023L2T-DLN: Learning to Teach with Dynamic Loss Network30 Oct 2023
- 2023Label Budget Allocation in Multi-Task Learning24 Aug 2023
- 2023OWAdapt: An adaptive loss function for deep learning using OWA operators30 May 2023
- 2023A comprehensive theoretical framework for the optimization of neural networks classification performance with respect to weighted metrics22 May 2023
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
21 more
- 2023Cost-Sensitive Self-Training for Optimizing Non-Decomposable Metrics28 Apr 2023
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awareness18 Jan 2023
- 2022Implicit Training of Energy Model for Structure Prediction21 Nov 2022
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learning19 Sep 2022
- 2022Reliable Decision from Multiple Subtasks through Threshold Optimization: Content Moderation in the Wild16 Aug 2022
- 2022Customizing ML Predictions for Online Algorithms18 May 2022
- 2022Reinforcement Teaching25 Apr 2022
- 2022The worst of both worlds: A comparative analysis of errors in learning from data in psychology and machine learning12 Mar 2022
- 2022FORML: Learning to Reweight Data for Fairness3 Feb 2022
- 2022Scalable Decision-Focused Learning in Restless Multi-Armed Bandits with Application to Maternal and Child Health2 Feb 2022
- 2021Meta-learning PINN loss functions12 Jul 2021
- 2021Training Over-parameterized Models with Non-decomposable Objectives9 Jul 2021
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Tasks6 Jul 2021
- 2021Implicit Acceleration and Feature Learning in Infinitely Wide Neural Networks with Bottlenecks1 Jul 2021
- 2021Learning MDPs from Features: Predict-Then-Optimize for Sequential Decision Problems by Reinforcement Learning6 Jun 2021
- 2021CCMN: A General Framework for Learning with Class-Conditional Multi-Label Noise16 May 2021
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metrics21 Apr 2021
- 2021Score-oriented loss (SOL) functions29 Mar 2021
- 2021Searching for Robustness: Loss Learning for Noisy Classification Tasks27 Feb 2021
- 2021FASA: Feature Augmentation and Sampling Adaptation for Long-Tailed Instance Segmentation25 Feb 2021
- 2021Optimizing Black-box Metrics with Iterative Example Weighting18 Feb 2021
- 2021FASA: Feature Augmentation and Sampling Adaptation for Long-Tailed Instance Segmentationcited by 76
- 2021Meta-learning PINN loss functionscited by 51
- 2022Customizing ML Predictions for Online Algorithmscited by 21
- 2022The worst of both worlds: A comparative analysis of errors in learning from data in psychology and machine learningcited by 17
- 2021Searching for Robustness: Loss Learning for Noisy Classification Taskscited by 13
- 2022Scalable Decision-Focused Learning in Restless Multi-Armed Bandits with Application to Maternal and Child Healthcited by 12
- 2022FORML: Learning to Reweight Data for Fairnesscited by 12
- 2021Score-oriented loss (SOL) functionscited by 11
- 2021Training Over-parameterized Models with Non-decomposable Objectivescited by 11
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metricscited by 9
- 2021CCMN: A General Framework for Learning with Class-Conditional Multi-Label Noisecited by 9
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Taskscited by 9
12 most cited of 30 All 30 (JSON)
- 2026Distribution-Free Pretraining of Classification Losses via Evolutionary Dynamics5 May 2026
- 2026Don't stop me now: How Validation Criteria Affect Checkpoint Selection and Early Stopping25 Feb 2026
- 2025Aligning the Evaluation of Probabilistic Predictions with Downstream Value25 Aug 2025
- 2025An Efficient Plugin Method for Metric Optimization of Black-Box Models3 Mar 2025
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learning1 Mar 2024
- 2023L2T-DLN: Learning to Teach with Dynamic Loss Network30 Oct 2023
- 2023Label Budget Allocation in Multi-Task Learning24 Aug 2023
- 2023OWAdapt: An adaptive loss function for deep learning using OWA operators30 May 2023
- 2023A comprehensive theoretical framework for the optimization of neural networks classification performance with respect to weighted metrics22 May 2023
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
- 2023Cost-Sensitive Self-Training for Optimizing Non-Decomposable Metrics28 Apr 2023
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awareness18 Jan 2023
12 newest of 30 All 30 (JSON)