MTI-Net: Multi-Scale Task Interaction Networks for Multi-Task Learning
Not on the map. Why?
Abstract (arXiv)
In this paper, we argue about the importance of considering task interactions at multiple scales when distilling task information in a multi-task learning setup. In contrast to common belief, we show that tasks with high affinity at a certain scale are not guaranteed to retain this behaviour at other scales, and vice versa. We propose a novel architecture, namely MTI-Net, that builds upon this finding in three ways. First, it explicitly models task interactions at every scale via a multi-scale multi-modal distillation unit. Second, it propagates distilled task information from lower to higher scales via a feature propagation module. Third, it aggregates the refined task features from all scales via a feature aggregation unit to produce the final per-task predictions. Extensive experiments on two multi-task dense labeling datasets show that, unlike prior work, our multi-task model delivers on the full potential of multi-task learning, that is, smaller memory footprint, reduced number of calculations, and better performance w.r.t. single-task learning. The code is made publicly available: https://github.com/SimonVandenhende/Multi-Task-Learning-PyTorch.
Key facts. Cited by 157 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 18; 2022: 28; 2023: 37; 2024: 32; 2025: 24; 2026: 18.
157 papers.
Sort:
By area:
- 2021Geometry Uncertainty Projection Network for Monocular 3D Object Detectioncited by 155
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-designcited by 97
- 2022InvPT: Inverted Pyramid Multi-task Transformer for Dense Scene Understandingcited by 89
- 2021Domain Adaptive Semantic Segmentation with Self-Supervised Depth Estimationcited by 80
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2021Exploring Relational Context for Multi-Task Dense Predictioncited by 71
- 2021InverseForm: A Loss Function for Structured Boundary-Aware Segmentationcited by 65
- 2023VMT-Adapter: Parameter-Efficient Transfer Learning for Multi-Task Dense Scene Understandingcited by 62
- 2024Multi-Task Dense Prediction via Mixture of Low-Rank Expertscited by 58
- 2021Fine-grained Semantics-aware Representation Enhancement for Self-supervised Monocular Depth Estimationcited by 55
- 2022Discovering Object Masks with Transformers for Unsupervised Semantic Segmentationcited by 54
- 2022Polyhistor: Parameter-Efficient Multi-Task Adaptation for Dense Vision Taskscited by 52
38 more
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Expertscited by 45
- 2024MTLoRA: A Low-Rank Adaptation Approach for Efficient Multi-Task Learningcited by 44
- 2021Learning Multiple Dense Prediction Tasks from Partially Annotated Datacited by 42
- 2022Multi-Task Learning with Multi-Query Transformer for Dense Predictioncited by 34
- 2021Learning to Relate Depth and Semantics for Unsupervised Domain Adaptationcited by 32
- 2022Efficient Multi-Task RGB-D Scene Analysis for Indoor Environmentscited by 32
- 2021Dual Transfer Learning for Event-based End-task Prediction via Pluggable Event to Image Translationcited by 31
- 2021Instance-wise Occlusion and Depth Orders in Natural Scenescited by 31
- 2023SA-BEV: Generating Semantic-Aware Bird's-Eye-View Feature for Multi-view 3D Object Detectioncited by 30
- 2022MGNet: Monocular Geometric Scene Understanding for Autonomous Drivingcited by 27
- 2023DeMT: Deformable Mixer Transformer for Multi-Task Learning of Dense Predictioncited by 27
- 2023Dual-Balancing for Multi-Task Learningcited by 27
- 2022DenseMTL: Cross-task Attention Mechanism for Dense Multi-task Learningcited by 25
- 2023Edge-MoE: Memory-Efficient Multi-Task Vision Transformer Architecture with Task-level Sparsity via Mixture-of-Expertscited by 23
- 2021Semi-supervised Multi-task Learning for Semantics and Depthcited by 22
- 2023InvPT++: Inverted Pyramid Multi-Task Transformer for Visual Scene Understandingcited by 22
- 2024MTMamba: Enhancing Multi-Task Dense Scene Understanding by Mamba-Based Decoderscited by 22
- 2023PanopticPartFormer++: A Unified and Decoupled View for Panoptic Part Segmentationcited by 19
- 2023Mitigating Task Interference in Multi-Task Learning via Explicit Task Routing with Non-Learnable Primitivescited by 18
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Datacited by 18
- 2024Spider: A Unified Framework for Context-dependent Concept Segmentationcited by 18
- 2023FedHCA2: Towards Hetero-Client Federated Multi-Task Learningcited by 17
- 2025Selective Task Group Updates for Multi-Task Optimizationcited by 15
- 2022PromptonomyViT: Multi-Task Prompt Learning Improves Video Transformers using Synthetic Scene Datacited by 14
- 2023Vision Transformer Adapters for Generalizable Multitask Learningcited by 14
- 2023Prompt Guided Transformer for Multi-Task Dense Predictioncited by 13
- 2024SwinMTL: A Shared Architecture for Simultaneous Depth Estimation and Semantic Segmentation from Monocular Camera Imagescited by 13
- 2024MICDrop: Masking Image and Depth Features via Complementary Dropout for Domain-Adaptive Semantic Segmentationcited by 13
- 2023TSP-Transformer: Task-Specific Prompts Boosted Transformer for Holistic Scene Understandingcited by 12
- 2023BridgeNet: Comprehensive and Effective Feature Interactions via Bridge Feature for Multi-task Dense Predictionscited by 12
- 2021Channel Exchanging Networks for Multimodal and Multitask Dense Image Predictioncited by 11
- 2024StainDiffuser: MultiTask Dual Diffusion Model for Virtual Stainingcited by 11
- 2024Quantifying Task Priority for Multi-Task Optimizationcited by 10
- 2024MTMamba++: Enhancing Multi-Task Dense Scene Understanding via Mamba-Based Decoderscited by 10
- 2021Spatio-Temporal Multi-Task Learning Transformer for Joint Moving Object Detection and Segmentationcited by 9
- 2023Performance-aware Approximation of Global Channel Pruning for Multitask CNNscited by 9
- 2023Transferring to Real-World Layouts: A Depth-aware Framework for Scene Adaptationcited by 9
- 2024A Backpack Full of Skills: Egocentric Video Understanding with Diverse Task Perspectivescited by 9
50 most cited of 157 All 157 (JSON)
- 2026An Exploratory Study of Frequency-Aware Task Weighting for YOLOv8-Based Unified Driving Perception31 Aug 2026
- 2026Task-State Adaptation with Prototype Memory for Multi-Task Dense Prediction28 Aug 2026
- 2026EMAN: Optimization-Driven Capacity Growth through Path Emergence in Multi-Task Learning5 Aug 2026
- 2026BTI-Net: Bidirectional Decoder-Level Task Interaction via Uncertainty-Aware Gating for Multi-Task Medical Image Analysis27 Jun 2026
- 2026Instance-Aware Knowledge Distillation for Semi-Supervised Learning of an On-Board Multi-Task Dense Prediction Model for Collision Avoidance System15 Jun 2026
- 2026Mono-Hydra++: Real-Time Monocular Scene Graph Construction with Multi-Task Learning for 3D Indoor Mapping17 May 2026
- 2026B3-Net: Controlled Posterior Bridge Learning for Multi-Task Dense Prediction7 May 2026
- 2026Radar-Camera BEV Multi-Task Learning with Cross-Task Attention Bridge for Joint 3D Detection and Segmentation14 Apr 2026
- 2026MTLSI-Net: A Linear Semantic Interaction Network for Parameter-Efficient Multi-Task Dense Prediction2 Apr 2026
- 2026M2H-MX: Multi-Task Semantic and Geometric Perception for Real-Time Monocular 3D Scene Graph Construction31 Mar 2026
- 2026Frequency Switching Mechanism for Parameter-E!cient Multi-Task Learning22 Mar 2026
- 2026Adaptive Bidirectional Task Interaction for Joint Segmentation and Classification of Breast Ultrasound1 Mar 2026
38 more
- 2026MTPano: Multi-Task Panoramic Scene Understanding via Label-Free Integration of Dense Prediction Priors5 Feb 2026
- 2026Multi-Task Learning for Robot Perception with Imbalanced Data2 Feb 2026
- 2026UV-M3TL: A Unified and Versatile Multimodal Multi-Task Learning Framework for Assistive Driving Perception2 Feb 2026
- 2026RobuMTL: Enhancing Multi-Task Learning Robustness Against Weather Conditions16 Jan 2026
- 2026FUME: Fused Unified Multi-Gas Emission Network for Livestock Rumen Acidosis Detection13 Jan 2026
- 2026FAR-AMTN: Attention Multi-Task Network for Face Attribute Recognition4 Jan 2026
- 2025Simplifying Multi-Task Architectures Through Task-Specific Normalization23 Dec 2025
- 2025NexusFlow: Unifying Disparate Tasks under Partial Supervision via Invertible Flow Networks6 Dec 2025
- 20253D-Aware Multi-Task Learning with Cross-View Correlations for Dense Scene Understanding25 Nov 2025
- 2025Parameter Aware Mamba Model for Multi-task Dense Prediction18 Nov 2025
- 2025UniSER: A Foundation Model for Unified Soft Effects Removal18 Nov 2025
- 2025Seabed-Net: A multi-task network for joint bathymetry estimation and seabed classification from remote sensing imagery in shallow waters22 Oct 2025
- 2025M2H: Multi-Task Learning with Efficient Window-Based Cross-Task Attention for Monocular Spatial Perception20 Oct 2025
- 2025Revisit the Imbalance Optimization in Multi-task Learning: An Experimental Analysis28 Sep 2025
- 2025Shared Neural Space: Unified Precomputed Feature Encoding for Multi-Task and Cross Domain Vision24 Sep 2025
- 2025Parameter-Efficient Multi-Task Learning via Progressive Task-Specific Adaptation23 Sep 2025
- 2025SPGen: Spherical Projection as Consistent and Flexible Representation for Single Image 3D Shape Generation16 Sep 2025
- 2025Enhancing Mamba Decoder with Bidirectional Interaction in Multi-Task Dense Prediction28 Aug 2025
- 2025Multi-Task Dense Prediction Fine-Tuning with Mixture of Fine-Grained Experts25 Jul 2025
- 2025Synchronizing Task Behavior: Aligning Multiple Tasks during Test-Time Training10 Jul 2025
- 2025Resolving Token-Space Gradient Conflicts: Token Space Manipulation for Transformer-Based Multi-Task Learning10 Jul 2025
- 2025Towards Unified Modeling in Federated Multi-Task Learning via Subspace Decoupling30 May 2025
- 2025Occlusion Boundary and Depth: Mutual Enhancement via Multi-Task Learning27 May 2025
- 2025MultiTaskVIF: Segmentation-oriented visible and infrared image fusion via multi-task learning10 May 2025
- 2025Co-SemDepth: Fast Joint Semantic Segmentation and Depth Estimation on Aerial Images23 Mar 2025
- 2025Selective Task Group Updates for Multi-Task Optimization17 Feb 2025
- 2025DiTASK: Multi-Task Fine-Tuning with Diffeomorphic Transformations9 Feb 2025
- 2025Hier-EgoPack: Hierarchical Egocentric Video Understanding with Diverse Task Perspectives4 Feb 2025
- 2025Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learning12 Jan 2025
- 2025TADFormer : Task-Adaptive Dynamic Transformer for Efficient Multi-Task Learning8 Jan 2025
- 2024Unprejudiced Training Auxiliary Tasks Makes Primary Better: A Multi-Task Learning Perspective27 Dec 2024
- 2024Optimizing Dense Visual Predictions Through Multi-Task Coherence and Prioritization4 Dec 2024
- 2024Multi-Task Label Discovery via Hierarchical Task Tokens for Partially Annotated Dense Predictions27 Nov 2024
- 2024Diff-2-in-1: Bridging Generation and Dense Perception with Diffusion Models7 Nov 2024
- 2024Swiss Army Knife: Synergizing Biases in Knowledge from Vision Foundation Models for Multi-Task Learning18 Oct 2024
- 2024Stylistic Multi-Task Analysis of Ukiyo-e Woodblock Prints16 Oct 2024
- 2024SGW-based Multi-Task Learning in Vision Tasks3 Oct 2024
- 2024MICDrop: Masking Image and Depth Features via Complementary Dropout for Domain-Adaptive Semantic Segmentation29 Aug 2024
50 newest of 157 All 157 (JSON)
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-designcited by 97
- 2024Multi-Task Dense Prediction via Mixture of Low-Rank Expertscited by 58
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Expertscited by 45
- 2023Edge-MoE: Memory-Efficient Multi-Task Vision Transformer Architecture with Task-level Sparsity via Mixture-of-Expertscited by 23
- 2025Multi-Task Dense Prediction Fine-Tuning with Mixture of Fine-Grained Expertscited by 1
- 2025Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learningcited by 0
- 2025Multi-Task Dense Prediction Fine-Tuning with Mixture of Fine-Grained Experts25 Jul 2025
- 2025Transforming Vision Transformer: Towards Efficient Multi-Task Asynchronous Learning12 Jan 2025
- 2024Multi-Task Dense Prediction via Mixture of Low-Rank Experts26 Mar 2024
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Experts28 Jul 2023
- 2023Edge-MoE: Memory-Efficient Multi-Task Vision Transformer Architecture with Task-level Sparsity via Mixture-of-Experts30 May 2023
- 2022M3ViT: Mixture-of-Experts Vision Transformer for Efficient Multi-task Learning with Model-Accelerator Co-design26 Oct 2022
- 2024MTMamba: Enhancing Multi-Task Dense Scene Understanding by Mamba-Based Decoderscited by 22
- 2024MTMamba++: Enhancing Multi-Task Dense Scene Understanding via Mamba-Based Decoderscited by 10
- 2025Enhancing Mamba Decoder with Bidirectional Interaction in Multi-Task Dense Predictioncited by 1
- 2025Parameter Aware Mamba Model for Multi-task Dense Predictioncited by 0
- 2026B3-Net: Controlled Posterior Bridge Learning for Multi-Task Dense Predictioncited by 0
- 2026B3-Net: Controlled Posterior Bridge Learning for Multi-Task Dense Prediction7 May 2026
- 2025Parameter Aware Mamba Model for Multi-task Dense Prediction18 Nov 2025
- 2025Enhancing Mamba Decoder with Bidirectional Interaction in Multi-Task Dense Prediction28 Aug 2025
- 2024MTMamba++: Enhancing Multi-Task Dense Scene Understanding via Mamba-Based Decoders27 Aug 2024
- 2024MTMamba: Enhancing Multi-Task Dense Scene Understanding by Mamba-Based Decoders2 Jul 2024
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Datacited by 18
- 2025SPGen: Spherical Projection as Consistent and Flexible Representation for Single Image 3D Shape Generationcited by 5
- 2024Diff-2-in-1: Bridging Generation and Dense Perception with Diffusion Modelscited by 3
- 2025SPGen: Spherical Projection as Consistent and Flexible Representation for Single Image 3D Shape Generation16 Sep 2025
- 2024Diff-2-in-1: Bridging Generation and Dense Perception with Diffusion Models7 Nov 2024
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Data22 Mar 2024
- 2021Geometry Uncertainty Projection Network for Monocular 3D Object Detectioncited by 155
- 2022InvPT: Inverted Pyramid Multi-task Transformer for Dense Scene Understandingcited by 89
- 2021Domain Adaptive Semantic Segmentation with Self-Supervised Depth Estimationcited by 80
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2021Exploring Relational Context for Multi-Task Dense Predictioncited by 71
- 2021InverseForm: A Loss Function for Structured Boundary-Aware Segmentationcited by 65
- 2023VMT-Adapter: Parameter-Efficient Transfer Learning for Multi-Task Dense Scene Understandingcited by 62
- 2021Fine-grained Semantics-aware Representation Enhancement for Self-supervised Monocular Depth Estimationcited by 55
- 2022Discovering Object Masks with Transformers for Unsupervised Semantic Segmentationcited by 54
- 2022Polyhistor: Parameter-Efficient Multi-Task Adaptation for Dense Vision Taskscited by 52
- 2024MTLoRA: A Low-Rank Adaptation Approach for Efficient Multi-Task Learningcited by 44
- 2021Learning Multiple Dense Prediction Tasks from Partially Annotated Datacited by 42
12 most cited of 143 All 143 (JSON)
- 2026An Exploratory Study of Frequency-Aware Task Weighting for YOLOv8-Based Unified Driving Perception31 Aug 2026
- 2026Task-State Adaptation with Prototype Memory for Multi-Task Dense Prediction28 Aug 2026
- 2026EMAN: Optimization-Driven Capacity Growth through Path Emergence in Multi-Task Learning5 Aug 2026
- 2026BTI-Net: Bidirectional Decoder-Level Task Interaction via Uncertainty-Aware Gating for Multi-Task Medical Image Analysis27 Jun 2026
- 2026Instance-Aware Knowledge Distillation for Semi-Supervised Learning of an On-Board Multi-Task Dense Prediction Model for Collision Avoidance System15 Jun 2026
- 2026Mono-Hydra++: Real-Time Monocular Scene Graph Construction with Multi-Task Learning for 3D Indoor Mapping17 May 2026
- 2026Radar-Camera BEV Multi-Task Learning with Cross-Task Attention Bridge for Joint 3D Detection and Segmentation14 Apr 2026
- 2026MTLSI-Net: A Linear Semantic Interaction Network for Parameter-Efficient Multi-Task Dense Prediction2 Apr 2026
- 2026M2H-MX: Multi-Task Semantic and Geometric Perception for Real-Time Monocular 3D Scene Graph Construction31 Mar 2026
- 2026Frequency Switching Mechanism for Parameter-E!cient Multi-Task Learning22 Mar 2026
- 2026Adaptive Bidirectional Task Interaction for Joint Segmentation and Classification of Breast Ultrasound1 Mar 2026
- 2026MTPano: Multi-Task Panoramic Scene Understanding via Label-Free Integration of Dense Prediction Priors5 Feb 2026
12 newest of 143 All 143 (JSON)