Training Convolutional ReLU Neural Networks in Polynomial Time: Exact Convex Optimization Formulations
Training Convolutional ReLU Neural Networks in Polynomial Time: Exact Convex Optimization Formulations
复制标题
在多项式时间内训练卷积 ReLU 神经网络:精确的凸优化公式
DOI:
--
复制
发表时间:
2020
期刊:
影响因子:
--
通讯作者:
Mert Pilanci
中科院分区:
文献类型:
--
作者:
Tolga Ergen;Mert Pilanci
We study training of Convolutional Neural Networks (CNNs) with ReLU activations and introduce exact convex optimization formulations with a polynomial complexity with respect to the number of data samples, the number of neurons and data dimension. Particularly, we develop a convex analytic framework utilizing semi-infinite duality to obtain equivalent convex optimization problems for several CNN architectures. We first prove that two-layer CNNs can be globally optimized via an $ell_2$ norm regularized convex program. We then show that certain three-layer CNN training problems are equivalent to an $ell_1$ regularized convex program. We also extend these results to multi-layer CNN architectures. Furthermore, we present extensions of our approach to different pooling methods.
影响因子:
6
作者:
Ergen, T.;Pilanci, M.
通讯作者:
Pilanci, M.
DOI:
--
发表时间:
2019-02
期刊:
--
影响因子:
--
作者:
Pedro H. P. Savarese;Itay Evron;Daniel Soudry;N. Srebro
通讯作者:
Pedro H. P. Savarese;Itay Evron;Daniel Soudry;N. Srebro
DOI:
--
发表时间:
2020
期刊:
International Conference on Machine Learning
影响因子:
--
作者:
Pilanci, Mert;Ergen, Tolga
通讯作者:
Ergen, Tolga