arXiv:2411.03672cs.CVcs.AI2024-11被引 9

用元学习和长序列建模提升自动驾驶3D语义场景补全效果

MetaSSC: Enhancing 3D Semantic Scene Completion for Autonomous Driving through Meta-Learning and Long-sequence Modeling

  • 基于元学习预训练,提取可迁移的语义几何知识
  • 在模拟协同感知数据上训练,标签更完整
  • 结合Mamba与可变形卷积,高效捕捉长距离依赖

语义场景补全(SSC)对自动驾驶全面感知至关重要。现有方法常忽视实际部署成本。传统3D CNN和自注意力机制难以有效捕捉3D体素网格中的长程依赖,影响性能。为此,我们提出MetaSSC,一种基于元学习的SSC框架,融合可变形卷积、大核注意力与Mamba(D-LKA-M)模型。首先通过体素级语义分割预训练任务,探索不完整区域的语义与几何特征,获取可迁移的元知识。利用模拟协同感知数据集,以多辆联网自动驾驶汽车(CAVs)的聚合传感器数据监督单车感知训练,生成更丰富完整的标签。该元知识通过无额外参数的双阶段训练策略适配目标域,实现高效部署。为进一步增强对3D体素网格中长序列关系的建模能力,将Mamba块与可变形卷积、大核注意力结合到主干网络中。大量实验表明,MetaSSC达到当前最优性能,显著优于对比模型,同时降低部署成本。

原文摘要 · Abstract (English)

Semantic scene completion (SSC) is essential for achieving comprehensive perception in autonomous driving systems. However, existing SSC methods often overlook the high deployment costs in real-world applications. Traditional architectures, such as 3D Convolutional Neural Networks (3D CNNs) and self-attention mechanisms, face challenges in efficiently capturing long-range dependencies within 3D voxel grids, limiting their effectiveness. To address these issues, we introduce MetaSSC, a novel meta-learning-based framework for SSC that leverages deformable convolution, large-kernel attention, and the Mamba (D-LKA-M) model. Our approach begins with a voxel-based semantic segmentation (SS) pretraining task, aimed at exploring the semantics and geometry of incomplete regions while acquiring transferable meta-knowledge. Using simulated cooperative perception datasets, we supervise the perception training of a single vehicle using aggregated sensor data from multiple nearby connected autonomous vehicles (CAVs), generating richer and more comprehensive labels. This meta-knowledge is then adapted to the target domain through a dual-phase training strategy that does not add extra model parameters, enabling efficient deployment. To further enhance the model's capability in capturing long-sequence relationships within 3D voxel grids, we integrate Mamba blocks with deformable convolution and large-kernel attention into the backbone network. Extensive experiments demonstrate that MetaSSC achieves state-of-the-art performance, significantly outperforming competing models while also reducing deployment costs.

3D语义补全元学习Mamba自动驾驶

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。