让不到30克的微型无人机在极低功耗下实现高精度3D重建。
Tiny-DroNeRF: Tiny Neural Radiance Fields aboard Federated Learning-enabled Nano-drones
- 基于Instant-NGP压缩模型,适配100毫瓦级超低功耗芯片。
- 内存减少96%,重建精度仅下降5.7分贝,仍保持高保真。
- 多机协同联邦学习,突破单机存储限制,提升整体建模效果。
重量低于30克的微型飞行机器人可凭借其敏捷性和紧凑形态,自主探索工业巡检与搜救任务中的复杂狭窄环境。然而,其资源受限——微控制器(MCU)功耗低于100毫瓦,算力约100亿次/秒,内存不足100兆字节。尽管如此,我们仍致力于在微型无人机上实现复杂的视觉任务,如密集3D场景重建,这是空间感知与运动规划的基础能力。当前高性能3D重建方法依赖神经辐射场(NeRF),需数GB内存和高功耗GPU支持。本文提出Tiny-DroNeRF,一种基于Instant-NGP的轻量化NeRF模型,专为搭载于纳米无人机上的GAP9超低功耗MCU优化。进一步结合分布式联邦学习框架,在多台纳米无人机间协同训练模型。实验表明,与Instant-NGP相比,Tiny-DroNeRF内存占用降低96%,重建精度仅下降5.7 dB;联邦学习使模型可利用远超单机存储容量的数据,显著提升整体重建质量。本工作首次实现超低功耗MCU上的NeRF训练与纳米无人机间的联邦学习协同。
原文摘要 · Abstract (English)
Sub-30g nano-sized aerial robots can leverage their agility and form factor to autonomously explore cluttered and narrow environments, like in industrial inspection and search and rescue missions. However, the price for their tiny size is a strong limit in their resources, i.e., sub-100 mW microcontroller units (MCUs) delivering $\sim$100 GOps/s at best, and memory budgets well below 100 MB. Despite these strict constraints, we aim to enable complex vision-based tasks aboard nano-drones, such as dense 3D scene reconstruction: a key robotic task underlying fundamental capabilities like spatial awareness and motion planning. Top-performing 3D reconstruction methods leverage neural radiance fields (NeRF) models, which require GBs of memory and massive computation, usually delivered by high-end GPUs consuming 100s of Watts. Our work introduces Tiny-DroNeRF, a lightweight NeRF model, based on Instant-NGP, and optimized for running on a GAP9 ultra-low-power (ULP) MCU aboard our nano-drones. Then, we further empower our Tiny-DroNeRF by leveraging a collaborative federated learning scheme, which distributes the model training among multiple nano-drones. Our experimental results show a 96% reduction in Tiny-DroNeRF's memory footprint compared to Instant-NGP, with only a 5.7 dB drop in reconstruction accuracy. Finally, our federated learning scheme allows Tiny-DroNeRF to train with an amount of data otherwise impossible to keep in a single drone's memory, increasing the overall reconstruction accuracy. Ultimately, our work combines, for the first time, NeRF training on an ULP MCU with federated learning on nano-drones.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。