4 个月前

SpinalNet:具有渐进输入的深度神经网络

SpinalNet:具有渐进输入的深度神经网络

摘要

深度神经网络(DNNs)在众多领域中已经达到了最先进的性能。然而,DNNs需要较高的计算时间,人们总是希望以更低的计算成本获得更好的性能。因此,我们研究了人类的体感系统,并设计了一种神经网络(脊柱网络,SpinalNet),以实现更高的精度和更少的计算量。传统的神经网络(NNs)中的隐藏层接收前一层的输入,应用激活函数,然后将结果传递给下一层。而在所提出的脊柱网络中,每一层被分为三个部分:1)输入部分,2)中间部分,3)输出部分。每层的输入部分接收一部分输入数据;每层的中间部分则接收前一层中间部分的输出以及当前层输入部分的输出。这使得传入权重的数量显著低于传统DNNs。脊柱网络还可以作为DNN的全连接层或分类层,并支持传统学习和迁移学习。我们在大多数DNNs中观察到,在较低计算成本的情况下,错误率有显著降低。使用脊柱网络分类层的传统学习方法在VGG-5网络上提供了QMNIST、Kuzushiji-MNIST、EMNIST(字母、数字和平衡)数据集上的最先进(SOTA)性能。而使用ImageNet预训练初始权重和脊柱网络分类层的传统学习方法,则在STL-10、Fruits 360、Bird225和Caltech-101数据集上提供了最先进(SOTA)性能。所提出的脊柱网络的脚本可在以下链接获取:https://github.com/dipuk0506/SpinalNet

代码仓库

Mechachleopteryx/SpinalNet
pytorch
GitHub 中提及
dipuk0506/SpinalNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
fine-grained-image-classification-on-bird-225VGG-19bn (Spinal FC)
Accuracy: 99.02
fine-grained-image-classification-on-bird-225VGG-19bn
Accuracy: 98.67
fine-grained-image-classification-on-caltechWide-ResNet-101 (Spinal FC)
Accuracy: 97.32
Top-1 Error Rate: 2.68%
fine-grained-image-classification-on-caltechWide-ResNet-101
Top-1 Error Rate: 2.89%
fine-grained-image-classification-on-caltechVGG-19bn (Spinal FC)
Top-1 Error Rate: 6.84%
fine-grained-image-classification-on-fruitsVGG-19bn
Accuracy (%): 99.90
fine-grained-image-classification-on-oxfordWide-ResNet-101 (Spinal FC)
Accuracy: 99.30%
image-classification-on-emnist-balancedCNN(Spinal FC)
Accuracy: 83.21
Trainable Parameters: 16050
image-classification-on-emnist-balancedCNN
Accuracy: 79.61
Trainable Parameters: 21840
image-classification-on-emnist-balancedVGG-5
Accuracy: 91.04
Trainable Parameters: 3646000
image-classification-on-emnist-balancedCNN(Spinal FC)
Accuracy: 82.77
Trainable Parameters: 13820
image-classification-on-emnist-balancedVGG-5(Spinal FC)
Accuracy: 91.05
Trainable Parameters: 3630000
image-classification-on-emnist-digitsVGG-5(Spinal FC)
Accuracy (%): 99.75
image-classification-on-emnist-lettersVGG-5
Accuracy: 95.86
image-classification-on-emnist-lettersVGG-5(Spinal FC)
Accuracy: 95.88
image-classification-on-flowers-102Wide-ResNet-101 (Spinal FC)
Accuracy: 99.30
image-classification-on-kuzushiji-mnistVGG-5 (Spinal FC)
Accuracy: 99.15
Error: 0.85
image-classification-on-mnistVGG-5 (Spinal FC)
Accuracy: 99.72
Percentage error: 0.28
image-classification-on-stl-10Wide-ResNet-101 (Spinal FC)
Percentage correct: 98.66
image-classification-on-stl-10VGG-19bn
Percentage correct: 95.44

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SpinalNet:具有渐进输入的深度神经网络 | 论文 | HyperAI超神经