When you access a web page, bidders such as Criteo must determine in a few dozens of milliseconds if they want to purchase the advertising space on the page. At that moment, a real-time auction takes place, and once you remove all the communication exchange delays, it leaves a handful of milliseconds to compute exactly how much to bid. In the past year, Criteo has put a large amount of effort into reshaping its in-house machine learning stack responsible for making such predictions—in particular, opening it to new technologies such as TensorFlow.
Unfortunately, even for simple logistic regression models and small neural networks, Criteo’s initial TensorFlow implementations saw inference time increase by 100, going from 300 microseconds to 30 milliseconds.
Nicolas Kowalski and Axel Antoniotti outline how Criteo approached this issue, discussing how Criteo profiled its model to understand its bottleneck; why commonly shared solutions such as optimizing TensorFlow build for the target hardware, freezing and cleaning up the model, and using accelerated linear algebra (XLA) ended up being lackluster; and how Criteo rewrote is models from scratch, reimplementing cross-features and hashing functions using low-level TF operations in order to factorize as much as possible all TensorFlow nodes in its model.
- A basic understanding of how TensorFlow and TensorFlow Serving work
- Experience optimizing TensorFlow models for serving (useful but not required)
What you'll learn
- Understand how to optimize a TensorFlow model before serving it online
- Discover how to profile a TensorFlow model with a complex preprocessing architecture
- Learn how and when to replace feature columns with custom cross-features and hashing functions to factorize and drastically reduce the number of nodes in the model
This session is from the 2019 O'Reilly TensorFlow World Conference in Santa Clara, CA.
- Title: How Criteo optimized and sped up its TensorFlow models by 10x and served them under 5 ms
- Release date: February 2020
- Publisher(s): O'Reilly Media, Inc.
- ISBN: 0636920372523
You might also like
TensorFlow model optimization: Quantization and pruning
Join Raziel Alverez (Google) to learn from TensorFlow performance experts who cover topics including optimization, quantization, …
Head First Design Patterns, 2nd Edition
You know you don’t want to reinvent the wheel, so you look to design patterns—the lessons …
O'Reilly Software Architecture Conference 2020 - New York, NY
O’Reilly Media’s Software Architecture Conference (SACON) New York 2020 was Maggie Carrol’s (MAG Aerospace) compelling examination …
40 Algorithms Every Programmer Should Know
Learn algorithms for solving classic computer science problems with this concise guide covering everything from fundamental …