压缩深层神经网络隐藏层维度对其分类性能的影响

作者：成凌飞; 贺扬; 张培玲; 李艳

来源：电子科技, 2019, 32(01): 72-75.

DOI：10.16180/j.cnki.issn1007-7820.2019.01.0015

• 深层神经网络
• 分类
• 压缩比例
• 泛化能力
• 训练时间
• 深度学习

为了使深层神经网络具有更好的泛化能力、少量训练时间的性能。文中汲取压缩神经网络的思想,通过将深层神经网络的输入层与隐藏层按照不同的比例将隐藏层的维度进行压缩,并在传统压缩深层神经网络的基础上,在其顶层添加一个分类层,使深层神经网络拥有分类的能力。实验将构建的深层神经网络应用于MNIST手写数据集的分类测试,结果表明,经过适当压缩的深层神经网络比未被压缩的深层神经网络具有更好的分类效果,而且节省了大量的训练时间。

• 单位
  河南理工大学; 自动化学院