Cited by 52Advanced Neural Network ApplicationsLow-power high-performance VLSI designNumerical Methods and Algorithms

SOLE: Hardware-Software Co-design of Softmax and LayerNorm for Efficient Transformer Inference

Open access
Wang, Wenxun, Zhou, Shuchang, Sun, Wenyu, Sun, Peiqin, Liu, Yongpan·2023·2023 IEEE/ACM International Conference on Computer Aided Design (ICCAD)

Abstract

Transformers have shown remarkable performance in both natural language processing (NLP) and computer vision (CV) tasks. However, their real-time inference speed and efficiency are limited due to the inefficiency in Softmax and Layer Normalization (LayerNorm). Previous works based on function approximation suffer from inefficient implementation as they place emphasis on computation while disregarding memory overhead concerns. Moreover, such methods rely on retraining to compensate for approximation error which can be costly and inconvenient. In this paper, we present SOLE, a hardware-software co-design for Softmax and LayerNorm which is composed of E2Softmax and AILayerNorm. E2Softmax utilizes log2 quantization of exponent function and log-based division to approximate Softmax while AILayerNorm adopts low-precision statistic calculation. Compared with state-of-the-art designs, we achieve both low-precision calculation and low bit-width storage on Softmax and LayerNorm. Experiments show that SOLE maintains inference accuracy without retraining while offering orders of magnitude speedup and energy savings over GPU, achieving 3.04×, 3.86× energy-efficiency improvements and 2.82×, 3.32× area-efficiency improvements over prior state-of-the-art custom hardware for Softmax and LayerNorm, respectively.

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

Crossref · 10.1109/iccad57390.2023.10323725

Cites

36 of 36 references verified0 citation errors · 0 not checkable

Statistics

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • SOLE's E2Softmax Unit avoids large lookup tables and multiplications, uses log2 quantization on exponent function outputs, and stores intermediate results in 4-bit representations, reducing memory usage.

    “As shown in Fig.11, its E2Softmax Unit avoids large lookup tables and multiplications, uses log2 quantization on exponent function outputs, and stores intermediate results in 4-bit representations, reducing memory usage.”
  • SOLE designs custom hardware for efficient transformer inference, optimizing Softmax and LayerNorm operations.

    “SOLE [23] designs custom hardware for efficient transformer inference, optimizes Softmax and LayerNorm operations.”
  • SOLE designs a two-stage LayerNorm Unit that facilitates pipelined computations and flexible batch processing.

    “This design facilitates pipelined computations and flexible batch processing, with robust error tolerance and reduced buffering and multiplication.”
  • SOLE employs dynamic compression for Layer Normalization statistics via Power-of-Two Factor.

    “Additionally, SOLE employs dynamic compression for Layer Normalization statistics via Power-of-Two Factor [43], and innovatively designs a two-stage LayerNorm Unit.”
  • SOLE introduces E2Softmax, which processes 8-bit quantized inputs using fixed-point arithmetic, employs log2 quantization post-exponentiation and an approximate logarithmic division.

    “SOLE [23] introduces E2Softmax, which processes 8-bit quantized inputs using fixed-point arithmetic. This method employs log2 quantization post-exponentiation and an approximate logarithmic division, eschewing traditional high-precision operations.”

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Cites

Flagged first, then by citations

3 of the 39 references couldn’t be matched to a known paper.

Research works printed but not linked: 2

  1. A. Dosovitskiy, L. Beyer, A. Kolesnikov, D. Weissenborn, X. Zhai, T. Unterthiner, M. Dehghani, M. Minderer, G. Heigold, S. Gelly, et al., "An image is worth 16x16 words: Transformers for image recognition at scale," arXiv preprint arXiv:2010.11929, 2020. · arXiv
  2. J. Yu, J. Park, S. Park, M. Kim, S. Lee, D. H. Lee, and J. Choi, "Nn-lut: neural approximation of non-linear operations for efficient transformer inference," in Proceedings of the 59th ACM/IEEE Design Automation Conference, pp. 577-582, 2022.

Cited by

Sorted by citations

Authors
Wang, W., Zhou, S., Sun, W., Sun, P., & Liu, Y.
Affiliations
  • Tsinghua University (CN)
  • Megvii (China) (CN)
Venue
2023 IEEE/ACM International Conference on Computer Aided Design (ICCAD)
Publisher
IEEE
Published
2023-10-28
Pages
1-9
OpenAlex
W4389166736
Cited by
51
References
39
Citations by year
2026: 10 · 2025: 37 · 2024: 4
FWCI
5.45
Type
conference paper
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
  • Physical Sciences › Engineering › Electrical and Electronic Engineering › Low-power high-performance VLSI design
  • Physical Sciences › Computer Science › Computational Theory and Mathematics › Numerical Methods and Algorithms
Keywords
hardware-software co-design, softmax approximation, transformer inference, low-precision arithmetic, energy-efficient inference, area efficiency

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review