Task-based End-to-end Model Learning

By Priya Lekha Donti, Brandon Amos, J. Kolter. 2017

25 citations recorded by the source corpus.

As machine learning techniques have become more ubiquitous, it has become common to see machine learning prediction algorithms operating within some larger process. However, the criteria by which we train machine learning algorithms often differ from the ultimate criteria on which we evaluate them. This paper proposes an end-to-end approach for learning probabilistic machine learning models within the context of stochastic programming, in a manner that directly captures the ultimate task-based objective for which they will be used. We then present two experimental evaluations of the proposed approach, one as applied to a generic inventory stock problem and the second to a real-world electrical grid scheduling task. In both cases, we show that the proposed approach can outperform both a traditional modeling approach and a purely black-box policy optimization approach.

Subfields: Artificial Intelligence, Electrical and Electronic Engineering, Management Science and Operations Research, Optimization & Training

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Adam N. Elmachtoub
  2. Paul Grigas
  3. Ryan McNellis
  4. Javier F. Peña

In Electrical and Electronic Engineering

  1. Adam N. Elmachtoub
  2. Paul Grigas

In Management Science and Operations Research

  1. Adam N. Elmachtoub
  2. Paul Grigas
  3. Ryan McNellis

In Optimization & Training

  1. Adam N. Elmachtoub
  2. Paul Grigas

Citations in the corpus