Giving Each Task what it Needs Leveraging Structured Sparsity for Tailored Multi-Task Learning
2025 (English)In: Computer Vision – ECCV 2024 Workshops: Milan, Italy, September 29–October 4, 2024, Proceedings / [ed] Alessio Del Bue; Cristian Canton; Jordi Pont-Tuset; Tatiana Tommasi, Springer Science and Business Media Deutschland GmbH , 2025, Vol. XI, p. 202-218Conference paper, Published paper (Refereed)
Abstract [en]
In the Multi-task Learning (MTL) framework, every task demands distinct feature representations, ranging from low-level to high-level attributes. It is vital to address the specific (feature/parameter) needs of each task, especially in computationally constrained environments. This work, therefore, introduces Layer-Optimized Multi-Task (LOMT) models that utilize structured sparsity to refine feature selection for individual tasks and enhance the performance of all tasks in a multi-task scenario. Structured or group sparsity systematically eliminates parameters from trivial channels and, sometimes, eventually, entire layers within a convolution neural network during training. Consequently, the remaining layers provide the most optimal features for a given task. In this two-step approach, we subsequently leverage this sparsity-induced optimal layer information to build the LOMT models by connecting task-specific decoders to these strategically identified layers, deviating from conventional approaches that uniformly connect decoders at the end of the network. This tailored architecture optimizes the network, focusing on essential features while reducing redundancy. We validate the efficacy of the proposed approach on two datasets, i.e., NYU-v2 and CelebAMask-HD datasets, for multiple heterogeneous tasks. A detailed performance analysis of the LOMT models, in contrast to the conventional MTL models, reveals that the LOMT models outperform for most task combinations. The excellent qualitative and quantitative outcomes highlight the effectiveness of employing structured sparsity for optimal layer (or feature) selection.
Place, publisher, year, edition, pages
Springer Science and Business Media Deutschland GmbH , 2025. Vol. XI, p. 202-218
Series
Lecture Notes in Computer Science (LNCS), ISSN 0302-9743, E-ISSN 1611-3349 ; 15633
Keywords [en]
Multi-task learning, group sparsity, feature selection, layer optimization
National Category
Computer graphics and computer vision
Research subject
Machine Learning
Identifiers
URN: urn:nbn:se:ltu:diva-113965DOI: 10.1007/978-3-031-91979-4_16ISI: 001544984800015Scopus ID: 2-s2.0-105008008892OAI: oai:DiVA.org:ltu-113965DiVA, id: diva2:1980162
Conference
18th European Conference on Computer Vision (ECCV 2024), Milan, Italy, September 29 - October 4, 2024
Note
ISBN for host publication: 978-3-031-91978-7, 978-3-031-91979-4
2025-07-012025-07-012025-11-28Bibliographically approved