In plain words: Cortex is a compiler that writes the entire code for deep learning models that call themselves repeatedly, instead of the usual two-step fix of graph tweaks plus ready-made chip kernels. That end-to-end control cut inference latency by up to 14 times across different hardware.
Abstract · Cortex: A Compiler for Recursive Deep Learning Models
Optimizing deep learning models is generally performed in two steps: (i) high-level graph optimizations such as kernel fusion and (ii) low level kernel optimizations such as those found in vendor libraries. This approach often leaves significant performance on the table, especially for the case of recursive deep learning models. In this paper, we present Cortex, a compiler-based approach to generate highly-efficient code for recursive models for low latency inference. Our compiler approach and low reliance on vendor libraries enables us to perform end-to-end optimizations, leading to up to 14X lower inference latencies over past work, across different backends.
Pratik Fegade, Tianqi Chen, Phillip B. Gibbons, Todd C. Mowry
arXiv:2011.01383 · cs.LG, cs.DC · submitted Nov 2, 2020 · updated Mar 5, 2021
abstract · pdf · html · 11 pages, 12 figures and 6 tables