Autonomous Learning Conference Paper 2021

CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints

Architecture sketch comboptnet

Bridging logical and algorithmic reasoning with modern machine learning techniques is a fundamental challenge with potentially transformative impact. On the algorithmic side, many NP-hard problems can be expressed as integer programs, in which the constraints play the role of their ``combinatorial specification.'' In this work, we aim to integrate integer programming solvers into neural network architectures as layers capable of learning both the cost terms and the constraints. The resulting end-to-end trainable architectures jointly extract features from raw data and solve a suitable (learned) combinatorial problem with state-of-the-art integer programming solvers. We demonstrate the potential of such layers with an extensive performance analysis on synthetic data and with a demonstration on a competitive computer vision keypoint matching benchmark.

Author(s): Anselm Paulus and Michal Rolínek and Vít Musil and Brandon Amos and Georg Martius
Book Title: Proceedings of the 38th International Conference on Machine Learning
Volume: 139
Pages: 8443--8453
Year: 2021
Month: July
Day: 18--24
Series: Proceedings of Machine Learning Research
Editors: Meila, Marina and Zhang, Tong
Publisher: PMLR
Project(s):
Bibtex Type: Conference Paper (inproceedings)
Event Name: The Thirty-eighth International Conference on Machine Learning (ICML)
Event Place: Virtual
URL: https://proceedings.mlr.press/v139/paulus21a.html
Electronic Archiving: grant_archive
Language: English
Talk Type: Spotlight
Links:

BibTex

@inproceedings{paulus2021:comboptnet,
  title = {CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints},
  booktitle = {Proceedings of the 38th International Conference on Machine Learning},
  abstract = {Bridging logical and algorithmic reasoning with modern machine learning techniques is a fundamental challenge with potentially transformative impact. On the algorithmic side, many NP-hard problems can be expressed as integer programs, in which the constraints play the role of their ``combinatorial specification.'' In this work, we aim to integrate integer programming solvers into neural network architectures as layers capable of learning both the cost terms and the constraints. The resulting end-to-end trainable architectures jointly extract features from raw data and solve a suitable (learned) combinatorial problem with state-of-the-art integer programming solvers. We demonstrate the potential of such layers with an extensive performance analysis on synthetic data and with a demonstration on a competitive computer vision keypoint matching benchmark.},
  volume = {139},
  pages = {8443--8453},
  series = {Proceedings of Machine Learning Research},
  editors = {Meila, Marina and Zhang, Tong},
  publisher = {PMLR},
  month = jul,
  year = {2021},
  slug = {comboptnet},
  author = {Paulus, Anselm and Rolínek, Michal and Musil, Vít and Amos, Brandon and Martius, Georg},
  url = {https://proceedings.mlr.press/v139/paulus21a.html},
  month_numeric = {7}
}