← Search

Benoit Steiner

9 accepted papers

2023

Learning Compiler Pass Orders using Coreset and Normalized Value Prediction

ICML 2023poster

Finding the optimal pass sequence of compilation can lead to a significant reduction in program size. Prior works on compilation pass ordering have two major drawbacks. They either require an excessive budget (in terms of the number of compilation passes) at compile time or fail to generalize to uns…

2023

Searching Large Neighborhoods for Integer Linear Programs with Contrastive Learning

ICML 2023poster

Integer Linear Programs (ILPs) are powerful tools for modeling and solving a large number of combinatorial optimization problems. Recently, it has been shown that Large Neighborhood Search (LNS), as a heuristic algorithm, can find high-quality solutions to ILPs faster than Branch and Bound. However,…

Cited by 40SourcePDFScholar
2023

SurCo: Learning Linear SURrogates for COmbinatorial Nonlinear Optimization Problems

ICML 2023poster

Optimization problems with nonlinear cost functions and combinatorial constraints appear in many real-world applications but remain challenging to solve efficiently compared to their linear counterparts. To bridge this gap, we propose $\textbf{\emph{\texttt{SurCo}}}$ that learns linear $\underline{\…

Cited by 32SourcePDFScholar
2022

Flashlight: Enabling Innovation in Tools for Machine Learning

ICML 2022spotlight

As the computational requirements for machine learning systems and the size and complexity of machine learning frameworks increases, essential framework innovation has become challenging. While computational needs have driven recent compiler, networking, and hardware advancements, utilization of tho…

2021

Learning Space Partitions for Path Planning

NeurIPS 2021poster

Path planning, the problem of efficiently discovering high-reward trajectories, often requires optimizing a high-dimensional and multimodal reward function. Popular approaches like CEM and CMA-ES greedily focus on promising regions of the search space and may get trapped in local maxima. DOO and VOO…

2019

PyTorch: An Imperative Style, High-Performance Deep Learning Library

NeurIPS 2019poster

Deep learning frameworks have often focused on either usability or speed, but not both. PyTorch is a machine learning library that shows that these two goals are in fact compatible: it was designed from first principles to support an imperative and Pythonic programming style that supports code as a…

2018

A Hierarchical Model for Device Placement

ICLR 2018poster

We introduce a hierarchical model for efficient placement of computational graphs onto hardware devices, especially in heterogeneous environments with a mixture of CPUs, GPUs, and other computational devices. Our method learns to assign graph operations to groups and to allocate those groups to avai…

Cited by 210SourcePDFScholar
2017

Device Placement Optimization with Reinforcement Learning

ICML 2017poster

The past few years have witnessed a growth in size and computational requirements for training and inference with neural networks. Currently, a common approach to address these requirements is to use a heterogeneous distributed environment with a mixture of hardware devices such as CPUs and GPUs. Im…

Cited by 556SourcePDFScholar