Patrick Cook, Danny Jammooa, Morten Hjorth-Jensen, Daniel D. Lee, Dean Lee
We present a general class of machine learning algorithms called parametric matrix models. In contrast with most existing machine learning models that imitate the biology of neurons, parametric matrix models use matrix equations that emulate physical systems. Similar to how physics problems are usually solved, parametric matrix models learn the governing equations that lead to the desired outputs. Parametric matrix models can be efficiently trained from empirical data, and the equations may use algebraic, differential, or integral relations. While originally designed for scientific computing, we prove that parametric matrix models are universal function approximators that can be applied to general machine learning problems. After introducing the underlying theory, we apply parametric matrix models to a series of different challenges that show their performance for a wide range of problems. For all the challenges tested here, parametric matrix models produce accurate results within an efficient and interpretable computational framework that allows for input feature extrapolation.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Image Classification | EMNIST-Balanced | Accuracy | 85.95 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | EMNIST-Balanced | Trainable Parameters | 349172 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | EMNIST-Balanced | Accuracy | 81.57 | PMM (Parametric Matrix Model) |
| Image Classification | EMNIST-Balanced | Trainable Parameters | 13792 | PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Accuracy | 90.89 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Percentage error | 9.11 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Trainable Parameters | 278280 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Accuracy | 88.58 | PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Percentage error | 11.42 | PMM (Parametric Matrix Model) |
| Image Classification | Fashion-MNIST | Trainable Parameters | 16744 | PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Accuracy | 98.99 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Percentage error | 1.01 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Trainable Parameters | 129416 | Convolutional PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Accuracy | 97.38 | PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Percentage error | 2.62 | PMM (Parametric Matrix Model) |
| Image Classification | MNIST | Trainable Parameters | 4990 | PMM (Parametric Matrix Model) |