基于凸聚类的多任务学习回归

Multi-task learning regression via convex clustering

Computational Statistics and Data Analysis · 2024
被引 3
ABS 3

中文导读

提出一种多任务学习方法,通过引入聚类中心参数来分离回归与聚类任务,避免不同簇间信息传递带来的估计偏差,提升回归系数向量的估计和预测精度。

Abstract

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 the transfer of 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, an MTL method is proposed 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, estimation and prediction accuracy for regression coefficient vectors are improved. The effectiveness of the proposed method is shown through Monte Carlo simulations and applications to real data.

多任务学习聚类分析回归分析机器学习