Multiscale Deep Equilibrium Models
Not on the map. Why?
Abstract (arXiv)
We propose a new class of implicit networks, the multiscale deep equilibrium model (MDEQ), suited to large-scale and highly hierarchical pattern recognition domains. An MDEQ directly solves for and backpropagates through the equilibrium points of multiple feature resolutions simultaneously, using implicit differentiation to avoid storing intermediate states (and thus requiring only O(1) memory consumption). These simultaneously-learned multi-resolution features allow us to train a single model on a diverse set of tasks and loss functions, such as using a single MDEQ to perform both image classification and semantic segmentation. We illustrate the effectiveness of this approach on two large-scale vision tasks: ImageNet classification and semantic segmentation on high-resolution images from the Cityscapes dataset. In both settings, MDEQs are able to match or exceed the performance of recent competitive computer vision models: the first time such performance and scale have been achieved by an implicit deep learning approach. The code and pre-trained models are at https://github.com/locuslab/mdeq .
Key facts. Cited by 149 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 37; 2022: 34; 2023: 30; 2024: 19; 2025: 10; 2026: 19.
149 papers.
Sort:
By area:
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2021Neural SDEs as Infinite-Dimensional GANscited by 159
- 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
- 2021You Only Learn One Representation: Unified Network for Multiple Taskscited by 86
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 84
- 2022Tutorial on amortized optimizationcited by 83
- 2021On Training Implicit Modelscited by 73
- 2021Stabilizing Equilibrium Models by Jacobian Regularizationcited by 71
- 2022Learning Deep Implicit Fourier Neural Operators (IFNOs) with Applications to Heterogeneous Material Modelingcited by 65
- 2022Deep Equilibrium Optical Flow Estimationcited by 63
- 2023One-Step Diffusion Distillation via Deep Equilibrium Modelscited by 60
38 more
- 2022Gradient Gating for Deep Multi-Rate Learning on Graphscited by 49
- 2021Vision Transformers with Patch Diversificationcited by 45
- 2022Nonlocal Kernel Network (NKN): a Stable and Resolution-Independent Deep Neural Networkcited by 45
- 2023Parallelizing non-linear sequential models over the sequence lengthcited by 45
- 2021CTNet: Context-based Tandem Network for Semantic Segmentationcited by 43
- 2022Deep Equilibrium Approaches to Diffusion Modelscited by 42
- 2021Long Expressive Memory for Sequence Modelingcited by 41
- 2021Training Feedback Spiking Neural Networks by Implicit Differentiation on the Equilibrium Statecited by 39
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNscited by 38
- 2021If your data distribution shifts, use self-learningcited by 36
- 2022Path Independent Equilibrium Models Can Better Exploit Test-Time Computationcited by 33
- 2022EIGNN: Efficient Infinite-Depth Graph Neural Networkscited by 32
- 2022Reversible Column Networkscited by 32
- 2022Optimization-Induced Graph Implicit Nonlinear Diffusioncited by 28
- 2021Constraint-based graph network simulatorcited by 27
- 2021SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit modelscited by 26
- 2021Resolution learning in deep convolutional networks using scale-space theorycited by 26
- 2021Sliced Recursive Transformercited by 26
- 2022Beyond backpropagation: bilevel optimization through implicit differentiation and equilibrium propagationcited by 26
- 2023Recurrence without Recurrence: Stable Video Landmark Detection with Deep Equilibrium Modelscited by 26
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
- 2022Alternating Differentiation for Optimization Layerscited by 22
- 2023TorchDEQ: A Library for Deep Equilibrium Modelscited by 21
- 2022MGNNI: Multiscale Graph Neural Networks with Implicit Layerscited by 19
- 2021Anytime Dense Prediction with Confidence Adaptivitycited by 18
- 2021Feasibility-based Fixed Point Networkscited by 17
- 2021A global convergence theory for deep ReLU implicit networks via over-parameterizationcited by 17
- 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiationcited by 17
- 2022Bi-level Physics-Informed Neural Networks for PDE Constrained Optimization using Broyden's Hypergradientscited by 17
- 2023Deep Equilibrium Diffusion Restoration with Parallel Samplingcited by 17
- 2022The least-control principle for local learning at equilibriumcited by 16
- 2021Joint inference and input optimization in equilibrium networkscited by 15
- 2021A Unified Efficient Pyramid Transformer for Semantic Segmentationcited by 14
- 2022Explainable AI via Learning to Optimizecited by 14
- 2024Fixed Point Diffusion Modelscited by 14
- 2021Memory-Efficient Differentiable Transformer Architecture Searchcited by 13
- 2022Revisiting Sparse Convolutional Model for Visual Recognitioncited by 13
- 2023A Closer Look at the Adversarial Robustness of Deep Equilibrium Modelscited by 13
50 most cited of 149 All 149 (JSON)
- 2026Mutual Equilibrium: Multimodal Representation Learning through Reciprocal Feedback30 Sep 2026
- 2026A Tilted Bowl Is Not a Slippery Slope: Compressing Looped Models30 Sep 2026
- 2026Q-DEQ: Discrete Solving and Quantization for Deep Equilibrium Models in Time Series Forecasting under Edge Deployment Coding Constraints21 Sep 2026
- 2026Certified Inference and Training for Deep Equilibrium Networks: A Continuation Framework with Polynomial Complexity Guarantees15 Sep 2026
- 2026Response Renormalization for Critical Deep Equilibrium Models24 Aug 2026
- 2026Training Crossroads for Recurrent Vision Transformers: Recurrence, Neural ODEs, and Deep Supervision5 Aug 2026
- 2026SILVA Networks as Structured Implicit Layers and Vector Attractors via Dynamic Interaction Fields31 Jul 2026
- 2026End-to-End Learning of Safe Optimal Feedback Control in High Dimensions with Control Barrier Function Layers22 Jul 2026
- 2026Loop the Loopies!17 Jul 2026
- 2026The Equilibrium Is the Initialization: Lazy Identity Collapse in Physics-Structured Deep Equilibrium Reasoning13 Jul 2026
- 2026Once-for-All: Scalable Simultaneous Forecasting via Equilibrium State Estimation11 Jun 2026
- 2026Training-Free Looped Transformers22 May 2026
38 more
- 2026Visual Implicit Autoregressive Modeling2 May 2026
- 2026Tiny Autoregressive Recursive Models9 Mar 2026
- 2026Understanding the Curse of Unrolling23 Feb 2026
- 2026Lipschitz Multiscale Deep Equilibrium Models: A Theoretically Guaranteed and Accelerated Approach3 Feb 2026
- 2026Consistency Deep Equilibrium Models3 Feb 2026
- 2026On the Convergence of Jacobian-Free Backpropagation for Optimal Control Problems with Implicit Hamiltonians31 Jan 2026
- 2026NewPINNs: Physics-Informing Neural Networks Using Conventional Solvers for Partial Differential Equations23 Jan 2026
- 2025Closed-Loop Transformers: Autoregressive Modeling as Iterative Latent Equilibrium26 Nov 2025
- 2025End-to-End Training of High-Dimensional Optimal Control with Implicit Hamiltonians via Jacobian-Free Backpropagation1 Oct 2025
- 2025Reversible Deep Equilibrium Models16 Sep 2025
- 2025Deep Equilibrium Convolutional Sparse Coding for Hyperspectral Image Denoising21 Aug 2025
- 2025Local Scale Equivariance with Latent Deep Equilibrium Canonicalizer19 Aug 2025
- 2025DyCAF-Net: Dynamic Class-Aware Fusion Network5 Aug 2025
- 2025Ergodic Generative Flows6 May 2025
- 2025Recovering Pulse Waves from Video Using Deep Unrolling and Deep Equilibrium Models21 Mar 2025
- 2025DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows3 Mar 2025
- 2025qNBO: quasi-Newton Meets Bilevel Optimization3 Feb 2025
- 2024Certified Robustness for Deep Equilibrium Models via Serialized Random Smoothing1 Nov 2024
- 2024Understanding Representation of Deep Equilibrium Models from Neural Collapse Perspective30 Oct 2024
- 2024Accelerating AI Performance using Anderson Extrapolation on GPUs25 Oct 2024
- 2024Automatic Differentiation of Optimization Algorithms with Time-Varying Updates21 Oct 2024
- 2024IGNN-Solver: A Graph Neural Solver for Implicit Graph Neural Networks11 Oct 2024
- 2024Differentiation Through Black-Box Quadratic Programming Solvers8 Oct 2024
- 2024On Logical Extrapolation for Mazes with Recurrent and Implicit Networks3 Oct 2024
- 2024Closed-Loop Long-Horizon Robotic Planning via Equilibrium Sequence Modeling2 Oct 2024
- 2024Constructing artificial life and materials scientists with accelerated AI using Deep AndersoNN29 Jul 2024
- 2024Towards Scalable and Stable Parallelization of Nonlinear RNNs26 Jul 2024
- 2024The Extrapolation Power of Implicit Models19 Jul 2024
- 2024LPGD: A General Framework for Backpropagation through Embedded Optimization Layers8 Jul 2024
- 2024Deep Implicit Optimization enables Robust Learnable Features for Deformable Image Registration11 Jun 2024
- 2024Infusing Self-Consistency into Density Functional Theory Hamiltonian Prediction via Deep Equilibrium Models6 Jun 2024
- 2024Subhomogeneous Deep Equilibrium Models1 Mar 2024
- 2024Positive concave deep equilibrium models6 Feb 2024
- 2024Deep Equilibrium Models are Almost Equivalent to Not-so-deep Explicit Models for High-dimensional Gaussian Mixtures5 Feb 2024
- 2024Training Implicit Networks for Image Deblurring using Jacobian-Free Backpropagation3 Feb 2024
- 2024Fixed Point Diffusion Models16 Jan 2024
- 2023One-Step Diffusion Distillation via Deep Equilibrium Models12 Dec 2023
- 2023Accelerating Hopfield Network Dynamics: Beyond Synchronous Updates and Forward Euler27 Nov 2023
50 newest of 149 All 149 (JSON)
- 2023One-Step Diffusion Distillation via Deep Equilibrium Modelscited by 60
- 2022Deep Equilibrium Approaches to Diffusion Modelscited by 42
- 2023Deep Equilibrium Diffusion Restoration with Parallel Samplingcited by 17
- 2024Fixed Point Diffusion Modelscited by 14
- 2025Ergodic Generative Flowscited by 2
- 2025Ergodic Generative Flows6 May 2025
- 2024Fixed Point Diffusion Models16 Jan 2024
- 2023One-Step Diffusion Distillation via Deep Equilibrium Models12 Dec 2023
- 2023Deep Equilibrium Diffusion Restoration with Parallel Sampling20 Nov 2023
- 2022Deep Equilibrium Approaches to Diffusion Models23 Oct 2022
- 2026Training-Free Looped Transformerscited by 12
- 2026Loop the Loopies!cited by 4
- 2026Loop the Loopies!17 Jul 2026
- 2026Training-Free Looped Transformers22 May 2026
- 2023Efficient Symbolic Reasoning for Neural-Network Verification23 Mar 2023
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2021Neural SDEs as Infinite-Dimensional GANscited by 159
- 2023Looped Transformers are Better at Learning Learning Algorithmscited by 135
- 2021You Only Learn One Representation: Unified Network for Multiple Taskscited by 86
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 84
- 2022Tutorial on amortized optimizationcited by 83
- 2021On Training Implicit Modelscited by 73
- 2021Stabilizing Equilibrium Models by Jacobian Regularizationcited by 71
- 2022Learning Deep Implicit Fourier Neural Operators (IFNOs) with Applications to Heterogeneous Material Modelingcited by 65
- 2022Deep Equilibrium Optical Flow Estimationcited by 63
- 2022Gradient Gating for Deep Multi-Rate Learning on Graphscited by 49
12 most cited of 140 All 140 (JSON)
- 2026Mutual Equilibrium: Multimodal Representation Learning through Reciprocal Feedback30 Sep 2026
- 2026A Tilted Bowl Is Not a Slippery Slope: Compressing Looped Models30 Sep 2026
- 2026Q-DEQ: Discrete Solving and Quantization for Deep Equilibrium Models in Time Series Forecasting under Edge Deployment Coding Constraints21 Sep 2026
- 2026Certified Inference and Training for Deep Equilibrium Networks: A Continuation Framework with Polynomial Complexity Guarantees15 Sep 2026
- 2026Response Renormalization for Critical Deep Equilibrium Models24 Aug 2026
- 2026Training Crossroads for Recurrent Vision Transformers: Recurrence, Neural ODEs, and Deep Supervision5 Aug 2026
- 2026SILVA Networks as Structured Implicit Layers and Vector Attractors via Dynamic Interaction Fields31 Jul 2026
- 2026End-to-End Learning of Safe Optimal Feedback Control in High Dimensions with Control Barrier Function Layers22 Jul 2026
- 2026The Equilibrium Is the Initialization: Lazy Identity Collapse in Physics-Structured Deep Equilibrium Reasoning13 Jul 2026
- 2026Once-for-All: Scalable Simultaneous Forecasting via Equilibrium State Estimation11 Jun 2026
- 2026Visual Implicit Autoregressive Modeling2 May 2026
- 2026Tiny Autoregressive Recursive Models9 Mar 2026
12 newest of 140 All 140 (JSON)