Source-linked AI summary
SVGA-Net: Sparse Voxel-Graph Attention Network for 3D Object Detection from Point Clouds
Qingdong He, Zhengning Wang, Hao Zeng, Yi Zeng, Yijun Liu
TL;DR
Point clouds provide useful depth and geometric information, but their unordered and sparse nature makes direct 3D detection challenging. SVGA-Net combines local and global voxel graphs with multiscale regression; it reports decent results alongside state-of-the-art methods on KITTI.
Problem
Point clouds offer accurate depth and geometric information, but their unordered and sparse properties make direct 3D object detection challenging.
Source: §1 · p. 1Method
SVGA-Net processes raw point clouds with local complete graphs in spherical voxels, a global KNN graph, and multiscale sparse-to-dense regression.
Source: §1 · p. 2, §3.2 · p. 4Results
SVGA-Net achieves decent experimental results with state-of-the-art methods on the KITTI 3D detection dataset.
Source: §1 · p. 2Takeaways & Limitations
Experiments demonstrate the efficiency of the network's design choices for extending graph representations to 3D object detection.
Source: §5 · p. 7
Abstract
from arXiv · showhide
Accurate 3D object detection from point clouds has become a crucial component in autonomous driving. However, the volumetric representations and the projection methods in previous works fail to establish the relationships between the local point sets. In this paper, we propose Sparse Voxel-Graph Attention Network (SVGA-Net), a novel end-to-end trainable network which mainly contains voxel-graph module and sparse-to-dense regression module to achieve comparable 3D detection tasks from raw LIDAR data. Specifically, SVGA-Net constructs the local complete graph within each divided 3D spherical voxel and global KNN graph through all voxels. The local and global graphs serve as the attention mechanism to enhance the extracted features. In addition, the novel sparse-to-dense regression module enhances the 3D box estimation accuracy through feature maps aggregation at different levels. Experiments on KITTI detection benchmark demonstrate the efficiency of extending the graph representation to 3D object detection and the proposed SVGA-Net can achieve decent detection accuracy.
1. Introduction
Point-cloud 3D detection can use depth and geometric information, but unordered, sparse data make direct detection challenging. SVGA-Net uses local and global voxel graphs and multiscale regression, with reported KITTI results comparable to state-of-the-art methods.
- Point clouds offer depth and geometric information, but their unordered, sparse properties make direct 3D detection challenging.Source: §1 · p. 1
- Earlier projection and voxel methods have limitations, while PointNet-based approaches fail to establish neighbor relationships between grouped point sets.Source: §1 · p. 1
- The end-to-end network processes raw point clouds directly with graph representations, without converting them to other formats.Source: §1 · pp. 1–2
- Its voxel-graph network constructs local complete graphs within spherical voxels and a global KNN graph across voxels to learn discriminative features.Source: §1 · p. 2
- SVGA-Net aggregates features at different scales to improve 3D localization accuracy.Source: §1 · p. 2
- SVGA-Net achieves decent experimental results with state-of-the-art methods on the KITTI 3D detection dataset.Source: §1 · p. 2
2. Related Work
Prior point-cloud detection methods use projections, regular voxel grids, or direct point processing; graph methods have also been explored, but few address 3D object detection.
2.1. Projection-based methods for point clouds
- Projection-based methods combine LiDAR views and, in some cases, RGB features to generate and refine 3D boxes, though high-level layers may lose geometric information.Source: §2.1 · p. 2
2.2. Volumetric methods for point clouds
- Volumetric methods group points in regular 3D grids, often relying on heavy 3D CNN layers that bring large computation.Source: §2.2 · p. 2
2.3. Pointnet-based methods for point clouds
- PointNet and PointNet++ use parallel MLPs on raw irregular data, but still suffer feature loss between local and global point sets.Source: §2.3 · p. 2
2.4. Graph-based methods for point clouds
- Graph approaches learn local and global point information, while Point-GNN applies a one-stage GNN to 3D detection using auto-registration, merging, and scoring.Source: §2.4 · p. 2
3. Proposed Method
The proposed method pairs a voxel-graph network for point-cloud features with a sparse-to-dense regression module for final classification and box prediction.
3. Proposed Method
- SVGA-Net's proposed architecture comprises a voxel-graph network and a sparse-to-dense regression module.The latter produces final classification and regression maps from voxel-graph features.Source: §3 · p. 2, Figure 3 · p. 4
3.1. Voxel-graph network architecture
- Fixed-radius spherical voxels group point features, which local complete-graph and global KNN attention aggregate before max pooling yields spatially aligned voxel features.The global graph connects voxel spheres; the resulting feature vectors feed the sparse-to-dense regression module.Source: §3.1 · pp. 3–4
3.2. Sparse-to-dense regression
- Three downsampling blocks produce multiscale features that are upsampled, concatenated, and fused with processed dense maps to form classification and box-regression inputs.The module combines original block outputs with three processed feature maps by element-wise addition, then concatenates the fused maps after a 3 × 3 convolution.Source: §3.2 · p. 4
3.3. Loss function
- Training combines binary-cross-entropy classification and Smooth L1 box regression losses, with separate weights for the loss terms and positive and negative anchors.The anchor-balance constants are 1.5 and 1; the classification and regression weights are 1 and 2, respectively.Source: §3.3 · p. 5
4. Experiments
Experiments evaluate SVGA-Net on KITTI, where it outperforms an RGB+Lidar baseline on car 3D detection and achieves better or comparable validation results. Qualitative results and ablations examine its detections, design choices, and runtime.
4. Experiments
- KITTI has 7,481 training and 7,518 test samples; the authors split training data roughly 1:1 for training and validation, and evaluate AP at category-specific 3D IoU thresholds.The reported IoU thresholds are 0.7 for cars and 0.5 for cyclists and pedestrians.Source: §4 · p. 5
4.1. Training
- The networks were trained end-to-end for 200 epochs with batch size 16 on four GPUs, using ADAM and data augmentation.Training used an initial learning rate of 0.001 for 140 epochs, then decayed it tenfold every 20 epochs.Source: §4.1 · p. 5
4.2. Comparing with state-of-the-art methods
- SVGA-Net outperforms RGB+Lidar MMF on KITTI car 3D detection by 0.52%, 3.72%, and 7.50% across the three difficulty levels.Table 1 reports average precision on the official test set.Source: §4.2 · p. 5, Table 1 · p. 6
- On the KITTI validation split, SVGA-Net achieves better or comparable car results than state-of-the-art methods across three difficulty levels.The paper reports comparisons for car 3D detection and bird’s-eye-view detection.Source: §4.2 · pp. 5–6, Table 2 · p. 6, Table 3 · p. 6
4.3. Qualitative results
- Qualitative KITTI test examples show predicted 3D boxes in different scenes, including poor lighting and severe occlusion.Figure 4 presents detections projected onto RGB images and corresponding point clouds.Source: §4.3 · p. 6, Figure 4 · p. 7
4.4. Ablation studies
- Detection accuracy improves as local point-attention layers increase from one to three, then decreases slightly at four.The authors attribute the decrease to possible over-learning.Source: §4.4 · p. 6, Table 4 · p. 7
- Removing global attention greatly reduces AP on both detection tasks, while the sparse-to-dense regression design ranks first among its tested alternatives for improving accuracy.The ablations examine global feature information and regression-module design choices.Source: §4.4 · p. 6, Table 4 · p. 7
- KNN graph performance improves as K rises from one to three, then declines at larger K; one-sample inference takes 62 ms.Of the 62 ms, 41 ms is spent aggregating local and global features, 9 ms on data reading and preprocessing, and 12 ms on box detection.Source: §4.4 · p. 7
5. Conclusions
SVGA-Net applies graph representation to point-cloud detection, pairing local graphs in spherical voxels and a global graph with sparse-to-dense feature-map fusion. Experiments demonstrate the efficiency of the design choices, and future work will extend the network to combine RGB images.
5. Conclusions
- Local complete graphs in divided spherical voxels provide a local representation of point features and fuse information between points and their neighborhoods.The conclusion also identifies the global graph as a way to supervise and learn point features.Source: §5 · p. 7
- A global graph helps supervise and learn point features.Source: §5 · p. 7
- The sparse-to-dense regression module fuses feature maps at different scales.Source: §5 · p. 7
- Experiments demonstrate the efficiency of the network's design choices; future work will extend it to combine RGB images to improve detection accuracy.Source: §5 · p. 7