arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.03397cs.LGstat.ML

Dropout: Explicit Forms and Capacity Control

Johns Hopkins University(约翰斯·霍普金斯大学);University of California, Berkeley(加州大学伯克利分校);TTI Chicago(芝加哥丰田技术研究所)

Raman Arora, Peter Bartlett, Poorya Mianjy, Nathan Srebro

英文摘要:

We investigate the capacity control provided by dropout in various machine learning problems. First, we study dropout for matrix completion, where it induces a data-dependent regularizer that, in expectation, equals the weighted trace-norm of the product of the factors. In deep learning, we show that the data-dependent regularizer due to dropout directly controls the Rademacher complexity of the underlying class of deep neural networks. These developments enable us to give concrete generalization error bounds for the dropout algorithm in both matrix completion as well as training deep neural networks. We evaluate our theoretical findings on real-world datasets, including MovieLens, MNIST, and Fashion-MNIST.

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

↑