English

Cluster-based classification with neural ODEs via control

Optimization and Control 2025-04-18 v2 Machine Learning

Abstract

We address binary classification using neural ordinary differential equations from the perspective of simultaneous control of NN data points. We consider a single-neuron architecture with parameters fixed as piecewise constant functions of time. In this setting, the model complexity can be quantified by the number of control switches. Previous work has shown that classification can be achieved using a point-by-point strategy that requires O(N)O(N) switches. We propose a new control method that classifies any arbitrary dataset by sequentially steering clusters of dd points, thereby reducing the complexity to O(N/d)O(N/d) switches. The optimality of this result, particularly in high dimensions, is supported by some numerical experiments. Our complexity bound is sufficient but often conservative because same-class points tend to appear in larger clusters, simplifying classification. This motivates studying the probability distribution of the number of switches required. We introduce a simple control method that imposes a collinearity constraint on the parameters, and analyze a worst-case scenario where both classes have the same size and all points are i.i.d. Our results highlight the benefits of high-dimensional spaces, showing that classification using constant controls becomes more probable as dd increases.

Keywords

Cite

@article{arxiv.2312.13807,
  title  = {Cluster-based classification with neural ODEs via control},
  author = {Antonio Álvarez-López and Rafael Orive-Illera and Enrique Zuazua},
  journal= {arXiv preprint arXiv:2312.13807},
  year   = {2025}
}

Comments

28 pages, 27 figures

R2 v1 2026-06-28T13:58:38.456Z