中文

一种求解二值激活函数深度神经网络的整数规划方法

最优化与控制 2020-08-10 v3 机器学习 神经与进化计算

摘要

我们研究具有二值激活函数(BDNN)的深度神经网络,即激活函数仅有两个状态。我们证明 BDNN 可重新表述为一个混合整数线性规划,该规划可通过经典整数规划求解器求得全局最优解。此外,我们给出了一种启发式求解算法,并在数据不确定性下研究该模型,采用两阶段鲁棒优化方法。我们在随机与真实数据集上实现了所提方法,结果表明 BDNN 的启发式版本在 Breast Cancer Wisconsin 数据集上优于经典深度神经网络,而在随机数据上表现较差。

关键词

引用

@article{arxiv.2007.03326,
  title  = {An Integer Programming Approach to Deep Neural Networks with Binary Activation Functions},
  author = {Bubacarr Bah and Jannis Kurtz},
  journal= {arXiv preprint arXiv:2007.03326},
  year   = {2020}
}