arXiv · 2603.11323
UNet-AF: An alias-free UNet for image restoration
Abstract
The simplicity and effectiveness of the UNet architecture makes it ubiquitous in image restoration, image segmentation, and diffusion models. They are often assumed to be equivariant to translations, yet they traditionally consist of layers that are known to be prone to aliasing, which hinders their equivariance in practice. To overcome this limitation, we propose a new alias-free UNet designed from a careful selection of state-of-the-art translation-equivariant layers. We evaluate the proposed equivariant architecture against non-equivariant baselines on image restoration tasks and observe competitive performance with a significant increase in measured equivariance. Through extensive ablation studies, we also demonstrate that each change is crucial for its empirical equivariance. Our implementation is available at https://github.com/jscanvic/UNet-AF
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jérémy Scanvic, Quentin Barthélemy, Julián Tachella. 2026-03-11. UNet-AF: An alias-free UNet for image restoration. https://arxiv.org/abs/2603.11323
Cite the original work for its findings. Save a collection to share your selection of sources.