PointViG: A Lightweight GNN-based Model for Efficient Point Cloud Analysis

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zheng, Qiang, Qi, Yafei, Wang, Chen, Zhang, Chao, Sun, Jian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912029959258112
author Zheng, Qiang
Qi, Yafei
Wang, Chen
Zhang, Chao
Sun, Jian
author_facet Zheng, Qiang
Qi, Yafei
Wang, Chen
Zhang, Chao
Sun, Jian
contents In the domain of point cloud analysis, despite the significant capabilities of Graph Neural Networks (GNNs) in managing complex 3D datasets, existing approaches encounter challenges like high computational costs and scalability issues with extensive scenarios. These limitations restrict the practical deployment of GNNs, notably in resource-constrained environments. To address these issues, this study introduce <b>Point<\b> <b>Vi<\b>sion <b>G<\b>NN (PointViG), an efficient framework for point cloud analysis. PointViG incorporates a lightweight graph convolutional module to efficiently aggregate local features and mitigate over-smoothing. For large-scale point cloud scenes, we propose an adaptive dilated graph convolution technique that searches for sparse neighboring nodes within a dilated neighborhood based on semantic correlation, thereby expanding the receptive field and ensuring computational efficiency. Experiments demonstrate that PointViG achieves performance comparable to state-of-the-art models while balancing performance and complexity. On the ModelNet40 classification task, PointViG achieved 94.3% accuracy with 1.5M parameters. For the S3DIS segmentation task, it achieved an mIoU of 71.7% with 5.3M parameters. These results underscore the potential and efficiency of PointViG in point cloud analysis.
format Preprint
id arxiv_https___arxiv_org_abs_2407_00921
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle PointViG: A Lightweight GNN-based Model for Efficient Point Cloud Analysis
Zheng, Qiang
Qi, Yafei
Wang, Chen
Zhang, Chao
Sun, Jian
Computer Vision and Pattern Recognition
In the domain of point cloud analysis, despite the significant capabilities of Graph Neural Networks (GNNs) in managing complex 3D datasets, existing approaches encounter challenges like high computational costs and scalability issues with extensive scenarios. These limitations restrict the practical deployment of GNNs, notably in resource-constrained environments. To address these issues, this study introduce <b>Point<\b> <b>Vi<\b>sion <b>G<\b>NN (PointViG), an efficient framework for point cloud analysis. PointViG incorporates a lightweight graph convolutional module to efficiently aggregate local features and mitigate over-smoothing. For large-scale point cloud scenes, we propose an adaptive dilated graph convolution technique that searches for sparse neighboring nodes within a dilated neighborhood based on semantic correlation, thereby expanding the receptive field and ensuring computational efficiency. Experiments demonstrate that PointViG achieves performance comparable to state-of-the-art models while balancing performance and complexity. On the ModelNet40 classification task, PointViG achieved 94.3% accuracy with 1.5M parameters. For the S3DIS segmentation task, it achieved an mIoU of 71.7% with 5.3M parameters. These results underscore the potential and efficiency of PointViG in point cloud analysis.
title PointViG: A Lightweight GNN-based Model for Efficient Point Cloud Analysis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.00921