Machine Learning: Science and Technology (Jan 2024)

Stable attractors for neural networks classification via ordinary differential equations (SA-nODE)

  • Raffaele Marino,
  • Lorenzo Buffoni,
  • Lorenzo Chicchi,
  • Lorenzo Giambagli,
  • Duccio Fanelli

DOI
https://doi.org/10.1088/2632-2153/ad7f26
Journal volume & issue
Vol. 5, no. 3
p. 035087

Abstract

Read online

A novel approach for supervised classification is presented which sits at the intersection of machine learning and dynamical systems theory. At variance with other methodologies that employ ordinary differential equations for classification purposes, the untrained model is a priori constructed to accommodate for a set of pre-assigned stationary stable attractors. Classifying amounts to steer the dynamics towards one of the planted attractors, depending on the specificity of the processed item supplied as an input. Asymptotically the system will hence converge on a specific point of the explored multi-dimensional space, flagging the category of the object to be eventually classified. Working in this context, the inherent ability to perform classification, as acquired ex post by the trained model, is ultimately reflected in the shaped basin of attractions associated to each of the target stable attractors. The performance of the proposed method is here challenged against simple toy models crafted for the purpose, as well as by resorting to well established reference standards. More precisely, we achieved an accuracy of $98.06 \%$ on the MNIST test set and 88.21% on the Fashion MNIST test set. Although this method does not reach the performance of state-of-the-art deep learning algorithms, it illustrates that continuous dynamical systems with closed analytical interaction terms can serve as high-performance classifiers.

Keywords