基于伪标签的弱监督迁移学习模型  

Weak Supervised Transfer Learning Model Based on Pseudo-label

作　　者：侯鑫烨 董增寿[1] 刘鑫[1] HOU Xinye;DONG Zengshou;LIU Xin(School of Electronic Information Engineering,Taiyuan University of Science and Technology,Taiyuan Shanxi 030024,China;Yangquan Regional Innovation Promotion Center,Yangquan Shanxi 045000,China)

机构地区：[1]太原科技大学电子信息工程学院,山西太原030024 [2]阳泉市区域创新促进中心,山西阳泉045000

出　　处：《机床与液压》2021年第24期185-189,共5页Machine Tool & Hydraulics

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

摘　　要：针对目标域标记数据少导致迁移模型泛化能力差的问题,提出基于伪标签的半监督迁移学习模型WSTLPL。卷积神经网络用于学习原始振动数据的可迁移特征,用源域数据预训练网络;利用该网络预测目标域数据类别,将分类概率最大的类标签作为数据的伪标签。根据域自适应和伪标签学习的正则化项,对神经网络的参数施加约束,以减少学习到的可迁移特征的分布差异。结果表明:与现有诊断模型相比,该迁移模型的准确率更高。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 convolu⁃tional 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 high⁃est classification probability was taken as the pseudo-label of data.According to the regularization term of domain adaptation and pseu⁃do-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.

关 键 词：弱监督学习 域自适应 伪标签 迁移学习 

分 类 号：TH165.3[机械工程—机械制造及自动化]