Sunshine to Rainstorm: Cross-Weather Knowledge Distillation for Robust 3D Object Detection (2402.18493v1)
Abstract: LiDAR-based 3D object detection models have traditionally struggled under rainy conditions due to the degraded and noisy scanning signals. Previous research has attempted to address this by simulating the noise from rain to improve the robustness of detection models. However, significant disparities exist between simulated and actual rain-impacted data points. In this work, we propose a novel rain simulation method, termed DRET, that unifies Dynamics and Rainy Environment Theory to provide a cost-effective means of expanding the available realistic rain data for 3D detection training. Furthermore, we present a Sunny-to-Rainy Knowledge Distillation (SRKD) approach to enhance 3D detection under rainy conditions. Extensive experiments on the WaymoOpenDataset large-scale dataset show that, when combined with the state-of-the-art DSVT model and other classical 3D detectors, our proposed framework demonstrates significant detection accuracy improvements, without losing efficiency. Remarkably, our framework also improves detection capabilities under sunny conditions, therefore offering a robust solution for 3D detection regardless of whether the weather is rainy or sunny
- Z. Yang, Y. Sun, S. Liu, and J. Jia, “3DSSD: Point-Based 3D Single Stage Object Detector,” in CVPR, 2020.
- X. Huang, P. Wang, X. Cheng, D. Zhou, Q. Geng, and R. Yang, “The ApolloScape Open Dataset for Autonomous Driving and its Application,” IEEE Trans. Pattern Anal. Mach. Intell., vol. 42, 2020.
- C. A. Diaz-Ruiz, Y. Xia, Y. You, J. Nino, J. Chen, J. Monica, X. Chen, K. Luo, Y. Wang, M. Emond, W.-L. Chao, B. Hariharan, K. Q. Weinberger, and M. Campbell, “Ithaca365: Dataset and Driving Perception Under Repeated and Challenging Weather Conditions,” in CVPR, 2022.
- M. Pitropov, D. Garcia, J. Rebello, M. Smart, C. Wang, K. Czarnecki, and S. Waslander, “Canadian Adverse Driving Conditions Dataset,” The International Journal of Robotics Research, vol. 40, 2021.
- W. Zheng, M. Hong, L. Jiang, and C.-W. Fu, “Boosting 3D Object Detection by Simulating Multimodality on Point Clouds,” in CVPR, 2022.
- W. Zheng, L. Jiang, F. Lu, Y. Ye, and C.-W. Fu, “Boosting Single-Frame 3D Object Detection by Simulating Multi-Frame Point Clouds,” in ACMMM, 2022.
- R. Heinzler, F. Piewak, P. Schindler, and W. Stork, “CNN-Based Lidar Point Cloud De-Noising in Adverse Weather,” IEEE Robotics and Automation Letters, vol. 5, 2020.
- H. Sheng, S. Cai, Y. Liu, B. Deng, J. Huang, X.-S. Hua, and M.-J. Zhao, “Improving 3D Object Detection With Channel-Wise Transformer,” in ICCV, 2021.
- N. Charron, S. Phillips, and S. L. Waslander, “De-noising of Lidar Point Clouds Corrupted by Snowfall,” in CRV, 2018.
- H. Wang, C. Shi, S. Shi, M. Lei, S. Wang, D. He, B. Schiele, and L. Wang, “DSVT: Dynamic Sparse Voxel Transformer With Rotated Sets,” in CVPR, 2023.
- M. Hahner, C. Sakaridis, D. Dai, and L. Van Gool, “Fog Simulation on Real LiDAR Point Clouds for 3D Object Detection in Adverse Weather,” in ICCV, 2021.
- Y. Zhang, Q. Hu, G. Xu, Y. Ma, J. Wan, and Y. Guo, “Not All Points Are Equal: Learning Highly Efficient Point-Based Detectors for 3D LiDAR Point Clouds,” in CVPR, 2022.
- V. Kilic, D. Hegde, V. A. Sindagi, A. Cooper, M. Foster, and V. M. Patel, “Lidar Light Scattering Augmentation (LISA): Physics-based Simulation of Adverse Weather Conditions for 3D Object Detection,” ArXiv, 2021.
- A. Piroli, V. Dallabetta, M. Walessa, D. Meissner, J. Kopp, and K. Dietmayer, “Robust 3D Object Detection in Cold Weather Conditions,” in 2022 IEEE Intelligent Vehicles Symposium (IV), 2022.
- A. H. Lang, S. Vora, H. Caesar, L. Zhou, J. Yang, and O. Beijbom, “PointPillars: Fast Encoders for Object Detection From Point Clouds,” in CVPR, 2019.
- S. Shi, X. Wang, and H. Li, “PointRCNN: 3D Object Proposal Generation and Detection From Point Cloud,” in CVPR, 2019.
- S. Shi, C. Guo, L. Jiang, Z. Wang, J. Shi, X. Wang, and H. Li, “PV-RCNN: Point-Voxel Feature Set Abstraction for 3D Object Detection,” in CVPR, 2020.
- S. Shi, L. Jiang, J. Deng, Z. Wang, C. Guo, J. Shi, X. Wang, and H. Li, “PV-RCNN++: Point-Voxel Feature Set Abstraction With Local Vector Representation for 3D Object Detection,” Int. J. Comput. Vision, vol. 131, 2022.
- C. He, H. Zeng, J. Huang, X.-S. Hua, and L. Zhang, “Structure Aware Single-Stage 3D Object Detection From Point Cloud,” in CVPR, 2020.
- Y. Yan, Y. Mao, and B. Li, “SECOND: Sparsely Embedded Convolutional Detection,” Sensors, vol. 18, 2018.
- W. Zheng, W. Tang, L. Jiang, and C.-W. Fu, “SE-SSD: Self-Ensembling Single-Stage Object Detector From Point Cloud,” in CVPR, 2021.
- X. Wu, L. Peng, H. Yang, L. Xie, C. Huang, C. Deng, H. Liu, and D. Cai, “Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion,” in CVPR, 2022.
- S. Teufel, G. Volk, A. Von Bernuth, and O. Bringmann, “Simulating Realistic Rain, Snow, and Fog Variations For Comprehensive Performance Characterization of LiDAR Perception,” in 2022 IEEE 95th Vehicular Technology Conference: (VTC2022-Spring), 2022.
- N. A. M. Mai, P. Duthon, L. Khoudour, A. Crouzil, and S. A. Velastin, “3D Object Detection with SLS-Fusion Network in Foggy Weather Conditions,” Sensors, vol. 21, 2021.
- J. Lin, H. Yin, J. Yan, W. Ge, H. Zhang, and G. Rigoll, “Improved 3D Object Detector Under Snowfall Weather Condition Based on LiDAR Point Cloud,” IEEE Sensors Journal, vol. 22, 2022.
- M. Hahner, C. Sakaridis, M. Bijelic, F. Heide, F. Yu, D. Dai, and L. Van Gool, “LiDAR Snowfall Simulation for Robust 3D Object Detection,” in CVPR, 2022.
- J. Yang, S. Shi, R. Ding, Z. Wang, and X. Qi, “Towards Efficient 3D Object Detection with Knowledge Distillation,” NIPS, vol. 35, 2022.
- Q. Xu, Y. Zhou, W. Wang, C. R. Qi, and D. Anguelov, “SPG: Unsupervised Domain Adaptation for 3D Object Detection via Semantic Point Generation,” in ICCV, 2021.
- Y.-C. Shih, W.-H. Liao, W.-C. Lin, S.-K. Wong, and C.-C. Wang, “Reconstruction and Synthesis of Lidar Point Clouds of Spray,” IEEE Robotics and Automation Letters, vol. 7, 2022.
- Z. Yang, Y. Sun, S. Liu, X. Shen, and J. Jia, “STD: Sparse-to-Dense 3D Object Detector for Point Cloud,” in ICCV, 2019.
- M. Bijelic, T. Gruber, F. Mannan, F. Kraus, W. Ritter, K. Dietmayer, and F. Heide, “Seeing Through Fog Without Seeing Fog: Deep Multimodal Sensor Fusion in Unseen Adverse Weather,” in CVPR, 2020.
- H. Wu, C. Wen, S. Shi, X. Li, and C. Wang, “Virtual Sparse Convolution for Multimodal 3D Object Detection,” in CVPR, 2023.
- J. Deng, S. Shi, P. Li, W. Zhou, Y. Zhang, and H. Li, “Voxel R-CNN: Towards High Performance Voxel-based 3D Object Detection,” AAAI, vol. 35, 2021.
- P. Sun, H. Kretzschmar, X. Dotiwalla, A. Chouard, V. Patnaik, P. Tsui, J. Guo, Y. Zhou, Y. Chai, B. Caine, V. Vasudevan, W. Han, J. Ngiam, H. Zhao, A. Timofeev, S. Ettinger, M. Krivokon, A. Gao, A. Joshi, Y. Zhang, J. Shlens, Z. Chen, and D. Anguelov, “Scalability in Perception for Autonomous Driving: Waymo Open Dataset,” in CVPR, 2020.
- R. Bridson, J. Houriham, and M. Nordenstam, “Curl-noise for procedural fluid flow,” ACM Trans. Graph., vol. 26, no. 3, p. 46–es, jul 2007. [Online]. Available: https://doi.org/10.1145/1276377.1276435
- G. Borgefors, “Distance transformations in digital images,” Computer vision, graphics, and image processing, vol. 34, no. 3, pp. 344–371, 1986.
- A. T. Do and M. Yoo, “Lossdistillnet: 3d object detection in point cloud under harsh weather conditions,” IEEE Access, vol. 10, pp. 84 882–84 893, 2022.
- Q. Hu, B. Yang, L. Xie, S. Rosa, Y. Guo, Z. Wang, N. Trigoni, and A. Markham, “Randla-net: Efficient semantic segmentation of large-scale point clouds,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, 2020, pp. 11 108–11 117.