arXiv · 2511.18667
Equivariant Deep Equilibrium Models for Imaging Inverse Problems
Abstract
Equivariant imaging (EI) enables training signal reconstruction models without requiring ground truth data by leveraging signal symmetries. Deep equilibrium models (DEQs) are a powerful class of neural networks where the output is a fixed point of a learned operator. However, training DEQs with complex EI losses requires implicit differentiation through fixed-point computations, whose implementation can be challenging. We show that backpropagation can be implemented modularly, simplifying training. Experiments demonstrate that DEQs trained with implicit differentiation outperform those trained with Jacobian-free backpropagation and other baseline methods. Additionally, we find evidence that EI-trained DEQs approximate the proximal map of an invariant prior.
Explore related subjects
Keep this discovery
Alexander Mehta, Ruangrawee Kitichotkul, Vivek K Goyal, Julián Tachella. 2025-11-24. Equivariant Deep Equilibrium Models for Imaging Inverse Problems. https://arxiv.org/abs/2511.18667
Cite the original work for its findings. Save a collection to share your selection of sources.