Sense Less, Generate More: Pre-training LiDAR Perception with Masked Autoencoders for Ultra-Efficient 3D Sensing

Sina Tayebati,Theja Tulabandhula,Amit R. Trivedi
2024-06-12
Abstract:In this work, we propose a disruptively frugal LiDAR perception dataflow that generates rather than senses parts of the environment that are either predictable based on the extensive training of the environment or have limited consequence to the overall prediction accuracy. Therefore, the proposed methodology trades off sensing energy with training data for low-power robotics and autonomous navigation to operate frugally with sensors, extending their lifetime on a single battery charge. Our proposed generative pre-training strategy for this purpose, called as radially masked autoencoding (R-MAE), can also be readily implemented in a typical LiDAR system by selectively activating and controlling the laser power for randomly generated angular regions during on-field operations. Our extensive evaluations show that pre-training with R-MAE enables focusing on the radial segments of the data, thereby capturing spatial relationships and distances between objects more effectively than conventional procedures. Therefore, the proposed methodology not only reduces sensing energy but also improves prediction accuracy. For example, our extensive evaluations on Waymo, nuScenes, and KITTI datasets show that the approach achieves over a 5% average precision improvement in detection tasks across datasets and over a 4% accuracy improvement in transferring domains from Waymo and nuScenes to KITTI. In 3D object detection, it enhances small object detection by up to 4.37% in AP at moderate difficulty levels in the KITTI dataset. Even with 90% radial masking, it surpasses baseline models by up to 5.59% in mAP/mAPH across all object classes in the Waymo dataset. Additionally, our method achieves up to 3.17% and 2.31% improvements in mAP and NDS, respectively, on the nuScenes dataset, demonstrating its effectiveness with both single and fused LiDAR-camera modalities. <a class="link-external link-https" href="https://github.com/sinatayebati/Radial_MAE" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
### What problem does this paper attempt to solve? This paper aims to solve the high - energy - consumption problem of LiDAR (Light Detection and Ranging) sensors in low - power - consumption robot applications. Specifically, the author proposes a novel generative pre - training method, which uses Masked Autoencoders (MAE) to reduce the energy consumption of the LiDAR perception system while improving the prediction accuracy. #### Main problems: 1. **High energy consumption of LiDAR**: Due to its active sensing mechanism (emitting lasers and measuring reflections), LiDAR sensors consume significantly more energy compared to passive sensors such as cameras. This makes LiDAR face challenges in low - power - consumption robot applications that need to run for a long time and have limited battery resources. 2. **Trade - off between energy and performance**: Traditional LiDAR systems often require higher energy consumption when pursuing high - precision and long - distance sensing. For example, as the sensing distance increases, the laser emission power \( P_{\text{laser}} \) needs to increase according to the relationship of \( R^4 \), which poses strict requirements for battery life and thermal management. 3. **Data processing efficiency**: Although existing methods use generative models to compress point - cloud data to accelerate downstream tasks, they fail to fundamentally reduce the energy consumption of the sensor itself. #### Solutions: To address the above challenges, the author proposes Radially Masked Autoencoding (R - MAE). R - MAE solves the problems in the following ways: - **Generate rather than sense**: R - MAE uses a generative model to predict the parts of the environment that are predictable or have a relatively small impact on the overall prediction accuracy, thereby reducing the need for actual sensing. - **Selectively activate lasers**: In actual operation, R - MAE can randomly generate data in specific angular regions by selectively activating and controlling the laser power, further reducing energy consumption. - **Efficient feature learning**: Through pre - training, R - MAE can capture spatial relationships and distances between objects, improving the accuracy of downstream tasks. Experimental results show that on datasets such as Waymo, nuScenes, and KITTI, the average precision of R - MAE in detection tasks has increased by more than 5%, especially with significant improvement in small - object detection. In conclusion, this paper effectively solves the high - energy - consumption problem of LiDAR in low - power - consumption robot applications and improves the performance of the perception system by introducing R - MAE.