基于变权重迁移学习的BN参数学习算法

TP311; 针对小数据集条件下的贝叶斯网络(Bayesian network,BN)参数估计困难问题,提出了一种基于变权重迁移学习(DWTL)的BN参数学习算法.首先,利用MAP和MLE方法学习得到目标域初始参数和各源域参数;然后根据不同源域数据样本贡献的不同计算源权重因子;接着基于目标域样本统计量与小数据集样本阈值的关系设计了目标域初始参数和源域参数的平衡系数;最后,基于上述参数、源权重因子和平衡系数计算得到新的目标参数.在实验研究中,通过对经典BN模型的参数学习问题验证了DWTL算法的有效性;针对小数据集下的轴承故障诊断问题,相较于传统迁移学习(LP)算法,DWTL算法学习精度提高了10...

Full description

Saved in:
Bibliographic Details
Published in计算机应用研究 Vol. 38; no. 1; pp. 110 - 114
Main Authors 郭文强, 徐成, 肖秦琨, 李梦然
Format Journal Article
LanguageChinese
Published 陕西科技大学电子信息与人工智能学院,西安710021%陕西科技大学电气与控制工程学院,西安710021%西安工业大学电子信息工程学院,西安710021 2021
Subjects
Online AccessGet full text

Cover

Loading…
More Information
Summary:TP311; 针对小数据集条件下的贝叶斯网络(Bayesian network,BN)参数估计困难问题,提出了一种基于变权重迁移学习(DWTL)的BN参数学习算法.首先,利用MAP和MLE方法学习得到目标域初始参数和各源域参数;然后根据不同源域数据样本贡献的不同计算源权重因子;接着基于目标域样本统计量与小数据集样本阈值的关系设计了目标域初始参数和源域参数的平衡系数;最后,基于上述参数、源权重因子和平衡系数计算得到新的目标参数.在实验研究中,通过对经典BN模型的参数学习问题验证了DWTL算法的有效性;针对小数据集下的轴承故障诊断问题,相较于传统迁移学习(LP)算法,DWTL算法学习精度提高了10%.实验结果表明:所提出的算法能够较好地解决样本数据集在相对稀缺条件下的目标参数建模问题.
ISSN:1001-3695
DOI:10.19734/j.issn.1001-3695.2019.10.0600