Paper

X-TIME: An in-memory engine for accelerating machine learning on tabular data with CAMs

Ziegler, Tobias, Roth, Ron M, Graves, Catherine E, Ignowski, Jim, Moon, John, Xu, Cong, Gajjar, Archit, Pedretti, Giacomo, Buonanno, Luca, Faraboschi, Paolo, Bruel, Pedro, Serebryakov, Sergey, Foltin, Martin

2024

Overview

Structured, or tabular, data is the most common format in data science. While deep learning models have proven formidable in learning from unstructured data such as images or speech, they are less accurate than simpler approaches when learning from tabular data. In contrast, modern tree-based Machine Learning (ML) models shine in extracting relevant information from structured data. An essential requirement in data science is to reduce model inference latency in cases where, for example, models are used in a closed loop with simulation to accelerate scientific discovery. However, the hardware acceleration community has mostly focused on deep neural networks and largely ignored other forms of machine learning. Previous work has described the use of an analog content addressable memory (CAM) component for efficiently mapping random forests. In this work, we focus on an overall analog-digital architecture implementing a novel increased precision analog CAM and a programmable network on chip allowing the inference of state-of-the-art tree-based ML models, such as XGBoost and CatBoost. Results evaluated in a single chip at 16nm technology show 119x lower latency at 9740x higher throughput compared with a state-of-the-art GPU, with a 19W peak power consumption.

• Publisher: Cornell University Library, arXiv.org
• Subject: Artificial neural networks / Associative memory / Closed loops / Data science / Deep learning / Inference / Machine learning / Network latency / Power consumption / State of the art / Structured data / System on chip / Tables (data) / Unstructured data