The paper introduces a class of neural controlled differential equation (CDE) models inspired by quantum mechanics, called neural quantum controlled differential equations (NQDEs). In NQDEs, the dynamics of the latent state are modeled by the Schrödinger equation, driven by the input sequence. The hidden state represents the wave function, and its "collapse" is used to interpret the classification probability.
The authors implement and compare four variants of NQDEs on a toy spiral classification problem. Two variants use unitary constraints via ProjUNN, while the other two use orthogonal constraints via GeoTorch. The models differ in how the class-specific representations are combined before or after the final linear layer.
All four NQDE variants are able to achieve 100% accuracy on the spiral classification task with very limited training data (128 spirals). The ProjUNN-based models with concatenation before the linear layer (NQDE1 unn) perform the best in terms of both final loss and number of function evaluations.
The authors conclude that neural CDE architectures emulating quantum evolutions can effectively learn relevant dynamics for this toy classification problem. They suggest exploring the approximation power of these models and comparing them to other models on larger datasets as future work.
Para Outro Idioma
do conteúdo original
arxiv.org
Principais Insights Extraídos De
by Lingyi Yang,... às arxiv.org 05-01-2024
https://arxiv.org/pdf/2404.19673.pdfPerguntas Mais Profundas