Latency-Aware Neural Architecture Search with Multi-Objective Bayesian Optimization

By David Eriksson, Pierce I-Jen Chuang, Sam Daulton, Ahmed Aly, Arun Babu, Akshat Shrivastava, Peng Xia, Shicong Zhao, Ganesh Venkatesh, M. Balandat. 2021

21 citations recorded by the source corpus.

When tuning the architecture and hyperparameters of large machine learning models for on-device deployment, it is desirable to understand the optimal trade-offs between on-device latency and model accuracy. In this work, we leverage recent methodological advances in Bayesian optimization over high-dimensional search spaces and multi-objective Bayesian optimization to efficiently explore these trade-offs for a production-scale on-device natural language understanding model at Facebook.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Management Science and Operations Research, Model Architectures

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. Carole-Jean Wu
  2. Bilge Acun
  3. Michael G. Rabbat
  4. Kim M. Hazelwood
  5. Hsien-Hsin S. Lee
  6. Udit Gupta
  7. Newsha Ardalani
  8. Maximilian Balandat

In Computer Vision and Pattern Recognition

  1. Carole-Jean Wu
  2. Michael G. Rabbat
  3. Kim M. Hazelwood
  4. Bilge Acun
  5. Hsien-Hsin S. Lee
  6. Salvatore Candido
  7. Udit Gupta
  8. Newsha Ardalani

In Management Science and Operations Research

  1. Michael G. Rabbat
  2. Maximilian Balandat
  3. Myle Ott
  4. Konstantinos I. Tsianos

In Model Architectures

  1. Michael G. Rabbat
  2. Udit Gupta
  3. Newsha Ardalani
  4. Carole-Jean Wu
  5. Myle Ott

Citations in the corpus