arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.00723cs.LGmath.OC

A Mixed-Integer Programming Approach to Training Dense Neural Networks

University of Wisconsin-Madison(威斯康星大学麦迪逊分校)

Vrishabh Patil, Yonatan Mintz

英文摘要:

Artificial Neural Networks (ANNs) are prevalent machine learning models that are applied across various real-world classification tasks. However, training ANNs is time-consuming and the resulting models take a lot of memory to deploy. In order to train more parsimonious ANNs, we propose a novel mixed-integer programming (MIP) formulation for training fully-connected ANNs. Our formulations can account for both binary and rectified linear unit (ReLU) activations, and for the use of a log-likelihood loss. We present numerical experiments comparing our MIP-based methods against existing approaches and show that we are able to achieve competitive out-of-sample performance with more parsimonious models.

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

↑