Journal of Computer Applications ›› 2025, Vol. 45 ›› Issue (8): 2530-2536.DOI: 10.11772/j.issn.1001-9081.2024071012

• Artificial intelligence • Previous Articles    

Partial label regression algorithm integrating feature attention and residual connection

Haifeng WU1,2, Liqing TAO1, Yusheng CHENG1,2()   

  1. 1.School of Computer and Information,Anqing Normal University,Anqing Anhui 246133,China
    2.University Key Laboratory of Intelligent Perception and Computing of Anhui Province (Anqing Normal University),Anqing Anhui 246133,China
  • Received:2024-07-18 Revised:2024-11-12 Accepted:2024-11-13 Online:2024-11-19 Published:2025-08-10
  • Contact: Yusheng CHENG
  • About author:WU Haifeng, born in 1982, M. S., professor. His research interests include machine learning, data mining.
    TAO Liqing, born in 1999, M. S. candidate. Her research interests include partial label learning, data mining.
  • Supported by:
    Anhui Provincial Natural Science Foundation(2108085MF216)

集成特征注意力和残差连接的偏标签回归算法

吴海峰1,2, 陶丽青1, 程玉胜1,2()   

  1. 1.安庆师范大学 计算机与信息学院,安徽 安庆 246133
    2.智能感知与计算安徽省高校重点实验室(安庆师范大学),安徽 安庆 246133
  • 通讯作者: 程玉胜
  • 作者简介:吴海峰(1982—),男,安徽安庆人,教授,硕士,主要研究方向:机器学习、数据挖掘
    陶丽青(1999—),女,安徽铜陵人,硕士研究生,主要研究方向:偏标签学习、数据挖掘
  • 基金资助:
    安徽省自然科学基金资助项目(2108085MF216);安徽省高校自然科学研究项目(2024AH040175);安徽省高校自然科学研究项目(2024AH051099)

Abstract:

Partial Label Regression (PLR) complements the current situation that Partial Label Learning (PLL) only focuses on classification tasks. To address the problem that the existing PLR algorithms ignore characteristic differences between instance features, a Partial Label Regression algorithm integrating Feature Attention and Residual Connection (PLR-FARC) was proposed. Firstly, labels of real datasets were expanded into a set of real-value candidate labels by label enhancement technique. Secondly, the attention mechanism was employed to generate contribution of individual features to labels automatically. Thirdly, the residual connection was introduced to reduce information loss and maintain feature integrity during feature transmission. Finally, prediction loss was calculated based on IDent (IDentification method) and PIDent (Progressive IDentification method), respectively. Experimental results on Abalone, Airfoil, Concrete, Cpu-act, Housing and Power-plant datasets show that compared to IDent and PIDent, PLR-FARC has the Mean Absolute Error (MAE) reduced by 2.15%, 38.38%, 8.86%, 4.19%, 15.71% and 15.55%, averagely and respectively, and the Mean Squared Error (MSE) reduced by 9.35%, 71.32%, 23.10%, 20.17%, 27.22% and 9.46%, averagely and respectively. It can be seen that the proposed algorithm is feasible and effective.

Key words: Partial Label Learning (PLL), Partial Label Regression (PLR), candidate label, attention mechanism, residual connection

摘要:

偏标签回归(PLR)弥补了偏标签学习(PLL)仅聚焦于分类任务的局限。针对现有的PLR算法忽略实例特征的特性差异的问题,提出一种集成特征注意力和残差连接的偏标签回归算法(PLR-FARC)。首先,通过标签增强技术将真实数据集的标签扩充为一组实值候选标签;其次,借助注意力机制自动生成每个特征对标签的贡献度;再次,引入残差连接以减少特征在传递过程中的信息丢失,从而维持特征的完整性;最后,分别基于IDent (IDentification method)和PIDent (Progressive IDentification method)计算预测损失。在Abalone、Airfoil、Concrete、Cpu-act、Housing和Power-plant数据集上的实验结果表明,相较于IDent和PIDent,PLR-FARC的平均绝对误差(MAE)分别平均降低了2.15%、38.38%、8.86%、4.19%、15.71%和15.55%,均方误差(MSE)分别平均降低了9.35%、71.32%、23.10%、20.17%、27.22%和9.46%。可见,所提算法是可行且有效的。

关键词: 偏标签学习, 偏标签回归, 候选标签, 注意力机制, 残差连接

CLC Number: