Papers
Topics
Authors
Recent
Search
2000 character limit reached

End-to-End Iterative Regularization

Updated 5 July 2026
  • 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

minx 12xy22+αk=1Khkx1.\underset{x}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \alpha \sum_{k=1}^K \|h_k * 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,

infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),

with empirical counterpart

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^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.

Study Iterative object Regularization mechanism
(Kofler et al., 2022) Unrolled iterative neural network Learned convolutional sparsifying filters with data consistency
(Mukherjee et al., 2021) Unrolled reconstruction operator plus refinement Jointly learned variational regularizer and warm-started iterative solve
(Rosasco et al., 2014) Incremental gradient over epochs Number of passes over the data acts as a regularization parameter
(Wu et al., 2020) Averaged optimization trajectory Weighted iterate averaging yields adjustable regularization

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 sks_k and rewrites the nonsmooth problem as

minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,

with λ>0\lambda>0. The unrolled network alternates between a regularization step and a data-consistency step:

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,

initialized by

x0:=Ay.x_0 := A^\sharp y.

The second step is equivalent to solving

(AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.

For non-Cartesian MRI with density compensation, infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),0 and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),1 are replaced by preconditioned versions using infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),4, and the network can also learn infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),5 and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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

infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),7

with infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),9 with a parametrized regularizer E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.0, and the two are trained jointly. The training objective is a min-max problem,

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.1

which is equivalent, via Kantorovich-Rubinstein duality, to

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^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

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.3

with

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.4

and

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.5

with

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.6

The paper also uses a gradient penalty,

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.7

to enforce the 1-Lipschitz condition (Mukherjee et al., 2021).

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)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.8

E^(w)=1ni=1n(w,xiyi)2.\widehat{\mathcal E}(w)=\frac1n\sum_{i=1}^n(\langle w,x_i\rangle-y_i)^2.9

initialized by

sks_k0

The learnable components include primal CNN blocks, dual CNN blocks, and step sizes sks_k1, sks_k2, initialized at sks_k3, with sks_k4 unrolled layers (Mukherjee et al., 2021).

After training, the learned reconstruction sks_k5 is used to initialize a separate variational refinement,

sks_k6

solved by gradient descent from

sks_k7

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 sks_k8 up to subsequences, and a principled tradeoff controlled by sks_k9: small minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,0 prioritizes measurement consistency, while large minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_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 minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_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:

minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,3

followed by

minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,4

Here minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,5 indexes epochs, not individual sample updates (Rosasco et al., 2014).

The paper’s central principle is that the step size minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,6 is fixed a priori and the number of epochs minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,7 is the only tuning parameter. Small minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_1,8 means that the algorithm has not fit the sample too closely; large minx,{sk}k 12xy22+λ2k=1Khkxsk22+αk=1Ksk1,\underset{x,\{s_k\}_k}{\min}\ \frac{1}{2}\|x-y\|_2^2 + \frac{\lambda}{2}\sum_{k=1}^K \|h_k * x - s_k\|_2^2 + \alpha \sum_{k=1}^K \|s_k\|_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 λ>0\lambda>00 with a population iterate λ>0\lambda>01, decomposing the error into sample error and approximation error through inequalities such as

λ>0\lambda>02

When the best linear solution exists,

λ>0\lambda>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 λ>0\lambda>04 satisfies

λ>0\lambda>05

then

λ>0\lambda>06

If λ>0\lambda>07, then also

λ>0\lambda>08

Finite-sample bounds make the same point quantitatively. For any λ>0\lambda>09,

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),0

In the attainable case zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),1, the excess risk obeys

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),2

with recommended stopping rule

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),3

yielding

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),4

In the non-attainable case zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),5,

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),6

and the optimal epoch choice is

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),7

leading to

zj=k=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),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=1KhkSα/λ(hkxj),z_j = \sum_{k=1}^K h_k^\ast * \mathcal{S}_{\alpha/\lambda}(h_k * x_j),9

and its explicitly regularized counterpart

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,0

with canonical choice

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,1

Given a probability weight sequence xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,2, cumulative weights xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,3, and averaged iterate

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,4

the paper’s central claim is that with the right weighting scheme, xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,5 behaves like the solution path of a regularized problem (Wu et al., 2020).

For SGD on an xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,6-strongly convex and xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,7-smooth objective, the regularized and unregularized learning rates are linked by

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,8

and

xj+1=argminx12Axy22+λ2xzj22,x_{j+1} = \arg\min_x \frac{1}{2}\|Ax-y\|_2^2 + \frac{\lambda}{2}\|x-z_j\|_2^2,9

The main theorem gives the exact relation in expectation

x0:=Ay.x_0 := A^\sharp y.0

together with

x0:=Ay.x_0 := A^\sharp y.1

In this sense, iterate averaging implements x0:=Ay.x_0 := A^\sharp 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:=Ay.x_0 := A^\sharp y.3 (Wu et al., 2020).

The result extends beyond vanilla SGD. For preconditioned SGD with positive definite preconditioner x0:=Ay.x_0 := A^\sharp y.4, the induced regularizer becomes

x0:=Ay.x_0 := A^\sharp y.5

and the same structural relation holds. For Nesterov’s accelerated SGD, the paper defines a modified weighting scheme

x0:=Ay.x_0 := A^\sharp 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:=Ay.x_0 := A^\sharp 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:=Ay.x_0 := A^\sharp y.8

When the objective is quadratic, x0:=Ay.x_0 := A^\sharp 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 (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.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 (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.1, split by subject into 12/3/4 for train/validation/test, retrospective undersampling at approximately (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.2, and added Gaussian noise with (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.3. Models were trained with (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.4, filter sizes (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.5, network depth (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.6, and (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.7 conjugate-gradient iterations, using ADAM with learning rate (AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.8. The proposed network was trained for 75 epochs ((AA+λI)x=Ay+λzj.\big(A^\ast A + \lambda I\big)x = A^\ast y + \lambda z_j.9 hours), while the DnCn3D baseline was trained for 500 epochs (infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),01. Reported quantitative results include about infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),02 dB PSNR and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),03 SSIM for AR, about infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),04 dB PSNR and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),05 SSIM for ACR, about infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),06 dB PSNR and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),07 SSIM for UAR with infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),08, and about infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),09 dB PSNR and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),10 SSIM for UAR with refinement. Reconstruction times are around infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),11 seconds per image for AR, around infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),12 seconds for ACR, around infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),13 seconds for UAR end-to-end, and around infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),14 seconds for UAR with refinement. The paper also reports an interpretable dependence on infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),15: infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),16 and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),17 are too weak, infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),18 gives the best tradeoff, and infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),20, and recomputes batchnorm statistics after averaging. Reported results improve test accuracy from infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),21 to infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),22 for VGG-16 on CIFAR-10, from infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),23 to infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),24 for ResNet-18 on CIFAR-10, and from infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(x,y),25 to infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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 infwHE(w),E(w)=H×R(w,xy)2dρ(x,y),\inf_{w\in \mathcal H} \mathcal E(w), \qquad \mathcal E(w)=\int_{\mathcal H\times\mathbb R}(\langle w,x\rangle-y)^2\,d\rho(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.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to End-to-End Training with Iterative Regularization.