基于自适应边际损失的小样本故障诊断方法
Fault diagnosis method for small sample based on adaptive margin loss作者机构:南昌航空大学无损检测技术教育部重点实验室南昌330063 哈德斯菲尔德大学效率与性能工程中心英国哈德斯菲尔德HD13DH
出 版 物:《兵器装备工程学报》 (Journal of Ordnance Equipment Engineering)
年 卷 期:2024年第45卷第9期
页 面:253-260页
核心收录:
学科分类:12[管理学] 1201[管理学-管理科学与工程(可授管理学、工学学位)] 08[工学] 081104[工学-模式识别与智能系统] 082503[工学-航空宇航制造工程] 0835[工学-软件工程] 0825[工学-航空宇航科学与技术] 0811[工学-控制科学与工程] 0812[工学-计算机科学与技术(可授工学、理学学位)]
基 金:国家自然科学基金项目(52075236) 江西省自然科学基金重点项目(20212ACB202005)
主 题:故障诊断 小样本 元度量学习 自适应边际损失 无人机
摘 要:针对基于度量学习的小样本故障诊断方法,在模型优化过程中对边际影响的忽略导致模型对训练数据过于敏感,进而产生过拟合。为此,构造了一种自适应边际损失函数,帮助模型学习样本之间的相对距离,以获得足够的距离信息,提高对新样本的泛化能力。另外,根据训练数据的分布和模型的性能,自动调整边际的大小,使其自适应地区分不同的故障类别。为更好地解决小样本问题,提出了元度量学习框架,采用元学习片段式训练模式。在度量模块中,引入余弦相似性以提高方法的表达能力,并指导模型的优化和训练,使其更好地适应小样本数据。为验证所提方法的有效性,使用带故障的无人机飞行日志数据构建了数据集,并将所提方法与传统的度量学习的方法进行了对比。实验结果表明,所提方法在无人机小样本故障诊断中表现出良好的诊断性能和稳定性,为小样本故障诊断提供了一种有效方案。