4 个月前

SplineCNN:基于连续B样条核的快速几何深度学习

SplineCNN:基于连续B样条核的快速几何深度学习

摘要

我们介绍了基于样条的卷积神经网络(SplineCNNs),这是一种适用于非规则结构和几何输入(如图或网格)的深度神经网络变体。我们的主要贡献是一种基于B样条的新卷积算子,由于B样条基函数的局部支持性质,该算子使得计算时间与核大小无关。因此,通过使用由固定数量的可训练权重参数化的连续核函数,我们获得了传统卷积神经网络(CNN)卷积算子的泛化形式。与在频谱域进行滤波的相关方法不同,所提出的方法完全在空间域内聚合特征。此外,SplineCNN允许对深度架构进行端到端的训练,仅以几何结构作为输入,而无需手工设计的特征描述符。为了验证该方法的有效性,我们在图像图分类、形状对应和图节点分类等任务中应用了该方法,并展示了其性能优于或匹敌现有最先进方法,同时具有显著更快的速度和一些有利特性,如领域独立性。

代码仓库

基准测试

基准方法指标
node-classification-on-citeseerSplineCNN
Accuracy: 79.20%
node-classification-on-coraSplineCNN
Accuracy: 89.48% ± 0.31%
node-classification-on-pubmedSplineCNN
Accuracy: 88.88%
superpixel-image-classification-on-75SplineCNN
Classification Error: 4.78

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SplineCNN:基于连续B样条核的快速几何深度学习 | 论文 | HyperAI超神经