Metadata-driven Task Relation Discovery for Multi-task Learning

Metadata-driven Task Relation Discovery for Multi-task Learning

Zimu Zheng, Yuqi Wang, Quanyu Dai, Huadi Zheng, Dan Wang

Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence
Main track. Pages 4426-4432. https://doi.org/10.24963/ijcai.2019/615

Task Relation Discovery (TRD), i.e., reveal the relation of tasks, has notable value: it is the key concept underlying Multi-task Learning (MTL) and provides a principled way for identifying redundancies across tasks. However, task relation is usually specifically determined by data scientist resulting in the additional human effort for TRD, while transfer based on brute-force methods or mere training samples may cause negative effects which degrade the learning performance. To avoid negative transfer in an automatic manner, our idea is to leverage commonly available context attributes in nowadays systems, i.e., the metadata. In this paper, we, for the first time, introduce metadata into TRD for MTL and propose a novel Metadata Clustering method, which jointly uses historical samples and additional metadata to automatically exploit the true relatedness. It also avoids the negative transfer by identifying reusable samples between related tasks. Experimental results on five real-world datasets demonstrate that the proposed method is effective for MTL with TRD, and particularly useful in complicated systems with diverse metadata but insufficient data samples. In general, this study helps in automatic relation discovery among partially related tasks and sheds new light on the development of TRD in MTL through the use of metadata as apriori information.
Keywords:
Machine Learning: Transfer, Adaptation, Multi-task Learning
Machine Learning Applications: Networks
Machine Learning Applications: Other Applications
Machine Learning: Clustering
Multidisciplinary Topics and Applications: Ubiquitous Computing Systems