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 of transformations, then measurement consistency plus equivariance uniquely determines the reconstruction --- even in the null space of the sensing operator .
Definition: Equivariant Imaging (EI)
Equivariant Imaging (EI)
Equivariant Imaging trains a reconstruction network using only unpaired measurements, by enforcing two losses:
1. Data consistency:
2. Equivariance:
where is a transformation from a group under which the signal class is invariant.
The total loss is:
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 : the equivariance constraint fills in the missing information that measurements cannot provide.
Theorem: EI Recovers the Null Space
Let with (underdetermined), and let be a group of transformations acting on . If:
-
The signal class is invariant under : for all and .
-
The group action is transitive on the null space: for any , there exist and such that for some .
Then the EI loss has a unique minimiser satisfying:
Data consistency constrains the reconstruction in the range of (the measured subspace). The equivariance loss constrains the null space: when we transform the reconstruction , the null-space component changes, and requiring the network to reconstruct it correctly from new measurements provides indirect observations of the null space.
Range-null decomposition
Any signal decomposes as where and .
Data consistency uniquely determines (in the noiseless case): implies .
Equivariance constrains the null space
For a transformation , the transformed signal has a different range-null decomposition. The requirement constrains how the network maps range-space information to null-space content.
If mixes the range and null spaces (transitive action), then different transformations provide complementary constraints on , collectively determining it uniquely.
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
Example: Equivariant Imaging for Partial Fourier Measurements
Apply equivariant imaging to an RF imaging system with partial Fourier sensing . Identify appropriate symmetries and explain why shifts are particularly effective.
Forward model
: the DFT followed by selection of frequencies in . The null space consists of signals whose Fourier support lies entirely in (unmeasured frequencies).
Why shifts work
By the Fourier shift theorem, a spatial shift by pixels corresponds to phase modulation in frequency: .
The EI loss for the shifted signal creates measurements at the same frequency locations , but with different phase factors. This creates a system of equations that constrains the unmeasured frequency content through the reconstruction network.
Practical protocol
For each training measurement :
- Reconstruct:
- Sample random shift
- Create virtual measurement:
- Re-reconstruct:
- EI loss:
Definition: Measurement Splitting
Measurement Splitting
Measurement splitting is a complementary self-supervised strategy that splits the measurements into two disjoint subsets:
where . Train the network on and validate on :
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 transmitters and receivers collects measurements from multiple viewpoints. How can equivariant imaging exploit the multi-view geometry?
Multi-view forward model
Each transmitter-receiver pair provides measurements where depends on the array geometry and the transmit/receive steering vectors.
The combined sensing matrix has a structured null space determined by the spatial frequency coverage.
Perspective equivariance
For scenes that are invariant under rotations about the vertical axis (common for urban/indoor environments), rotating the scene is equivalent to permuting the transmitter-receiver pairs.
This geometric equivariance provides a natural group for EI: rotate the reconstructed scene, re-measure with the appropriate permuted sensing matrix, and require consistency.
Practical considerations
The rotation group for RF imaging is typically discrete (e.g., 4-fold or 8-fold symmetry for planar arrays). Additional symmetries may include:
- Translations (shift-invariant scenes)
- Reflections (symmetric array configurations)
- Scale invariance (for multi-frequency measurements)
The effectiveness depends on how well these symmetries mix the null space of .
Self-Supervised Methods: Data Requirements and Assumptions
| Method | Data Needed | Key Assumption | Null-Space Info | Best For |
|---|---|---|---|---|
| DIP | 1 measurement | CNN spectral bias | Architecture bias | No data available |
| Noise2Noise | Noisy pairs | Zero-mean independent noise | None (denoising only) | Repeated measurements |
| SURE | Noisy images | Gaussian noise, known | None (denoising only) | Gaussian noise regime |
| EI | Unpaired measurements | Signal symmetries | Group action on null space | Structured forward models |
| Meas. splitting | Redundant measurements | Enough per split | Cross-validation | Over-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 , 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 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
Correct. For , the system has infinitely many solutions differing by null-space components. Without additional constraints (symmetries, priors, regularisation), the network can produce any null-space content.
Historical Note: From Edinburgh to DeepInverse
2021Equivariant 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 , 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.