Multi-Task Clustering with Model Relation Learning

Multi-Task Clustering with Model Relation Learning

Xiaotong Zhang, Xianchao Zhang, Han Liu, Jiebo Luo

Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence
Main track. Pages 3132-3140. https://doi.org/10.24963/ijcai.2018/435

Multi-task clustering improves the clustering performance of each task by transferring knowledge among the related tasks. An important aspect of multi-task clustering is to assess the task relatedness. However, to our knowledge, only two previous works have assessed the task relatedness, but they both have limitations. In this paper, we propose a multi-task clustering with model relation learning (MTCMRL) method, which automatically learns the model parameter relatedness between each pair of tasks. The objective function of MTCMRL consists of two parts: (1) within-task clustering: clustering each task by introducing linear regression model into symmetric nonnegative matrix factorization; (2) cross-task relatedness learning: updating the parameter of the linear regression model in each task by learning the model parameter relatedness between the clusters in each pair of tasks. We present an effective alternating algorithm to solve the non-convex optimization problem. Experimental results show the superiority of the proposed method over traditional single-task clustering methods and existing multi-task clustering methods.
Keywords:
Machine Learning: Transfer, Adaptation, Multi-task Learning
Machine Learning: Clustering