Equivariant Imaging

Learning from Measurements Alone

DIP reconstructs from a single measurement but requires per-image optimisation. Noise2Noise and SURE train amortised networks but need either paired data or Gaussian noise assumptions. Equivariant Imaging (EI) offers a fundamentally different approach: it learns a reconstruction network from unpaired measurements alone, using the symmetries of the signal class as self-supervision.

The key insight, due to Tachella, Chen, and Davies (2021), is that if the signal class is invariant under a group G\mathcal{G} of transformations, then measurement consistency plus equivariance uniquely determines the reconstruction --- even in the null space of the sensing operator A\mathbf{A}.

Definition:

Equivariant Imaging (EI)

Equivariant Imaging trains a reconstruction network using only unpaired measurements, by enforcing two losses:

1. Data consistency: LDC=Ey[βˆ₯AfΞΈ(y)βˆ’yβˆ₯2]\mathcal{L}_{\text{DC}} = \mathbb{E}_\mathbf{y}\bigl[\|\mathbf{A}f_\theta(\mathbf{y}) - \mathbf{y}\|^2\bigr]

2. Equivariance: LEI=Ey,g[βˆ₯fΞΈ(ATgfΞΈ(y))βˆ’TgfΞΈ(y)βˆ₯2]\mathcal{L}_{\text{EI}} = \mathbb{E}_{\mathbf{y}, g}\bigl[\|f_\theta(\mathbf{A}T_g f_\theta(\mathbf{y})) - T_g f_\theta(\mathbf{y})\|^2\bigr]

where TgT_g is a transformation from a group G\mathcal{G} under which the signal class is invariant.

The total loss is: L=LDC+λ LEI.\mathcal{L} = \mathcal{L}_{\text{DC}} + \lambda\,\mathcal{L}_{\text{EI}}.

EI requires no paired training data and no pretrained model. It learns to reconstruct from measurements alone, using the group symmetries as a self-supervisory signal. The key insight is that the symmetries provide information about the null space of A\mathbf{A}: the equivariance constraint fills in the missing information that measurements cannot provide.

Theorem: EI Recovers the Null Space

Let A∈RMΓ—N\mathbf{A} \in \mathbb{R}^{M \times N} with M<NM < N (underdetermined), and let G\mathcal{G} be a group of transformations acting on RN\mathbb{R}^N. If:

  1. The signal class is invariant under G\mathcal{G}: Tgx∈XT_g \mathbf{x} \in \mathcal{X} for all x∈X\mathbf{x} \in \mathcal{X} and g∈Gg \in \mathcal{G}.

  2. The group action is transitive on the null space: for any v∈null(A)\mathbf{v} \in \text{null}(\mathbf{A}), there exist g1,…,gK∈Gg_1, \ldots, g_K \in \mathcal{G} and Ξ±1,…,Ξ±K\alpha_1, \ldots, \alpha_K such that v=βˆ‘kΞ±k Pnull Tgk Prange u\mathbf{v} = \sum_k \alpha_k\, \mathbf{P}_{\text{null}}\, T_{g_k}\, \mathbf{P}_{\text{range}}\, \mathbf{u} for some u\mathbf{u}.

Then the EI loss LDC+λ LEI\mathcal{L}_{\text{DC}} + \lambda\,\mathcal{L}_{\text{EI}} has a unique minimiser fΞΈβˆ—f_\theta^* satisfying:

fΞΈβˆ—(Ax+w)β†’xasΒ Οƒ2β†’0.f_\theta^*(\mathbf{A}\mathbf{x} + \mathbf{w}) \to \mathbf{x} \qquad \text{as } \sigma^2 \to 0.

Data consistency constrains the reconstruction in the range of AH\mathbf{A}^H (the measured subspace). The equivariance loss constrains the null space: when we transform the reconstruction Tgx^T_g \hat{\mathbf{x}}, the null-space component changes, and requiring the network to reconstruct it correctly from new measurements provides indirect observations of the null space.

,

Equivariant Imaging: Effect of Symmetry Group

Visualise how equivariant imaging uses symmetries to learn reconstruction without ground truth. The plot shows the reconstruction PSNR as a function of the number of group transformations used in training.

More transformations provide more self-supervisory signal, improving reconstruction quality. Rotations are the most informative for isotropic scenes; shifts are best for systems with partial Fourier measurements (SAR, diffraction tomography).

Parameters
4
0.25

Example: Equivariant Imaging for Partial Fourier Measurements

Apply equivariant imaging to an RF imaging system with partial Fourier sensing A=PΞ©F\mathbf{A} = \mathbf{P}_\Omega\mathbf{F}. Identify appropriate symmetries and explain why shifts are particularly effective.

Definition:

Measurement Splitting

Measurement splitting is a complementary self-supervised strategy that splits the measurements into two disjoint subsets:

y1=A1x+w1,y2=A2x+w2\mathbf{y}_1 = \mathbf{A}_1\mathbf{x} + \mathbf{w}_1, \qquad \mathbf{y}_2 = \mathbf{A}_2\mathbf{x} + \mathbf{w}_2

where A=[A1;A2]\mathbf{A} = [\mathbf{A}_1; \mathbf{A}_2]. Train the network on y1\mathbf{y}_1 and validate on y2\mathbf{y}_2:

Lsplit=βˆ₯A2fΞΈ(y1)βˆ’y2βˆ₯2.\mathcal{L}_{\text{split}} = \|\mathbf{A}_2 f_\theta(\mathbf{y}_1) - \mathbf{y}_2\|^2.

This is a Noise2Noise-like loss applied to subsets of measurements.

Measurement splitting requires no symmetry assumptions but needs enough measurements in each split. It can be combined with EI for additional self-supervision.

Example: Equivariant Imaging for Multi-View RF Imaging

A multi-static RF imaging system with II transmitters and JJ receivers collects measurements from multiple viewpoints. How can equivariant imaging exploit the multi-view geometry?

,

Self-Supervised Methods: Data Requirements and Assumptions

MethodData NeededKey AssumptionNull-Space InfoBest For
DIP1 measurementCNN spectral biasArchitecture biasNo data available
Noise2NoiseNoisy pairsZero-mean independent noiseNone (denoising only)Repeated measurements
SURENoisy imagesGaussian noise, known Οƒ2\sigma^2None (denoising only)Gaussian noise regime
EIUnpaired measurementsSignal symmetriesGroup action on null spaceStructured forward models
Meas. splittingRedundant measurementsEnough per splitCross-validationOver-sampled systems

Common Mistake: Wrong Symmetry Group Degrades EI Performance

Mistake:

Using rotation equivariance for RF scenes that are not rotationally invariant (e.g., corridor environments, elongated structures, or scenes with strong directional features).

Correction:

The EI loss penalises violations of the assumed symmetry. If the scene is not truly invariant under G\mathcal{G}, the equivariance constraint becomes a wrong prior that biases the reconstruction.

Best practice: Choose symmetries that match the scene statistics. For urban RF imaging, use discrete rotations (90-degree) and reflections. For general scenes, use shifts (which assume stationarity, a weaker assumption). When in doubt, use measurement splitting instead.

Quick Check

Why does data consistency alone (without equivariance) fail to produce good reconstructions for underdetermined inverse problems?

Data consistency is not a valid loss function

The null space of A\mathbf{A} is unconstrained --- infinitely many signals produce the same measurements

Neural networks cannot learn linear inverse problems

Gradient descent fails to converge for the data consistency loss

Historical Note: From Edinburgh to DeepInverse

2021

Equivariant imaging was developed by Julian Tachella, Dongdong Chen, and Mike Davies at the University of Edinburgh. Davies --- a leading figure in compressed sensing and computational imaging --- saw that the missing piece in self-supervised imaging was a principled way to handle the null space.

The EI framework drew on ideas from geometric deep learning (group equivariant networks) and classical sampling theory (the role of symmetries in Fourier analysis). The team subsequently developed the DeepInverse library, an open-source PyTorch toolkit that implements EI alongside DIP, Noise2Noise, SURE, and many other methods discussed in this chapter.

Mike Davies was also referenced by Caire as a key figure in the intersection of compressed sensing and RF imaging.

Equivariant Imaging (EI)

A self-supervised framework for inverse problems that learns reconstruction from unpaired measurements by exploiting known symmetries of the signal class, providing information about the null space of the sensing operator.

Related: Measurement Splitting

Measurement Splitting

A self-supervised strategy that splits measurements into disjoint subsets, training on one and validating on the other, providing a Noise2Noise-like loss for inverse problems.

Related: Equivariant Imaging (EI)

Key Takeaway

Equivariant imaging trains reconstruction networks from measurements alone, using signal symmetries as self-supervision. The equivariance loss constrains the null space of A\mathbf{A}, complementing data consistency that constrains the range. For RF imaging, translations and rotations are natural symmetries for scenes with spatial stationarity or isotropy. The choice of symmetry group must match the scene statistics --- wrong symmetries act as a harmful prior.