arXiv:2411.06346cs.LGcs.CV2024-11NeurIPS被引 7

用张量分解压缩激活图,让嵌入式设备也能高效训练深度模型

Activation Map Compression through Tensor Decomposition for Deep Learning

  • 用SVD和高阶SVD压缩激活图存储,降低内存占用
  • 实验显示在主流模型上兼顾泛化性能与内存节省
  • 适合资源受限的边缘AI设备训练场景

物联网与深度学习正协同快速发展,亟需统一为边缘人工智能(Edge AI)框架。尽管设备端推理已广泛研究,但反向传播因计算与内存开销过大,仍面临挑战。本文基于张量分解技术,解决反向传播中激活图存储带来的主要瓶颈。通过比较奇异值分解(SVD)及其高阶变体(HOSVD)对激活图的压缩效果,发现低秩分解可在保留关键学习特征的同时显著减少内存占用,并提供收敛性理论保证。在主流架构与任务上的实验表明,该方法在泛化能力与内存占用的权衡上优于现有先进方案,具有帕累托优势。

原文摘要 · Abstract (English)

Internet of Things and Deep Learning are synergetically and exponentially growing industrial fields with a massive call for their unification into a common framework called Edge AI. While on-device inference is a well-explored topic in recent research, backpropagation remains an open challenge due to its prohibitive computational and memory costs compared to the extreme resource constraints of embedded devices. Drawing on tensor decomposition research, we tackle the main bottleneck of backpropagation, namely the memory footprint of activation map storage. We investigate and compare the effects of activation compression using Singular Value Decomposition and its tensor variant, High-Order Singular Value Decomposition. The application of low-order decomposition results in considerable memory savings while preserving the features essential for learning, and also offers theoretical guarantees to convergence. Experimental results obtained on main-stream architectures and tasks demonstrate Pareto-superiority over other state-of-the-art solutions, in terms of the trade-off between generalization and memory footprint.

边缘计算张量分解模型压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。