arXiv · 2503.19640
An Efficient Data Reuse with Tile-Based Adaptive Stationary for Transformer Accelerators
Abstract
Transformer-based models have become the \textit{de facto} backbone across many fields, such as computer vision and natural language processing. However, as these models scale in size, external memory access (EMA) for weight and activations becomes a critical bottleneck due to its significantly higher energy consumption compared to internal computations. While most prior work has focused on optimizing the self-attention mechanism, little attention has been given to optimizing data transfer during linear projections, where EMA costs are equally important. In this paper, we propose the Tile-based Adaptive Stationary (TAS) scheme that selects the input or weight stationary in a tile granularity, based on the input sequence length. Our experimental results demonstrate that TAS can significantly reduce EMA by more than 97\% compared to traditional stationary schemes, while being compatible with various attention optimization techniques and hardware accelerators.
Explore related subjects
Keep this discovery
Tseng-Jen Li, Tian-Sheuan Chang. 2025-03-25. An Efficient Data Reuse with Tile-Based Adaptive Stationary for Transformer Accelerators. https://arxiv.org/abs/2503.19640
Cite the original work for its findings. Save a collection to share your selection of sources.