Source-linked AI summary

Risk and parameter convergence of logistic regression

Ziwei Ji, Matus Telgarsky

arXiv:1803.07300v3cs.LGmath.OCstat.ML

TL;DR

Gradient descent’s path for logistic regression is difficult to characterize when data are neither strongly convex nor linearly separable. The paper characterizes that path as a data-defined ray, with a maximum-margin direction and a bounded optimum for the remaining data.

  • Problem

    Prior analyses treat strongly convex risk and linearly separable data as distinct cases, while general data may fit neither setting.

    Source: §1 · p. 1
  • Method

    The paper characterizes the data-defined subspaces and studies gradient descent’s direction and projected iterates to identify its limiting ray.

    Source: §1 · p. 1
  • Results

    Gradient descent follows a unique ray whose direction is the maximum-margin predictor for a separable subset, while its projection onto S converges to the remaining data’s unique optimum.

    Source: §1 · p. 1
  • Takeaways & Limitations

    The characterization captures both implicit bias toward the ray and implicit regularization through tracking constrained optima.

    Source: §1 · p. 2

    The separable-case proof scheme does not directly extend to the general case because γ_i tends to zero.

    Source: §4.3 · p. 9

Abstract

from arXiv · show

Gradient descent, when applied to the task of logistic regression, outputs iterates which are biased to follow a unique ray defined by the data. The direction of this ray is the maximum margin predictor of a maximal linearly separable subset of the data; the gradient descent iterates converge to this ray in direction at the rate $\mathcal{O}(\ln\ln t / \ln t)$. The ray does not pass through the origin in general, and its offset is the bounded global optimum of the risk over the remaining data; gradient descent recovers this offset at a rate $\mathcal{O}((\ln t)^2 / \sqrt{t})$.

1 Introduction

The paper studies gradient descent for logistic regression and characterizes its path on data that need not be strongly convex or linearly separable. The iterates follow a data-defined ray combining a maximum-margin direction with an optimum for the remaining data.

1 Introduction

  • For logistic-risk gradient descent, the paper studies iterates generated by repeated gradient steps from an initial vector using step-size parameters.The theorem setup specifies initialization at zero, examples with norm at most one, and logistic or exponential loss.Source: §1 · p. 1
  • The paper addresses data that are neither strongly convex-risk cases nor linearly separable, where standard analyses treat distinct regimes.For separable data, the risk infimum may be zero without being attained; strongly convex risk instead has a unique bounded optimum.Source: §1 · p. 1
  • Gradient descent follows a unique ray: its direction converges to the maximum-margin predictor for a separable subset, while its projection onto S converges to the remaining data’s unique optimum.The remaining-data risk is strongly convex over bounded subsets of S.Source: §1 · p. 1
  • The theorem captures both implicit bias toward the ray and implicit regularization by showing that iterates track a sequence of constrained optima.The risk itself may be minimized by any vector in a problem-defined convex cone.Source: §1 · p. 2

1.1 Notation

  • The notation represents the sample with a matrix A and decomposes the data into S, the strongly convex part, and S⊥, the separable part.The text defines orthogonal projections onto these subspaces.Source: §1.1 · pp. 2–3

2 Problem structure

The section characterizes the gradient-descent path by separating a maximal linearly separable subset from a remaining strongly convex component, then combines these structures in an optimal ray.

2 Problem structure

  • In the separable example, gradient descent follows the unique maximum-margin separator, although scaling a separating vector can drive risk to zero.Source: §2 · pp. 3–4
  • When the class circles overlap, the risk is coercive and strongly convex over bounded subsets, with a unique bounded optimum.Source: §2 · p. 3
  • When the circles touch at the origin, the other examples remain separable, but the unavoidable origin-point loss gives the offset v̄=0.Source: §2 · p. 3
  • The greedy construction builds a maximal linearly separable subset Z, while the remaining examples form the strongly convex component.Source: §2 · p. 4
  • For mixed data, the remaining examples can shift the ray’s offset without affecting its maximum-margin direction.Source: §2 · p. 4
  • Theorem 2.1 uniquely partitions the data into a strongly convex component with a unique minimizer and a separable component whose projection has a maximum-margin direction.Source: §2 · p. 4

3 Risk convergence

The risk-convergence proof combines a smoothness-based descent bound with a comparison point supplied by the problem structure. It refines smoothness estimates to support the parameter-convergence analysis.

3 Risk convergence

  • The proof combines a generalized smoothness-based descent bound with reference point v̄+ū, enabled by Theorem 2.1.Source: §3 · pp. 4–5
  • A step-dependent R(wj)-smoothness bound refines the logistic-loss analysis and is essential for proving parameter convergence.Source: §3 · p. 5
  • The condition ℓ′′≤ℓ lets the Taylor remainder be bounded using R(w), and applying ln then gives ln R constant smoothness.Source: §3 · p. 6
  • Since R(w0) ≤ 1, choosing ηj ≤ 1 ensures η̂j ≤ 1, and the passage states that Lemma 3.4 then holds under this condition.Source: §3 · p. 6

4 Parameter convergence

The analysis establishes convergence toward the optimal ray in direction and convergence to its bounded component over S. It treats bounds and parameter convergence separately for the separable and general cases.

4 Parameter convergence

  • The iterates converge in direction to the separable component ū and over S to the bounded component v̄.The section states that the corresponding parameter convergence rates are given in the subsequent case analyses.Source: §4 · p. 6

4.2 Parameter convergence over S⊥: separable case

  • In the separable case, the exponential-loss analysis uses unit step sizes; the logistic-loss argument uses a warm start so small risk makes the two losses comparable.The comparison is used once the risk is at most ϵ/n, so every point’s loss is at most ϵ.Source: §4 · p. 6, §4.2 · p. 9

4 Parameter convergence

  • Strong convexity of the risk over S turns risk convergence into convergence of the iterates’ S-components.The section invokes a positive strong-convexity modulus over the 1-sublevel set.Source: §4 · p. 6

4.1 Bounding |Π⊥wt|

  • Bounds on iterate norms support the later parameter-rate arguments, which use both lower and upper bounds and adapt a perceptron-proof quantity.The proof controls a weighted sum of loss derivatives using the inner product with ū and the choice r = ln(t)/γ.Source: §4.1 · p. 7

4.3 Parameter convergence over S⊥: general case

  • For general data, the analysis replaces risk by excess risk R(w) − R̄ to recover a separable-style argument, while accounting for cross-terms.A case analysis uses strong convexity when the error over S is larger, and a step in the ū direction when the error over S⊥ is larger.Source: §4.3 · pp. 9–10

5 Related work

The paper relates its structural and convergence analysis to AdaBoost, convex optimization, perceptron bounds, and prior logistic-regression work. It distinguishes its general-instance treatment from analyses requiring separability or finite optima.

  • The structural result builds on AdaBoost analyses [Freund and Schapire, 1997; Mukherjee et al., 2011; Telgarsky, 2012, 2013], while adding the spaces S and S⊥ and a unique ū.The paper also supplies an elementary proof of the structural result.Source: §5 · p. 10
  • The analysis also draws on smoothness-based gradient-descent proofs [Bubeck, 2015] and perceptron convergence ideas to bound iterate norms.Source: §5 · p. 10
  • Prior separable-data analyses [Soudry et al., 2017; Gunasekar et al., 2018; Nacson et al., 2018] use a different tracking approach and achieve a better rate under separability.The paper notes that whether that rate improvement is possible for nonseparable data is unclear.Source: §5 · pp. 10–11
  • Risk-convergence methods assuming bounded domains or existing optima [Bach and Moulines, 2013; Bach, 2014] acquire polynomial-in-t factors in the unbounded setting described here.The paper attributes these factors to exponential-concavity dependence on the domain bound or Hessian eigenvalue dependence when no finite optimum exists.Source: §5 · p. 11
  • Unlike adversarial online-learning settings, the fixed training set allows the structural result to provide a good, small comparator.The cited comparator consequence is tied to Theorem 2.1.Source: §5 · p. 11

A Omitted proofs from Section 2

The omitted proofs establish the data-defined decomposition and the existence and uniqueness properties needed for the optimal ray. They also show that the risk restricted to S has a unique optimum.

B Omitted proofs from Section 3

The omitted proofs establish the risk-decrease and smoothness tools used in the convergence analysis. They derive a cumulative bound and apply it in the proof of Theorem 3.1.

C Omitted proofs from Section 4

The appendix section is divided into subsections paralleling Section 4.

C.1 Bounding |Π⊥wt|

The proof develops a smoothness-based convergence guarantee sensitive to Ac, splitting the bound in Lemma 4.3 into upper and lower bounds.

C.1 Bounding |Π⊥wt|

  • The section begins a proof of a smoothness-based convergence guarantee that accounts for Ac.Source: §C.1 · p. 16
  • Lemma 4.3 is proved by establishing separate upper and lower bounds.Source: §C.1 · p. 16
  • The upper-bound strategy rewrites quantities using ℓ′ so terms cancel after applying Lemma 4.4.Source: §C.1 · p. 17
  • The lower-bound argument invokes Theorem 2.1 and uses smoothness from Lemma 3.4.Source: §C.1 · pp. 16, 18

C.2 Parameter convergence when separable

The proof extends parameter convergence to the general separable case beyond the exponential loss and uses a technical inequality for the logistic and exponential losses.

C.2 Parameter convergence when separable

  • Lemma 4.7 establishes a technical inequality for ℓlog using r := ℓlog(z), while its claims for ℓexp are immediate.Source: §C.2 · p. 19
  • The proof handles Theorem 4.1 for the general separable case, not only for ℓexp.Source: §C.2 · p. 19
  • After a warm-start time t0 with risk at most ϵ/n, the loss remains at most ϵ/n at every later iterate.Source: §C.2 · p. 19
  • The proof combines upper and lower bounds on |wt|, with the constants hidden in the O notation independent of the problem.Source: §C.2 · p. 20

C.3 Parameter convergence in general

The general parameter-convergence proof applies a Fenchel–Young scheme to wt and its comparison iterate, with a controlled warm start.

C.3 Parameter convergence in general

  • Lemma 4.11 combines Lemmas 4.9 and 4.10 after the loss has decreased to at most ϵ/n.Source: §C.3 · p. 22
  • The general proof applies Lemma 4.11’s Fenchel–Young scheme to wt or its comparison iterate, whose risk is no greater than R(wt).Source: §C.3 · p. 22
  • A warm-start time t0 is chosen to meet conditions determined by an arbitrary ϵ in (0, 1).Source: §C.3 · p. 22
Loading 1803.07300v3…