Learning twofold heterogeneous multi-task by sharing similar convolution kernel pairs

作者:

Highlights:

摘要

Heterogeneous multi-task learning (HMTL) is an important topic in multi-task learning (MTL). Most existing HMTL methods usually solve either scenario where all tasks reside in the same input (feature) space yet unnecessarily the consistent output (label) space or scenario where their input (feature) spaces are heterogeneous while the output (label) space is consistent. However, to the best of our knowledge, there is limited study on twofold heterogeneous MTL (THMTL) scenario where the input and the output spaces are both inconsistent or heterogeneous. In order to handle this complicated scenario, in this paper, we design a simple and effective multi-task adaptive learning (MTAL) network to learn multiple tasks in such THMTL setting. Specifically, we explore and utilize the inherent relationship between tasks for knowledge sharing from similar convolution kernels in individual layers of the MTAL network. To realize the sharing, we weightedly aggregate any pair of convolutional kernels with their similarity greater than some threshold . Consequently, our model effectively performs cross-task learning while suppresses the intra-redundancy of the entire network. Finally, we conduct end-to-end training. Our experimental results demonstrate the significant competitiveness of our method in comparison with the current state-of-the-art methods.

论文关键词:Heterogeneous tasks,Multi-task learning,Convolution kernel sharing

论文评审过程:Received 23 November 2021, Revised 2 June 2022, Accepted 6 July 2022, Available online 16 July 2022, Version of Record 25 July 2022.

论文官网地址:https://doi.org/10.1016/j.knosys.2022.109396