用短时傅里叶变换改进神经网络,让3D点云树种分类更高效
Introducing the Short-Time Fourier Kolmogorov Arnold Network: A Dynamic Graph CNN Approach for Tree Species Classification in 3D Point Clouds
- 用STFT替代线性层,构建轻量动态图卷积网络
- 参数量减少87%,性能接近顶级模型PointMLP lite
- 适合需要低计算资源的森林监测应用
基于地面激光扫描(TLS)和机载激光扫描(ALS)的树种准确分类对生物多样性保护至关重要。尽管先进的3D点云分类深度学习模型表现优异,但其高复杂度常阻碍高效、低计算量架构的发展。本文提出STFT-KAN,一种融合短时傅里叶变换(STFT)的新型柯尔莫哥洛夫-阿诺德网络,可替代标准线性层与激活函数。我们将STFT-KAN嵌入轻量版DGCNN(liteDGCNN),用于TLS数据的树种分类。实验表明,STFT-KAN在保持性能的同时显著降低模型复杂度,参数量大幅减少;混合架构在边卷积中使用MLP、其余层采用STFT-KAN,性能媲美MLP模型,参数量分别减少50%和75%;相比主流方法PointMLP lite,STFT-KAN实现87%的参数量下降,仍保持竞争力。
原文摘要 · Abstract (English)
Accurate classification of tree species based on Terrestrial Laser Scanning (TLS) and Airborne Laser Scanning (ALS) is essential for biodiversity conservation. While advanced deep learning models for 3D point cloud classification have demonstrated strong performance in this domain, their high complexity often hinders the development of efficient, low-computation architectures. In this paper, we introduce STFT-KAN, a novel Kolmogorov-Arnold network that integrates the Short-Time Fourier Transform (STFT), which can replace the standard linear layer with activation. We implemented STFT-KAN within a lightweight version of DGCNN, called liteDGCNN, to classify tree species using the TLS data. Our experiments show that STFT-KAN outperforms existing KAN variants by effectively balancing model complexity and performance with parameter count reduction, achieving competitive results compared to MLP-based models. Additionally, we evaluated a hybrid architecture that combines MLP in edge convolution with STFT-KAN in other layers, achieving comparable performance to MLP models while reducing the parameter count by 50% and 75% compared to other KAN-based variants. Furthermore, we compared our model to leading 3D point cloud learning approaches, demonstrating that STFT-KAN delivers competitive results compared to the state-of-the-art method PointMLP lite with an 87% reduction in parameter count.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。