Abstract:Many areas of science and engineering encounter data defined on spherical manifolds. Modelling and analysis of spherical data often necessitates spherical harmonic transforms, at high degrees, and increasingly requires efficient computation of gradients for machine learning or other differentiable programming tasks. We develop novel algorithmic structures for accelerated and differentiable computation of generalised Fourier transforms on the sphere $\mathbb{S}^2$ and rotation group $\text{SO}(3)$, i.e. spherical harmonic and Wigner transforms, respectively. We present a recursive algorithm for the calculation of Wigner $d$-functions that is both stable to high harmonic degrees and extremely parallelisable. By tightly coupling this with separable spherical transforms, we obtain algorithms that exhibit an extremely parallelisable structure that is well-suited for the high throughput computing of modern hardware accelerators (e.g. GPUs). We also develop a hybrid automatic and manual differentiation approach so that gradients can be computed efficiently. Our algorithms are implemented within the JAX differentiable programming framework in the S2FFT software code. Numerous samplings of the sphere are supported, including equiangular and HEALPix sampling. Computational errors are at the order of machine precision for spherical samplings that admit a sampling theorem. When benchmarked against alternative C codes we observe up to a 400-fold acceleration. Furthermore, when distributing over multiple GPUs we achieve very close to optimal linear scaling with increasing number of GPUs due to the highly parallelised and balanced nature of our algorithms. Provided access to sufficiently many GPUs our transforms thus exhibit an unprecedented effective linear time complexity.

Fast expansion into harmonics on the ball

Fast and backward stable transforms between spherical harmonic expansions and bivariate Fourier series

Fast Calculation of Laplace-Beltrami Eigenproblems Via Subdivision Linear Subspace

Numerical reconstruction from the Fourier transform on the ball using prolate spheroidal wave functions

Differentiable and accelerated spherical harmonic and Wigner transforms

Fast Discrete Algorithms for Sparse Fourier Expansions of High Dimensional Functions

Efficient Exact Quadrature of Regular Solid Harmonics Times Polynomials Over Simplices in $\mathbb{R}^3$

An adaptive fast Gauss transform in two dimensions

Benchmarking Numerical Algorithms for Harmonic Maps into the Sphere

A fast direct solver for nonlocal operators in wavelet coordinates

Optimization Algorithms for Faster Computational Geometry.

Orthogonal Polynomial Expansions on Sparse Grids.

The Radon Transform over Cones with Vertices on the Sphere and Orthogonal Axes

Fast and Accurate Spherical Harmonics Products.

Convergence in Total Variation for nonlinear functionals of random hyperspherical harmonics

A Fast Fourier-Galerkin Method Solving Boundary Integral Equations for the Helmholtz Equation with Exponential Convergence

Fast and accurate evaluation of Biot-Savart integrals over spatial curves

Solving parabolic equations on the unit sphere via Laplace transforms and radial basis functions

An efficient solver for volumetric scattering based on fast spherical harmonics transforms

Subspace recursive Fermi-operator expansion strategies for large-scale DFT eigenvalue problems on HPC architectures