Dynamic Neural Network for Multi-Task Learning Searching across Diverse Network Topologies
Not on the map. Why?
Abstract (arXiv)
In this paper, we present a new MTL framework that searches for structures optimized for multiple tasks with diverse graph topologies and shares features among tasks. We design a restricted DAG-based central network with read-in/read-out layers to build topologically diverse task-adaptive structures while limiting search space and time. We search for a single optimized network that serves as multiple task adaptive sub-networks using our three-stage training process. To make the network compact and discretized, we propose a flow-based reduction algorithm and a squeeze loss used in the training process. We evaluate our optimized network on various public MTL datasets and show ours achieves state-of-the-art performance. An extensive ablation study experimentally validates the effectiveness of the sub-module and schemes in our framework.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 32 of its 42 references that are in the map. First posted on arXiv 13 Mar 2023.
Builds on
32 of 42 references.
References · 32
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,444
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
- 2018DARTS: Differentiable Architecture Searchcited by 1,798
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,499
- 2017Learning Transferable Architectures for Scalable Image Recognitioncited by 1,101
- 2018Efficient Neural Architecture Search via Parameter Sharingcited by 837
- 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 803
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 673
- 2018End-to-End Multi-Task Learning with Attentioncited by 534
20 more
- 2016Cross-stitch Networks for Multi-task Learningcited by 467
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 375
- 2017PathNet: Evolution Channels Gradient Descent in Super Neural Networkscited by 364
- 2017Large-Scale Evolution of Image Classifierscited by 348
- 2018SNAS: Stochastic Neural Architecture Searchcited by 289
- 2016Designing Neural Network Architectures using Reinforcement Learningcited by 279
- 2019AdaShare: Learning What To Share For Efficient Deep Multi-Task Learningcited by 160
- 2018NDDR-CNN: Layerwise Feature Fusing in Multi-Task CNNs by Neural Discriminative Dimensionality Reductioncited by 127
- 2020Learning to Branch for Multi-Task Learningcited by 115
- 2017Evolving Deep Neural Networkscited by 95
- 2017A Genetic Programming Approach to Designing Convolutional Neural Network Architecturescited by 59
- 2015Cross-domain Image Retrieval with a Dual Attribute-aware Ranking Networkcited by 53
- 2020MTL-NAS: Task-Agnostic Neural Architecture Search towards General-Purpose Multi-Task Learningcited by 50
- 2017Latent Multi-task Architecture Learningcited by 29
- 2017Beyond Shared Hierarchies: Deep Multitask Learning through Soft Layer Orderingcited by 28
- 2016Integrated perception with recurrent multi-task neural networkscited by 24
- 2018Evolutionary Architecture Search For Deep Multitask Networkscited by 23
- 2022Controllable Dynamic Multi-Task Architecturescited by 21
- 2019Deep Elastic Networks with Model Selection for Multi-Task Learningcited by 20
- 2016Deep Cross Residual Learning for Multitask Visual Recognitioncited by 16
Built on by
Citing papers per year: 2024: 6; 2025: 2; 2026: 1.
9 papers.
- 2024DINO-Foresight: Looking into the Future with DINOcited by 45
- 2024A Comprehensive Survey of Convolutions in Deep Learning: Applications, Challenges, and Future Trendscited by 16
- 2024Multi-task Learning for Real-time Autonomous Driving Leveraging Task-adaptive Attention Generatorcited by 6
- 2024Joint-Task Regularization for Partially Labeled Multi-Task Learningcited by 6
- 2024Continual Panoptic Perception: Towards Multi-modal Incremental Interpretation of Remote Sensing Imagescited by 4
- 2025StableMTL: Repurposing Latent Diffusion Models for Multi-Task Learning from Partially Annotated Synthetic Datasetscited by 2
- 2024Giving each task what it needs -- leveraging structured sparsity for tailored multi-task learningcited by 0
- 2025A Survey on Deep Multi-Task Learning in Connected Autonomous Vehiclescited by 0
- 2026Evolving Without Ending: Unifying Multimodal Incremental Learning for Continual Panoptic Perceptioncited by 0
- 2026Evolving Without Ending: Unifying Multimodal Incremental Learning for Continual Panoptic Perception22 Jan 2026
- 2025A Survey on Deep Multi-Task Learning in Connected Autonomous Vehicles29 Jul 2025
- 2025StableMTL: Repurposing Latent Diffusion Models for Multi-Task Learning from Partially Annotated Synthetic Datasets9 Jun 2025
- 2024DINO-Foresight: Looking into the Future with DINO16 Dec 2024
- 2024Continual Panoptic Perception: Towards Multi-modal Incremental Interpretation of Remote Sensing Images19 Jul 2024
- 2024Giving each task what it needs -- leveraging structured sparsity for tailored multi-task learning5 Jun 2024
- 2024Joint-Task Regularization for Partially Labeled Multi-Task Learning2 Apr 2024
- 2024Multi-task Learning for Real-time Autonomous Driving Leveraging Task-adaptive Attention Generator6 Mar 2024
- 2024A Comprehensive Survey of Convolutions in Deep Learning: Applications, Challenges, and Future Trends23 Feb 2024