Low-Precision Hardware Architectures Meet Recommendation Model Inference at Scale

By Zhaoxia Deng, Jongsoo Park, Ping Tak Peter Tang, Haixin Liu, J. Yang, Hector Yuen, Jianyu Huang, Daya Shanker Khudia, Xiaohan Wei, Ellie Wen and 8 more. 2021

14 citations recorded by the source corpus.

Tremendous success of machine learning (ML) and the unabated growth in model complexity motivated many ML-specific designs in hardware architectures to speed up the model inference. While these architectures are diverse, highly optimized low-precision arithmetic is a component shared by most. Nevertheless, recommender systems important to Facebook’s personalization services are demanding and complex: They must serve billions of users per month responsively with low latency while maintaining high prediction accuracy. Do these low-precision architectures work well with our production recommendation systems? They do. But not without significant effort. In this article, we share our search strategies to adapt reference recommendation models to low-precision hardware, our optimization of low-precision compute kernels, and the tool chain to maintain our models’ accuracy throughout their lifespan. We believe our lessons from the trenches can promote better codesign between hardware architecture and software engineering, and advance the state of the art of ML in industry.

Subfields: Artificial Intelligence, Computer Networks and Communications, Computer Vision and Pattern Recognition, Hardware and Architecture, Efficiency & Compression

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 Networks and Communications

  1. Carole-Jean Wu
  2. Udit Gupta
  3. Hsien-Hsin S. Lee
  4. Kiwan Maeng
  5. Michael G. Rabbat
  6. Kim M. Hazelwood
  7. Newsha Ardalani
  8. Myle Ott

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 Hardware and Architecture

  1. Kiwan Maeng
  2. Michael G. Rabbat
  3. Kim M. Hazelwood
  4. Newsha Ardalani
  5. Carole-Jean Wu
  6. Ramya Raghavendra
  7. Myle Ott

In Efficiency & Compression

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

Citations in the corpus