欢迎访问机床与液压官方网站!

咨询热线:020-32385312 32385313 RSS EMAIL-ALERT
基于伪标签的弱监督迁移学习模型
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家留学基金资助项目(201808140235);山西省留学归国人员择优资助项目(2020-127);山西省重点研发计划项目(201903D321012)


Weak Supervised Transfer Learning Model Based on Pseudo-label
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对目标域标记数据少导致迁移模型泛化能力差的问题,提出基于伪标签的半监督迁移学习模型WSTLPL。卷积神经网络用于学习原始振动数据的可迁移特征,用源域数据预训练网络;利用该网络预测目标域数据类别,将分类概率最大的类标签作为数据的伪标签。根据域自适应和伪标签学习的正则化项,对神经网络的参数施加约束,以减少学习到的可迁移特征的分布差异。结果表明:与现有诊断模型相比,该迁移模型的准确率更高。

    Abstract:

    A semi-supervised transfer learning model WSTLPL (weakly supervised transfer learning based on pseudo-label) was proposed to solve the problem of poor generalization ability of transfer model due to the lack of target domain label data. The convolutional neural network was used to learn the transferable characteristics of the original vibration data, and the network was pre-trained with the source domain data;the trained network was used to predict the target domain data category, and the class label with the highest classification probability was taken as the pseudo-label of data. According to the regularization term of domain adaptation and pseudo-label learning, the parameters of neural network were constrained to reduce the distribution differences of the learned transferable features. The results show that the proposed transfer model is more accurate than the existing diagnostic model.

    参考文献
    相似文献
    引证文献
引用本文

侯鑫烨,董增寿,刘鑫.基于伪标签的弱监督迁移学习模型[J].机床与液压,2021,49(24):185-189.
HOU Xinye, DONG Zengshou, LIU Xin. Weak Supervised Transfer Learning Model Based on Pseudo-label[J]. Machine Tool & Hydraulics,2021,49(24):185-189

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-04-28
  • 出版日期: 2021-12-28