用语义感知图网络提升单车视角转向估计精度
Enhancing Steering Estimation with Semantic-Aware GNNs
- 融合语义信息构建图结构,仅20%跨类别连接降低计算成本
- 基于单目图像重建伪3D点云,性能媲美激光雷达模型
- 在KITTI数据集上相较2D模型提升71%,适合低成本自动驾驶
转向估计是自动驾驶中的关键任务,传统方法依赖2D图像模型。本文探索结合3D空间信息的混合架构,使用激光雷达点云作为输入,将3D神经网络与循环神经网络(RNN)结合进行时序建模。系统评估四种混合3D模型,均优于仅使用2D的基线模型,其中图神经网络(GNN)-RNN模型表现最佳。为减少对激光雷达的依赖,利用预训练统一模型从单目图像估计深度,重建伪3D点云,并将原为激光雷达设计的GNN-RNN模型适配至伪3D表示,实现相当甚至更优性能。该统一模型还为每个点提供语义标签,增强场景结构化表达。为优化图构建,提出高效连接策略:主要在同语义类点间建立连接,仅保留20%跨类别连接,显著降低图复杂度与计算开销,同时保持关键空间关系。最终在KITTI数据集上验证,相较2D-only模型实现71%的性能提升。研究证明了3D空间信息与高效图构造在转向估计中的优势,同时保持单目图像的成本效益,避免激光雷达系统的高成本。
原文摘要 · Abstract (English)
Steering estimation is a critical task in autonomous driving, traditionally relying on 2D image-based models. In this work, we explore the advantages of incorporating 3D spatial information through hybrid architectures that combine 3D neural network models with recurrent neural networks (RNNs) for temporal modeling, using LiDAR-based point clouds as input. We systematically evaluate four hybrid 3D models, all of which outperform the 2D-only baseline, with the Graph Neural Network (GNN) - RNN model yielding the best results. To reduce reliance on LiDAR, we leverage a pretrained unified model to estimate depth from monocular images, reconstructing pseudo-3D point clouds. We then adapt the GNN-RNN model, originally designed for LiDAR-based point clouds, to work with these pseudo-3D representations, achieving comparable or even superior performance compared to the LiDAR-based model. Additionally, the unified model provides semantic labels for each point, enabling a more structured scene representation. To further optimize graph construction, we introduce an efficient connectivity strategy where connections are predominantly formed between points of the same semantic class, with only 20\% of inter-class connections retained. This targeted approach reduces graph complexity and computational cost while preserving critical spatial relationships. Finally, we validate our approach on the KITTI dataset, achieving a 71% improvement over 2D-only models. Our findings highlight the advantages of 3D spatial information and efficient graph construction for steering estimation, while maintaining the cost-effectiveness of monocular images and avoiding the expense of LiDAR-based systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。