Improved Iteration Complexities for Overconstrained $p$-Norm Regression

TL;DR

This paper presents improved iteration complexity algorithms for solving overconstrained $ ext{l}_p$ regression problems, achieving faster convergence rates and extending results to $ ext{l}_ty$ and dual norms, with two novel iterative frameworks.

Contribution

It introduces new algorithms with better iteration complexities for $ ext{l}_p$ regression, including $ ext{l}_ty$ and dual norm cases, using innovative row reweighting and acceleration techniques.

Findings

Achieved $ ilde{O}_p(d^{(p-2)/(3p-2)})$ iteration complexity for $ ext{l}_p$ regression.

Derived $ ilde{O}(d^{1/3}psilon^{-2/3})$ complexity for approximate $ ext{l}_ty$ regression.

Provided two frameworks: iterative refinement with width reduction and smooth acceleration, both improving prior methods.

Abstract

In this paper we obtain improved iteration complexities for solving $\ell_p$ regression. We provide methods which given any full-rank $\mathbf{A} \in \mathbb{R}^{n \times d}$ with $n \geq d$, $b \in \mathbb{R}^n$, and $p \geq 2$ solve $\min_{x \in \mathbb{R}^d} \left\|\mathbf{A} x - b\right\|_p$ to high precision in time dominated by that of solving $\widetilde{O}_p(d^{\frac{p-2}{3p-2}})$ linear systems in $\mathbf{A}^\top \mathbf{D} \mathbf{A}$ for positive diagonal matrices $\mathbf{D}$. This improves upon the previous best iteration complexity of $\widetilde{O}_p(n^{\frac{p-2}{3p-2}})$ (Adil, Kyng, Peng, Sachdeva 2019). As a corollary, we obtain an $\widetilde{O}(d^{1/3}\epsilon^{-2/3})$ iteration complexity for approximate $\ell_\infty$ regression. Further, for $q \in (1, 2]$ and dual norm $q = p/(p-1)$ we provide an algorithm that solves $\ell_q$ regression in $\widetilde{O}(d^{\frac{p-2}{2p-2}})$ iterations. To obtain this result we analyze row reweightings (closely inspired by $\ell_p$-norm Lewis weights) which allow a closer connection between $\ell_2$ and $\ell_p$ regression. We provide adaptations of two different iterative optimization frameworks which leverage this connection and yield our results. The first framework is based on iterative refinement and multiplicative weights based width reduction and the second framework is based on highly smooth acceleration. Both approaches yield $\widetilde{O}_p(d^{\frac{p-2}{3p-2}})$ iteration methods but the second has a polynomial dependence on $p$ (as opposed to the exponential dependence of the first algorithm) and provides a new alternative to the previous state-of-the-art methods for $\ell_p$ regression for large $p$.