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.

BitNN: A Bit-Serial Accelerator for K-Nearest Neighbor Search in Point Clouds

Constrained All-k-Nearest-Neighbor Search

BiPointNet: Binary Neural Network for Point Clouds

FusionArch: A Fusion-Based Accelerator for Point-Based Point Cloud Neural Networks

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

JUNO: Optimizing High-Dimensional Approximate Nearest Neighbour Search with Sparsity-Aware Algorithm and Ray-Tracing Core Mapping

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

Efficient parallel processing for K-nearest-neighbor search in spatial databases

Efficient Parallel Processing of High-Dimensional Spatial K NN Queries

Exploring Bit-Difference for Approximate KNN Search in High-Dimensional Databases

Bit-balance: Model-Hardware Co-design for Accelerating NNs by Exploiting Bit-level Sparsity

Nearest Neighbors Meet Deep Neural Networks for Point Cloud Analysis

Efficient index-based KNN join processing for high-dimensional data

FuseKNA: Fused Kernel Convolution based Accelerator for Deep Neural Networks

MPR — A Partitioning-Replication Framework for Multi-Processing kNN Search on Road Networks

Double-Bit Quantization and Index Hashing for Nearest Neighbor Search

An Efficient Accelerator for Point-based and Voxel-based Point Cloud Neural Networks

Fast Point Cloud Geometry Compression with Context-based Residual Coding and INR-based Refinement

A Point Transformer Accelerator with Distribution-Aware Heuristic Distance Calculation

A 200M-Query-vector/s Computing-in-RRAM ADC-less K-Nearest-neighbor Accelerator with Time-Domain Winner-Takes-All Circuits

SparseByteNN: A Novel Mobile Inference Acceleration Framework Based on Fine-Grained Group Sparsity