Composite Learning for Robust and Effective Dense Predictions
Not on the map. Why?
Abstract (arXiv)
Multi-task learning promises better model generalization on a target task by jointly optimizing it with an auxiliary task. However, the current practice requires additional labeling efforts for the auxiliary task, while not guaranteeing better model performance. In this paper, we find that jointly training a dense prediction (target) task with a self-supervised (auxiliary) task can consistently improve the performance of the target task, while eliminating the need for labeling auxiliary tasks. We refer to this joint training as Composite Learning (CompL). Experiments of CompL on monocular depth estimation, semantic segmentation, and boundary detection show consistent performance improvements in fully and partially labeled datasets. Further analysis on depth estimation reveals that joint training with self-supervision outperforms most labeled auxiliary tasks. We also find that CompL can improve model robustness when the models are evaluated in new domains. These results demonstrate the benefits of self-supervision as an auxiliary task, and establish the design of novel task-specific self-supervised methods as a new axis of investigation for future multi-task learning research.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 58 of its 72 references that are in the map. First posted on arXiv 13 Oct 2022.
Builds on
58 of 72 references.
References · 58
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Layer Normalizationcited by 5,579
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,239
- 2018Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentationcited by 3,210
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
38 more
- 2020Supervised Contrastive Learningcited by 2,803
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,282
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,586
- 2018Group Normalizationcited by 1,514
- 2019Theoretically Principled Trade-off between Robustness and Accuracycited by 1,510
- 2015SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentationcited by 1,483
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,430
- 2019Deep High-Resolution Representation Learning for Visual Recognitioncited by 1,417
- 2018BDD100K: A Diverse Driving Dataset for Heterogeneous Multitask Learningcited by 1,282
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,187
- 2015Multi-Scale Context Aggregation by Dilated Convolutionscited by 1,128
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,062
- 2020What Makes for Good Views for Contrastive Learning?cited by 843
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
- 2018Deep Ordinal Regression Network for Monocular Depth Estimationcited by 721
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 672
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
- 2015Holistically-Nested Edge Detectioncited by 590
- 2020Dense Contrastive Learning for Self-Supervised Visual Pre-Trainingcited by 512
- 2019Using Self-Supervised Learning Can Improve Model Robustness and Uncertaintycited by 511
- 2018Rethinking ImageNet Pre-trainingcited by 376
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
- 2019S4L: Self-Supervised Semi-Supervised Learningcited by 335
- 2017Dilated Residual Networkscited by 293
- 2021Exploring Cross-Image Pixel Contrast for Semantic Segmentationcited by 242
- 2020Quasi-Dense Similarity Learning for Multiple Object Trackingcited by 220
- 2016UberNet: Training a `Universal' Convolutional Neural Network for Low-, Mid-, and High-Level Vision using Diverse Datasets and Limited Memorycited by 200
- 2019Attentive Single-Tasking of Multiple Taskscited by 148
- 2020Anomaly Detection in Video via Self-Supervised and Multi-Task Learningcited by 147
- 2019Boosting Few-Shot Visual Learning with Self-Supervisioncited by 140
- 2020Semantically-Guided Representation Learning for Self-Supervised Monocular Depthcited by 134
- 2015Instance-aware Semantic Segmentation via Multi-task Network Cascadescited by 120
- 2016HyperFace: A Deep Multi-task Learning Framework for Face Detection, Landmark Localization, Pose Estimation, and Gender Recognitioncited by 103
- 2019Self-Supervised Generalisation with Meta Auxiliary Learningcited by 88
- 2021Multi-Task Self-Training for Learning General Representationscited by 85
- 2020How Useful is Self-Supervised Pretraining for Visual Tasks?cited by 80
- 2021Exploring Relational Context for Multi-Task Dense Predictioncited by 71
- 2020Three Ways to Improve Semantic Segmentation with Self-Supervised Depth Estimationcited by 66
50 most cited of 58 All 58 (JSON)
Built on by
Citing papers per year: 2023: 2; 2024: 2.
4 papers.
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Expertscited by 45
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Datacited by 18
- 2023Video Task Decathlon: Unifying Image and Video Tasks in Autonomous Drivingcited by 8
- 2024Unprejudiced Training Auxiliary Tasks Makes Primary Better: A Multi-Task Learning Perspectivecited by 1
- 2024Unprejudiced Training Auxiliary Tasks Makes Primary Better: A Multi-Task Learning Perspective27 Dec 2024
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Data22 Mar 2024
- 2023Video Task Decathlon: Unifying Image and Video Tasks in Autonomous Driving8 Sep 2023
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Experts28 Jul 2023