按数据维度参数化的 ReLU 网络训练的计算复杂性
机器学习
2022-08-24 v3 计算复杂性
数据结构与算法
神经与进化计算
机器学习
摘要
理解训练带修正线性单元(ReLU)的简单神经网络的计算复杂性近来已成为一个研究热点。为弥补文献中的空白并补充相关结果,我们针对以各种损失函数训练两层 ReLU 网络的参数化复杂性给出了若干结果。在简要讨论其他参数后,我们重点分析训练数据维度 对计算复杂性的影响。我们给出了关于参数 的 W[1]-困难性意义上的运行时间下界,并证明已知的暴力搜索策略本质上是最优的(在指数时间假设下)。与先前工作相比,我们的结果对更广泛的损失函数成立,包括对所有 的 损失。特别地,我们将一个针对常数 和凸损失函数的已知多项式时间算法推广到更一般的损失函数类,并在这些情形下也与我们的运行时间下界相匹配。
引用
@article{arxiv.2105.08675,
title = {The Computational Complexity of ReLU Network Training Parameterized by Data Dimensionality},
author = {Vincent Froese and Christoph Hertrich and Rolf Niedermeier},
journal= {arXiv preprint arXiv:2105.08675},
year = {2022}
}