Home >Technology peripherals >AI >Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

王林
王林forward
2023-10-27 11:17:04936browse

Original title: GraphAlign: Enhancing Accurate Feature Alignment by Graph matching for Multi-Modal 3D Object Detection

The content that needs to be rewritten is: Paper link: https://arxiv.org/pdf/2310.08261. pdf

Author affiliation: Beijing Jiaotong University Hebei University of Science and Technology Tsinghua University

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

##Thesis idea:

LiDAR and Cameras are complementary sensors for 3D object detection in autonomous driving. However, studying unnatural interactions between point clouds and images is challenging, and the key lies in how to perform feature alignment of heterogeneous modalities. Currently, many methods only achieve feature alignment through projection calibration and ignore the issue of coordinate conversion accuracy errors between sensors, resulting in suboptimal performance. This paper proposes a more accurate feature alignment strategy called GraphAlign for 3D object detection through graph matching. Specifically, this paper fuses the image features of the semantic segmentation encoder in the image branch with the point cloud features of the 3D sparse CNN in the LiDAR branch. In order to reduce the amount of calculation, this paper uses Euclidean distance calculation to construct the nearest neighbor relationship in the point cloud feature subspace. Through projection calibration between the image and the point cloud, the nearest neighbors of the point cloud features are projected onto the image features. We then search for a more suitable feature alignment by matching the nearest neighbor of a single point cloud to multiple images. In addition, this paper also provides a self-attention module to enhance the weight of important relationships to fine-tune feature alignment between heterogeneous modalities. A large number of experiments were conducted in the nuScenes benchmark to prove the effectiveness and efficiency of GraphAlign proposed in this article.

Main contributions:

This article proposed GraphAlign, a graph-based A graph matching feature alignment framework to solve the misalignment problem in multi-modal 3D object detection.

This article proposes Graph Feature Alignment (GFA) and Self-Attention Feature Alignment (SAFA) modules to achieve precise alignment of image features and point cloud features, which can Feature alignment between point clouds and image modalities is further enhanced, thereby improving detection accuracy.

By conducting experiments using two benchmarks, KITTI and nuScenes, we have proven that GraphAlign can effectively improve the accuracy of point cloud detection, especially in long-distance target detection

Network design:

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Figure 1. Comparison of feature alignment strategies

(a) The projection-based method can quickly establish the relationship between modal features, However, misalignment may occur due to sensor error. (b) Attention-based methods retain semantic information by learning alignment, but are computationally expensive. (c) GraphAlign proposed in this paper uses graph-based feature alignment to match more reasonable alignments between modalities, thereby reducing computational effort and improving accuracy.

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Figure 2. The framework of GraphAlign.

Rewritten in Chinese as follows: It consists of the graph feature alignment (GFA) module and the self-attention feature alignment (SAFA) module. The GFA module receives image and point cloud features as input, uses a projection calibration matrix to convert 3D positions into 2D pixel positions, builds local neighborhood information to find nearest neighbors, and combines image and point cloud features. The SAFA module models the contextual relationships between K nearest neighbors through a self-attention mechanism to enhance the importance of fused features and ultimately selects the most representative features

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Figure 3. GFA processing flow

(a) Sensor accuracy error causes misalignment. (b) GFA establishes proximity relationships through graphs in point cloud features. (c) This article projects point cloud features onto image features and obtains the K nearest neighbors of image features. (d) This paper performs one-to-many fusion, specifically, by fusing each individual point cloud feature with K neighboring image features to achieve better alignment.

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Figure 4. SAFA module process

We have simplified the head and max modules. The purpose of the SAFA module is to improve the global context information between K neighbors. , to enhance the representation of fused features

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Experimental results:

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Citation:

Song, Z., Wei, H., Bai, L., Yang, L., & Jia, C. (2023) . GraphAlign: Enhancing Accurate Feature Alignment by Graph matching for Multi-Modal 3D Object Detection.

ArXiv. /abs/2310.08261

Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign

Original link: https: //mp.weixin.qq.com/s/eN6THT2azHvoleT1F6MoSw

The above is the detailed content of Accurate feature alignment to enhance multimodal 3D object detection: Application of GraphAlign. For more information, please follow other related articles on the PHP Chinese website!

Statement:
This article is reproduced at:51cto.com. If there is any infringement, please contact admin@php.cn delete