arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.14845cs.CVcs.LG
收藏

自适应点Transformer

Adaptive Point Transformer

Sapienza University of Rome(罗马第一大学);Leonardo Labs(莱昂纳多实验室)

Alessandro Baiocchi, Indro Spinelli, Alessandro Nicolosi, Simone Scardapane

AI总结:

针对点云Transformer随点云规模二次缩放带来的可扩展性问题,提出带自适应令牌选择和预算机制的AdaPT,在保证精度的同时降低计算复杂度,推理时可灵活调计算成本。

摘要:

近年来3D数据采集的迅猛发展,推动了用于点云处理的几何深度学习模型的开发,这一进程也得益于Transformer在自然语言处理领域的显著成功。尽管点云Transformer(PTs)近期已取得令人瞩目的成果,但它们随点云规模呈二次方增长的计算量,给实际应用带来了严峻的可扩展性挑战。为解决这一问题,我们提出自适应点云Transformer(AdaPT),这是一种通过自适应令牌选择机制增强的标准PT模型。AdaPT在推理过程中动态减少令牌数量,从而实现对大型点云的高效处理。此外,我们引入了一种预算机制,可在推理时灵活调整模型的计算成本,无需重新训练或微调单独的模型。我们在点云分类任务上开展的大量实验评估表明,与标准PT相比,AdaPT在保持具有竞争力的准确率的同时,显著降低了计算复杂度。AdaPT的代码已公开。

英文摘要:

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 made publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑