arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.03250stat.MEcs.LGstat.ML

基于非凸组惩罚的鲁棒正则化聚类的多任务学习

Multi-task learning via robust regularized clustering with non-convex group penalties

Graduate School of Mathematics, Kyushu University(九州大学大学院数理学府);Faculty of Mathematics, Kyushu University(九州大学数理学部)

Akira Okazaki, Shuichi Kawano

AI总结:

针对多任务学习中忽略异常任务的问题,提出MTLRRC方法,结合鲁棒正则化聚类与非凸组惩罚,实现鲁棒任务聚类与异常检测,并通过模拟和实际数据验证其有效性。

摘要:

多任务学习(MTL)旨在通过共享相关任务之间的共同信息来提高估计和预测性能。MTL中的一个自然假设是,任务根据其特征被划分为若干聚类。然而,基于这一假设的现有MTL方法往往忽略了具有较大任务特定成分或与其他任务无关的异常任务。为了解决这一问题,我们提出了一种新的MTL方法,称为基于鲁棒正则化聚类的多任务学习(MTL-RRC)。MTLRRC引入了受鲁棒凸聚类启发的鲁棒正则化项,并进一步扩展以处理非凸和组稀疏惩罚。该扩展使MTLRRC能够同时进行鲁棒任务聚类和异常任务检测。我们还建立了扩展鲁棒聚类与多元M估计器之间的联系,这为MTLRRC对异常任务的鲁棒性提供了解释。我们开发了一种基于修正交替方向乘子法的高效算法来估计参数。通过模拟研究和实际数据应用,验证了MTLRRC的有效性。

英文摘要:

Multi-task learning (MTL) aims to improve estimation and prediction performance by sharing common information among related tasks. One natural assumption in MTL is that tasks are classified into clusters based on their characteristics. However, existing MTL methods based on this assumption often ignore outlier tasks that have large task-specific components or no relation to other tasks. To address this issue, we propose a novel MTL method called Multi-Task Learning via Robust Regularized Clustering (MTLRRC). MTLRRC incorporates robust regularization terms inspired by robust convex clustering, which is further extended to handle non-convex and group-sparse penalties. The extension allows MTLRRC to simultaneously perform robust task clustering and outlier task detection. The connection between the extended robust clustering and the multivariate M-estimator is also established. This provides an interpretation of the robustness of MTLRRC against outlier tasks. An efficient algorithm based on a modified alternating direction method of multipliers is developed for the estimation of the parameters. The effectiveness of MTLRRC is demonstrated through simulation studies and application to real data.

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

↑