Clifford-Steerable Convolutional Neural Networks
About
We present Clifford-Steerable Convolutional Neural Networks (CS-CNNs), a novel class of $\mathrm{E}(p, q)$-equivariant CNNs. CS-CNNs process multivector fields on pseudo-Euclidean spaces $\mathbb{R}^{p,q}$. They cover, for instance, $\mathrm{E}(3)$-equivariance on $\mathbb{R}^3$ and Poincar\'e-equivariance on Minkowski spacetime $\mathbb{R}^{1,3}$. Our approach is based on an implicit parametrization of $\mathrm{O}(p,q)$-steerable kernels via Clifford group equivariant neural networks. We significantly and consistently outperform baseline methods on fluid dynamics as well as relativistic electrodynamics forecasting tasks.
Maksim Zhdanov, David Ruhe, Maurice Weiler, Ana Lucic, Johannes Brandstetter, Patrick Forr\'e• 2024
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 1-step forecasting | Navier-Stokes (NS) R2 (test) | MSE4.50e-4 | 15 | |
| 1-step forecasting | Shallow-water equations SWE-1 R2 (test) | MSE0.1132 | 15 | |
| 1-step forecasting | 3D Maxwell (MW3) R3 (test) | MSE6.82e-4 | 15 | |
| 1-step forecasting | 2D relativistic Maxwell (MW2) R1,2 (test) | MSE0.3494 | 15 |
Showing 4 of 4 rows