不平衡小样本基于局部域对抗适应网络的发动机振动预测模型  

Partial domain adversarial adaptation networks for imbalanced small samples in aeroengine vibration prediction

作　　者：季友昌 袁伟伟[1] 毛善斌 任春红[2] 关东海 JI Youchang;YUAN Weiwei;MAO Shanbin;REN Chunhong;GUAN Donghai(College of Computer Science and Technology,Nanjing University of Aeronautics and Astronautics,Nanjing 211106,China;Beijing Power Machinery Institute,Beijing 100074,China)

机构地区：[1]南京航空航天大学计算机科学与技术学院,江苏南京211106 [2]北京动力机械研究所,北京100074

出　　处：《智能系统学报》2023年第5期1005-1016,共12页CAAI Transactions on Intelligent Systems

基　　金：基础科研项目(JCKY2020204C009)。

摘　　要：在发动机振动预测中,实际装配数据样本量小且类别不平衡,难以直接建立有效的预测模型。迁移学习方法能够通过迁移源域知识来提高目标域模型性能,为此,本文提出了基于局部域对抗适应网络的发动机振动预测模型。将领域按标签分为多个局部域,通过多个局部域对抗适应网络将目标域样本映射到源域,保证小样本中的少数类得到正确迁移。并通过伪标签来解决目标样本的域转换,使用源域分类器给出可靠的预测结果。本文在多个真实数据集上验证了所提方法的有效性和泛化性,与其他方法相比,振动预测准确率能够平均提升15%左右。In aeroengine vibration prediction,due to the imbalanced small samples of the actual assembly data,directly establishing an effective prediction model is difficult.Transfer learning can improve model performance in the target do-main by transferring the knowledge of the source domain.Therefore,this study proposes an aeroengine vibration predic-tion model based on a partial domain adversarial adaptation network.Each domain is divided into multiple local do-mains according to labels.Through multiple local domain adversarial adaptation networks,the samples in the target do-main can be mapped onto the source domain so that the minority class samples can be transferred correctly.The pseudo label is used to solve the domain transformation of the samples in the target domain,and the classifier of the source do-main is used to provide a reliable prediction result.In this study,the validity and generalization of the proposed method are verified on several real datasets.Compared with other methods,the area under the curve and F1 of the vibration pre-diction model can be improved by approximately 15%on average.

关 键 词：不平衡数据 域适应 对抗学习 振动预测 异构迁移学习 目标域 小样本学习 特征空间 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TP18[自动化与计算机技术—计算机科学与技术]