Simplifying Multi-Task Architectures Through Task-Specific Normalization
Not on the map. Why?
Abstract (arXiv)
Multi-task learning (MTL) aims to leverage shared knowledge across tasks to improve generalization and parameter efficiency, yet balancing resources and mitigating interference remain open challenges. Architectural solutions often introduce elaborate task-specific modules or routing schemes, increasing complexity and overhead. In this work, we show that normalization layers alone are sufficient to address many of these challenges. Simply replacing shared normalization with task-specific variants already yields competitive performance, questioning the need for complex designs. Building on this insight, we propose Task-Specific Sigmoid Batch Normalization (TSσBN), a lightweight mechanism that enables tasks to softly allocate network capacity while fully sharing feature extractors. TSσBN improves stability across CNNs and Transformers, matching or exceeding performance on NYUv2, Cityscapes, CelebA, and PascalContext, while remaining highly parameter-efficient. Moreover, its learned gates provide a natural framework for analyzing MTL dynamics, offering interpretable insights into capacity allocation, filter specialization, and task relationships. Our findings suggest that complex MTL architectures may be unnecessary and that task-specific normalization offers a simple, interpretable, and efficient alternative.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 75 of its 87 references that are in the map. First posted on arXiv 23 Dec 2025.
Builds on
75 of 87 references.
References · 75
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Layer Normalizationcited by 5,669
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2017Rethinking Atrous Convolution for Semantic Image Segmentationcited by 2,161
- 2022Visual Prompt Tuningcited by 1,681
- 2015SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentationcited by 1,487
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,358
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2021BitFit: Simple Parameter-efficient Fine-tuning for Transformer-based Masked Language-modelscited by 1,245
38 more
- 2021Towards a Unified View of Parameter-Efficient Transfer Learningcited by 872
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 821
- 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 676
- 2017Learning multiple visual domains with residual adapterscited by 557
- 2018End-to-End Multi-Task Learning with Attentioncited by 534
- 2016Cross-stitch Networks for Multi-task Learningcited by 467
- 2021Compacter: Efficient Low-Rank Hypercomplex Adapter Layerscited by 460
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 459
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 375
- 2018How Does Batch Normalization Help Optimization?cited by 357
- 2015Understanding Neural Networks Through Deep Visualizationcited by 334
- 2016Revisiting Batch Normalization For Practical Domain Adaptationcited by 288
- 2021Parameter-efficient Multi-task Fine-tuning for Transformers via Shared Hypernetworkscited by 284
- 2014Detect What You Can: Detecting and Representing Objects using Holistic Models and Body Partscited by 232
- 2018PAD-Net: Multi-Tasks Guided Prediction-and-Distillation Network for Simultaneous Depth Estimation and Scene Parsingcited by 224
- 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 222
- 2022Multi-Task Learning as a Bargaining Gamecited by 215
- 2020Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropoutcited by 201
- 2019Domain-Specific Batch Normalization for Unsupervised Domain Adaptationcited by 197
- 2020Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Modelscited by 169
- 2019AdaShare: Learning What To Share For Efficient Deep Multi-Task Learningcited by 160
- 2018Understanding Batch Normalizationcited by 159
- 2020MTI-Net: Multi-Scale Task Interaction Networks for Multi-Task Learningcited by 157
- 2019Attentive Single-Tasking of Multiple Taskscited by 148
- 2020A Survey on Negative Transfercited by 122
- 2020Learning to Branch for Multi-Task Learningcited by 115
- 2021DSelect-k: Differentiable Selection in the Mixture of Experts with Applications to Multi-Task Learningcited by 115
- 2020Training BatchNorm and Only BatchNorm: On the Expressive Power of Random Features in CNNscited by 104
- 2021Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learningcited by 101
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-designcited by 98
- 2020Normalization Techniques in Training DNNs: Methodology, Analysis and Applicationcited by 90
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 78
- 2019Branched Multi-Task Networks: Deciding What Layers To Sharecited by 76
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2023Tuning LayerNorm in Attention: Towards Efficient Multi-Modal LLM Finetuningcited by 64
- 2024Multi-Task Dense Prediction via Mixture of Low-Rank Expertscited by 58
- 2018Towards Understanding Regularization in Batch Normalizationcited by 56
50 most cited of 75 All 75 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.