arXiv:2605.09357cs.DCcs.LG2026-05

将CNN模型拆分到多微控制器协同推理,突破单设备内存瓶颈

Split CNN Inference on Networked Microcontrollers

论文配图:Split CNN Inference on Networked Microcontrollers
图 1 · 摘自论文原文
  • 在子层粒度拆分模型,实现核级与神经元级的分布式计算
  • 8个MCU协同运行MobileNetV2,单设备峰值内存降低至原方案1/4
  • 适用于资源受限场景下的边缘智能部署,如物联网设备集群

在微控制器(MCU)上运行深度神经网络受制于有限的内存资源。尽管TinyML技术可缩减模型规模与计算量,但推理过程中中间激活值导致的峰值随机存取内存(RAM)占用仍使许多模型无法在独立MCU上运行。本文提出一种细粒度的网络化MCU协同推理系统,支持跨多设备的卷积神经网络(CNN)推理。核心思想是打破内存瓶颈需在子层而非层边界进行拆分。通过重解释预训练模型,实现核级与神经元级的参数与激活分布,并由轻量级资源感知协调器调度异构设备上的推理任务。我们在真实测试平台上实现了该系统,使用MobileNetV2在最多8个MCU上进行了评估。实验表明,原本无法在单个MCU上运行的CNN模型可通过网络化MCU协同执行,显著降低单设备峰值内存使用,同时保持实际端到端推理延迟。所有源代码见:https://github.com/shashsuresh/split-inference-on-MCUs。

原文摘要 · Abstract (English)

Running deep neural networks on microcontroller units (MCUs) is severely constrained by limited memory resources. While TinyML techniques reduce model size and computation, they often fail in practice due to excessive peak Random Access Memory (RAM) usage during inference, dominated by intermediate activations. As a result, many models remain infeasible on standalone MCUs. In this work, we present a fine-grained split inference system for networked MCUs that enables collaborative inference of Convolutional Neural Networks (CNN) models across multiple devices. Our key insight is that breaking the memory bottleneck requires splitting inference at sub-layer granularity rather than at layer boundaries. We reinterpret pre-trained models to enable kernel-wise and neuron-wise partitioning, and distribute both model parameters and intermediate activations across multiple MCUs. A lightweight, resource-aware coordinator orchestrates the inference across MCU devices with heterogeneous resources. We implement the proposed system on a real testbed and evaluate it on up to 8 MCUs using MobileNetV2, a representative CNN model. Our experimental results show that CNN models infeasible on a single MCU can be executed across networked MCUs, reducing the per-MCU peak RAM usage while maintaining the practical end-to-end inference latency. All the source code of this work can be found here: https://github.com/shashsuresh/split-inference-on-MCUs.

边缘计算模型拆分MCUTinyML

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