The paper demonstrates that iterative regularization is achieved by embedding unrolled iterative solvers within the training pipeline, combining learned filters with explicit data-consistency steps.
It employs methods such as early stopping, epoch control, and iterate averaging to adjust the bias-variance tradeoff and serve as implicit regularizers in both inverse problems and empirical risk minimization.
Empirical evidence in MRI, CT, and deep-network settings confirms that integrating physical forward models with learned regularizers yields competitive performance with enhanced interpretability.
End-to-end training with iterative regularization denotes a family of learning procedures in which the reconstruction operator, the iterative solver, or the optimization trajectory is treated as part of a single training pipeline, while regularization is supplied by explicit penalties, learned priors, the update structure, the stopping rule, or weighted averaging of iterates. In inverse problems, the central pattern is to combine a physical forward model with learned regularization inside an unrolled architecture and, in some cases, to use the learned output as a warm start for a variational refinement. In statistical learning, closely related analyses show that the number of passes over the data or the averaging scheme over optimization iterates can itself act as the effective regularization parameter (Kofler et al., 2022, Mukherjee et al., 2021, Rosasco et al., 2014, Wu et al., 2020).
1. Conceptual scope and problem classes
The cited literature covers two closely connected settings. The first is ill-posed inverse problems, where reconstruction must be stabilized because the system may be underdetermined or unstable due to poor conditioning. The second is empirical risk minimization, where repeated optimization steps can fit sample noise unless the trajectory is controlled. In both settings, the key question is not only which objective is minimized, but also how the iterative procedure itself shapes the final estimator (Kofler et al., 2022, Rosasco et al., 2014).
In the inverse-problem setting, one representative formulation assumes an image that is sparse with respect to convolutional filters and considers an objective of the form
xmin21∥x−y∥22+αk=1∑K∥hk∗x∥1.
The regularizer is explicit, but its practical effect depends on the iterative reconstruction rule that uses it. In the statistical-learning setting, the canonical objective is expected square loss,
w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),
with empirical counterpart
E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.
There, iterative regularization arises because the optimization dynamics over epochs alter the bias-variance tradeoff (Kofler et al., 2022, Rosasco et al., 2014).
A useful cross-sectional comparison is the following.
Taken together, these works place “iterative regularization” in a broader sense than classical explicit penalization alone. The iteration can carry the regularizing effect through architecture, solver coupling, stopping time, or post hoc averaging.
2. Physics-informed unrolling and convolutional analysis operator learning
In "Convolutional Analysis Operator Learning by End-To-End Training of Iterative Neural Networks" (Kofler et al., 2022), convolutional analysis operator learning (CAOL) is reformulated so that the learned filters are optimized for the reconstruction algorithm that actually uses them. Classical CAOL trains filters separately on ground-truth images by minimizing a transform-learning objective, but that decoupled procedure ignores the subsequently employed reconstruction algorithm as well as the physical model responsible for image formation. The paper’s central claim is that iterative neural networks can overcome this limitation because they contain the physical model (Kofler et al., 2022).
The variational model introduces auxiliary variables sk and rewrites the nonsmooth problem as
with λ>0. The unrolled network alternates between a regularization step and a data-consistency step:
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,
initialized by
x0:=A♯y.
The second step is equivalent to solving
(A∗A+λI)x=A∗y+λzj.
For non-Cartesian MRI with density compensation, w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),0 and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),1 are replaced by preconditioned versions using w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),2 (Kofler et al., 2022).
This architecture is model-based rather than a generic CNN. The reconstruction is constrained by the measurement operator w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),3, the learnable part is the regularizer or filter bank, and the architecture mirrors a proximal splitting method. Under an orthonormal-basis assumption, the iteration corresponds to a backward-backward splitting algorithm for the regularized problem. The trainable parameter set is w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),4, and the network can also learn w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),5 and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),6; to enforce positivity, the paper applies a Soft-Plus nonlinearity to them. Because soft-thresholding is not differentiable with respect to its threshold, the paper uses the smooth approximation
w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),7
with w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),8, and trains all methods with ADAM (Kofler et al., 2022).
The principal significance of this formulation is that the learned filters are optimized through the full unrolled reconstruction pipeline. They are therefore tuned to the finite number of unrolled iterations, the exact measurement operator, and the specific data-consistency step, rather than to a standalone sparsity objective.
3. Jointly learned regularizers and warm-started variational refinement
A different but closely related formulation appears in "End-to-end reconstruction meets data-driven regularization for inverse problems" (Mukherjee et al., 2021), which proposes Unrolled Adversarial Regularization (UAR). The method combines an unrolled reconstruction operator w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),9 with a parametrized regularizer E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.0, and the two are trained jointly. The training objective is a min-max problem,
E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.1
which is equivalent, via Kantorovich-Rubinstein duality, to
E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.2
The first term enforces measurement consistency, while the second matches the distribution of reconstructions to the ground-truth distribution through the Wasserstein-1 distance (Mukherjee et al., 2021).
The regularizer acts adversarially. The alternating updates are
The reconstruction network itself is iterative and unrolled. Inspired by a primal-dual-style algorithm, it replaces proximal operators in the image and measurement spaces with trainable CNNs:
E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.8
E(w)=n1i=1∑n(⟨w,xi⟩−yi)2.9
initialized by
sk0
The learnable components include primal CNN blocks, dual CNN blocks, and step sizes sk1, sk2, initialized at sk3, with sk4 unrolled layers (Mukherjee et al., 2021).
After training, the learned reconstruction sk5 is used to initialize a separate variational refinement,
The paper states that the end-to-end reconstruction gives an excellent initial point and that it takes significantly fewer iterations for gradient-descent to recover the optimal solution. It also claims well-posedness under compactness and continuity assumptions, noise stability in the sense that sk8 up to subsequences, and a principled tradeoff controlled by sk9: small x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,0 prioritizes measurement consistency, while large x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,1 prioritizes matching the image distribution (Mukherjee et al., 2021).
4. Epoch count and early stopping as iterative regularization
"Learning with incremental iterative regularization" (Rosasco et al., 2014) develops a statistical theory in which the number of epochs acts as the effective regularization parameter. The setting is least-squares learning in a separable Hilbert space x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,2, with bounded inputs and outputs. The algorithm is an incremental gradient method over the empirical risk, run one point at a time and repeated for multiple epochs:
Here x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,5 indexes epochs, not individual sample updates (Rosasco et al., 2014).
The paper’s central principle is that the step size x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,6 is fixed a priori and the number of epochs x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,7 is the only tuning parameter. Small x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,8 means that the algorithm has not fit the sample too closely; large x,{sk}kmin21∥x−y∥22+2λk=1∑K∥hk∗x−sk∥22+αk=1∑K∥sk∥1,9 reduces approximation error but increases estimation error. In that sense, epochs control the bias-variance tradeoff and act exactly like a regularization parameter. The analysis compares the empirical iterate λ>00 with a population iterate λ>01, decomposing the error into sample error and approximation error through inequalities such as
λ>02
When the best linear solution exists,
λ>03
These identities formalize the claim that optimization dynamics and statistical regularization are inseparable in multi-epoch training (Rosasco et al., 2014).
The strong universal consistency theorem states that if the stopping rule λ>04 satisfies
λ>05
then
λ>06
If λ>07, then also
λ>08
Finite-sample bounds make the same point quantitatively. For any λ>09,
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),0
In the attainable case zj=k=1∑Khk∗∗Sα/λ(hk∗xj),1, the excess risk obeys
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),2
with recommended stopping rule
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),3
yielding
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),4
In the non-attainable case zj=k=1∑Khk∗∗Sα/λ(hk∗xj),5,
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),6
and the optimal epoch choice is
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),7
leading to
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),8
The paper therefore shows that multiple epochs are not merely an optimization detail; they are part of the statistical procedure (Rosasco et al., 2014).
5. Iterate averaging and adjustable regularization
"Obtaining Adjustable Regularization for Free via Iterate Averaging" (Wu et al., 2020) studies a different mechanism by which an optimization trajectory can be converted into a regularized solution. The starting point is unregularized empirical risk minimization,
zj=k=1∑Khk∗∗Sα/λ(hk∗xj),9
and its explicitly regularized counterpart
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,0
with canonical choice
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,1
Given a probability weight sequence xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,2, cumulative weights xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,3, and averaged iterate
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,4
the paper’s central claim is that with the right weighting scheme, xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,5 behaves like the solution path of a regularized problem (Wu et al., 2020).
For SGD on an xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,6-strongly convex and xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,7-smooth objective, the regularized and unregularized learning rates are linked by
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,8
and
xj+1=argxmin21∥Ax−y∥22+2λ∥x−zj∥22,9
The main theorem gives the exact relation in expectation
x0:=A♯y.0
together with
x0:=A♯y.1
In this sense, iterate averaging implements x0:=A♯y.2-regularization. The practical interpretation is explicit: one can train only the unregularized model, choose the averaging weights accordingly, and obtain the effect of the regularized solution without retraining for each x0:=A♯y.3 (Wu et al., 2020).
The result extends beyond vanilla SGD. For preconditioned SGD with positive definite preconditioner x0:=A♯y.4, the induced regularizer becomes
x0:=A♯y.5
and the same structural relation holds. For Nesterov’s accelerated SGD, the paper defines a modified weighting scheme
x0:=A♯y.6
and proves analogous convergence, thereby answering the open question posed by Neu and Rosasco on whether an averaging scheme exists for NSGD that yields adjustable x0:=A♯y.7-regularization (Wu et al., 2020).
The paper also treats general strongly convex and smooth objectives. There the result becomes approximate rather than exact: the averaged solution lies between two regularized solutions with tunable parameters
x0:=A♯y.8
When the objective is quadratic, x0:=A♯y.9, and the approximate statement becomes exact regularization. This establishes iterate averaging as a form of iterative regularization rather than merely a variance-reduction device (Wu et al., 2020).
6. Empirical demonstrations and methodological trade-offs
The empirical studies span MRI reconstruction, CT reconstruction, and deep networks. In accelerated radial cardiac cine MRI, the CAOL-unrolling study evaluates a multi-coil, non-Cartesian encoding model with (A∗A+λI)x=A∗y+λzj.0 coil sensitivity maps, golden-angle radial sampling, density compensation preconditioning, and TorchKbNufft for the NUFFT implementation. The dataset consists of 15 healthy volunteers and 4 patients, with 216 cine MR images total, image size (A∗A+λI)x=A∗y+λzj.1, split by subject into 12/3/4 for train/validation/test, retrospective undersampling at approximately (A∗A+λI)x=A∗y+λzj.2, and added Gaussian noise with (A∗A+λI)x=A∗y+λzj.3. Models were trained with (A∗A+λI)x=A∗y+λzj.4, filter sizes (A∗A+λI)x=A∗y+λzj.5, network depth (A∗A+λI)x=A∗y+λzj.6, and (A∗A+λI)x=A∗y+λzj.7 conjugate-gradient iterations, using ADAM with learning rate (A∗A+λI)x=A∗y+λzj.8. The proposed network was trained for 75 epochs ((A∗A+λI)x=A∗y+λzj.9 hours), while the DnCn3D baseline was trained for 500 epochs (w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),00 days). Compared against decoupled CAOL and DnCn3D, the proposed method produced visibly better reconstructions than CAOL, clearly outperformed CAOL quantitatively, and achieved performance comparable to DnCn3D while remaining more interpretable and using fewer trainable parameters (Kofler et al., 2022).
In sparse-view CT on Mayo Clinic low-dose CT data, the UAR study uses 2250 slices from 9 patients for training and 128 slices from 1 patient for testing, with parallel-beam geometry, 200 projection angles, and Gaussian noise with w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),01. Reported quantitative results include about w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),02 dB PSNR and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),03 SSIM for AR, about w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),04 dB PSNR and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),05 SSIM for ACR, about w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),06 dB PSNR and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),07 SSIM for UAR with w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),08, and about w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),09 dB PSNR and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),10 SSIM for UAR with refinement. Reconstruction times are around w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),11 seconds per image for AR, around w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),12 seconds for ACR, around w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),13 seconds for UAR end-to-end, and around w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),14 seconds for UAR with refinement. The paper also reports an interpretable dependence on w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),15: w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),16 and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),17 are too weak, w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),18 gives the best tradeoff, and w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),19 is over-regularized (Mukherjee et al., 2021).
In deep-network experiments, the iterate-averaging study trains VGG-16 on CIFAR-10, ResNet-18 on CIFAR-10, and ResNet-18 on CIFAR-100 with SGD for 300 epochs, averages checkpoints from epochs 61 to 300, uses geometric weighting w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),20, and recomputes batchnorm statistics after averaging. Reported results improve test accuracy from w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),21 to w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),22 for VGG-16 on CIFAR-10, from w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),23 to w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),24 for ResNet-18 on CIFAR-10, and from w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),25 to w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),26 for ResNet-18 on CIFAR-100, with very small additional computation time (Wu et al., 2020).
These experiments highlight a recurring trade-off. End-to-end unrolling can deliver fast inference and competitive quality, but the strongest formulations in this literature retain explicit data consistency, variational structure, or physically grounded initialization rather than replacing them with unconstrained feedforward mappings (Kofler et al., 2022, Mukherjee et al., 2021).
7. Interpretation, misconceptions, and scope
A common misconception is that end-to-end training with iterative regularization is equivalent to training an arbitrary CNN on paired input-output examples. The inverse-problem studies considered here do not support that equation. In one case, the architecture alternates a learned thresholding-based regularization step with a data-consistency solve that explicitly uses the forward model w∈HinfE(w),E(w)=∫H×R(⟨w,x⟩−y)2dρ(x,y),27; in the other, the unrolled operator is coupled to a learned 1-Lipschitz regularizer and may initialize a separate variational optimization. Both are model-based constructions rather than generic feedforward surrogates (Kofler et al., 2022, Mukherjee et al., 2021).
A second misconception is that regularization must always be an explicit penalty coefficient in the loss. The least-squares and iterate-averaging analyses show otherwise. With fixed step size, the number of passes over the data acts as a regularization parameter, and weighted averaging of an unregularized trajectory can reproduce or approximate a regularized solution path. In these settings, stopping time and averaging weights are not ancillary implementation details; they are part of the estimator definition (Rosasco et al., 2014, Wu et al., 2020).
A third misconception is that interpretability and performance are necessarily opposed. The MRI study reports performance comparable to DnCn3D while emphasizing a more transparent regularization structure and fewer trainable parameters, and the CT study combines end-to-end speed with the well-posedness and noise-stability guarantees of the variational setting. This does not imply that interpretability always improves performance, but it does show that the two objectives are not treated as mutually exclusive in the cited work (Kofler et al., 2022, Mukherjee et al., 2021).
Taken together, the literature suggests a unifying principle: iterative procedures can be made part of the learning problem itself. In some formulations, the regularizer is learned through the full unrolled solver; in others, the unrolled solver initializes a convergent variational refinement; in still others, the optimization trajectory is regularized by early stopping or transformed into a regularized solution by iterate averaging. A plausible implication is that “end-to-end” in this domain is best understood not as the elimination of iterative structure, but as its incorporation into training, model selection, and inference.
“Emergent Mind helps me see which AI papers have caught fire online.”
Philip
Creator, AI Explained on YouTube
Sign up for free to explore the frontiers of research
Discover trending papers, chat with arXiv, and track the latest research shaping the future of science and technology.Discover trending papers, chat with arXiv, and more.