通过稀疏化提升线性RNN推理效率,边缘设备上实现42倍低延迟。
Accelerating Linear Recurrent Neural Networks for the Edge with Unstructured Sparsity
- 采用无结构稀疏化压缩线性RNN,降低计算与内存开销
- 在同精度下减少2倍计算量、36%内存占用,性能更优
- 部署于英特尔Loihi 2芯片,实测延迟降42倍、能耗降149倍
线性循环神经网络可在推理时保持恒定内存和每标记时间,适合边缘端流式应用。然而资源受限环境下的部署需硬件感知优化以降低延迟与功耗。无结构稀疏化提供了有效解决方案,在兼容硬件上可大幅减少计算与内存需求。本文通过缩放研究分析了不同计算预算下的性能-效率权衡。结果表明,高度稀疏的线性RNN始终优于稠密基线,在同精度下计算量减少2倍、内存降低36%。模型在真实音频去噪流任务中达到最先进水平。通过将稀疏模型量化为定点数并部署于Intel Loihi 2类脑芯片进行实时处理,相比边缘GPU上的稠密模型,实现42倍更低延迟和149倍更低能耗。研究展示了无结构稀疏化的变革潜力,为资源受限环境中的高效循环网络铺平道路。
原文摘要 · Abstract (English)
Linear recurrent neural networks enable powerful long-range sequence modeling with constant memory usage and time-per-token during inference. These architectures hold promise for streaming applications at the edge, but deployment in resource-constrained environments requires hardware-aware optimizations to minimize latency and energy consumption. Unstructured sparsity offers a compelling solution, enabling substantial reductions in compute and memory requirements--when accelerated by compatible hardware platforms. In this paper, we conduct a scaling study to investigate the Pareto front of performance and efficiency across inference compute budgets. We find that highly sparse linear RNNs consistently achieve better efficiency-performance trade-offs than dense baselines, with 2x less compute and 36% less memory at iso-accuracy. Our models achieve state-of-the-art results on a real-time streaming task for audio denoising. By quantizing our sparse models to fixed-point arithmetic and deploying them on the Intel Loihi 2 neuromorphic chip for real-time processing, we translate model compression into tangible gains of 42x lower latency and 149x lower energy consumption compared to a dense model on an edge GPU. Our findings showcase the transformative potential of unstructured sparsity, paving the way for highly efficient recurrent neural networks in real-world, resource-constrained environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。