Search papers, labs, and topics across Lattice.
The paper introduces Tiny-DroNeRF, a lightweight neural radiance field (NeRF) model based on Instant-NGP, optimized to run on ultra-low-power microcontrollers (MCUs) aboard nano-drones for dense 3D scene reconstruction. To further enhance the model, they implement a federated learning scheme to distribute training across multiple nano-drones, addressing memory limitations and improving reconstruction accuracy. Experimental results demonstrate a 96% reduction in memory footprint compared to Instant-NGP with a small accuracy drop, and improved overall reconstruction accuracy through federated learning.
NeRFs can now fly: Tiny-DroNeRF slashes the memory footprint of neural radiance fields by 96% to enable 3D scene reconstruction on nano-drones with ultra-low-power MCUs, and then uses federated learning to boost accuracy.
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.
