基于改进梯形网络的半监督虚拟对抗训练模型

莫建文, 贾鹏

桂林电子科技大学学报 ›› 2020, Vol. 40 ›› Issue (04) : 321-327. DOI: 10.16725/j.cnki.cn45-1351/tn.2020.04.011

基于改进梯形网络的半监督虚拟对抗训练模型

  • 莫建文, 贾鹏
作者信息 +
History +

摘要

为了提高半监督深层生成模型的分类精度,提出一种基于改进梯形网络的半监督虚拟对抗训练模型。该模型在梯形网络框架的基础上,以mixup数据增强和虚拟对抗训练相结合的模式训练分类器。用mixup对训练数据做增强处理得到新的扩展数据,以解决半监督分类模型有标记样本较少的问题,并且对梯形网络框架施加虚拟对抗噪声,通过构建平滑性正则化约束,提高模型的泛化能力。模型以有标记数据的分类损失、未标记数据的重构损失和虚拟对抗损失相结合的方式调整参数,训练得到分类器。模型分别在MNIST数据库、SVHN数据库上进行实验,并且与其他半监督深层生成模型进行对比,结果表明,该模型能增强泛化能力,提高分类精度。

关键词

半监督分类 / 梯形网络 / 数据增强 / 虚拟对抗训练

中图分类号

TP181 / TP311.13

引用本文

导出引用
莫建文, 贾鹏. 基于改进梯形网络的半监督虚拟对抗训练模型. 桂林电子科技大学学报. 2020, 40(04): 321-327 https://doi.org/10.16725/j.cnki.cn45-1351/tn.2020.04.011

基金

国家自然科学基金(61661017,61967005,U1501252); 桂林电子科技大学研究生教育创新计划(2019YCXS020)

评论

Accesses

Citation

Detail

段落导航
相关文章

/