Towards 3D Semantic Scene Completion for Autonomous Driving: A Meta-Learning Framework Empowered by Deformable Large-Kernel Attention and Mamba Model
Authors:
Preprint
Abstract
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.Keywords
All Related Versions
This publication has 20 references indexed in Scilit:
- OccGen: Generative Multi-modal 3D Occupancy Prediction for Autonomous Driving2024
- Traffic expertise meets residual RL: Knowledge-informed model-based residual reinforcement learning for CAV trajectory controlCommunications in Transportation Research, 2024
- Collaborative Semantic Occupancy Prediction with Hybrid Feature Fusion in Connected Automated Vehicles2024
- Symphonize 3D Semantic Scene Completion with Contextual Instance Queries2024
- Human as AI mentor: Enhanced human-in-the-loop reinforcement learning for safe and efficient autonomous drivingCommunications in Transportation Research, 2024
- Visual attention networkComputational Visual Media, 2023
- Towards real-world traffic prediction and data imputation: A multi-task pretraining and fine-tuning approachInformation Sciences, 2023
- ST-A-PGCL: Spatiotemporal adaptive periodical graph contrastive learning for traffic prediction under real scenariosKnowledge-Based Systems, 2023
- Semantic Scene Completion from a Single Depth Image2017
- Are we ready for autonomous driving? The KITTI vision benchmark suite2012