Abstract:Point clouds are an important form of 3D data used in applications such as computer vision and autonomous driving, but the irregular and disordered nature of point clouds makes processing them severely challenging. Recently, point-based neural networks for point clouds have been widely used in various 3D applications. Notably, transformer-based models have demonstrated state-of-the-art accuracy. However, three significant challenges exist: (1) Data interdependence hinders parallel execution in networks like Point Transformer. (2) The Farthest Point Sampling (FPS) involves redundant memory access and computational overhead. (3) Intermediate results require repetitive memory access and calculations between FPS and kNN operators. This limits Point Transformer’s processing speed to 17.80 frames per second on NVIDIA Jetson Orin, below the real-time requirement of around 30 frames per second. In this paper, we introduce, an innovative Point Transformer Accelerator to address the aforementioned three challenges from the following three levels. On the computation graph level, our investigation reveals that the Point Transformer’s performance suffers minimal degradation when operating within a constrained receptive field. Leveraging this insight, strategically frees the MaxPool and attention-kNN layers, along with their associated data dependencies, achieving an inconsequential loss in accuracy. On the operator level, we identify that the variability for distance computation among accessed points during Farthest Point Sampling (FPS) iterations contributes to redundant memory accesses and computational overhead. proposes a distribution-aware heuristic for distance calculation to minimize unnecessary memory accesses and computational redundancies within the FPS operator. On the architecture level, we recognize that the transition down process (encompassing FPS and kNN operations) constitutes 71.77% of the total inference time, proposes an integrated FPS-kNN architecture to select error-driven k neighbors, reducing repeated memory accesses and distance recalculations of intermediate results. Through extensive experimentation, demonstrates remarkable performance improvements, achieving end-to-end speedups of up to 2.96×, 1.70×, and 1.19× when compared to the state-of-the-art accelerators PointAcc pointacc, MARS mars, and PTrAcc ptracc, respectively, across a variety of point cloud neural networks.

TorchSparse plus plus : Efficient Point Cloud Engine

TorchSparse++: Efficient Training and Inference Framework for Sparse Convolution on GPUs

SSF: Sparse Point Cloud Object Detection Based on Self-Adaptive Voxel Encoding and Focal-Sparse Convolution

Minuet: Accelerating 3D Sparse Convolutions on GPUs

SpOctA: A 3D Sparse Convolution Accelerator with Octree-Encoding-Based Map Search and Inherent Sparsity-Aware Processing

Optimizing Sparse Convolution on GPUs with CUDA for 3D Point Cloud Processing in Embedded Systems

A Demonstration Platform for Large-Scaled Point Cloud Network Based on 28nm 2D/3D Unified Sparse Convolution Accelerator.

Torch-Points3D: A Modular Multi-Task Frameworkfor Reproducible Deep Learning on 3D Point Clouds

An Efficient FPGA Accelerator for Point Cloud

Rendering-Oriented 3D Point Cloud Attribute Compression using Sparse Tensor-based Transformer

Accelerating DNN-based 3D point cloud processing for mobile computing

Scorch: A Library for Sparse Deep Learning

An Efficient Sparse Inference Software Accelerator for Transformer-based Language Models on CPUs

SparsePipe: Parallel Deep Learning for 3D Point Clouds

TSC-PCAC: Voxel Transformer and Sparse Convolution Based Point Cloud Attribute Compression for 3D Broadcasting

A 28-Nm Energy-Efficient Sparse Neural Network Processor for Point Cloud Applications Using Block-Wise Online Neighbor Searching

Sparse Tensor-Based Multiscale Representation for Point Cloud Geometry Compression.

A Point Transformer Accelerator with Distribution-Aware Heuristic Distance Calculation

DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets

Sparse Fuse Dense: Towards High Quality 3D Detection with Depth Completion

Exploring Fine-Grained Sparsity in Convolutional Neural Networks for Efficient Inference