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

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

BN parameter learning algorithm based on dynamic weighted transfer learning

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

郭文强、徐成、肖秦琨、李梦然

展开 >

陕西科技大学电子信息与人工智能学院,西安710021

陕西科技大学电气与控制工程学院,西安710021

西安工业大学电子信息工程学院,西安710021

小数据集 贝叶斯网络 迁移学习 参数学习

国家自然科学基金资助项目国家自然科学基金资助项目陕西省科技厅重点研发计划资助项目陕西省科技厅自然科学基金资助项目陕西省教育厅产业化研究项目西安市科技计划资助项目

61271362620713662020SF-2862017JM605718JC0032019216514GXRC001CG002GXYD11

2021

计算机应用研究
四川省电子计算机应用研究中心

计算机应用研究

CSTPCDCSCD北大核心
影响因子:0.93
ISSN:1001-3695
年,卷(期):2021.38(1)
  • 4
  • 6