arrow
Return

Adaptive token selection for scalable point cloud transformers

delete2025-05-27
delete0
delete
OA
AI
A
Alessandro Baiocchi
I
Indro Spinelli
A
Alessandro Nicolosi
S
Simone Scardapane *
DOI:10.1016/j.neunet.2025.107477delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
The recent surge in 3D data acquisition has spurred the development of geometric deep learning models for point cloud processing, boosted by the remarkable success of transformers in natural language processing. While point cloud transformers (PTs) have achieved impressive results recently, their quadratic scaling with respect to the point cloud size poses a significant scalability challenge for real-world applications. To address this issue, we propose the Adaptive Point Cloud Transformer (AdaPT), a standard PT model augmented by an adaptive token selection mechanism. AdaPT dynamically reduces the number of tokens during inference, enabling efficient processing of large point clouds. Furthermore, we introduce a budget mechanism to flexibly adjust the computational cost of the model at inference time without the need for retraining or fine-tuning separate models. Our extensive experimental evaluation on point cloud classification tasks demonstrates that AdaPT significantly reduces computational complexity while maintaining competitive accuracy compared to standard PTs. The code for AdaPT is publicly available at https://github.com/ispamm/adaPT.
Keywords:
Transformer
Geometric deep learning
Point clouds
Gumbel-Softmax
Token selection
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Neural Networks cover
Neural Networks
IF:
6.3
Papers:
7.8K
Citations:
3.0W

Organization

L
leonardo labs
Scholars:
4
Papers: 2
Citations: 0
S
sapienza university rome
Scholars:
6.3W
Papers: 4.7W
Citations: 381