Spiking point transformer for point cloud classification

Abstract

Spiking Neural Networks (SNNs) offer an attractive and energy-efficient alternative to conventional Artificial Neural Networks (ANNs) due to their sparse binary activation. When SNN meets Transformer, it shows great potential in 2D image processing. However, their application for 3D point cloud remains underexplored. To this end, we present Spiking Point Transformer (SPT), the first transformer-based SNN framework for point cloud classification. Specifically, we first design Queue-Driven Sampling Direct Encoding for point cloud to reduce computational costs while retaining the most effective support points at each time step. We introduce the Hybrid Dynamics Integrate-and-Fire Neuron (HD-IF), designed to simulate selective neuron activation and reduce over-reliance on specific artificial neurons. SPT attains state-of-the-art results on three benchmark datasets that span both real-world and synthetic datasets in the SNN domain. Meanwhile, the theoretical energy consumption of SPT is at least 6.4 x less than its ANN counterpart.

Publication
In AAAI 39(20)

本文提出Spiking Point Transformer(SPT),首个基于Transformer的脉冲神经网络框架,用于三维点云分类。SPT设计队列驱动采样直接编码,在降低计算成本的同时保留关键支撑点;并引入混合动力学积分发放神经元(HD-IF),模拟选择性神经元激活,减少对特定人工神经元的过度依赖。在多个真实与合成点云基准上取得领先结果,理论能耗较ANN对应模型降低至少6.4倍。