计算机应用 ›› 2014, Vol. 34 ›› Issue (7): 2061-2065.DOI: 10.11772/j.issn.1001-9081.2014.07.2061

• 人工智能 • 上一篇    下一篇

可自适应分组的超球多任务学习算法

毛文涛1,王海成2,刘尚旺1   

  1. 1. 河南师范大学 计算机与信息工程学院, 河南 新乡 453007;
    2. 清华大学 工程物理系,北京 100084
  • 收稿日期:2014-01-14 修回日期:2014-03-10 出版日期:2014-07-01 发布日期:2014-08-01
  • 通讯作者: 毛文涛
  • 作者简介:毛文涛(1980-),男,河南新乡人,副教授,博士,主要研究方向:机器学习、模式识别;王海成(1979-),男, 河南辉县人,硕士研究生, 主要研究方向:智能控制、模式识别;刘尚旺(1973-),男, 河南新乡人, 讲师,博士, 主要研究方向:模式识别、神经网络。
  • 基金资助:

    国家自然科学基金资助项目;河南省基础与前沿技术研究计划项目

Hyper-spherical multi-task learning algorithm with adaptive grouping

MAO Wentao1,WANG Haicheng2,LIU Shangwang1   

  1. 1. College of Computer and Information Engineering, Henan Normal University, Xinxiang Henan 453007, China;
    2. Department of Engineering Physics, Tsinghua University, Beijing 100084, China
  • Received:2014-01-14 Revised:2014-03-10 Online:2014-07-01 Published:2014-08-01
  • Contact: MAO Wentao

摘要:

针对现有回归多任务学习中各任务独立评估风险、缺乏统一约束条件的缺点,提出了一种具有自适应分组能力的超球多任务学习算法。该算法以极限学习机(ELM)为基础形式,首先引入超球损失函数对所有任务的风险进行统一评估,并采用迭代再权最小二乘法求解;其次,考虑到任务之间关联度存在差异,基于相关性强的任务其权重向量也较相似的假设,构建带分组结构的正则项,使得同组内的任务独立进行训练,最终将优化目标转为混合0-1规划问题,并采用多目标优化方法自动确定模型参数和最优分组结构。基于仿真数据和圆柱壳振动信号数据的测试结果表明,该算法可有效识别出任务中的分组结构,同时与现有算法相比,可明显提高回归模型的泛化能力。

Abstract:

To solve the problem in most of conventional multi-task learning algorithms which evaluate risk independently for single task and lack uniform constraint across all tasks, a new hyper-spherical multi-task learning algorithm with adaptive grouping was proposed in this paper. Based on Extreme Learning Machine (ELM) as basic framework, this algorithm introduced hyper-spherical loss function to evaluate the risks of all tasks uniformly, and got decision model via iterative reweighted least squares solution. Furthermore, considering the existence of relatedness between tasks, this paper also constructed regularizer with grouping structure based on the assumption that related tasks had more similar weight vector, which would make the tasks in same group be trained independently. Finally, the optimization object was transformed into a mixed 0-1 programming problem, and a multi-objective method was utilized to identify optimal grouping structure and get model parameters. The simulation results on toy data and cylindrical vibration signal data show that the proposed algorithm outperforms state-of-the-art methods in terms of generalization performance and the ability of identifying inner structure in tasks.

中图分类号: