English

A-IO: Adaptive Inference Orchestration for Memory-Bound NPUs

Distributed, Parallel, and Cluster Computing 2026-04-16 v2 Artificial Intelligence

Abstract

During the deployment of Large Language Models (LLMs), the autoregressive decoding phase on heterogeneous NPU platforms (e.g., Ascend 910B) faces severe memory-bound challenges. This study reveals the ``Model Scaling Paradox'' caused by the static deployment of single-sized models. It also points out the kernel synchronization overhead of fine-grained speculative decoding \cite{leviathan2023fast, chen2023speculative} under NPU computational graph compilation, and the severe limitations of purely relying on micro-level acceleration algorithms like Prompt LookUp Decoding (PLD)

Keywords

Cite

@article{arxiv.2604.09752,
  title  = {A-IO: Adaptive Inference Orchestration for Memory-Bound NPUs},
  author = {Chen Zhang and Yan Ding and Haotian Wang and Chubo Liu and Keqin Li and Kenli Li},
  journal= {arXiv preprint arXiv:2604.09752},
  year   = {2026}
}
R2 v1 2026-07-01T12:03:35.904Z