arXiv · 2009.01717
Multi-Loss Weighting with Coefficient of Variations
Abstract
Many interesting tasks in machine learning and computer vision are learned by optimising an objective function defined as a weighted linear combination of multiple losses. The final performance is sensitive to choosing the correct (relative) weights for these losses. Finding a good set of weights is often done by adopting them into the set of hyper-parameters, which are set using an extensive grid search. This is computationally expensive. In this paper, we propose a weighting scheme based on the coefficient of variations and set the weights based on properties observed while training the model. The proposed method incorporates a measure of uncertainty to balance the losses, and as a result the loss weights evolve during training without requiring another (learning based) optimisation. In contrast to many loss weighting methods in literature, we focus on single-task multi-loss problems, such as monocular depth estimation and semantic segmentation, and show that multi-task approaches for loss weighting do not work on those single-tasks. The validity of the approach is shown empirically for depth estimation and semantic segmentation on multiple datasets.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Rick Groenendijk, Sezer Karaoglu, Theo Gevers, Thomas Mensink. 2020-11-10. Multi-Loss Weighting with Coefficient of Variations. https://arxiv.org/abs/2009.01717
Cite the original work for its findings. Save a collection to share your selection of sources.