arXiv reader

On the Complexity of Finding Stationary Points in Stochastic (L_0, L_1)-Smooth Nonconvex Optimization

Yanpeng Liang and Luo Luo

arXiv
2610.21692v1 · PDF · 15 October 2026
Categories
math.OC
Licence
CC BY 4.0 The site shows the text.

Abstract

We study the stochastic first-order oracle (SFO) complexity of finding \epsilon-stationary points of (L_0,L_1)-smooth nonconvex functions. Under bounded-variance noise, existing analyses of normalized and clipped stochastic gradient descent (SGD) establish optimal dependence on \epsilon and L_0, but the jointly optimal dependence on L_0, L_1, and \epsilon remains unclear. We resolve this question for an unbiased SFO whose gradient noise has a pth central moment bounded by \sigma^p for some \sigma\geq 0, where p\in(1,2].Specifically, we establish an SFO lower bound of \Omega\big(\big(\sigma^{\frac{p}{p-1}}\epsilon^{-\frac{p}{p-1}}+1\big) (L_0 \Delta\epsilon^{-2}+L_1\Delta\epsilon^{-1})\big), where \Delta>0 is the initial optimality gap.This bound identifies an unavoidable contribution proportional to L_1\epsilon^{-1}\Delta. We also prove that mini-batch (smoothed) normalized SGD can attain a matching upper bound, thereby establishing the optimal SFO complexity.

Items 38

Lemma

Lemma 3.1

V0MD

The functions \psi and \phi defined in equation equation (?) satisfy the following properties.

  1. The function \psi satisfies \psi(t)=0 for all t \le 1/2, and \psi(t)=1 for all t\ge1.

  2. The function \psi satisfies 0\le\psi(t)\le1, 0\le\psi'(t)\le4, and |\psi''(t)|\le24 for all t\in{\mathbb{R}}.

  3. The function \phi satisfies 0<\phi(t)<2, \phi'(t)=\operatorname{sech}^2(t), and |\phi''(t)|\le2\phi'(t) for all t\in{\mathbb{R}}.

  4. The derivative \phi' is even and holds \phi'(t)>2/5 for all t\in[-1, 1].

Remark

Remark 1

V9JJ

We illustrate the derivatives of \psi and \phi in Figure ?. Note that the derivative \psi'(t) is non-zero only for t\in(1/2, 1), which is useful to construct the zero-chain. Additionally, the manganite of \phi''(t) always can be bounded by 2\phi'(t), corresponding to the (L_0, L_1)-smoothness.

Lemma

Lemma 3.2

FYAN

The function \bar f_T defined in equation (?) satisfies the following properties.

  1. The optimality gap holds \bar f_T({\bf{0}}) - \inf_{{\bf{y}}\in{\mathbb{R}}^T}\bar f_T({\bf{y}}) =2T-1.

  2. For all {\bf{y}}\in{\mathbb{R}}^T, it holds \left\| \nabla\bar f_T({\bf{y}}) \right\| \le9\sqrt T and \left\| \nabla^2\bar f_T({\bf{y}}) \right\| \le 58.

  3. For given {\bf{y}}=[y_1,\dots,y_T]\in{\mathbb{R}}^T, if there exists i\in[T] such that |y_i|\le1, then \left\| \nabla\bar f_T({\bf{y}}) \right\| \ge\frac25.

  4. For all {\bf{y}}\in{\mathbb{R}}^T, it holds \left\| \nabla^2 \bar f_T({\bf{y}}) \right\| \le 160\left\| \nabla\bar f_T({\bf{y}}) \right\|.

  5. For all {\bf{y}}\in{\mathbb{R}}^T, it holds \operatorname{prog}_0(\nabla\bar f_T({\bf{y}}))\le\operatorname{prog}_{1/4}({\bf{y}})+1.

Used by Remark 2, Proof
Remark

Remark 2

GLEU

Lemma 3.2(d) implies that function \bar f_T is (0, 320)-smooth, which is essential different with the analysis for standard smooth case [carmon2020lower]. In fact, the ratio of the gradient norm to the Hessian norm in the classical hard instance for standard smooth optimization is unbounded, so that it cannot characterize the specific hardness of (L_0, L_1)-smoothness. Please refer to discussion in Appendix ?.

Lemma

Lemma 3.3

PX0T

The function H_{{\bf U},\mu} defined in equation (?) satisfies

H_{{\bf U},\mu}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}H_{{\bf U},\mu}({\bf{x}}) \le2T.

Moreover, if there exists some i\in[T] such that \left| [{\bf U}^\top\rho_R({\bf{x}})]_i\right|\le1, then it holds

\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| \ge\frac14.

Furthermore, all {\bf{x}}\in{\mathbb{R}}^d, it holds

\left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| \le 240\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right).
Used by Proof, Proof, Proof
Remark

Remark 3

9KH5

Following Remark 2, inequality (3.3) indicates function H_{{\bf U},\mu} defined in equation (?) is (\mu\log(1/\mu),300)-smooth, going beyond the hard instance for standard smooth optimization [carmon2020lower].

Lemma

Lemma 3.4

GVMR

For all L_0,L_1,\epsilon>0, we follow definitions in equations (?)–(?) by taking

\mu:=\frac{\delta_0}{4\log({\mathrm{e}}/\delta_0)}, \quad a:=\frac{7680\epsilon}{L_1} \quad b:=\frac{480}{L_1}, \quad \text{with}~~ \delta_0:= \min\left\{10^{-2}, \frac{L_0}{16L_1\epsilon} \right\}.

Then the scaled function f_{{\bf U}}({\bf{x}}):=aH_{{\bf U},\mu}\left({{\bf{x}}}/{b}\right) has the following properties.

  1. The function f_{{\bf U}} is (L_0,L_1)-smooth.

  2. The optimality gap holds f_{{\bf U}}({\bf{0}})-\inf_{{\bf{x}}\in{\mathbb{R}}^d}f_{{\bf U}}({\bf{x}}) \le2aT.

  3. For given {\bf{x}}=[x_1,\dots,x_T]\in{\mathbb{R}}^d, if there exists i\in[T] such that \left|\left[{\bf U}^\top\rho_R({\bf{x}}/b)\right]_i\right|\le1, then \left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| \ge4\epsilon.

Used by Proof, Proof, Proof
Definition

Definition 4.1

D5GG

A stochastic gradient {\bf{g}}:{\mathbb{R}}^T\times{\mathcal{Z}}\to{\mathbb{R}}^T is a probability-\theta zero-chain with \theta\in[0,1] if it holds

\mathbb P \Big(\forall\,{\bf{y}}\in{\mathbb{R}}^T: \operatorname{prog}_0 \big({\bf{g}}({\bf{y}};Z)\big) \le \operatorname{prog}_{1/4}({\bf{y}}), {\bf{g}}({\bf{y}};Z) ={\bf{g}}\big(\Pi_{\operatorname{prog}_{1/4}({\bf{y}})}({\bf{y}}); Z\big) \Big) \ge 1-\theta,

and

\!\!\!\mathbb P_Z \Big(\forall\,{\bf{y}}\in{\mathbb{R}}^T\!: \operatorname{prog}_0 \big({\bf{g}}({\bf{y}};Z)\big) \le 1+\operatorname{prog}_{1/4}({\bf{y}}), {\bf{g}}({\bf{y}};Z) = {\bf{g}}\big(\Pi_{1+\operatorname{prog}_{1/4}({\bf{y}})}({\bf{y}}); Z\big) \Big)=1.

Here, we denote \Pi_s({\bf{y}})=[y_1,\dots,y_s,0,\dots,0]^\top for given {\bf{y}}=[y_1,\dots,y_d]^\top\in{\mathbb{R}}^d, where we let \Pi_s({\bf{y}})={\bf{y}} when s>T for convention.

Used by Proof
Lemma

Lemma 4.2

M5HX

The functions \bar F_T and Q_T defined in equations (?) and (?) satisfy

  1. For all {\bf{y}}\in{\mathbb{R}}^T, it holds \left\| \nabla Q_T({\bf{y}}) \right\|\le249.

  2. The stochastic gradient evaluation {\bf{g}}_T({\bf{y}};Z):=\nabla\bar F_T({\bf{y}};Z) is a probability-q zero-chain and holds

    {\mathbb{E}}_Z[{\bf{g}}_T({\bf{y}};Z)]=\nabla\bar f_T({\bf{y}})

Used by Proof
Lemma

Lemma 4.3

PWCO

For given p\in(1,2] and \sigma\ge0, we let

q:=\begin{cases} 1, & \sigma = 0, \\[0.25cm] \displaystyle{\min\left\{ 1,\, \left(\frac{2\cdot(3984)^p\epsilon^p}{\sigma^p}\right)^{\frac{1}{p-1}} \right\}}, & \sigma > 0. \end{cases}

Then for all {\bf{x}}\in{\mathbb{R}}^d, it holds

\begin{aligned} {\mathbb{E}}_Z[\mathsf O_{f_{{\bf U}}}({\bf{x}};Z)] =\nabla f_{{\bf U}}({\bf{x}}), \qquad \text{and} \qquad {\mathbb{E}}_Z\left[\left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z) -\nabla f_{{\bf U}}({\bf{x}}) \right\|^p \right] \le\sigma^p. \end{aligned}
Used by Proof
Lemma

Lemma 4.4

TFEB

We let

\delta\in(0,1), \qquad {\bf U}\sim\operatorname{Unif}(\operatorname{St}(d,T)), \qquad\text{and}\qquad d\ge \left\lceil T+32R^2\log\left(\frac{2T^2}{q\delta}\right) \right\rceil.

Then for any randomized algorithm \mathsf A\in\mathcal A_{\mathrm{rand}} associated with the oracle \mathsf O_{f_{{\bf U}}}({\bf{x}};Z) defined in equations (?) with probability at least 1-\delta, it holds

\left\| \nabla f_{{\bf U}}({\bf{x}}^{(t)}) \right\| \ge4\epsilon

for all t\le(T-\log(2/\delta))/(2q), where {\bf{x}}^{(t)} is the tth query generated by randomized algorithm \mathsf A.

Used by Proof
Theorem

Theorem 4.5

0L33

For any given p\in(1,2], there exists numerical constant \hat c>0 such that for all L_0,L_1,\Delta,\sigma>0, and \epsilon\leq {\hat c} L_1\Delta, it holds

\begin{aligned} m_{\epsilon}^{\mathrm{rand}} (L_0,L_1,\Delta,\sigma,p) \ge \Omega\left( \bigg(\frac{\sigma^{\frac{p}{p-1}}}{\epsilon^{\frac{p}{p-1}}}+1\bigg)\frac{L_1\Delta}{\epsilon}\right). \end{aligned}
Used by Proof
Corollary

Corollary 4.6

NA7V

For any given p\in(1,2], there exists numerical constant c>0 such that for all L_0,L_1,\Delta,\sigma>0, and \epsilon\leq c \min\{\sqrt{L_0\Delta}, L_1\Delta\}, it holds

m_{\epsilon}^{\mathrm{rand}}(L_0,L_1,\Delta,\sigma,p) \ge \Omega\left(\bigg(\frac{\sigma^{\frac{p}{p-1}}}{\epsilon^{\frac{p}{p-1}}}+1\bigg)\bigg(\frac{L_0\Delta}{\epsilon^2}+\frac{L_1\Delta}{\epsilon}\bigg)\right).
Theorem

Theorem 5.1

DUMY

We run Algorithm ? to solve problem (?) with f\in{\mathcal{F}}(L_0, L_1, \Delta) by using stochastic first-order oracle {\bf{g}}:{\mathbb{R}}^d\times{\mathcal{Z}}\to{\mathbb{R}}^d such that

\begin{aligned} {\mathbb{E}}_Z[{\bf{g}}({\bf{x}};Z)]=\nabla f({\bf{x}}) \qquad\text{and} \qquad {\mathbb{E}}_Z\left[\left\| {\bf{g}}({\bf{x}};Z)-\nabla f({\bf{x}}) \right\|^p\right] \leq \sigma^p \end{aligned}

for some \sigma\geq 0 and p\in(1,2]. By taking parameters

\begin{aligned} \eta =\frac{\epsilon}{4(L_0+L_1\epsilon)},\quad \gamma\in\left[0,\frac{\epsilon}{4}\right],\quad b=\bigg\lceil \bigg(\frac{32\sqrt2\,\sigma}{\epsilon}\bigg)^{\frac{p}{p-1}}+1 \bigg\rceil, \quad \text{and}\quad T=\left\lceil\frac{4\Delta}{\eta\epsilon}\right\rceil. \end{aligned}

the output satisfies {\mathbb{E}}[\left\| \nabla f(\hat{\bf{x}}) \right\|]\leq\epsilon.

Proof

Proof

OP0A

Recall that the functions \psi and \phi are defined as

\psi(t):=S(2t-1) \qquad \text{and} \qquad \phi(t):=1+\tanh(t).

respectively, where

S(u)= \begin{cases} 0, & u\le 0,\\ 10u^3-15u^4+6u^5, & u\in(0,1),\\ 1, & u\ge 1. \end{cases}

We verify the first-order and second-order derivatives of S as

\begin{aligned} S'(u) = \begin{cases} 30u^2(u-1)^2, & \text{if}~u\in(0,1), \\ 0, & \text{otherwise}, \end{cases} \end{aligned}

and

\begin{aligned} S''(u) =\begin{cases} 60u(u-1)(2u-1), & \text{if}~u\in(0,1) \\ 0, & \text{otherwise}. \end{cases} \end{aligned}

We now prove each statement of the lemma.

The proof of statement (a). The definitions of \psi and S in equations (Proof) and (Proof) directly shows that \psi(t)=0 for all t \le 1/2, and \psi(t)=1 for all t\ge1.

The proof of statement (b). We first consider the value of \psi(t). By equation equation (Proof), we know that S'(t)\geq 0 on {\mathbb{R}}, which implies the function S is non-decreasing. Combining the facts S(0)=0 and S(1)=1, we obtain 0\le S(u)\le1 for all u\in[0,1]. Combing the definitions S(u)=0 for all u\leq 0 and S(u)=1 for all u\geq 1, we know that S(u)\in[0,1] on {\mathbb{R}}. Hence, the definition of \psi(t)=S(2t-1) implies

0\le\psi(t)\le1.

We then consider the value of \psi'(t). By the chain rule, we can write

\begin{aligned} \psi'(t) = \begin{cases} 60(2t-1)^2(2t-2)^2, & t\in[ 1/2,1], \\ 0 & \text{otherwise}. \end{cases} \end{aligned}

Clearly, it holds \psi'(t)\ge0. For t\in[1/2,1], we let

u:=2t-1\in[0,1],

which leads to

\begin{aligned} \psi'(t)= 60(2t-1)^2(2t-2)^2 = 60u^2(u-1)^2 . \end{aligned}

Since for all u\in[0,1], it holds

u(u-1) = \left(u-\frac12\right)^2 - \frac14 \in \left[-\frac14, 0\right],

which implies

\psi'(t) \le 60\cdot\frac1{16} = \frac{15}{4} <4.

Recall that \psi'(t)=0 for all t\in[0, 1/2], we obtain 0\le\psi'(t)\le4 for all t\in{\mathbb{R}}.

We now consider the value of \psi''(t). By the chain rule, we can write

\psi''(t)=4S''(2t-1).

Equation (Proof) indicates \psi''(t)=0 for all t\not\in(1/2,1). Hence, we focus on the case of t\in(1/2,1), that is u=2t-1\in(0,1). We write

\psi''(t)=240u(1-u)(1-2u).

and define

g(u):=u(1-u)(1-2u) = u-3u^2+2u^3.

Then we have

g'(u)=1-6u+6u^2,

which has critical points

u_{+} = \frac{3+\sqrt3}{6} \qquad\text{and}\qquad u_{-} = \frac{3-\sqrt3}{6}.

Since it holds g(0)=g(1)=0, the maximum of |g(u)| on [0,1] is attained at one of these critical points. We can verify

\begin{aligned} |g(u_+)| = |g(u_-)|=\frac{\sqrt3}{18}, \end{aligned}

which implies

|\psi''(t)| \le 240 \cdot \frac{\sqrt3}{18} <24.

The proof of statement (c). The fact \tanh t\in(-1,1) directly implies

\begin{aligned} \phi(t)=1+\tanh t\in(0,2). \end{aligned}

We can also verify \phi'(t)=\operatorname{sech}^2t and \phi''(t) = -2\operatorname{sech}^2(t)\tanh(t), which implies

\begin{aligned} |\phi''(t)| = 2\operatorname{sech}^2(t)|\tanh(t)| < 2\operatorname{sech}^2(t) = 2\phi'(t). \end{aligned}

The proof of statement (d). The statement that \phi' is even can be verifies as

\phi'(-t)=\operatorname{sech}^2(-t)=\operatorname{sech}^2(t)=\phi'(t).

It remains to show that \phi'(t)>2/5 for |t|\le1. For all t\geq 0, it holds

\phi''(t) = -2\operatorname{sech}^2(t)\,\tanh (t) \le0,

which implies \phi' is non-increasing on [0,\infty). Therefore, for all t\in[0,1], it holds

\phi'(t) =\operatorname{sech}^2(t) \ge \operatorname{sech}^2(1) > \frac{2}{5}.

Since \phi'(t) is even, we have |\phi'(t)|>2/5 when |t|\leq 1.

Used by Proof, Proof, Proof
Lemma

Lemma A.1

GMFB

For every t\in{\mathbb{R}}, the function \psi defined in equation (?) satisfies

\begin{aligned} \psi(t)+\psi(-t) &\le1, \\ \psi'(t)+\psi'(-t) &\le4, \\ |\psi''(t)|+|\psi''(-t)| &\le24. \end{aligned}
Used by Proof, Proof
Proof of Lemma A.1

The expression of \psi'(\cdot) in equation (?) implies

\psi'(t)=\psi''(t)=0 \qquad \text{whenever } t\leq 1/2 \text{~~or~~} t \geq 1.

We prove this lemma by considering the following three cases.

The Case of |t|\le1/2. The condition |t|\le1/2 implies both t and -t are at most 1/2. Hence, the definitions of \psi(\cdot) and S(\cdot) in equations (Proof) and (Proof) implies \psi(t)=\psi(-t)=0. This results

\psi(t)+\psi(-t)=0\le1.

Furthermore, equation (Proof) indicates \psi'(t)=\psi'(-t)=0 and \psi''(t)=\psi''(-t)=0, which imply

\psi'(t)+\psi'(-t)=0\le4 \qquad\text{and}\qquad |\psi''(t)|+|\psi''(-t)|=0\le24.

The Case of t>1/2. The condition t>1/2 implies -t<-1/2. Hence, the definition (Proof) and equation (Proof) implies

\begin{aligned} \psi(-t)=\psi'(-t)=\psi''(-t)=0. \end{aligned}

Combining with Lemma 3.1(b), we obtain

\psi(t)+\psi(-t)=\psi(t)\le1,~~ \psi'(t)+\psi'(-t)=\psi'(t)\le4,~~\text{and}~~ |\psi''(t)|+|\psi''(-t)|=|\psi''(t)|\le24.

The Case of t<-1/2. The condition t<-1/2 implies -t>1/2. Hence, the definition (Proof) and equation (Proof) implies

\begin{aligned} \psi(t)=\psi'(t)=\psi''(t)=0. \end{aligned}

Combining with Lemma 3.1(b), we obtain

\psi(t)+\psi(-t)=\psi(-t)\le1,~\psi'(t)+\psi'(-t)=\psi'(-t)\le4,~\text{and}~ |\psi''(t)|+|\psi''(-t)|=|\psi''(-t)|\le24.

Hence, equations (A.1), (A.1), and (A.1) hold for all t\in{\mathbb{R}}.

Proof

Proof

MIL6

Recall that the function \bar f_T is defined as

\bar f_T({\bf{y}}) := \sum_{i=1}^{T}A_i({\bf{y}}), ~~~~\text{where}~~ A_i({\bf{y}}):= \begin{cases} -\phi(y_1), & \text{if}~i=1, \\[0.1cm] \psi(-y_{i-1})\phi(-y_i) - \psi(y_{i-1})\phi(y_i), & \text{otherwise}. \end{cases}

The proof of statement (a). According to definition (Proof) and Lemma 3.1(a), we have \phi(0)=1 and \psi(0)=0. Hence, definition (Proof) implies

A_i({\bf{0}})=\begin{cases} -1, & \text{if}~i=1, \\ 0, & \text{otherwise}. \end{cases} \qquad\text{and}\qquad \bar f_T({\bf{0}})=-1.

For all {\bf{y}}\in{\mathbb{R}}^T, Lemma 3.1(c) implies

A_1({\bf{y}})=-\phi(y_1)>-2.

Moreover, for every i\ge2, Lemma 3.1(b) and (c) implies

\psi(-y_{i-1})\phi(-y_i)\ge0 \qquad\text{and}\qquad 0\le\psi(y_{i-1})\phi(y_i)<2.

It follows from equation (Proof) that

A_i({\bf{y}})>-2 \qquad\text{and}\qquad \bar f_T({\bf{y}}) = \sum_{i=1}^T A_i({\bf{y}})>-2T.

Thus, it holds

\inf_{{\bf{y}}\in{\mathbb{R}}^T}\bar f_T({\bf{y}})\ge-2T.

We then consider the vector

{\bf{y}}^{(r)}= r[1,\ldots,1]^\top, \qquad \text{where}~~r\ge1.

Lemma 3.1(a) implies \psi(r)=1 and \psi(-r)=0. Consequently, definition equation (Proof) implies

A_i({\bf{y}}^{(r)}) = -\phi(r).

for all i\in[T], which results

\lim_{r\to+\infty} \bar f_T({\bf{y}}^{(r)}) = \lim_{r\to+\infty} -T\phi(r) = -2T.

Combining equations (?) and (?) yields

\bar f_T({\bf{0}})-\inf_{{\bf{y}}\in{\mathbb{R}}^T}\bar f_T({\bf{y}})=2T-1.

The proof of statement (b). We first establish the uniform bound on the gradient of \bar f_T. Taking the derivative on equation (Proof) and using the evenness of \phi' from Lemma 3.1(d), we obtain

\begin{split} \frac{\partial\bar f_T({\bf{y}})}{\partial y_i} = \begin{cases} -\phi'(y_1) -\psi'(-y_1)\phi(-y_2) -\psi'(y_1)\phi(y_2), & i = 1, \\ -\big(\psi(-y_{i-1})+\psi(y_{i-1})\big)\phi'(y_i) -\psi'(-y_i)\phi(-y_{i+1})-\psi'(y_i)\phi(y_{i+1}), & i=2,\dots,T-1, \\ -\big(\psi(-y_{T-1})+\psi(y_{T-1})\big)\phi'(y_T), & i=T. \\ \end{cases} \end{split}

According to Lemma 3.1(b) and (c) and Lemma A.1, we have

0\le (\psi(-y_{i-1})+\psi(y_{i-1}))\phi'(y_i) \le1 ~~~~~~\text{and}~~~~~~ 0\le \psi'(-y_i)\phi(-y_{i+1}) +\psi'(y_i)\phi(y_{i+1}) \le8.

Combing with equation (?) implies

\left\| \nabla\bar f_T({\bf{y}}) \right\| = \sqrt{T} \max_{i\in[T]}\left|\frac{\partial\bar f_T}{\partial y_i}\right| \le9\sqrt T.

We now consider the bound of Hessian. Lemma 3.1(c) implies

\left|\frac{\partial^2 A_1({\bf{y}})}{\partial y_1^2}\right| =|\phi''(y_1)|\le2.

For all i\ge2, Lemma 3.1(c) and A.1 imply

\begin{aligned} & \left|\frac{\partial^2 A_i({\bf{y}})}{\partial y_{i-1}^2}\right| = |\psi''(-y_{i-1})\phi(-y_i) -\psi''(y_{i-1})\phi(y_i)| \le2(|\psi''(-y_{i-1})|+|\psi''(y_{i-1})|) \le48, \\[0.15cm] & \left|\frac{\partial^2 A_i({\bf{y}})}{\partial y_i^2}\right| = |-(\psi(-y_{i-1})+\psi(y_{i-1}))\phi''(y_i)| \le2, \\[0.15cm] & \left|\frac{\partial^2 A_i({\bf{y}})}{\partial y_{i-1}\partial y_i}\right| = |(\psi'(-y_{i-1})-\psi'(y_{i-1}))\phi'(y_i)| \le4. \end{aligned}

Note that each A_i({\bf{y}}) depends only on y_{i-1} and y_i for i\ge2, which implies the Hessian of \bar f_T({\bf{y}}) is tridiagonal matrix, i.e.,

\frac{\partial^2\bar f_T({\bf{y}})}{\partial y_i\partial y_j}=0 ~~~\text{for all}~~~i,j\in[T]~~~\text{such that}~~~|i-j|\ge2.

According to equations (?) and (?), the absolute value of each diagonal entry of \nabla^2\bar f_T({\bf{y}}) is bounded by 48+2=50. Moreover, equation (?) implies that each nonzero off-diagonal entry of \nabla^2\bar f_T({\bf{y}}) has absolute value at most 4. Hence, every row of \nabla^2\bar f_T({\bf{y}}) has absolute row sum at most 50+2\cdot4=58. Since the Hessian is symmetric, its spectral norm is bounded by its maximum absolute row sum, which implies

\left\| \nabla^2\bar f_T({\bf{y}}) \right\| \le\left\| \nabla^2\bar f_T({\bf{y}}) \right\|_\infty \le58

The proof of statement (c). We let i\in[T] be the smallest index such that |y_i|\le1. If i=1, then equation (?) and the nonnegativity of \psi' and \phi from Lemma 3.1(b) and (c) implies

\frac{\partial\bar f_T({\bf{y}})}{\partial y_1} \le-\phi'(y_1).

Combining with |y_1|\le1 and Lemma 3.1(d), we achieve

\left|\frac{\partial\bar f_T({\bf{y}})}{\partial y_1}\right| \ge\phi'(y_1)>\frac25.

If i\ge2, the definition of i implies |y_{i-1}|>1. According to Lemma 3.1(a), we obtain

\psi(-y_{i-1})+\psi(y_{i-1})=1.

Substituting equation (?) into equation (?) and using the non-negativity of \psi' and \phi from Lemma 3.1(b) and (c), we have

\frac{\partial\bar f_T({\bf{y}})}{\partial y_i} \le-\phi'(y_i).

Since |y_i|\le1, Lemma 3.1(d) implies

\left|\frac{\partial\bar f_T({\bf{y}})}{\partial y_i}\right| \ge\phi'(y_i)>\frac25.

Combing above two inequality, we obtain

\left\| \nabla\bar f_T({\bf{y}}) \right\| \ge \left|\frac{\partial\bar f_T}{\partial y_i}\right| > \frac25.

The proof of statement (d). We now establish the gradient-dependent upper bound for Hessian. We first consider the case that there exists some i\in[T] such that |y_i|\le1. By equation (?) and equation (?), we obtain

\left\| \nabla^2\bar f_T({\bf{y}}) \right\| \le 58 \leq 160\cdot\frac25 \le160\left\| \nabla\bar f_T({\bf{y}}) \right\|.

We then consider the case that |y_i|>1 holds for all i\in[T]. According to Lemma 3.1(a), we have

\psi(-y_i)+\psi(y_i)=1.

Moreover, since both y_i and -y_i do not lie in the interval (1/2,1), equation (Proof) implies

\psi'(-y_i)=\psi'(y_i)=0.

Substituting the above two identities into equation (?), we obtain

\frac{\partial\bar f_T({\bf{y}})}{\partial y_i} =-\phi'(y_i) \quad \text{for all}~~i\in[T].

Hence, the Hessian of \bar f_T({\bf{y}}) is diagonal and satisfies

\nabla^2\bar f_T({\bf{y}})=-\operatorname{diag}\big(\phi''(y_1),\ldots,\phi''(y_T)\big).

Consequently, Lemma 3.1(c) and equation (?) imply

\begin{aligned} \left\| \nabla^2\bar f_T({\bf{y}}) \right\|= \max_{i\in[T]}|\phi''(y_i)| \le2\max_{i\in[T]}\phi'(y_i) \le2\left\| \nabla\bar f_T({\bf{y}}) \right\| \le160\left\| \nabla\bar f_T({\bf{y}}) \right\|. \end{aligned}

Combining equations (?) and (?), we conclude that

\left\| \nabla^2\bar f_T({\bf{y}}) \right\| \le160\left\| \nabla\bar f_T({\bf{y}}) \right\| \qquad

for all {\bf{y}}\in{\mathbb{R}}^T.

The proof of statement (e). Follow the notation defined in Section ?, we let

\begin{aligned} \mathrm{prog}_{1/4}({\bf{y}}):=\max\{i\in{\mathbb{N}} : |y_i| > 1/4\}. \end{aligned}

for given {\bf{y}}=[y_1, \dots, y_T]\in{\mathbb{R}}^T. For each i>\operatorname{prog}_{1/4}({\bf{y}})+1, we have

\begin{aligned} |y_i| \le 1/4 \qquad\text{and}\qquad |y_{i-1}| \le 1/4. \end{aligned}

Combining Lemma 3.1(a) and equation equation (Proof), equation (?) implies

\psi(-y_{i-1})=\psi(y_{i-1})=0 \qquad\text{and}\qquad \psi'(-y_i)=\psi'(y_i)=0.

Substituting the above results into equation (?), we obtain

\frac{\partial\bar f_T({\bf{y}})}{\partial y_i}=0

for all i>\operatorname{prog}_{1/4}({\bf{y}})+1. Hence, it holds

\operatorname{prog}_0\big(\nabla\bar f_T({\bf{y}})\big) \le\operatorname{prog}_{1/4}({\bf{y}})+1.
Uses Proof, Lemma 3.1, Lemma A.1, Proof · Used by Proof
Lemma

Lemma A.2

MQT1

For all {\bf{x}}\in{\mathbb{R}}^d, the mapping \rho_R:{\mathbb{R}}^d\to{\mathbb{R}}^d defined in equation (?) holds

\begin{aligned} \left\| \rho_R({\bf{x}}) \right\|\le R, \qquad \left\| D\rho_R({\bf{x}}) \right\|\le1, \qquad \text{and} \qquad \left\| D^2\rho_R({\bf{x}}) \right\|\le\frac3R. \end{aligned}
Proof of Lemma A.2

Following [arjevani2023lower, Lemma 13 and its proof], we directly obtain

\left\| \rho_R({\bf{x}}) \right\|\le R \qquad\text{and}\qquad \left\| D\rho_R({\bf{x}}) \right\|\le1.

Here, we focus on the upper bound of \left\| D^2\rho_R({\bf{x}}) \right\|. We denote

\alpha({\bf{x}}):= \left(1+\frac{\left\| {\bf{x}} \right\|^2}{R^2}\right)^{-1/2}.

Then we can write \rho_R({\bf{x}})=\alpha({\bf{x}}){\bf{x}} and it holds [arjevani2023lower, Lemma 13]

D\rho_R({\bf{x}}) = \alpha({\bf{x}}){\bf I}_d - \frac{(\alpha({\bf{x}}))^3}{R^2}\cdot{\bf{x}}{\bf{x}}^\top.

According to [arjevani2023lower, Lemma 13], it holds

\left\| D\rho_R({\bf{x}})-D\rho_R({\bf{y}}) \right\| \le\frac3R\left\| {\bf{x}}-{\bf{y}} \right\| \qquad\text{for all }{\bf{x}},{\bf{y}}\in{\mathbb{R}}^d.

Since the mapping \rho_R is twice continuously differentiable, for any {\bf{h}},{\bf{k}}\in{\mathbb{R}}^d, we have

\begin{aligned} \left\| D^2\rho_R({\bf{x}})[{\bf{h}},{\bf{k}}] \right\| = \lim_{t\to0} \left\| \frac{ \big(D\rho_R({\bf{x}}+t{\bf{k}})-D\rho_R({\bf{x}})\big){\bf{h}} }{t} \right\|\le \frac3R\left\| {\bf{h}} \right\|\left\| {\bf{k}} \right\|, \end{aligned}

where the inequality follows from equation (?). Taking the supremum over all unit vectors {\bf{h}} and {\bf{k}}, we achieve

\left\| D^2\rho_R({\bf{x}}) \right\| \le\frac3R.

This completes the proof.

Proof

Proof

RS33

Recall that equation (?) has defined

H_{{\bf U},\mu}({\bf{x}}) = \bar f_T\big({\bf U}^\top\rho_R({\bf{x}})\big) +\frac{\mu}{2}\left\| {\bf{x}} \right\|^2 \quad \text{with} \quad R=\frac{1000\sqrt T}{\mu} \quad\text{and}\quad \mu\in\left(0,\frac{1}{100}\right],

where {\bf U}=[{\bf{u}}_1,\ldots,{\bf{u}}_T]\in\operatorname{St}(d,T). Throughout the proof of this lemma, we denote

\begin{aligned} {\bf{y}}:={\bf U}^\top\rho_R({\bf{x}}). \end{aligned}

and recall from equation (?) that \rho_R({\bf{x}})=\alpha({\bf{x}}){\bf{x}}. By Lemma 3.2(b), we have

\left\| \nabla\bar f_T({\bf{y}}) \right\| \le9\sqrt T \qquad \text{and} \qquad \left\| \nabla^2\bar f_T({\bf{y}}) \right\|\le 58.

The optimality gap. Note that equation (?) implies \rho_R({\bf{0}})={\bf{0}}. Combining with equation (?), we obtain

\begin{aligned} H_{{\bf U},\mu}({\bf{0}})=\bar f_T({\bf{0}})=-1. \end{aligned}

Moreover, the lower bound of \bar f_T({\bf{y}})\ge-2T in equation (?) implies

H_{{\bf U},\mu}({\bf{x}}) \ge\bar f_T({\bf U}^\top\rho_R({\bf{x}})) \ge-2T \qquad\text{for all }{\bf{x}}\in{\mathbb{R}}^d.

Combining above two results, we have

H_{{\bf U},\mu}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}H_{{\bf U},\mu}({\bf{x}}) \le -1+2T \le2T.

The lower bound of the Hessian. According to the chain rule and equation (?), the gradient of H_{{\bf U},\mu} has the form of

\nabla H_{{\bf U},\mu}({\bf{x}}) = D\rho_R({\bf{x}}){\bf U}\nabla\bar f_T({\bf{y}}) +\mu{\bf{x}}.

Since we have assumed there exists some i\in[T] such that |y_i|=\left| [{\bf U}^\top\rho_R({\bf{x}})]_i\right|\le1, we denote

i_\star:=\min\{i\in[T]:|y_i|\le1\}.

If i_\star=1, then equation (?) and the non-negativity of \psi' and \phi from Lemma 3.1(b) and (c) imply

\begin{aligned} \frac{\partial\bar f_T({\bf{y}})}{\partial y_1} \le-\phi'(y_1). \end{aligned}

If i_\star\ge2, the minimality of i_\star gives |y_{i_\star-1}|>1. Hence, Lemma 3.1(a) implies

\begin{aligned} \psi(-y_{i_\star-1})+\psi(y_{i_\star-1})=1. \end{aligned}

Substituting the above result into equation (?) also yields equation (?). Thus, Lemma 3.1(d) implies

\frac{\partial\bar f_T({\bf{y}})}{\partial y_{i_\star}} \le-\phi'(y_{i_\star}) <-\frac25.

In the case of \left\| {\bf{x}} \right\|\ge20\sqrt T/\mu, we apply equation (?), (Proof), and Proof to achieve

\begin{aligned} \begin{split} \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| &\ge 20\sqrt{T} -\left\| D\rho_R({\bf{x}}){\bf U}\nabla\bar f_T({\bf{y}}) \right\| \\ &\ge 20\sqrt{T} -\left\| \nabla\bar f_T({\bf{y}}) \right\| \\ &\ge 20\sqrt{T} -9\sqrt T, \\ &\ge 11. \end{split} \end{aligned}

In the case of \left\| {\bf{x}} \right\|<20\sqrt T/\mu. The setting R=1000\sqrt T/\mu implies \left\| {\bf{x}} \right\|/R<1/50. Therefore, equation (?) implies

\alpha({\bf{x}})> \left(1+\frac1{2500}\right)^{-1/2} >\frac{99}{100} \qquad \text{and} \qquad \frac1{\alpha({\bf{x}})}<\frac{101}{100}.

Let {\bf{u}}_{i_\star}\in{\mathbb{R}}^T be the i_\starth column of {\bf U}. Recall that {\bf U}^\top{\bf U}={\bf I}_T, we then apply equations (?) and (?) to give

\begin{aligned} \langle{\bf{u}}_{i_\star},\nabla H_{{\bf U},\mu}({\bf{x}})\rangle = \frac{\alpha({\bf{x}})\partial\bar f_T({\bf{y}})}{\partial y_{i_\star}} - \frac{(\alpha({\bf{x}}))^3}{R^2} \langle{\bf{u}}_{i_\star},{\bf{x}}\rangle \langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle + \mu\langle{\bf{u}}_{i_\star},{\bf{x}}\rangle. \end{aligned}

Note that the definition of i_\star indicates y_{i_\star}=\alpha({\bf{x}})\langle{\bf{u}}_{i_\star},{\bf{x}}\rangle and |y_{i_\star}|\le1, which results

|\langle{\bf{u}}_{i_\star},{\bf{x}}\rangle| \le\frac1{\alpha({\bf{x}})}.

For the second term on the right-hand side of inequality (?), we have

\begin{aligned} \begin{split} & \frac{(\alpha({\bf{x}}))^3}{R^2} \left| \langle{\bf{u}}_{i_\star},{\bf{x}}\rangle \langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle \right| \le \frac{(\alpha({\bf{x}}))^3}{R^2} \left| \langle{\bf{u}}_{i_\star},{\bf{x}}\rangle \right| \left\| {\bf{x}}\right\| \left\|{\bf U}\nabla\bar f_T({\bf{y}}) \right\| \\ \le & \frac{(\alpha({\bf{x}}))^2\left\| {\bf{x}}\right\| }{R^2} \left\|{\bf U}\nabla\bar f_T({\bf{y}}) \right\| \le \frac{9\sqrt T\left\| {\bf{x}} \right\|}{R^2} \leq \frac{9\sqrt T(20\sqrt T/\mu)}{(1000\sqrt T/\mu)^2} =\frac{\mu}{5000}\le \frac{2}{10^{6}}. \end{split} \end{aligned}

where the first inequality is based on Cauchy–Schwarz inequality, the second inequality is based on equation (?), the third inequality is based on equation (Proof) and the definition (?), and the last inequality is based on the condition \left\| {\bf{x}} \right\|<20\sqrt T/\mu and the setting R=1000\sqrt T/\mu. For the last term on the right-hand side of inequality (?), equation (?) and the setting \mu\le10^{-2} implies

\begin{aligned} \mu|\langle{\bf{u}}_{i_\star},{\bf{x}}\rangle| &\le\frac{\mu}{\alpha({\bf{x}})} <\frac{101}{100}\mu \leq \frac{101}{10000} \end{aligned}

Combining equations (?), (?), (?), and (?) with equation (?), we obtain

\begin{aligned} \left| \langle{\bf{u}}_{i_\star},\nabla H_{{\bf U},\mu}({\bf{x}})\rangle \right| \ge \frac{2\alpha({\bf{x}})}5 - 2\times10^{-6} -\frac{101}{10000} > \frac25\cdot\frac{99}{100} -2\times10^{-6} -\frac{101}{10000} > \frac14. \end{aligned}

Since we have \left\| {\bf{u}}_{i_\star} \right\|=1, the Cauchy–Schwarz inequality yields

\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| \ge \left| \langle{\bf{u}}_{i_\star},\nabla H_{{\bf U},\mu}({\bf{x}})\rangle \right| >\frac14.

Combining equation (?) and equation (?), we conclude that whenever there exists some i\in[T] such that |[{\bf U}^\top\rho_R({\bf{x}})]_i|\le1, it holds \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|\ge 1/4.

The upper bound for the Hessian. We now establish the gradient-dependent upper bound for the Hessian. For all vectors {\bf{h}},{\bf{k}}\in{\mathbb{R}}^d, the chain rule implies

\begin{aligned} \begin{split} D^2H_{{\bf U},\mu}({\bf{x}})[{\bf{h}},{\bf{k}}] &= \left\langle \nabla^2\bar f_T({\bf{y}}) {\bf U}^\top D\rho_R({\bf{x}})[{\bf{h}}], {\bf U}^\top D\rho_R({\bf{x}})[{\bf{k}}] \right\rangle \\ &~~\quad + \left\langle {\bf U}\nabla\bar f_T({\bf{y}}), D^2\rho_R({\bf{x}})[{\bf{h}},{\bf{k}}] \right\rangle + \mu\langle{\bf{h}},{\bf{k}}\rangle. \end{split} \end{aligned}

Taking the supremum on {\bf{h}} and {\bf{k}} over the unit ball and and using the fact \left\| {\bf U} \right\|=1, we obtain

\begin{aligned} \begin{split} \left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| \le &{} \left\| \nabla^2\bar f_T({\bf{y}}) \right\|\left\| D\rho_R({\bf{x}}) \right\|^2+\left\| \nabla\bar f_T({\bf{y}}) \right\|\left\| D^2\rho_R({\bf{x}}) \right\|+\mu \\ \le &{} 58+\frac{27\sqrt T}{R}+\mu \leq 58+\frac{1027\mu}{1000} < 59, \end{split} \end{aligned}

where the second inequality is based on equations (Proof), (Proof), and (?), the second last two inequalities are based on the settings R=1000\sqrt T/\mu and \mu\le10^{-2}.

If there exists some i\in[T] such that |y_i|\le1. According to equation (3.3) from Lemma 3.3, we have \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|\ge1/4. Combining this with equation (?) and the setting \mu<1 yields

\left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| <59 <240\cdot\frac14 \le240\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right).

We then focus on the case that |y_i|>1 holds for all i\in[T]. If \left\| {\bf{x}} \right\|\ge20\sqrt T/\mu, then equations (?) and (?) imply

\begin{aligned} \left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| &<59\le 240\cdot 11 \le 240 \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|\leq 240\left(\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+\mu\log \frac1\mu \right). \end{aligned}

If \left\| {\bf{x}} \right\|<20\sqrt T/\mu, then equation (?) gives

\begin{aligned} \alpha({\bf{x}})>\frac{99}{100} \qquad\text{and}\qquad \frac{1}{\alpha({\bf{x}})}<\frac{100}{99}. \end{aligned}

The condition |y_i|>1 holds for all i\in[T] indicates that we can combine equation (?) and Lemma 3.1(c) to achive

\left\| \nabla^2\bar f_T({\bf{y}}) \right\| = \max_{i\in[T]}|\phi''(y_i)| \le2\max_{i\in[T]}\phi'(y_i).

We denote the index j\in[T] such that \phi'(y_j)=\max_{i\in[T]}\phi'(y_i) and analyze the right-hand side of inequality (?) by the following two cases.

  1. If |y_j|\ge\log(1/\mu), then Lemma 3.1(c) implies

    \max_{i\in[T]}\phi'(y_i) = \operatorname{sech}^2(y_j) \le4\mathrm e^{-2|y_j|} \le4\mu^2 \le\mu\log\frac1\mu,

    where the first inequality follows from \cosh t\ge\mathrm e^{|t|}/2, and the last inequality uses \mu\le10^{-2}.

  2. If |y_j|<\log(1/\mu), then the identity y_j=\alpha({\bf{x}})\langle{\bf{u}}_j,{\bf{x}}\rangle implies

    |\langle{\bf{u}}_j,{\bf{x}}\rangle| < \frac{\log(1/\mu)}{\alpha({\bf{x}})}.

    Following the derivation of equation (?) by replacing {\bf{u}}_{i_\star} with {\bf{u}}_{j}, we obtain

    \begin{aligned} \langle {\bf{u}}_j ,\nabla H_{{\bf U},\mu}({\bf{x}})\rangle &=\frac{\alpha({\bf{x}})\partial\bar f_T({\bf{y}})}{\partial y_{j}}-\frac{(\alpha({\bf{x}}))^3}{R^2}\langle{\bf{u}}_j,{\bf{x}}\rangle\langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle +\mu\langle{\bf{u}}_j,{\bf{x}}\rangle \\ &=-\alpha({\bf{x}})\phi'(y_j)-\frac{(\alpha({\bf{x}}))^3}{R^2}\langle{\bf{u}}_j,{\bf{x}}\rangle\langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle +\mu\langle{\bf{u}}_j,{\bf{x}}\rangle, \end{aligned}

    where the last step is based on equation (?). Since we have \phi'(y_j)=\max_{i\in[T]}\phi'(y_i) and \left\| {\bf{u}}_j \right\|=1, applying the triangle inequality on above equation implies

    \begin{aligned} \alpha({\bf{x}})\max_{i\in[T]}\phi'(y_i)\le\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+\frac{(\alpha({\bf{x}}))^3}{R^2}|\langle{\bf{u}}_j,{\bf{x}}\rangle| \left|\langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle\right| + \mu|\langle{\bf{u}}_j,{\bf{x}}\rangle|. \end{aligned}

    For the second term on the right-hand side of equation (?), applying Cauchy–Schwarz inequality, equation (Proof) with the fact {\bf U}^\top{\bf U}={\bf I}_T imply

    \begin{aligned} |\langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle| \le \left\| {\bf{x}} \right\|\left\| \nabla\bar f_T({\bf{y}}) \right\|\le9\sqrt T\left\| {\bf{x}} \right\|. \end{aligned}

    Combining with equation (?), we obtain

    \begin{aligned} \begin{split} & \frac{(\alpha({\bf{x}}))^3}{R^2}|\langle{\bf{u}}_j,{\bf{x}}\rangle|\left|\langle{\bf{x}},{\bf U}\nabla\bar f_T({\bf{y}})\rangle\right| \\ \le & \frac{9(\alpha({\bf{x}}))^2\sqrt T\left\| {\bf{x}} \right\|\log(1/\mu)}{R^2} \\ < & \frac{9\mu\log(1/\mu)}{50000}<\frac{\mu\log(1/\mu)}{5000}, \end{split} \end{aligned}

    where the second inequality follows from \alpha({\bf{x}})\le1, \left\| {\bf{x}} \right\|<20\sqrt T/\mu, and R=1000\sqrt T/\mu. Substituting equations (?) and (?) into equation (?), we obtain

    \begin{aligned} \alpha({\bf{x}})\max_{i\in[T]}\phi'(y_i) \le \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| + \left(\frac1{5000}+\frac1{\alpha({\bf{x}})} \right)\mu\log\frac1\mu. \end{aligned}

    Dividing equation (?) by \alpha({\bf{x}})>0 and applying equation (?), we obtain

    \begin{aligned} \begin{split} \max_{i\in[T]}\phi'(y_i) &\le \frac1{\alpha({\bf{x}})} \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+ \left(\frac1{5000\alpha({\bf{x}})} + \frac1{(\alpha({\bf{x}}))^2} \right)\mu\log\frac1\mu \\ &\le 2\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right). \end{split} \end{aligned}

Combining equations (?) and (?), we conclude that

\max_{i\in[T]}\phi'(y_i) \le 2\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right).

We now return to the bound of Hessian. The first inequality in equation (?) implies

\begin{aligned} \begin{split} \left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| &\le \left\| \nabla^2\bar f_T({\bf{y}}) \right\|\left\| D\rho_R({\bf{x}}) \right\|^2+\left\| \nabla\bar f_T({\bf{y}}) \right\|\left\| D^2\rho_R({\bf{x}}) \right\|+\mu\\ &\le2\max_{i\in[T]}\phi'(y_i)+\frac{27\sqrt T}{R}+\mu \\ &\le4\left(\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+\mu\log\frac1\mu\right)+\frac{1027}{1000}\mu \\ &\le\frac{5027}{1000}\left(\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+\mu\log\frac1\mu\right) \\ &<240\left(\left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\|+\mu\log\frac1\mu\right), \end{split} \end{aligned}

where the second inequality is based on equations (?), (Proof), (Proof),and (?); the third inequality inequality follows from equation (?) and setting R=1000\sqrt T/\mu; the inequality uses the fact \mu\le10^{-2} which implies \mu\le\mu\log(1/\mu).

Combining equations (?), (?), and (?), we conclude that for every {\bf{x}}\in{\mathbb{R}}^d, it holds

\left\| \nabla^2H_{{\bf U},\mu}({\bf{x}}) \right\| \le 240\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right).

This completes the proof.

Proof

Proof

NY4K

Recall that the definition of f_{{\bf U}}({\bf{x}})=aH_{{\bf U},\mu}({\bf{x}}/b) in equation (?) implies

\begin{aligned} \nabla f_{{\bf U}}({\bf{x}}) =\frac{a}{b}\nabla H_{{\bf U},\mu}\left(\frac{{\bf{x}}}{b}\right), \qquad\text{and}\qquad \nabla^2 f_{{\bf U}}({\bf{x}}) = \frac{a}{b^2}\nabla^2 H_{{\bf U},\mu}\left(\frac{{\bf{x}}}{b}\right). \end{aligned}

We verify the properties of the function f_{{\bf U}}({\bf{x}}) as follows.

The proof of statement (a). Combining the gradient-dependent curvature bound equation (3.3) and equation (Proof), we have

\begin{aligned} \begin{split} & \left\| \nabla^2 f_{{\bf U}}({\bf{x}}) \right\| = \frac{a}{b^2}\left\| \nabla^2 H_{{\bf U},\mu}\left(\frac{{\bf{x}}}{b}\right) \right\| \\ \leq & \frac{240a}{b^2}\left( \left\| \nabla H_{{\bf U},\mu}({\bf{x}}) \right\| +\mu\log\frac1\mu \right) \\ =& \frac{240}{b}\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| +\frac{240a\mu}{b^2}\log\frac1\mu \\ =& \frac{L_1}{2}\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| +8L_1\epsilon\mu\log\frac1\mu, \end{split} \end{aligned}

where the last step follows from the setting b=480/L_1 and a/b=16\epsilon in equation (3.4). For the term \mu\log(1/\mu), recall we set

\begin{aligned} \delta_0=\min\left\{10^{-2},\frac{L_0}{16L_1\epsilon}\right\}\in(0, 10^{-2}] \qquad \text{and} \qquad \mu=\frac{\delta_0}{4\log(\mathrm e/\delta_0)} \end{aligned}

in equation (3.4), which implies

\begin{aligned} \log\frac1\mu = \log\frac1{\delta_0} +\log\left(4\log\frac{\mathrm e}{\delta_0}\right) \le 2\log\frac1{\delta_0}+\log4 <2\log\frac{\mathrm e}{\delta_0}, \end{aligned}

where the first inequality uses \log t\le t-1 for all t>0. Combining with the definition of \delta_0 and \mu in equation (?), it holds

\mu\log\frac1\mu \le\frac{\delta_0}{2} \le\frac{L_0}{32L_1\epsilon}.

Combining equation (?) and equation (?), we know that for all {\bf{x}}\in{\mathbb{R}}^d, it holds

\begin{aligned} \left\| \nabla^2 f_{{\bf U}}({\bf{x}}) \right\| &\le \frac{L_1}{2}\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| +8L_1\epsilon\mu\log\frac1\mu \\ &\le \frac{L_1}{2}\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| +4L_1\epsilon\delta_0 \\ &\le \frac12\left( L_0+L_1\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| \right). \end{aligned}

Following Corollary A.4 of [zhang2020improved] by replacing their parameters L_0, L_1, and c with L_0/2, L_1/2, and 1, respectively, we obtain

\begin{aligned} \left\| \nabla f_{{\bf U}}({\bf{y}})-\nabla f_{{\bf U}}({\bf{x}}) \right\| &\le \left( L_0+\frac{(\mathrm e-1)L_1 \left\| \nabla f_{{\bf U}}({\bf{x}}) \right\|}{2} \right)\left\| {\bf{y}}-{\bf{x}} \right\| \\ &\le \left( L_0+L_1\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| \right)\left\| {\bf{y}}-{\bf{x}} \right\|, \end{aligned}

for all {\bf{x}},{\bf{y}}\in{\mathbb{R}}^d with \left\| {\bf{y}}-{\bf{x}} \right\|\le 1/L_1\le 2/L_1. Thus, the function f_{{\bf U}} is (L_0,L_1)-smooth.

The proof of statement (b). Since a,b>0, the definition of the function f_{{\bf U}} in equation (?) and the initial optimality gap bound equation (3.3) imply

\begin{aligned} f_{{\bf U}}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}f_{{\bf U}}({\bf{x}}) &= a\left( H_{{\bf U},\mu}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}H_{{\bf U},\mu}\left(\frac{{\bf{x}}}{b}\right) \right) \\ &= a\left( H_{{\bf U},\mu}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}H_{{\bf U},\mu}({\bf{x}}) \right) \le2aT, \end{aligned}

where the second equality follows from the fact that the mapping {\bf{x}}\mapsto{\bf{x}}/b is a bijection on {\mathbb{R}}^d.

The proof of statement (c). Let i be the index i\in[T] such that

\left| \left[ {\bf U}^\top\rho_R\left(\frac{{\bf{x}}}{b}\right) \right]_i \right|\le1.

By the gradient formula in equation (Proof) and equation (3.3), we obtain

\left\| \nabla f_{{\bf U}}({\bf{x}}) \right\| = \frac{a}{b} \left\| \nabla H_{{\bf U},\mu}\left(\frac{{\bf{x}}}{b}\right) \right\| \ge \frac{a}{4b} = 4\epsilon,

where the last equality follows from the setting a/b=16\epsilon in equation (3.4).

Uses Lemma 3.3, Lemma 3.4 · Used by Proof
Lemma

Lemma B.1

D4XD

Based on the definition of \chi(t) in equation (?), for all t\in{\mathbb{R}}, it holds

\begin{aligned} 0\le\chi(t)\le1 \qquad\text{and}\qquad |\chi'(t)|^2\le14400\bigl(1-(\chi(t))^2\bigr). \end{aligned}
Used by Proof
Proof of Lemma B.1

According to definition of \chi(t) in equation (?), it holds

\chi(t)= \begin{cases} 1, & \text{if }|t|\le\dfrac14, \\[0.3cm] 0, & \text{if }|t|\ge\dfrac12, \end{cases}

which implies

\chi'(t)=0 \quad\text{whenever}~~ |t|\le\frac14 ~~\text{ or }~~|t|\ge\frac12.

Moreover, the bounds on S established in the proof of Lemma 3.1(b) imply

0\le\chi(t)\le1 \qquad\text{for all }t\in{\mathbb{R}}.

Combining equations (Proof) and (Proof) indicates equation (B.1) holds whenever |t|\le1/4 or |t|\ge1/2.

It remains to prove equation (B.1) holds when |t|\in(1/4,1/2). We let

\begin{aligned} u:=4|t|-1\in(0,1). \end{aligned}

for t such that |t|\in(1/4,1/2). Applying the chain rule and equation (Proof) give

\begin{aligned} |\chi'(t)| = 4S'(u) = 120u^2(1-u)^2. \end{aligned}

Moreover, the definition of S(u) in equation (?) gives

S(u)=u^3(10-15u+6u^2)\ge u^3= u^3(3(1-u)(3-2u)+1) \geq u^3,

Hence, combining the facts u\in(0,1) and S(u)\in(0,1) with above two results leads to

\begin{aligned} |\chi'(t)|^2 &=14400u^4(1-u)^4 \\ &\le14400u^3 \le14400S(u)(2-S(u)) \\ &=14400\bigl(1-\chi(t)^2\bigr). \end{aligned}

whenever |t|\in(1/4,1/2). Thus equation (B.1) holds for every t\in{\mathbb{R}}.

Lemma

Lemma B.2

UHES

The functions \Theta_i({\bf{y}}) defined in equation (?) satisfy

\left\| \nabla\Theta_i({\bf{y}}) \right\|\le120

for all i\in[T] and {\bf{y}}\in{\mathbb{R}}^T.

Used by Proof
Proof of Lemma B.2

The definition \Theta_i({\bf{y}}):=\prod_{k=i}^{T}\chi(y_k) implies

\frac{\partial\Theta_i({\bf{y}})}{\partial y_j} = \begin{cases} 0, & j<i,\\[0.1cm] \displaystyle \chi'(y_j) \prod_{\substack{k=i\\k\ne j}}^{T} \chi(y_k), & j\ge i. \end{cases}

Consequently, Lemma B.1 implies

\begin{aligned} \left\| \nabla\Theta_i({\bf{y}}) \right\|^2 = \sum_{j=i}^{T} |\chi'(y_j)|^2 \prod_{\substack{k=i\\k\ne j}}^{T} \chi(y_k)^2 \le 14400\sum_{j=i}^{T} \bigl(1-\chi(y_j)^2\bigr) \prod_{\substack{k=i\\k\ne j}}^{T} \chi(y_k)^2 \le14400. \end{aligned}

Taking square roots on above result yields

\left\| \nabla\Theta_i({\bf{y}}) \right\|\le120,

which completes the proof.

Lemma

Lemma B.3

CW9L

For given {\bf{y}}=[y_1,\dots,y_T]\in{\mathbb{R}}^T, suppose there exists some s\in\{0,\ldots,T-1\} such that |y_{s+1}|,\ldots,|y_T|\le1/4. Then the stochastic component \bar F_T defined in equation (?) satisfies:

  1. If Z=0, then \operatorname{supp}(\nabla\bar F_T({\bf{y}};0)) \subseteq[s] and \nabla\bar F_T({\bf{y}};0) =\nabla\bar F_T(\Pi_s({\bf{y}});0).

  2. If Z=1, then \operatorname{supp}(\nabla\bar F_T({\bf{y}};1)) \subseteq[s+1] and \nabla\bar F_T({\bf{y}};1) =\nabla\bar F_T(\Pi_{s+1}({\bf{y}});1).

Here, we denote \Pi_s({\bf{y}})=[y_1,\dots,y_s,0,\dots,0]^\top.

Used by Proof
Proof of Lemma B.3

The assumption on the index s means

|y_{s+1}|,\ldots,|y_T|\le\frac14.

According to equations (?) and (Proof), we have

\chi(y_k)=1 \qquad \text{and}\qquad \chi'(y_k)=0 \qquad \text{for all}~~k=s+1,\ldots,T.

Consequently, the definition of \Theta_i({\bf{y}}) in equation (?) implies

\Theta_i({\bf{y}}) =\begin{cases} \displaystyle{\prod_{k=i}^{s}\chi(y_k)}, & \text{if}~i\le s+1, \\[0.35cm] 1 & \text{otherwise}. \end{cases}

Hence, we have

\Theta_{s+1}({\bf{y}})=1, \qquad\text{and}\qquad \nabla\Theta_{s+1}({\bf{y}})={\bf{0}},

It also holds

\frac{\partial\Theta_i({\bf{y}})}{\partial y_j}=0 \qquad \text{whenever} \quad i\le s+1 \quad\text{and}\quad j\ge s+1.

Moreover, for all i\ge s+2, condition equation (Proof) gives |y_{i-1}|\le1/4 and |y_i|\le1/4. Hence, Lemma 3.1(a) implies

\psi(y_{i-1})=\psi(-y_{i-1})=\psi'(y_{i-1})=\psi'(-y_{i-1})=0.

Hence, the definition of A_i({\bf{y}}) in equation (?) implies

A_i({\bf{y}})=0 \qquad\text{and}\quad \nabla A_i({\bf{y}})={\bf{0}} \qquad \text{for all}~~i\ge s+2.

The proof of statement (a). For Z=0, equations (?) and (?) give

\bar F_T({\bf{y}};0) = \sum_{i=1}^{T} A_i({\bf{y}})\bigl(1-\Theta_i({\bf{y}})\bigr).

Combining with equations (?) and (?), we have

\nabla\bar F_T({\bf{y}};0) = \sum_{i=1}^{s} \nabla\left( A_i({\bf{y}})\bigl(1-\Theta_i({\bf{y}})\bigr) \right).

Under condition (Proof), we know that equation (?) indicates all functions A_1,\dots,A_s only depend on y_1,\ldots,y_s, and equations (?) and (?) imply that the function values and the gradients of all \Theta_1,\dots,\Theta_s only depend on the y_1,\ldots,y_s. Hence, equation (?) yields

\begin{aligned} \operatorname{supp} \bigl(\nabla\bar F_T({\bf{y}};0)\bigr) \subseteq[s] \qquad\text{and}\qquad \nabla\bar F_T({\bf{y}};0) =\nabla\bar F_T(\Pi_s({\bf{y}});0). \end{aligned}

The proof of statement (b). For Z=1, equations equation (?) and equation (?) give

\bar F_T({\bf{y}};1) =\sum_{i=1}^{T} A_i({\bf{y}})\left(1+\left(\frac1q-1\right)\Theta_i({\bf{y}})\right).

Recall that equation (?) indicates that all terms with i\ge s+2 in above summation have zero gradient. Thus, it holds

\nabla\bar F_T({\bf{y}};1) = \sum_{i=1}^{s+1} \nabla\left( A_i({\bf{y}}) \left( 1+\left(\frac1q-1\right)\Theta_i({\bf{y}}) \right) \right).

Under condition (Proof), we know equation (?) implies all function A_1,\dots,A_{s+1} only depends on y_1,\dots,y_{s+1}, and equations (?), (?), and (?) imply that the function values and the gradients of \Theta_1,\dots,\Theta_{s+1} only depend on y_1,\dots,y_{s+1}. Hence, equation (?) yields

\begin{aligned} \operatorname{supp} \bigl(\nabla\bar F_T({\bf{y}};1)\bigr) \subseteq[s+1] \qquad\text{and}\qquad \nabla\bar F_T({\bf{y}};1) =\nabla\bar F_T(\Pi_{s+1}({\bf{y}});1). \end{aligned}
Proof

Proof

2FGV

The verify properties of functions \bar F_T and Q_T.

The proof of statement (a). According to equation (?) and the product rule, we know that

\nabla Q_T({\bf{y}}) = \sum_{i=1}^{T} \left( \Theta_i({\bf{y}})\nabla A_i({\bf{y}}) +A_i({\bf{y}})\nabla\Theta_i({\bf{y}}) \right).

We denote

m:=\operatorname{prog}_{1/2}({\bf{y}}) \in\{0,1,\ldots,T\}.

For any m\in\{0,1,\ldots,T\} and i\ge m+2, we have |y_{i-1}|\le1/2. Hence, Lemma 3.1(a) and equation (Proof) implies

\psi(y_{i-1}) = \psi(-y_{i-1}) =\psi'(y_{i-1})=\psi'(-y_{i-1})=0.

Combing above results with equation (?) gives A_i({\bf{y}})=0 and \nabla A_i({\bf{y}})={\bf{0}}, which implies

\Theta_i({\bf{y}})\nabla A_i({\bf{y}})+A_i({\bf{y}})\nabla\Theta_i({\bf{y}})={\bf{0}} ~~~~~\text{for all}~i\ge m+2.

In the case of m=0, combing equations (Proof) and (?) implies

\begin{aligned} \nabla Q_T({\bf{y}}) = \Theta_1({\bf{y}})\nabla A_1({\bf{y}})+A_1({\bf{y}})\nabla\Theta_1({\bf{y}}) \end{aligned}

In the case of m\in[T], we have |y_m|>1/2. Hence, equation (Proof) and equation (Proof) imply

\chi(y_m)=\chi'(y_m)=0.

Note that for all i\le m, equations (?) and (Proof) yield

\Theta_i({\bf{y}})=\prod_{k=i}^{T}\chi(y_k)=0 \qquad\text{and}\qquad \nabla\Theta_i({\bf{y}})={\bf{0}},

respectively. This implies

\Theta_i({\bf{y}})\nabla A_i({\bf{y}})+A_i({\bf{y}})\nabla\Theta_i({\bf{y}})={\bf{0}} ~~~~~\text{for all}~i\le m.

Combining equations (?), (?) and (?), we obtain

\nabla Q_T({\bf{y}})= \begin{cases} \Theta_{m+1}({\bf{y}})\nabla A_{m+1}({\bf{y}})+A_{m+1}({\bf{y}})\nabla\Theta_{m+1}({\bf{y}}) & \text{if}~m<T,\\[0.1cm] {\bf{0}}, & \text{if}~m=T. \end{cases}

We now establish the upper bound on \left\| \nabla Q_T({\bf{y}}) \right\| by considering the following three cases.

  1. In the case of m=0, we have A_1({\bf{y}})=-\phi(y_1), so that Lemma 3.1(c) implies |A_1({\bf{y}})|<2 and \left\| \nabla A_1({\bf{y}}) \right\|\le1. By equation (?), equation (Proof), and Lemma B.2, we obtain

    \begin{aligned} \left\| \nabla Q_T({\bf{y}}) \right\| &\le |\Theta_1({\bf{y}})|\left\| \nabla A_1({\bf{y}}) \right\|+|A_1({\bf{y}})|\left\| \nabla\Theta_1({\bf{y}}) \right\| \\ &\le1+2\cdot120 =241<249. \end{aligned}
  2. In the case of m\in[T-1], combining the definition of A_i({\bf{y}}) in equation (?) with equation (A.1) and Lemma 3.1(c) imply

    |A_{m+1}({\bf{y}})| = |\psi(-y_{m})\phi(-y_{m+1})-\psi(y_{m})\phi(y_{m+1})| \le 2(\psi(-y_m)+\psi(y_m)) \le 2.

    Moreover, combining the definition of A_i({\bf{y}}) in equation (?) with equation (A.1), equation (A.1), and Lemma 3.1(c), we have

    \begin{aligned} &\left|\frac{\partial A_{m+1}({\bf{y}})}{\partial y_m}\right| = |\psi'(-y_{m})\phi(-y_{m+1})+\psi'(y_{m})\phi(y_{m+1})| \le 2|\psi'(-y_m)+\psi'(y_m)| \le8,\\ &\left|\frac{\partial A_{m+1}({\bf{y}})}{\partial y_{m+1}}\right| = |\psi(-y_{m})\phi'(-y_{m+1})+\psi(y_{m})\phi'(y_{m+1})| \leq |\psi(-y_m)+\psi(y_m)| \le1, \end{aligned}

    Since the function A_{m+1}({\bf{y}}) depends only on y_m and y_{m+1}, the above results leads to

    \left\| \nabla A_{m+1}({\bf{y}}) \right\| \le\sqrt{8^2+1^2} =\sqrt{65}<9.

    Therefore, combining equation (?) with equations (?), (Proof), (?), and (?), and Lemma B.2, we have

    \begin{aligned} \begin{split} \left\| \nabla Q_T({\bf{y}}) \right\| &\le |\Theta_{m+1}({\bf{y}})| \left\| \nabla A_{m+1}({\bf{y}}) \right\| +|A_{m+1}({\bf{y}})| \left\| \nabla\Theta_{m+1}({\bf{y}}) \right\| \\ &\le9+2\cdot120 =249. \end{split} \end{aligned}
  3. In the case of m=T, equation (?) directly gives \left\| \nabla Q_T({\bf{y}}) \right\|=0\le249.

Therefore, the above derivations conclude \left\| \nabla Q_T({\bf{y}}) \right\|\le249 holds for all {\bf{y}}\in{\mathbb{R}}^T.

The proof of statement (b). Differentiating equation (?) obtains

{\bf{g}}_T({\bf{y}};Z) = \nabla\bar f_T({\bf{y}}) + \left(\frac Zq-1\right)\nabla Q_T({\bf{y}}).

The distribution Z\sim\operatorname{Bernoulli}(q) indicates {\mathbb{E}}_Z[Z/q-1]=0, resulting

{\mathbb{E}}_Z[{\bf{g}}_T({\bf{y}};Z)] = \nabla\bar f_T({\bf{y}}).

We then verify the property of probability-q zero-chain for {\bf{g}}_T({\bf{y}};Z). Fix given {\bf{y}}\in{\mathbb{R}}^T, we let

\begin{aligned} s=\operatorname{prog}_{1/4}({\bf{y}}). \end{aligned}

We show that {\bf{g}}_T({\bf{y}};Z) is q-probability zero-chain by verifying conditions (4.1) and (4.1) as follows.

  1. If s<T, the definition of s implies |y_{s+1}|,\ldots,|y_T|\le1/4. Hence, when Z=0, we can apply Lemma B.3(a) to achieve

    \begin{aligned} \operatorname{prog}_0({\bf{g}}_T({\bf{y}};0)) \le s \quad\text{and}\quad {\bf{g}}_T({\bf{y}};0) = {\bf{g}}_T(\Pi_s({\bf{y}});0) \quad \text{for all } {\bf{y}}\in{\mathbb{R}}^T. \end{aligned}

    Recall that \mathbb P(Z=0)=1-q, then equation (?) verifies that condition equation (4.1) with \theta=q. When Z=1, wen can apply Lemma B.3(b) to achieve

    \begin{aligned} \operatorname{prog}_0({\bf{g}}_T({\bf{y}};1)) \le s+1 \quad\text{and}\quad {\bf{g}}_T({\bf{y}};1) = {\bf{g}}_T(\Pi_{s+1}({\bf{y}});1) \quad \text{for all } {\bf{y}}\in{\mathbb{R}}^T. \end{aligned}

    According to Lemma B.3(a), we know that equation (?) also holds when Z=0. Hence, we have verifies condition (4.1) with \theta=q.

  2. If s=T, we have \operatorname{prog}_{1/4}({\bf{y}})=T and \Pi_{1+\operatorname{prog}_{1/4}}({\bf{y}})=\Pi_{1+\operatorname{prog}_{1/4}}({\bf{y}})={\bf{y}}. Therefore, for all {\bf{y}}\in{\mathbb{R}}^T, all conditions

    \begin{aligned} \operatorname{prog}_0({\bf{g}}_T({\bf{y}};Z)) \le \operatorname{prog}_{1/4}({\bf{y}}),\quad {\bf{g}}_T({\bf{y}};Z) = {\bf{g}}_T\bigl( \Pi_{\operatorname{prog}_{1/4}({\bf{y}})}({\bf{y}});Z \bigr), \\ \operatorname{prog}_0({\bf{g}}_T({\bf{y}};Z)) \le1+\operatorname{prog}_{1/4}({\bf{y}}),\quad {\bf{g}}_T({\bf{y}};Z) = {\bf{g}}_T\bigl( \Pi_{1+\operatorname{prog}_{1/4}({\bf{y}})}({\bf{y}});Z \bigr) \end{aligned}

    holds in both cases of Z=0 and Z=1. This verifies equations (4.1) and (4.1).

Hence, we conclude that {\bf{g}}_T is a probability-q zero-chain.

Proof

Proof

6RE9

We let {\bf{y}}:={\bf U}^\top\rho_R({\bf{x}}/b). By the chain rule, the stochastic oracle defined in equation (?) has the form of

\mathsf O_{f_{{\bf U}}}({\bf{x}};Z) = \frac{a}{b} D\rho_R\left(\frac{{\bf{x}}}{b}\right)^\top {\bf U}\nabla\bar F_T({\bf{y}};Z) +\frac{a\mu{\bf{x}}}{b^2}

Combining the above result with equations (?), (?), (Proof), and (?), we obtain

\mathsf O_{f_{{\bf U}}}({\bf{x}};Z) = \nabla f_{{\bf U}}({\bf{x}}) + \frac{a}{b}\left(\frac Zq-1\right) D\rho_R\left(\frac{{\bf{x}}}{b}\right)^\top {\bf U}\nabla Q_T({\bf{y}}).

Unbiasedness. Since Z\sim\operatorname{Bernoulli}(q) with q\in(0,1], taking the expectation on equation (Proof) achieves

{\mathbb{E}}_Z[\mathsf O_{f_{{\bf U}}}({\bf{x}};Z)] = \nabla f_{{\bf U}}({\bf{x}}).

Bound on the pth central moment. Combing equations (Proof), (Proof), and (?), we have

\begin{aligned} \begin{split} \left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z)-\nabla f_{{\bf U}}({\bf{x}}) \right\| &\le \frac{a}{b} \left|\frac Zq-1\right| \left\| D\rho_R({\bf{x}}/b) \right\| \left\| {\bf U} \right\| \left\| \nabla Q_T({\bf{y}}) \right\| \\ &\le 16\epsilon\cdot249 \left|\frac Zq-1\right| = 3984\epsilon \left|\frac Zq-1\right|, \end{split} \end{aligned}

where we use the setting \left\| {\bf U} \right\|=1 and a/b=16\epsilon from equation (3.4). Since Z\sim\operatorname{Bernoulli}(q) and p\in(1,2], we have

\begin{aligned} {\mathbb{E}}_Z\left[\left|\frac Zq-1\right|^p\right] &= (1-q)^p q^{1-p}+1-q\le q^{1-p}+1 \le2q^{1-p}, \end{aligned}

where the inequalities follow from 0<q\le1 and p>1. Consequently, combining equations (?) and (?) imply

{\mathbb{E}}_Z\left[ \left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z)-\nabla f_{{\bf U}}({\bf{x}}) \right\|^p \right] \le 2(3984\epsilon)^p q^{1-p}.

If q<1, then \sigma>0 and the definition of q gives q^{p-1}=2(3984)^p\epsilon^p\sigma^{-p}. Substituting this identity into equation (?), we obtain

\begin{aligned} {\mathbb{E}}_Z\left[ \left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z)-\nabla f_{{\bf U}}({\bf{x}}) \right\|^p \right] \le 2(3984\epsilon)^p q^{1-p}=2(3984\epsilon)^p\cdot\frac{\sigma^p}{2(3984)^p\epsilon^p}=\sigma^p. \end{aligned}

If q=1, then Z=1 almost surely. Hence, equation (?) implies

{\mathbb{E}}_Z\left[ \left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z)-\nabla f_{{\bf U}}({\bf{x}}) \right\|^p \right] =0\le\sigma^p.

Combining equation (?) and equation (?), we conclude that for every {\bf{x}}\in{\mathbb{R}}^d,

{\mathbb{E}}_Z\left[ \left\| \mathsf O_{f_{{\bf U}}}({\bf{x}};Z)-\nabla f_{{\bf U}}({\bf{x}}) \right\|^p \right] \le\sigma^p.
Proof

Proof

H8P1

According to Lemma 4.2, we know that the gradient evaluation {\bf{g}}_T({\bf{y}};Z) is a probability-q zero-chain. Moreover, the definition of \rho implies \left\| \rho_R({\bf{x}}/b) \right\|\le R for every {\bf{x}}\in{\mathbb{R}}^d. Note that the proof of Lemma 5 of [arjevani2023lower] only require the oracle is a probability zero-chain, so that we can use this lemma with K=1 to achieve

\operatorname{prog}_{{1}/{4}}\big({\bf U}^{\top}\rho_{R}({\bf{x}}_{\mathsf{A}}^{(t)}/b) \big)< T

for all t\le(T-\log(2/\delta))/(2q). Hence, we have

\left| [{\bf U}^\top\rho_R({\bf{x}}^{(t)}/b)]_T \right| \le\frac14.

According to Lemma 3.3, we obtain

\left\| \nabla H_{{\bf U},\mu}({\bf{x}}^{(t)}/b) \right\| \ge\frac14.

Based on the definition

\nabla f_{{\bf U}}({\bf{x}}) = \frac ab\nabla H_{{\bf U},\mu}({\bf{x}}/b),

with a/b=16\epsilon, it follows that

\left\| \nabla f_{{\bf U}}({\bf{x}}^{(t)}) \right\| \ge4\epsilon.

This completes the proof.

Proof

Proof

IM2G

We follow the setting of a, b, \mu, and R by equations (3.4) and (?). For \epsilon\le\hat c L_1\Delta with \hat c=1/122880, we let

T:= \left\lfloor\frac{\Delta}{2a}\right\rfloor = \left\lfloor \frac{L_1\Delta}{15360\epsilon} \right\rfloor \geq 8.

Fix p\in(1,2]. According to Lemma 3.4(a) and (b), the function f_{{\bf U}} is (L_0,L_1)-smooth and satisfies

f_{{\bf U}}({\bf{0}}) -\inf_{{\bf{x}}\in{\mathbb{R}}^d}f_{{\bf U}}({\bf{x}}) \le2aT\le\Delta,

where the last inequality follows from equation (Proof). Thus, we conclude f_{{\bf U}}\in\mathcal F(L_0,L_1,\Delta).

Following the setting of q in equation (4.3), Lemma 4.3 implies the oracle \mathsf O_{f_{{\bf U}}}({\bf{x}};Z) satisfies is unbiased and has pth bounded central moment.

We let {\bf U}\sim\operatorname{Unif}(\operatorname{St}(d,T)) and use P_f to denote the induced distribution of f_{{\bf U}}. Taking sufficient large d such that

d\ge \left\lceil T+32R^2\log\left(\frac{4T^2}{q}\right) \right\rceil,

which satisfies equation (?) with \delta=1/2. Applying Lemma 4.4, we know that for all randomized algorithm \mathsf A\in\mathcal A_{\mathrm{rand}} and t\le N with

N:= \left\lfloor \frac{T-\log4}{2q} \right\rfloor,

we have

\mathbb P\left( \left\| \nabla f_{{\bf U}}({\bf{x}}^{(t)}) \right\| \ge4\epsilon \right) \ge\frac12.

This implies

\begin{aligned} {\mathbb{E}}\left[\left\| \nabla f_{{\bf U}}({\bf{x}}^{(t)}) \right\|\right] \ge 4\epsilon\, \mathbb P\left( \left\| \nabla f_{{\bf U}}({\bf{x}}^{(t)}) \right\| \ge4\epsilon \right)\ge2\epsilon>\epsilon, \end{aligned}

where the expectation and probability are taken over {\bf U}, the oracle randomness, and the internal randomness of \mathsf A. Since the distribution P_f is supported on \mathcal F(L_0,L_1,\Delta) and the oracle \mathsf O_{f_{{\bf U}}} condition equation (4.3), the distributional complexity defined in equation (?) implies

m_{\epsilon}^{\mathrm{rand}} (L_0,L_1,\Delta,\sigma,p)>N.

We now provide the lower bound for N. According to equation (?), we have

\begin{aligned} N\ge\frac{T-\log4}{2q}-1=\frac{T-\log4-2q}{2q}\ge\frac{T-4}{2q}\ge\frac{T}{4q}, \end{aligned}

where the first inequality uses \lfloor x\rfloor\ge x-1, and the last two inequalities follow from q\le1, \log4<2, and T\ge8. Moreover, by equation (Proof) and L_1\Delta/(15360\epsilon)\ge8, we obtain

T \ge \frac{L_1\Delta}{15360\epsilon}-1 \ge \frac{L_1\Delta}{30720\epsilon}.

Additionally, the definition of q implies

\begin{aligned} \begin{split} \frac1q =\max\left\{ 1,\, 2^{-\frac1{p-1}} (3984)^{-\frac p{p-1}}\cdot \frac{\sigma^{\frac p{p-1}}}{\epsilon^{\frac p{p-1}}} \right\} \ge \frac{ 2^{-\frac1{p-1}} (3984)^{-\frac p{p-1}} }{2} \left(\frac{\sigma^{\frac p{p-1}}}{\epsilon^{\frac p{p-1}}}+1 \right), \end{split} \end{aligned}

where the inequality follows from \max\{u,v\}\ge(u+v)/2 for u,v\ge0 and 2^{-\frac1{p-1}}(3984)^{-\frac p{p-1}}\le1. For any given p\in(1,2], combining equations (?), (?), (?), and (?) achieves

\begin{aligned} m_{\epsilon}^{\mathrm{rand}} (L_0,L_1,\Delta,\sigma,p) \ge \Omega\left(\left( \frac{\sigma^{\frac p{p-1}}}{\epsilon^{\frac p{p-1}}}+1 \right) \frac{L_1\Delta}{\epsilon}\right). \end{aligned}

This finishes the proof.

Proof

Proof

2DBK

For any given p\in(1,2], recall that [liu2025nonconvex, Theorem 3.3] have provided a lower bound for standard smooth optimization under the p-BCM condition, i.e., there exists \tilde c>0 such that for all L_0,\Delta,\sigma>0, and \epsilon\le\tilde c\sqrt{L_0\Delta}, it holds

m_{\epsilon}^{\mathrm{rand}}(L_0,0,\Delta,\sigma,p) \ge \Omega\left(\bigg(\frac{\sigma^{\frac{p}{p-1}}}{\epsilon^{\frac{p}{p-1}}}+1 \bigg)\frac{L_0\Delta}{\epsilon^2}\right),

Since \mathcal F(L_0,0,\Delta)\subset\mathcal F(L_0,L_1,\Delta), the above lower bound implies

m_{\epsilon}^{\mathrm{rand}}(L_0,L_1,\Delta,\sigma,p) \ge \Omega\left(\bigg(\frac{\sigma^{\frac{p}{p-1}}}{\epsilon^{\frac{p}{p-1}}}+1 \bigg)\frac{L_0\Delta}{\epsilon^2}\right).

Combining the result of Theorem 4.5, we know that for any given p\in(1,2], there exists numerical constant c=\min\{\hat c, \tilde c\} such that for all L_0,L_1,\Delta,\sigma>0, and \epsilon\le\hat c\min\{\sqrt{L_0\Delta},L_1\Delta\}, it holds

\begin{aligned} m_{\epsilon}^{\mathrm{rand}} (L_0,L_1,\Delta,\sigma,p) \ge \Omega\left(\bigg(\frac{\sigma^{\frac{p}{p-1}}}{\epsilon^{\frac{p}{p-1}}}+1\bigg)\bigg(\frac{L_0\Delta}{\epsilon^2}+\frac{L_1\Delta}{\epsilon}\bigg)\right). \end{aligned}

This finishes the proof.

Lemma

Lemma C.1

WPFI

If f is (L_0,L_1)-smooth, then it holds

\begin{aligned} f({\bf{y}}) \leq f({\bf{x}}) + \left\langle \nabla f({\bf{x}}), {\bf{y}}-{\bf{x}} \right\rangle + \frac{L_0+L_1\left\| \nabla f({\bf{x}}) \right\|}{2}\left\| {\bf{y}}-{\bf{x}} \right\|^2 \end{aligned}

for all {\bf{x}},{\bf{y}}\in{\mathbb{R}}^d with \left\| {\bf{y}}-{\bf{x}} \right\|\leq 1/L_1.

Lemma

Lemma C.2

8T1P

For all \gamma \geq 0, {\bf{g}},{\bf{h}}\in{\mathbb{R}}^d, it holds

\begin{aligned} \langle {\bf{h}},\varphi_\gamma({\bf{g}})\rangle \ge \|{\bf{h}}\|-2\|{\bf{g}}-{\bf{h}}\|-\gamma. \end{aligned}
Proof of Lemma C.2

If {\bf{g}}={\bf{0}} and \gamma=0, the result is trivial. Hence, we focus on the case of \|{\bf{g}}\|+\gamma>0. According to Cauchy–Schwarz inequality and triangle inequality, we achieve

\begin{aligned} \langle {\bf{h}},\varphi_\gamma({\bf{g}})\rangle &=\frac{\|{\bf{g}}\|^2+\langle {\bf{h}}-{\bf{g}},{\bf{g}}\rangle}{\|{\bf{g}}\|+\gamma}\\ &\ge \frac{\|{\bf{g}}\|^2}{\|{\bf{g}}\|+\gamma} -\frac{\|{\bf{h}}-{\bf{g}}\|\cdot\|{\bf{g}}\|}{\|{\bf{g}}\|+\gamma}\\ &\ge \|{\bf{g}}\|-\gamma-\|{\bf{h}}-{\bf{g}}\|\\ &\ge \|{\bf{h}}\|-\gamma-2\|{\bf{h}}-{\bf{g}}\|. \end{aligned}

This finishes the proof.

Lemma

Lemma C.3

CBUZ

For Algorithm ?, it holds that

\begin{aligned} {\mathbb{E}}[\left\| {\bf{g}}_t-\nabla f({\bf{x}}_t) \right\|] \leq \frac{2 \sqrt{2}\sigma}{b^{1-1 / p}} \end{aligned}
Proof of Lemma C.3

Recall that the stochastic gradient evaluation in the algorithm is

\begin{aligned} {\bf{g}}_t = \frac{1}{b}\sum_{i=1}^b {\bf{g}}({\bf{x}}_t;Z_{t,i}), \qquad\text{where}~~ Z_{t,1},\dots,Z_{t,b}\overset{\mathrm{i.i.d}}\sim {\mathcal{Z}}. \end{aligned}

Hence, it holds

\begin{aligned} \begin{split} & \mathbb{E}\left[\left\| {\bf{g}}_t-\nabla f({\bf{x}}_t) \right\|\right] \\ = & \mathbb{E}\left[\left\| \sum_{i=1}^{b} \frac{1}{b}\left({\bf{g}}({\bf{x}}_t;Z_{t,i})-\nabla f({\bf{x}}_t)\right) \right\|\right] \\ \leq & 2 \sqrt{2} \mathbb{E}\left [\left(\frac{1}{b^{p}} \sum_{i=1}^{b}\left\| {\bf{g}}({\bf{x}}_t;Z_{t,i})-\nabla f({\bf{x}}_i) \right\|^{p}\right)^{1 / p}\right] \\ = &\frac{2 \sqrt{2}}{b} \mathbb{E}\left[\left(\sum_{i=1}^{b}\left\| {\bf{g}}({\bf{x}}_t;Z_{t,i})-\nabla f({\bf{x}}_t) \right\|^{p}\right)^{1 / p}\right]\\ \leq & \frac{2 \sqrt{2}}{b} \left(\mathbb{E}\left[\sum_{i=1}^{b}\left\| {\bf{g}}({\bf{x}}_t;Z_{t,i})-\nabla f({\bf{x}}_t) \right\|^{p}\right]\right)^{1 / p}\\ \leq & \frac{2 \sqrt{2}}{b} \cdot b^{1 / p} \sigma= \frac{2 \sqrt{2}\sigma}{b^{1-1 / p}}, \end{split} \end{aligned}

where the first inequality is based on Lemma 4.3 of [liu2025nonconvex], second inequality is based on Jensen’s inequality, and the last inequality is based on p-BCM assumption.

Source and licence

The text on this page is from arXiv:2610.21692v1 by Yanpeng Liang and Luo Luo, under the licence CC BY 4.0. The corpus made these changes: it read the items out of the LaTeX source, it expanded the macros of the paper, it put each proof after its statement, and it converted the text to HTML. The formulas are the TeX of the paper, and KaTeX renders them. A formula or a reference can look different from the PDF. The page is published under CC BY 4.0.

The record of each item is in analysis/objects/2610/2610.21692.jsonl.