Exact SO(3)-Equivariant Isotropic Kernels for Rotation-Robust Neural Dynamics
Abstract
Neural surrogates for vector-valued partial differential equations can fit training data yet change their predictions when the same physical state is expressed in a rotated coordinate frame. We study this failure on three-dimensional Navier--Stokes dynamics observed at irregularly placed points. We introduce the Invariant-Conditioned Isotropic Kernel Neural Operator (IKNO), a compact graph model that builds local interactions from scalar quantities unchanged by rotation and vector directions that rotate with the data. Consequently, rotating the positions and velocities rotates the predicted velocity change in exactly the same way. On a held-out test set fixed after model design, training unconstrained graph models on randomly rotated examples reduces but does not eliminate their coordinate dependence. In contrast, IKNO is consistent to numerical precision, matches the forecasting accuracy of a general rotation-aware Tensor Field Network with $5.6$ times fewer parameters, and outperforms a parameter-matched graph simulator. These results show that a compact, PDE-specialized model can remove coordinate dependence without sacrificing forecasting accuracy.
Related
- Optimal Control Under Signal Temporal Logic Constraints: A Robust Reformulation via Augmented Dynamics
- On the Gradient Heterogeneity Dynamics of Adversarially Robust Federated Regression
- Kernel-Robust Dynamics for Reaction-Diffusion Equations with Measure-Valued Delay
- Minimal Two-Spheres and Manifolds with Positive Isotropic Curvature
- The exact number of nonconstant positive solutions to the planar isotropic Lp dual Minkowski problem
BibTeX
@misc{patel2026exact,
title = {{Exact SO(3)-Equivariant Isotropic Kernels for Rotation-Robust Neural Dynamics}},
author = {Ridham Patel},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.10626},
eprint = {2610.10626},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.10626},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Exact SO(3)-Equivariant Isotropic Kernels for Rotation-Robust Neural Dynamics
AU - Ridham Patel
PY - 2026
DA - 2026/10/07
PB - arXiv
DO - 10.48550/arxiv.2610.10626
UR - https://arxiv.org/abs/2610.10626
AB - Neural surrogates for vector-valued partial differential equations can fit training data yet change their predictions when the same physical state is expressed in a rotated coordinate frame. We study this failure on three-dimensional Navier--Stokes dynamics observed at irregularly placed points. We introduce the Invariant-Conditioned Isotropic Kernel Neural Operator (IKNO), a compact graph model that builds local interactions from scalar quantities unchanged by rotation and vector directions that rotate with the data. Consequently, rotating the positions and velocities rotates the predicted velocity change in exactly the same way. On a held-out test set fixed after model design, training unconstrained graph models on randomly rotated examples reduces but does not eliminate their coordinate dependence. In contrast, IKNO is consistent to numerical precision, matches the forecasting accuracy of a general rotation-aware Tensor Field Network with $5.6$ times fewer parameters, and outperforms a parameter-matched graph simulator. These results show that a compact, PDE-specialized model can remove coordinate dependence without sacrificing forecasting accuracy.
ER -