arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.13342stat.MEstat.ML
收藏

通过凸聚类进行多任务学习回归

Multi-Task Learning Regression via Convex Clustering

Akira Okazaki, Shuichi Kawano

AI总结:

针对分组融合正则化会跨聚类迁移信息的问题,提出引入质心参数、分离回归与聚类参数的多任务回归方法,并通过模拟和真实数据验证其精度提升。

摘要:

多任务学习(MTL)是一种旨在通过在相关任务之间共享共同信息来提高估计和预测总体性能的方法。在 MTL 中,存在若干关于任务关系的假设以及纳入这些关系的方法。在实际情形中,一个自然假设是任务根据其特征被划分为若干聚类。针对这一假设,分组融合正则化方法通过收缩任务之间的差异来执行任务聚类。这使我们能够在同一聚类内迁移共同信息。然而,该方法也会在不同聚类之间迁移信息,从而导致估计和预测性能恶化。为克服这一问题,我们提出了一种带有质心参数的 MTL 方法,该参数表示任务的聚类中心。由于该模型将参数分离为用于回归的参数和用于聚类的参数,因此我们能够提高回归系数向量的估计和预测精度。我们通过 Monte Carlo 模拟以及对真实数据的应用展示了所提方法的有效性。

英文摘要:

Multi-task learning (MTL) is a methodology that aims to improve the general performance of estimation and prediction by sharing common information among related tasks. In the MTL, there are several assumptions for the relationships and methods to incorporate them. One of the natural assumptions in the practical situation is that tasks are classified into some clusters with their characteristics. For this assumption, the group fused regularization approach performs clustering of the tasks by shrinking the difference among tasks. This enables us to transfer common information within the same cluster. However, this approach also transfers the information between different clusters, which worsens the estimation and prediction. To overcome this problem, we propose an MTL method with a centroid parameter representing a cluster center of the task. Because this model separates parameters into the parameters for regression and the parameters for clustering, we can improve estimation and prediction accuracy for regression coefficient vectors. We show the effectiveness of the proposed method through Monte Carlo simulations and applications to real data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑