Abstract:Stochastic gradient descent (SGD) is widely used for training generalized linear models (GLMs), such as support vector machine and logistic regression, on large industry datasets. Such a training consumes plenty of computing power and therefore plenty of accelerators are proposed to accelerate the GLM training. However, real-world datasets are always highly sparse. For example, YouTube’s social network connectivity contains only 2.31% non-zero elements. It is not trivial to design an accelerator that is able to efficiently train on a sparse dataset that is stored in a compressed sparse format (e.g., Compressed Sparse Row (CSR) format). The design of such an accelerator faces three challenges: (1) bank conflicts, which may happen when multiple processing engines in the accelerator access multiple memory banks; (2) complex interconnections, which are necessary to allow all processing engines to access any memory bank; and (3) high synchronization overhead, since each sample in sparse dataset has a different number of non-zero elements and these elements have different distributions, thus it is hard to overlap gradient computation and model update of neighboring batches. To this end, we propose SparseACC, a sparsity-aware accelerator for training generalized linear models. SparseACC is based on two key mechanisms. First, a software/hardware co-design approach solves the first two design challenges by proposing a novel bank-conflict-free and bank-balanced CSR format. Second, a weight-aware ping-pong model solves the third challenge, thus maximizing the utilization of the processing engines. SparseACC leverages these two mechanisms to orchestrate training over sparse datasets, such that the training time decreases linearly with the sparsity of the dataset. We prototype SparseACC on a Xilinx Alveo U280 FPGA 1. The experimental evaluation shows that SparseACC converges up to 3.5×, 18×, 38×, and 110× faster than the state-of-the-art counterparts on a sparse accelerator, a Tesla V100 GPU, an Intel i9-10900k CPU, and a dense accelerator, respectively.

Sparse-HeteroCL: from Sparse Tensor Algebra to Highly Customized Accelerators on FPGAs.

HeteroFlow: An Accelerator Programming Model with Decoupled Data Placement for Software-Defined FPGAs

Lasa: Abstraction and Specialization for Productive and Performant Linear Algebra on FPGAs

Extending High-Level Synthesis for Task-Parallel Programs

Sgap: Towards Efficient Sparse Tensor Algebra Compilation for GPU

An Optimization Toolchain Design Of Deep Learning Deployment Based On Heterogeneous Computing Platform

SPARTA: High-Level Synthesis of Parallel Multi-Threaded Accelerators

Compilation of Modular and General Sparse Workspaces

AHA: An Agile Approach to the Design of Coarse-Grained Reconfigurable Accelerators and Compilers

Automatically harnessing sparse acceleration

Compiling Halide Programs to Push-Memory Accelerators

A Method for Efficient Heterogeneous Parallel Compilation: A Cryptography Case Study

HLSPilot: LLM-based High-Level Synthesis

Programming Heterogeneous Systems from an Image Processing DSL

UniSparse: An Intermediate Language for General Sparse Format Customization

FEASTA: A Flexible and Efficient Accelerator for Sparse Tensor Algebra in Machine Learning

LLM-Aided Compilation for Tensor Accelerators

Spatial: a language and compiler for application accelerators

GAHLS: an optimized graph analytics based high level synthesis framework

Allo: A Programming Model for Composable Accelerator Design

SparseACC: A Generalized Linear Model Accelerator for Sparse Datasets