arXiv:2603.09511cs.ARcs.LG2026-03中稿 · DATE 2026被引 1

在超低功耗设备上实现小型Transformer模型的高效训练,兼顾隐私与性能。

TrainDeeploy: Hardware-Accelerated Parameter-Efficient Fine-Tuning of Small Transformer Models at the Extreme Edge

  • 融合高效推理与设备端训练,支持CNN与Transformer统一部署。
  • LoRA降低动态内存23%,参数量减少15倍,训练速度达每秒11张图像。
  • 首次在极边缘RISC-V芯片上完成Transformer端到端微调,适合嵌入式场景。

设备端神经网络微调可在保护数据隐私的同时实现长期适应。然而,反向传播带来的高计算与内存开销对超低功耗、内存受限的极边缘设备构成重大挑战,注意力机制模型因架构复杂性和计算规模而更加严峻。本文提出TrainDeeploy框架,统一异构超低功耗SoC上的高效推理与设备端训练。该框架首次为极边缘SoC提供完整的端到端训练流水线,支持CNN与Transformer模型,并集成层选择性微调与低秩适配(LoRA)等多种策略。在基于RISC-V的异构SoC上,成功实现紧凑卷积变压器(CCT)的端到端微调,最高达每秒11张图像。实验表明,与全反向传播相比,LoRA可降低动态内存使用23%,减少可训练参数与梯度数量15倍,内存传输量减少1.6倍。TrainDeeploy在CCT(0.28M参数,71-126M FLOPs)上达到最高4.6 FLOP/周期,在Deep-AE(0.27M参数,0.8M FLOPs)上达到13.4 FLOP/周期,扩展了前序框架在极边缘平台支持参数高效微调的能力。

原文摘要 · Abstract (English)

On-device tuning of deep neural networks enables long-term adaptation at the edge while preserving data privacy. However, the high computational and memory demands of backpropagation pose significant challenges for ultra-low-power, memory-constrained extreme-edge devices. These challenges are further amplified for attention-based models due to their architectural complexity and computational scale. We present TrainDeeploy, a framework that unifies efficient inference and on-device training on heterogeneous ultra-low-power System-on-Chips (SoCs). TrainDeeploy provides the first complete on-device training pipeline for extreme-edge SoCs supporting both Convolutional Neural Networks (CNNs) and Transformer models, together with multiple training strategies such as selective layer-wise fine-tuning and Low-Rank Adaptation (LoRA). On a RISC-V-based heterogeneous SoC, we demonstrate the first end-to-end on-device fine-tuning of a Compact Convolutional Transformer (CCT), achieving up to 11 trained images per second. We show that LoRA reduces dynamic memory usage by 23%, decreases the number of trainable parameters and gradients by 15x, and reduces memory transfer volume by 1.6x compared to full backpropagation. TrainDeeploy achieves up to 4.6 FLOP/cycle on CCT (0.28M parameters, 71-126M FLOPs) and up to 13.4 FLOP/cycle on Deep-AE (0.27M parameters, 0.8M FLOPs), while expanding the scope of prior frameworks to support both CNN and Transformer models with parameter-efficient tuning on extreme-edge platforms.

边缘计算模型微调LoRARISC-V

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