Skip to content
Preprint

A Prediction--Correction Analysis of Two-Way Block Splitting in Distributed Learning

Aug 2026 · 0 citations · 16 references
Mathematics

TL;DR

This note revisits the convergence of the Parikh--Boyd two-way block-splitting algorithm for large-scale distributed learning through the He--Yuan prediction--correction framework and clarifies which algebraic initialization conditions allow the reduced implementation to inherit the full-space convergence and complexity guarantees without modifying its local updates.

Abstract

This note revisits the convergence of the Parikh--Boyd two-way block-splitting algorithm for large-scale distributed learning through the He--Yuan prediction--correction framework. Simultaneous row--column partitioning is also relevant to hybrid federated learning, where data may be heterogeneous in both samples and features. We lift the reduced iteration to an equal-dimensional product space and reconstruct the primal and dual coordinates omitted by its implementation. The induced orthogonal-complement structure establishes exact iteration-by-iteration equivalence with the published updates. A mixed variational-inequality representation then yields a fundamental descent inequality, global convergence, an ergodic complexity bound, and current-iterate residual estimates under standard convexity, solvability, exact-subproblem, and invariant-initialization assumptions. The analysis also shows that the reduced state recursion is a metric proximal point iteration. No strong convexity, differentiability, or full-rank condition is imposed. The derivation clarifies which algebraic initialization conditions allow the reduced implementation to inherit the full-space convergence and complexity guarantees without modifying its local updates.

View source

Similar papers

#machine learning Preprint Aug 2026

Sample-Weighted End-to-End Trace-Norm Geometry for Multitask Learning

The sample-size-weighted trace norm of the end-to-end map from task coefficients to input-space predictors is studied instead the sample-size-weighted trace norm of the end-to-end map from task coefficients to input-space predictors.

Mahdi Mohammadigohari · 0 citations
Preprint Aug 2026

Generalization Error Estimation for Primal--Dual Algorithms in Non-Smooth Regression

A general recursive framework that includes the Chambolle--Pock algorithm and related primal--dual splitting methods is developed, which proves finite-sample guarantees for both estimators and establishes a matched-Gaussian universality result beyond Gaussian designs.

Kai Tan, Pierre C. Bellec · 0 citations
#machine learning Preprint Sep 2026

Exact information accounting for SGD methods

As an alternative to the standard geometric analyses, we give an exact, information-theoretic analysis of stochastic gradient descent (SGD) and its variants. We show that a preconditioned SGD step is the posterior-mean update of a Gaussian Bayes model, and that its one-step regret splits into an intrinsic-time cost and...

Akshay Balsubramani · 0 citations
Preprint Sep 2026

Near-Optimal Single-Loop Predictor--Corrector Extragradient Method for Strongly Convex--Strongly Concave Minimax Optimization

The Single-Loop Predictor--Corrector Extragradient Method with Damped Momentum (PCE-DM), which combines an extragradient prediction--correction scheme with a novel auxiliary feedback recursion for the weaker-curvature variable, closes the condition-number complexity gap between standard single-loop methods and near-opt...

Min-hao Zhang, Zi Xu · 0 citations
#machine learning Preprint Sep 2026

High-Probability Convergence of SGD via Batched Updates

This work proposes Batched SGD, a simple variant that partitions online samples into epochs and performs a single update per epoch using a refined, low-variance gradient estimate, and provides the first high-probability guarantees for FL.

Feng Zhu, Robert W. Heath, Aritra Mitra · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.