在超低功耗设备上实现单目深度估计的本地自适应,仅用3000样本17.8分钟即大幅提效
Multi-modal On-Device Learning for Monocular Depth Estimation on Ultra-low-power MCUs
- 设备端多模态学习:相机与微型深度传感器协同采集伪标签用于本地微调
- 仅需1.2MB内存即可完成训练,相比全量更新减少54%内存占用
- 首次实现在真实场景中仅用3000样本、17.8分钟完成优化,误差从4.9米降至0.6米
单目深度估计(MDE)对超低功耗物联网平台的空间感知应用至关重要。然而,为物联网节点设计的深度神经网络参数量有限,在实际部署中若传感器数据与训练集差异较大时,精度会显著下降。为此,本文提出一种部署于集成Greenwaves GAP9 MCU、80mW单目相机与8×8像素深度传感器的设备端多模态学习技术,整体功耗约300mW。正常运行时,使用仅107千参数的μPyD-Net模型进行推理;当进入新环境时,激活深度传感器同步采集伪标签,随后在MCU上完成全部微调。为优化基于反向传播的设备端训练,提出新型内存驱动的稀疏更新机制,将微调内存降低至1.2MB(较全量更新减少54%),同时保持精度稳定(KITTI和NYUv2数据集分别仅下降2%和1.5%)。实地测试首次证明,该方法可在17.8分钟内完成设备端学习,仅用3000个自标注样本,将均方根误差从4.9米降至0.6米。
原文摘要 · Abstract (English)
Monocular depth estimation (MDE) plays a crucial role in enabling spatially-aware applications in Ultra-low-power (ULP) Internet-of-Things (IoT) platforms. However, the limited number of parameters of Deep Neural Networks for the MDE task, designed for IoT nodes, results in severe accuracy drops when the sensor data observed in the field shifts significantly from the training dataset. To address this domain shift problem, we present a multi-modal On-Device Learning (ODL) technique, deployed on an IoT device integrating a Greenwaves GAP9 MicroController Unit (MCU), a 80 mW monocular camera and a 8 x 8 pixel depth sensor, consuming $\approx$300mW. In its normal operation, this setup feeds a tiny 107 k-parameter $μ$PyD-Net model with monocular images for inference. The depth sensor, usually deactivated to minimize energy consumption, is only activated alongside the camera to collect pseudo-labels when the system is placed in a new environment. Then, the fine-tuning task is performed entirely on the MCU, using the new data. To optimize our backpropagation-based on-device training, we introduce a novel memory-driven sparse update scheme, which minimizes the fine-tuning memory to 1.2 MB, 2.2x less than a full update, while preserving accuracy (i.e., only 2% and 1.5% drops on the KITTI and NYUv2 datasets). Our in-field tests demonstrate, for the first time, that ODL for MDE can be performed in 17.8 minutes on the IoT node, reducing the root mean squared error from 4.9 to 0.6m with only 3 k self-labeled samples, collected in a real-life deployment scenario.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。