Timezone: »

Fast and Flexible Monotonic Functions with Ensembles of Lattices
Mahdi Milani Fard · Kevin Canini · Andrew Cotter · Jan Pfeifer · Maya Gupta

Wed Dec 07 09:00 AM -- 12:30 PM (PST) @ Area 5+6+7+8 #74 #None

For many machine learning problems, there are some inputs that are known to be positively (or negatively) related to the output, and in such cases training the model to respect that monotonic relationship can provide regularization, and makes the model more interpretable. However, flexible monotonic functions are computationally challenging to learn beyond a few features. We break through this barrier by learning ensembles of monotonic calibrated interpolated look-up tables (lattices). A key contribution is an automated algorithm for selecting feature subsets for the ensemble base models. We demonstrate that compared to random forests, these ensembles produce similar or better accuracy, while providing guaranteed monotonicity consistent with prior knowledge, smaller model size and faster evaluation.

Author Information

Mahdi Milani Fard (Google)
Kevin Canini (Google)
Andy Cotter (Google)
Jan Pfeifer Pfeifer (Google)
Maya Gupta (Google)

More from the Same Authors