Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SE-UNet: Singular Equivariant Imaging for Real-World Constrained Generation

About

While diffusion models have revolutionized image synthesis, their application to real-world inverse problems is often hampered by the need for massive datasets and the difficulty of imposing strict physical constraints. In this work, we introduce \textbf{SE-UNet} (Singular Equivariant UNet), a framework designed to solve ill-posed imaging tasks without extensive pre-training. By treating generation as an optimization problem constrained by geometric equivariance ($D_4$ group) and singular value gating, SE-UNet effectively standardizes the solution space. We demonstrate that these strong inductive biases allow for state-of-the-art zero-shot inpainting results (80\% missing pixels) on CIFAR-10. Our method surpasses Deep Image Prior (DIP) baselines by over 4 dB in PSNR and exhibits a characteristic "singular snap" convergence -- rapidly locking into the signal manifold. SE-UNet thus offers a data-efficient pathway for constrained generation, aligning with the ReALM-GEN goal of bridging theoretical priors with practical deployment.

Kanishk Awadhiya• 2026

Related benchmarks

TaskDatasetResultRank
Image InpaintingCIFAR-10 (test)
PSNR25.34
9
Showing 1 of 1 rows

Other info

Follow for update