中文

基于 Pytorch 的深度自编码器模型构建

机器学习 2022-08-18 v1

摘要

本文提出一种基于 Pytorch 的深度自编码器模型。该算法将 Pytorch 的思想引入自编码器,并以一定概率随机清除连接到隐藏层神经元的输入权重,从而达到网络稀疏的效果,这与稀疏自编码器的出发点类似。新算法有效解决了模型可能过拟合的问题,并提高了图像分类的准确率。最后进行实验,并将实验结果与 ELM、RELM、AE、SAE、DAE 进行比较。

关键词

引用

@article{arxiv.2208.08231,
  title  = {Deep Autoencoder Model Construction Based on Pytorch},
  author = {Junan Pan and Zhihao Zhao},
  journal= {arXiv preprint arXiv:2208.08231},
  year   = {2022}
}

备注

16 pages, 10 figures