University of Bergen's GsCNN Mixes Scalars and Vectors for Spherical AI

A new SO(3)-equivariant architecture lifts the kernel and nonlinearity constraints of spherical CNNs, improving wind forecasts and spherical MNIST tasks.

·
·
·
University of Bergen's GsCNN Mixes Scalars and Vectors for Spherical AIPRO
  • Bergen team introduces GsCNN, an SO(3)-equivariant architecture for scalar and vector fields on the sphere (paper).
  • Removes kernel and nonlinearity constraints that limited previous spherical CNNs.
  • Each layer: SO(3) conv, inverse FT, nonlinearity, FT, smoothing in output spin.
  • Outperforms standard UNet CNN on ERA5 wind forecasting without data augmentation.
  • Matches or exceeds spin-weighted sCNNs; gap closes when spins interact in hidden layers.
  • Code and Triton kernels available in the S3CNN repository.

Weather forecasts, cosmological maps, and panoramic vision systems process signals whose natural domain is a sphere. Researchers at the University of Bergen have introduced a General spherical CNN, or GsCNN, in a new paper. The architecture preserves three-dimensional rotational symmetry while processing scalar fields such as temperature, vector fields such as wind, and higher-order tensor fields.

Rotation equivariance means that rotating an input produces a corresponding rotation in the output. Encoding that behavior directly can reduce the need for rotation-based data augmentation and prevent predictions from changing because a spherical field was expressed in a different orientation.

Flat grids lose globe rotations

A planar convolution applies the same translated filter across a rectangular grid. On a latitude-longitude map, grid cells shrink toward the poles, longitude wraps around, and neighborhoods change shape with latitude. Longitude shifts correspond to rotations around the globe’s vertical axis, but the grid does not preserve arbitrary rotations in three dimensions.

Earlier spherical CNNs addressed this geometry with Fourier-domain convolutions over SO(3), the mathematical group containing every three-dimensional rotation. Spin-weighted spherical CNNs extended the approach to vector fields, including work by Esteves and collaborators.

Spin describes how a local quantity changes when the tangent coordinate frame rotates around the surface normal. A scalar has spin zero, while tangent vectors such as wind can be represented through nonzero spin components. Previous spin-weighted architectures preserved these transformation rules by limiting their kernels and nonlinear activation functions.

Pro article

This story is for Pro members

You've reached the end of the free preview. Upgrade to AlphaSignal Pro to read the full article - and everything else behind the paywall.

Trending
  • No trending articles

Comments

avatar

Next Reads