Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Simplifying Multi-Task Architectures Through Task-Specific Normalization

About

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$\sigma$BN), a lightweight mechanism that enables tasks to softly allocate network capacity while fully sharing feature extractors. TS$\sigma$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.

Mihai Suteu, Ovidiu Serban• 2025

Related benchmarks

TaskDatasetResultRank
Multi-task LearningPascal Context
mIoU (Semantic Segmentation)77.12
47
Multi-task LearningNYU V2
mIoU53.78
19
Multi-task Learning (Segmentation, Part Segmentation, Disparity)Cityscapes
Semantic Segmentation mIoU70.17
16
Multi-task LearningNYU LibMTL v2 (test)
Segmentation Score53.78
14
Multi-task Learning (Semantic Segmentation, Surface Normals, Depth Estimation)NYU V2--
7
Multi-task Learning (40 facial attribute classification)CelebA
F1 Score0.6945
6
Showing 6 of 6 rows

Other info

Follow for update