Search papers, labs, and topics across Lattice.
This paper introduces a novel SO(3)-equivariant convolutional network (CNN) specifically designed for HEALPix discretized spherical data, leveraging steerable filters, parallel transport, and spin-weighted features. By encoding exact physical symmetry, the network avoids costly learning of orientation invariance, achieving sub-percent SO(3)-invariance error with controllable hyperparameters. This architecture demonstrates state-of-the-art performance in dense gradient regression and competitive results in CMB foreground removal and denoising, significantly improving simulation efficiency for scientific inference tasks.
Spherical CNNs learn physical symmetry exactly, slashing simulation costs and unlocking new levels of accuracy in cosmology and beyond.
Many of the most important datasets in the physical sciences live natively on the sphere, and these data often carry no preferred position or orientation. Using methods that do not reflect this symmetry leads to far higher simulation requirements, which can be catastrophic for scientific analyses (forcing reliance on cheap, imperfect models that bias results). We present a novel spherical implementation of a gauge equivariant mesh architecture that is specifically designed for the Hierarchical Equal Area isoLatitude Pixelization (\healpix{}) of the sphere. The $\mathrm{SO}(3)$-equivariant convolution is built from steerable filters, parallel transport and spin-weighted features. The architecture supplies the exact symmetry, so the network does not spend training data learning it. The convolution operations are constructed to give provably exact equivariance, given a band-limit. Our full CNNs can achieve SO(3)-invariance error at per-cent-level and lower, which is controlled through well-understood hyperparameters. On benchmarks, our technique gives the most accurate dense gradient regression and competitive cosmic microwave background foreground removal and denoising. The improved simulation efficiency matters most in simulation-based inference (SBI), where high-fidelity simulations are expensive and neural compressors require large training sets, so a fixed budget forces analyses with less accurate simulators. In cosmology lensing SBI, our technique yields the tightest posteriors at matched simulation count. This approach is a novel method that will enable new convolutional neural network analyses, including generative and inference tasks, for processes and data on the sphere.