cNKZ Projection: Nonlinear & Constrained Kaczmarz
- Constrained Nonlinear Kaczmarz (cNKZ) projection is an iterative method for solving high-dimensional nonlinear equations with convex constraints.
- It generalizes the classical Kaczmarz method by integrating row-action updates with metric and Bregman projections and diverse sampling rules.
- Empirical studies demonstrate that cNKZ variants achieve faster convergence and computational efficiency in inverse problems, robotics, and large-scale systems.
The constrained nonlinear Kaczmarz (cNKZ) projection encompasses a class of iterative projection methods for solving high-dimensional systems of nonlinear equations with side constraints , where is a (possibly composite) closed convex set. These methods generalize the classical Kaczmarz projection from linear systems to nonlinear and constrained contexts, employing row-action updates derived from residuals and one-row Jacobians, and enforcing constraint feasibility via (metric or Bregman) projections. Modern cNKZ variants admit stochastic sampling, greedy selection, Bregman geometry, and manifold-based formulations, and are supported by sharp convergence theory under local tangential cone, convexity, or full-rank Jacobian assumptions (Zhang et al., 2022, Gower et al., 2023, Zeng et al., 2020, Agrawal et al., 2024, Leitao et al., 2020, Xiao et al., 2024).
1. Problem Setting and General Framework
The prototypical cNKZ problem is to compute such that for a differentiable and a nonempty closed convex . can be a Cartesian product of sets, an -ball, the probability simplex, or more complex polyhedral/semialgebraic constraints. The solution set is assumed nonempty.
Standard cNKZ methods operate via sequential projections onto the tangent hyperplanes of (possibly linearized at the current iterate), interleaved with (metric or Bregman) projections onto or its components. Variants include cyclic, random, greedy, and sampling-based selection of the equation index; projection types depend on the geometry of and the chosen divergence.
Main update equation (Euclidean metric):
At iteration ,
where is chosen by a sampling, residual maximization, or cyclic rule. For general Bregman divergences , the step is
which is equivalently solved by a dual update involving the convex conjugate (Gower et al., 2023, Xiao et al., 2024).
2. Algorithmic Variants and Bregman Extensions
Metric vs. Bregman Projections
- Metric (Euclidean) projection: Suitable when is simple (e.g., box, ball, simplex). In this case, projections onto are and projection onto linearized equation hyperplanes admits closed form; standard Kaczmarz steps result.
- Bregman projection: Used for structured (e.g., ball, sparsity, simplex, entropic constraints) or non-Euclidean geometry. Each update projects onto a tangent affine hyperplane wrt. the Bregman distance induced by a strictly convex distance-generating function . The resulting problem is a one-dimensional convex program in the dual space and is solved by a globalized Newton step per iteration (Gower et al., 2023, Xiao et al., 2024).
Algorithmic sampling/selection rules
- Cyclic: Step through in order.
- Randomized: Sample uniformly or with nonuniform weights.
- Greedy-residual: Choose to maximize expected decrease per step (Zeng et al., 2020, Xiao et al., 2024).
- Greedy-distance: Maximize the step length to the projected constraint.
Accelerated and Motzkin Variants
- Projected Sampling Kaczmarz–Motzkin (PSKM): At each iteration, sample a batch of row indices, select the one with maximal residual, perform one-row Kaczmarz update, and project onto a randomly chosen component constraint. The accelerated variant (APSKM) uses sequential projections plus extrapolation to further reduce distance to the feasible set (Zhang et al., 2022).
3. Analytical Assumptions and Convergence Theory
Key assumptions enabling convergence include:
- Local tangential cone condition (TCC): Controls the local nonlinearity of each (Zhang et al., 2022, Leitao et al., 2020, Gower et al., 2023, Xiao et al., 2024).
- Row boundedness below: Bounds (Zhang et al., 2022).
- Full-rank Jacobian / convexity: Ensures global linear convergence; else, monotonic decrease (Zhang et al., 2022, Gower et al., 2023, Zeng et al., 2020).
Central convergence statements:
- Linear convergence in expectation: Under TCC with constant and full-rank Jacobian, for both metric and Bregman cNKZ,
for explicit rate dependent on , , , and norm structure (Zhang et al., 2022, Gower et al., 2023, Xiao et al., 2024).
- Monotonic decrease: If each is convex nonnegative, the error norm strictly decreases per update until convergence (Zhang et al., 2022).
- Bregman descent property: Each Bregman projection step strictly reduces up to a term proportional to squared residual (Gower et al., 2023, Xiao et al., 2024).
- Regularization and noise tolerance: Landweber-Kaczmarz/cNKZ admits robust regularization in inverse problems with noisy data and converges weakly/strongly to the solution set, halting when residual falls below noise floor (Leitao et al., 2020).
4. Computational Structure and Practical Complexity
Each cNKZ iteration for a single constraint has per-step cost determined by:
- Computing and : or per-component cost .
- (Bregman) projection step: plus a one-dimensional Newton solve for Bregman variants.
- Projection onto (if nontrivial): or a prox operator.
- APSKM/accelerated steps increase per-iteration cost marginally due to additional projections or extrapolation steps.
The overall complexity per iteration is for single-row methods, and for batch sampling of size (Zhang et al., 2022, Gower et al., 2023). Empirical studies show that batch sampling () balances convergence speed and work per update.
5. Applications and Comparative Empirical Performance
Recent cNKZ implementations have proven especially effective for:
- Large-scale nonlinear equation systems (e.g. , ), including equations with strong nonlinearity or sparsity constraints (Zhang et al., 2022, Gower et al., 2023, Xiao et al., 2024).
- Inverse problems under noisy data: Demonstrated robust performance and fewer forward/adjoint solves compared to Landweber–Kaczmarz with line search (Leitao et al., 2020).
- Constrained motion planning and coordinated multi-robot manipulation: Enabling rapid, robust projection onto the intersection of up to 80 nonlinear manifolds, with order-of-magnitude speed-ups in feasibility projection and success rates exceeding 90% in complex settings (Agrawal et al., 2024).
- Graph realization and quadratic systems: Locally linear convergence established for systems of quadratic (or sphere) constraints (Zeng et al., 2020).
Empirically, cNKZ and its variants (especially those using greedy-residual sampling or Bregman geometry) outperform classical randomized Kaczmarz, SGD, and projected gradient methods by factors of 2–10 in iterations and 10–50 in wall-clock time for large systems under convex or nearly convex constraints (Zhang et al., 2022, Xiao et al., 2024, Gower et al., 2023).
6. Parameter and Implementation Guidelines
Tuning critical parameters:
- Step size : Basic Kaczmarz step uses ; can enforce descent; for Bregman steps, the scalar is adaptively solved per-iteration (Zhang et al., 2022, Gower et al., 2023, Xiao et al., 2024).
- Batch/sample size : Empirically, intermediate batch sizes () yield favorable trade-offs (Zhang et al., 2022).
- Bregman generator : Must be strictly convex with as its domain; entropy, , and block-simplex penalties are common (Gower et al., 2023, Xiao et al., 2024).
- Projection thresholds/tolerances: Set according to application specifics, e.g., for equality, or application-specific physical tolerances (Agrawal et al., 2024).
- Randomization vs. greedy selection: Greedy-residual or distance selection provides significant iteration savings in practice; uniform random selection is simple and effective for homogeneous systems (Zeng et al., 2020, Xiao et al., 2024).
7. Extensions, Limitations, and Current Research Directions
cNKZ methods can be further generalized to:
- Handle general nonlinear manifold intersection via manifold-based projection formulation, as in complex robotic systems (Agrawal et al., 2024).
- Incorporate acceleration (e.g., Motzkin or APSKM extrapolation steps) to improve convergence in ill-conditioned or nearly inconsistent systems (Zhang et al., 2022).
- Integrate relaxed/adaptive projection step sizes (mirror-SPS, adaptive Bregman steps) for further empirical gains (Gower et al., 2023, Xiao et al., 2024).
- Support additional side constraints via Bregman-generated domains (sparsity, simplex, entropy).
- Employ scalable parallel implementations and composite constraint handling for high-dimensional applications.
Observed limitations include sensitivity of convergence to initializations in nonconvex settings, slowdown if projections onto are expensive, and the need for empirical tuning of per-constraint tolerances and step sizes in complex applications (e.g., multi-robot systems). Extensions to force/torque constraints, adaptive projection rules, and coupling with trajectory optimization are active research areas (Agrawal et al., 2024).
Key references:
- (Zhang et al., 2022): Introduction of sampling Kaczmarz–Motzkin methods and their constrained variants (PSKM/APSKM)
- (Gower et al., 2023): Bregman–Kaczmarz cNKZ with adaptive steps and strong convergence guarantees
- (Zeng et al., 2020): Successive projection and greedy/greedy-residual cNKZ for nonlinear/inequality systems
- (Agrawal et al., 2024): Manifold-based cNKZ for multi-robot manipulation
- (Leitao et al., 2020): Projective Landweber–Kaczmarz with regularization for ill-posed constrained systems
- (Xiao et al., 2024): Greedy randomized Bregman–Kaczmarz with sharp convergence for large-scale, structured, constraint-rich nonlinear systems