arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1908.02400cs.LGcs.NAmath.NAstat.ML
收藏

用矩阵条件化方法精化神经网络的结构

Refining the Structure of Neural Networks Using Matrix Conditioning

University of Maryland(马里兰大学)

Roozbeh Yousefzadeh, Dianne P O'Leary

摘要:

深度学习模型已被证明在执行许多机器学习任务时非常有用。然而,对于每个新数据集,选择模型的有效大小和结构可能是一个耗时的试错过程。虽然一个只有很少神经元的小型网络可能无法捕捉给定任务的复杂性,但拥有太多神经元可能会导致过度拟合和泛化能力差。在这里,我们提出了一种实用的方法,采用矩阵条件自动设计的前馈网络的层的结构,首先调整网络的层之间的神经元的比例,然后缩放网络的大小向上或向下。在样本图像和非图像数据集上的结果表明,我们的方法可以产生具有高精度的小网络。最后,在矩阵条件作用的指导下,我们提供了一种方法来有效地挤压已经训练好的模型。我们的技术降低了设计深度学习模型的人力成本,还可以减少训练时间和将神经网络用于应用程序的费用。

英文摘要:

Deep learning models have proven to be exceptionally useful in performing many machine learning tasks. However, for each new dataset, choosing an effective size and structure of the model can be a time-consuming process of trial and error. While a small network with few neurons might not be able to capture the intricacies of a given task, having too many neurons can lead to overfitting and poor generalization. Here, we propose a practical method that employs matrix conditioning to automatically design the structure of layers of a feed-forward network, by first adjusting the proportion of neurons among the layers of a network and then scaling the size of network up or down. Results on sample image and non-image datasets demonstrate that our method results in small networks with high accuracies. Finally, guided by matrix conditioning, we provide a method to effectively squeeze models that are already trained. Our techniques reduce the human cost of designing deep learning models and can also reduce training time and the expense of using neural networks for applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑