Timezone: »
Combining discrete probability distributions and combinatorial optimization problems with neural network components has numerous applications but poses several challenges. We propose Implicit Maximum Likelihood Estimation (I-MLE), a framework for end-to-end learning of models combining discrete exponential family distributions and differentiable neural components. I-MLE is widely applicable as it only requires the ability to compute the most probable states and does not rely on smooth relaxations. The framework encompasses several approaches such as perturbation-based implicit differentiation and recent methods to differentiate through black-box combinatorial solvers. We introduce a novel class of noise distributions for approximating marginals via perturb-and-MAP. Moreover, we show that I-MLE simplifies to maximum likelihood estimation when used in some recently studied learning settings that involve combinatorial solvers. Experiments on several datasets suggest that I-MLE is competitive with and often outperforms existing approaches which rely on problem-specific relaxations.
Author Information
Mathias Niepert (University of Stuttgart / NEC Labs Europe)
Pasquale Minervini (University College London)
Luca Franceschi (University College London)
More from the Same Authors
-
2022 : Discrete Learning Of DAGs Via Backpropagation »
Andrew Wren · Pasquale Minervini · Luca Franceschi · Valentina Zantedeschi -
2022 : Discrete Learning Of DAGs Via Backpropagation »
Andrew Wren · Pasquale Minervini · Luca Franceschi · Valentina Zantedeschi -
2022 Poster: ReFactor GNNs: Revisiting Factorisation-based Models from a Message-Passing Perspective »
Yihong Chen · Pushkar Mishra · Luca Franceschi · Pasquale Minervini · Pontus Lars Erik Saito Stenetorp · Sebastian Riedel -
2021 Poster: Efficient Learning of Discrete-Continuous Computation Graphs »
David Friede · Mathias Niepert -
2017 Poster: Learning Graph Representations with Embedding Propagation »
Alberto Garcia Duran · Mathias Niepert -
2016 Poster: Discriminative Gaifman Models »
Mathias Niepert