A Bidirectional Feedforward Neural Network Architecture Using the Discretized Neural Memory Ordinary Differential Equation

Deep Feedforward Neural Networks (FNNs) with skip connections have revolutionized various image recognition tasks. In this paper, we propose a novel architecture called bidirectional FNN (BiFNN), which utilizes skip connections to aggregate features between its forward and backward paths. The BiFNN accepts any FNN as a plugin that can incorporate any general FNN model into its forward path, introducing only a few additional parameters in the cross-path connections. The backward path is implemented as a nonparameter layer, utilizing a discretized form of the neural memory Ordinary Differential Equation (nmODE), which is named [Formula: see text]-net. We provide a proof of convergence for the [Formula: see text]-net and evaluate its initial value problem. Our proposed architecture is evaluated on diverse image recognition datasets, including Fashion-MNIST, SVHN, CIFAR-10, CIFAR-100, and Tiny-ImageNet. The results demonstrate that BiFNNs offer significant improvements compared to embedded models such as ConvMixer, ResNet, ResNeXt, and Vision Transformer. Furthermore, BiFNNs can be fine-tuned to achieve comparable performance with embedded models on Tiny-ImageNet and ImageNet-1K datasets by loading the same pretrained parameters.

Medienart:

E-Artikel

Erscheinungsjahr:

2024

Erschienen:

2024

Enthalten in:

Zur Gesamtaufnahme - volume:34

Enthalten in:

International journal of neural systems - 34(2024), 4 vom: 27. Feb., Seite 2450015

Sprache:

Englisch

Beteiligte Personen:

Niu, Hao [VerfasserIn]
Yi, Zhang [VerfasserIn]
He, Tao [VerfasserIn]

Links:

Volltext

Themen:

Bidirectional FNN
Image recognition
Journal Article
Ordinary differential equation
Skip connection

Anmerkungen:

Date Completed 29.02.2024

Date Revised 29.02.2024

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1142/S0129065724500151

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM368079759