A Convergence Analysis of Gradient Descent for Deep Linear Neural Networks

Abstract: We analyze speed of convergence to global optimum for gradient descent training a deep linear neural network (parameterized as $x \mapsto W_N W_{N-1} \cdots W_1 x$) by minimizing the $\ell_2$ loss over whitened data. Convergence at a linear rate is guaranteed when the following hold: (i) dimensions of hidden layers are at least the minimum of the input and output dimensions; (ii) weight matrices at initialization are approximately balanced; and (iii) the initial loss is smaller than the loss of any rank-deficient solution. The assumptions on initialization (conditions (ii) and (iii)) are necessary, in the sense that violating any one of them may lead to convergence failure. Moreover, in the important case of output dimension 1, i.e. scalar regression, they are met, and thus convergence to global optimum holds, with constant probability under a random initialization scheme. Our results significantly extend previous analyses, e.g., of deep linear residual networks (Bartlett et al., 2018).

\section{Approximate Balancedness and Deficiency Margin Under Customary Initialization} \label{app:balance_margin_init}

Two assumptions concerning initialization facilitate our main convergence result (Theorem~\ref{theorem:converge}):
\emph{(i)}~the initial weights $W_1(0),\ldots,W_N(0)$ are approximately balanced (see Definition~\ref{def:balance});
and \emph{(ii)}~the initial end-to-end matrix~$W_{1:N}(0)$ has positive deficiency margin with respect to the target~$\Phi$ (see Definition~\ref{def:margin}).
The current appendix studies the likelihood of these assumptions being met under customary initialization of random (layer-wise independent) Gaussian perturbations centered at zero.

For approximate balancedness we have the following claim, which shows that it becomes more and more likely the smaller the standard deviation of initialization is:
\vspace{1mm}
\begin{claim}
\label{claim:balance_init}
Assume all entries in the matrices $W_j\in\R^{d_j\times{d}_{j-1}}$, $j=1,\ldots,N$, are drawn independently at random from a Gaussian distribution with mean zero and standard deviation~$s>0$.
Then, for any~$\delta>0$, the probability of $W_1,\ldots,W_N$ being $\delta$-balanced is at least~$\max\{0,1-10\delta^{-2}Ns^4d_{max}^3\}$, where $d_{max}:=\max\{d_0,\ldots,d_N\}$.
\end{claim}
\vspace{-3mm}
\begin{proof}
See Appendix~\ref{app:proofs:balance_init}.
\end{proof}

In terms of deficiency margin, the claim below treats the case of a single output model (scalar regression), and shows that if the standard deviation of initialization is sufficiently small, with probability close to~$0.5$, a deficiency margin will be met.
However, for this deficiency margin to meet a chosen threshold~$c$, the standard deviation need be sufficiently large.
\vspace{1mm}
\begin{claim}
\label{claim:margin_init}
There is a constant $C_1 > 0$ such that the following holds. Consider the case where~$d_N=1$, $d_0\geq20$,\note{
	The requirement $d_0\geq20$ is purely technical, designed to simplify expressions in the claim.
} and suppose all entries in the matrices $W_j\in\R^{d_j\times{d}_{j-1}}$, $j=1,\ldots,N$, are drawn independently at random from a Gaussian distribution with mean zero, whose standard deviation~$s>0$ is small with respect to the target, \ie~$s\leq \| \Phi\|_F^{1/N} \big/ (10^5 d_0^3d_1 \cdots d_{N-1} C_1)^{1/(2N)}$.
Then, for any~$c$ with $0<c\leq  \norm{\Phi}_{F} \big/ \big(10^{5}d_0^{3}C_1(C_{1}N)^{2N}\big)$, the probability of the end-to-end matrix~$W_{1:N}$ having deficiency margin~$c$ with respect to~$\Phi$ is at least~$0.49$ if:\,\footnote{
	The probability $0.49$ can be increased to any $p < 1/2$ by increasing the constant $10^5$ in the upper bounds for~$s$ and~$c$.
}
\footnote{
	It is not difficult to see that the latter threshold is never greater than the upper bound for~$s$, thus sought-after standard deviations always exist.
}
$$s\geq{c}^{1/(2N)}\cdot\big(C_{1}N\norm{\Phi}_F^{1/(2N)}/(d_1\cdots{d}_{N-1})^{1/(2N)}\big)
\text{~.}$$
\end{claim}
\vspace{-3mm}
\begin{proof}
See Appendix~\ref{app:proofs:margin_init}.
\end{proof}

\section{Convergence Failures} \label{app:fail}

In this appendix we show that the assumptions on initialization facilitating our main convergence result (Theorem~\ref{theorem:converge})~---~approximate balancedness and deficiency margin~---~are both necessary, by demonstrating cases where violating each of them leads to convergence failure.
This accords with widely observed empirical phenomena, by which successful optimization in deep learning crucially depends on careful initialization (\cf~\citet{sutskever2013importance}).

Claim~\ref{claim:diverge_balance} below shows\note{
	For simplicity of presentation, the claim treats the case of even depth and uniform dimension across all layers.
	It can easily be extended to account for arbitrary depth and input/output/hidden dimensions.
}
that if one omits from Theorem~\ref{theorem:converge} the assumption of approximate balancedness at initialization, no choice of learning rate can guarantee convergence:
\vspace{1mm}
\begin{claim}
\label{claim:diverge_balance}
Assume gradient descent with some learning rate~$\eta>0$ is a applied to a network whose depth~$N$ is even, and whose input, output and hidden dimensions $d_0,\ldots,d_N$ are all equal to some $d\in\N$.
Then, there exist target matrices~$\Phi$ such that the following holds.
For any~$c$ with $0<c< \sigma_{min}(\Phi)$, there are initializations for which the end-to-end matrix~$W_{1:N}(0)$ has deficiency margin~$c$ with respect to~$\Phi$, and yet convergence will fail~---~objective will never go beneath a positive constant.
\end{claim}
\vspace{-3mm}
\begin{proof}
See Appendix~\ref{app:proofs:diverge_balance}.
\end{proof}

In terms of deficiency margin, we provide (by adapting Theorem~4 in~\citet{bartlett2018gradient}) a different, somewhat stronger result~---~there exist settings where initialization violates the assumption of deficiency margin, and despite being perfectly balanced, leads to convergence failure, for any choice of learning rate:\note{
	This statement becomes trivial if one allows initialization at a suboptimal stationary point, \eg~$W_j(0)=0,~j=1,\ldots,N$.
	Claim~\ref{claim:diverge_margin} rules out such trivialities by considering only non-stationary initializations.
}
\vspace{1mm}
\begin{claim}
\label{claim:diverge_margin}
Consider a network whose depth~$N$ is even, and whose input, output and hidden dimensions $d_0,\ldots,d_N$ are all equal to some $d\in\N$.
Then, there exist target matrices~$\Phi$ for which there are non-stationary initializations $W_1(0),\ldots,W_N(0)$ that are $0$-balanced, and yet lead gradient descent, under any learning rate, to fail~---~objective will never go beneath a positive constant.
\end{claim}
\vspace{-3mm}
\begin{proof}
See Appendix~\ref{app:proofs:diverge_margin}.
\end{proof}

\section{Implementation Details} \label{app:impl}

Below we provide implementation details omitted from our experimental report (Section~\ref{sec:exper}).

The platform used for running the experiments is PyTorch \citep{paszke2017automatic}.
For compliance with our analysis, we applied PCA whitening to the numeric regression dataset from UCI Machine Learning Repository.
That is, all instances in the dataset were preprocessed by an affine operator that ensured zero mean and identity covariance matrix.
Subsequently, we rescaled labels such that the uncentered cross-covariance matrix~$\Lambda_{yx}$ (see Section~\ref{sec:prelim}) has unit Frobenius norm (this has no effect on optimization other than calibrating learning rate and standard deviation of initialization to their conventional ranges).
With the training objective taking the form of Equation~\eqref{eq:loss_whitened}, we then computed~$c$~---~the global optimum~---~in accordance with the formula derived in Appendix~\ref{app:whitened}.
In our experiments with linear neural networks, balanced initialization was implemented with the assignment written in step~\emph{(iii)} of Procedure~\ref{proc:balance_init}.
In the non-linear network experiment, we added, for each $j\in\{1,\ldots,N-1\}$, a random orthogonal matrix to the right of~$W_j$, and its transpose to the left of~$W_{j+1}$~---~this assignment maintains the properties required from balanced initialization (see Footnote~\ref{note:balance_init_props}).
During all experiments, whenever we applied grid search over learning rate, values between $10^{-4}$ and~$1$ (in regular logarithmic intervals) were tried.

\section{Deferred Proofs} \label{app:proofs}

We introduce some additional notation here
in addition to the notation specified in Section~\ref{sec:prelim}. We use~$\norm{A}_\sigma$ to denote the spectral norm (largest singular value) of a matrix~$A$, and sometimes~$\norm{\vv}_2$ as an alternative to~$\norm{\vv}$~---~the Euclidean norm of a vector~$\vv$.
Recall that for a matrix $A$, $vec(A)$ is its vectorization in column-first order.
We let $F(\cdot)$ denote the cumulative distribution function of the standard normal distribution, \ie~$F(x) = \int_{-\infty}^x\frac{1}{\sqrt{2\pi}}e^{-\frac12u^2}du$ ($x\in\R$).

To simplify the presentation we will oftentimes use~$W$ as an alternative (shortened) notation for~$W_{1:N}$~---~the end-to-end matrix of a linear neural network.
We will also use~$L(\cdot)$ as shorthand for~$L^1(\cdot)$~---~the loss associated with a (directly parameterized) linear model, \ie~$L(W) := \frac12\norm{W-\Phi}_F^2$.
Therefore, in the context of gradient descent training a linear neural network, the following expressions all represent the loss at iteration $t$:
\[
\ell(t)=L^N(W_1(t),\ldots,W_N(t))=L^1(W_{1:N}(t)) = L^1(W(t)) = L(W(t)) = \frac12\norm{W(t)-\Phi}_F^2\,.
\]
Also, for weights $W_j\in\R^{d_j\times{d}_{j-1}}$, $j=1,\ldots,{N}$ of a linear neural network, we generalize the notation~$W_{1:N}$, and define $W_{j:j'}:=W_{j'}W_{j'-1}\cdots{W}_j$ for every $1\leq{j}\leq{j'}\leq{N}$.
Note that $W_{j:j'}^{\top}=W_{j}^{\top}W_{j+1}^{\top}\cdots{W}_{j'}^{\top}$.
Then, by a simple gradient calculation, the gradient descent updates (\ref{eq:gd}) can be written as
\begin{equation}
\label{eq:wjupdate}
W_j(t+1) = W_j(t) - \eta W_{j+1:N}^\top(t) \cdot \frac{dL}{dW}(W(t)) \cdot W_{1:j-1}^\top(t) \quad , 1\le j \le N\,,
\end{equation}
where we define $W_{1:0}(t) := I_{d_0}$ and $W_{N+1:N}(t) := I_{d_N}$ for completeness.

Finally, recall the standard definition of the tensor product of two matrices (also known as the Kronecker product):
 for matrices $A \in \BR^{m_A \times n_A}, B \in \BR^{m_B \times n_B}$, their tensor product $A \otimes B \in \BR^{m_A m_B \times n_A n_B}$ is defined as
	$$
	A \otimes B = \left ( \begin{matrix}
	a_{1,1} B& \cdots &a_{1,n_A} B \\
	\vdots & \ddots & \vdots \\
	a_{m_A,1} B & \cdots & a_{m_A,n_A} B
	\end{matrix} \right),
	$$
	where $a_{i,j}$ is the element in the $i$-th row and $j$-th column of $A$.

\subsection{Proof of Claim~\ref{claim:margin_interp}} \label{app:proofs:margin_interp}

\begin{proof}
Recall that for any matrices $A$ and~$B$ of compatible sizes $\sigma_{min}(A+B)\geq\sigma_{min}(A)-\sigma_{max}(B)$, and that the Frobenius norm of a matrix is always lower bounded by its largest singular value (\citet{horn1990matrix}).
Using these facts, we have:
\beas
&&\sigma_{min}(W')=\sigma_{min}\big(\Phi+(W'-\Phi)\big)\geq\sigma_{min}(\Phi)-\sigma_{max}(W'-\Phi) \\[0.5mm]
&&~\qquad\qquad\geq\sigma_{min}(\Phi)-\norm{W'-\Phi}_F\geq\sigma_{min}(\Phi)-\norm{W-\Phi}_F \\[1mm]
&&~\qquad\qquad\geq\sigma_{min}(\Phi)-(\sigma_{min}(\Phi)-c)=c\,.
\eeas
\end{proof}

\subsection{Proof of Lemma~\ref{lemma:descent}} \label{app:proofs:descent}

To prove Lemma \ref{lemma:descent}, we will in fact prove a stronger result, Lemma \ref{lem:remainsmooth} below, which states that for each iteration $t$, in addition to (\ref{eq:descent}) being satisfied, certain other properties are also satisfied, namely:
\emph{(i)} the weight matrices $W_1(t), \ldots, W_N(t)$ are $2\delta$-balanced, and
\emph{(ii)} $W_1(t), \ldots, W_N(t)$ have bounded spectral norms.

\begin{lemma}
  \label{lem:remainsmooth}
  Suppose the conditions of Theorem \ref{theorem:converge} are satisfied. Then for all $t \in \BN \cup \{0\}$,
\begin{enumerate}
\item[($\MA(t)$)] For $1 \leq j \leq N-1$, $\| W_{j+1}^\top(t) W_{j+1}(t) - W_j(t) W_j^\top(t) \|_F \leq 2\delta$.
\item[($\MA'(t)$)] If $t \geq 1$, then for $1 \leq j \leq N-1$,
  \begin{eqnarray}
    \label{eq:sumbalancet}
    &&\| W_{j+1}^\top(t) W_{j+1}(t) - W_j(t) W_j^\top(t) \|_F \nonumber\\
    &\leq& \| W_{j+1}^\top(t-1) W_{j+1}(t-1) - W_j(t-1) W_j^\top(t-1) \|_F \nonumber \\
                                                              && + \eta^2 \left\| \frac{dL^1}{dW} W(t-1)\right\|_F \cdot \left\| \frac{dL^1}{dW} W(t-1) \right\|_\sigma \cdot 4 \cdot (2 \| \Phi\|_F)^{2(N-1)/N}\nonumber.
  \end{eqnarray}
\item[($\MB(t)$)] If $t = 0$, then $\ell(t) \leq \frac 12 \| \Phi\|_F^2$. If $t \geq 1$, then   \begin{equation*}
     \ell(t) \leq
     \ell(t-1)  -\frac \eta 2 \sigma_{min}(W(t-1))^{\frac{2(N-1)}{N}} \left\| \frac{dL^1}{dW} (W(t-1)) \right\|_F^2.\quad
  \end{equation*}
\item[($\MC(t)$)] For $1 \leq j \leq N$, $\| W_j(t) \|_\sigma \leq (4 \| \Phi\|_F)^{1/N}$.
\end{enumerate}
\end{lemma}

First we observe that Lemma \ref{lemma:descent} is an immediate consequence of Lemma \ref{lem:remainsmooth}.
\begin{proof}[Proof of Lemma \ref{lemma:descent}]
  Notice that condition $\MB(t)$ of Lemma \ref{lem:remainsmooth} for each $t \geq 1$ immediately establishes the conclusion of Lemma \ref{lemma:descent} at time step $t-1$.
\end{proof}

\subsubsection{Preliminary lemmas}

We next prove some preliminary lemmas which will aid us in the proof of Lemma \ref{lem:remainsmooth}. The first is a matrix inequality that follows from Lidskii's theorem. For a matrix $A$, let $\Sing(A)$ denote the rectangular diagonal matrix {of the same size,} whose diagonal elements are the singular values of $A$ arranged in non-increasing order (starting from the $(1,1)$ position).

\begin{lemma}[\cite{bhatia_matrix_1997}, Exercise IV.3.5]
  \label{lemma:bhatiaexc}
For any two matrices $A,B$ of the same size, $\| \Sing(A) - \Sing(B) \|_\sigma \leq \| A-B \|_\sigma$ and $\| \Sing(A) - \Sing(B) \|_F \leq \| A-B\|_F$.
\end{lemma}

Using Lemma \ref{lemma:bhatiaexc}, we get:
\begin{lemma}
\label{lem:rearrange}
Suppose $D_1, D_2 \in \BR^{d \times d}$ are non-negative diagonal matrices with non-increasing values along the diagonal and $O \in \BR^{d \times d}$ is an orthogonal matrix. Suppose that $\| D_1 - OD_2O^\top\|_F \leq \ep$, for some $\ep > 0$. Then:
\begin{enumerate}
\item $\| D_1 - OD_1O^\top \|_F \leq 2\ep$.
\item $\| D_1 - D_2 \|_F \leq \ep$.
\end{enumerate}
\end{lemma}
\begin{proof}
  Since $D_1$ and $OD_2O^T$ are both symmetric positive semi-definite matrices, their singular values are equal to their eigenvalues. Moreover, the singular values of $D_1$ are simply its diagonal elements and the singular values of $OD_2O^T$ are simply the diagonal elements of $D_2$. Thus by Lemma \ref{lemma:bhatiaexc} we get that $\|D_1 - D_2 \|_F \leq \| D_1 - OD_2O^T\|_F \leq \ep$. Since the Frobenius norm is unitarily invariant, $\| D_1 - D_2 \|_F = \| OD_1O^T - OD_2O^T \|_F$, and by the triangle inequality it follows that
  \begin{align*}
\| D_1 - OD_1O^T\|_F \leq \| OD_1O^T - OD_2O^T\|_F + \| D_1 - OD_2O^T\|_F \leq 2\ep.
  \end{align*}
\end{proof}

Lemma \ref{lem:boundcommute} below states that if $W_1, \ldots, W_N$ are approximately balanced matrices, \ie~$W_{j+1}^\top W_{j+1} - W_j W_j^\top$ has small Frobenius norm for $1 \leq j \leq N-1$, then we can bound the Frobenius distance between $W_{1:j}^\top W_{1:j}$ and $(W_1^\top W_1)^j$ (as well as between $W_{j:N}W_{j:N}^\top$ and $(W_NW_N^\top)^{N-j+1}$).

\begin{lemma}
\label{lem:boundcommute}
Suppose that $d_N \leq d_{N-1}, d_0 \leq d_1$, and that for some $\nu > 0,M>0$, the matrices $W_j \in \BR^{d_j \times d_{j-1}}$, $1 \leq j \leq N$ satisfy, for $1 \leq j \leq N-1$,
\begin{equation}
\label{eq:pcndiff}
\| W_{j+1}^\top W_{j+1} - W_j W_j^\top \|_F \leq \nu,
\end{equation}
and for $1 \leq j \leq N$, $\| W_j\|_\sigma \leq M$. Then, for $1 \leq j \leq N$,
\begin{equation}
  \label{eq:wj1w}
\| W_{1:j}^\top W_{1:j} - (W_1^\top W_1)^{j} \|_F \leq \frac 32 \nu \cdot M^{2(j-1)} j^2,
\end{equation}
and
\begin{equation}
  \label{eq:wjnw}
\| W_{j:N} W_{j:N}^\top - (W_NW_N^\top)^{N-j+1}\|_F \leq \frac 32 \nu \cdot M^{2(N-j)} (N-j+1)^2.
\end{equation}
Moreover, if $ \sigma_{min}$ denotes the minimum singular value of $W_{1:N}$, $\sigma_{1,min}$ denotes the minimum singular value of $W_1$ and $\sigma_{N,min}$ denotes the minimum singular value of $W_N$, then
\begin{equation}
  \label{eq:singvallb}
   \sigma_{min}^2 - \frac 32 \nu M^{2(N-1)} N^2 \leq
  \begin{cases}
    \sigma_{N,min}^{2N} \quad : \quad d_N \geq d_0.\\
    \sigma_{1,min}^{2N} \quad : \quad d_N \leq d_0.
    \end{cases}
\end{equation}
\end{lemma}
\begin{proof}
For $1 \leq j \leq N$, let us write the singular value decomposition of $W_j$ as $W_j = U_j \Sigma_j V_j^\top$, where $U_j\in \BR^{d_j\times d_j}$ and $V_j\in \BR^{d_{j-1}\times d_{j-1}}$ are orthogonal matrices and $\Sigma_j\in \BR^{d_j\times d_{j-1}}$ is diagonal. We may assume without loss of generality that the singular values of $W_j$ are non-increasing along the diagonal of $\Sigma_j$. Then we can write (\ref{eq:pcndiff}) as
$$
\| V_{j+1}\Sigma_{j+1}^\top \Sigma_{j+1} V_{j+1}^\top - U_j \Sigma_j \Sigma_j^\top U_j^\top \|_F \leq \nu.
$$
Since the Frobenius norm is invariant to orthogonal transformations, we get that
$$
\| \Sigma_{j+1}^\top\Sigma_{j+1} - V_{j+1}^\top U_j \Sigma_j \Sigma_j^\top U_j^\top V_{j+1} \|_F \leq \nu.
$$
By Lemma \ref{lem:rearrange}, we have that $\| \Sigma_{j+1}^\top \Sigma_{j+1} - \Sigma_j \Sigma_j^\top \|_F \leq \nu$ and $\| \Sigma_j \Sigma_j^\top - V_{j+1}^\top U_j \Sigma_j \Sigma_j^\top U_j^\top V_{j+1} \|_F \leq 2\nu$. We may rewrite the latter of these two inequalities as
$$
\| [\Sigma_j \Sigma_j^\top ,V_{j+1}^\top U_j] \|_F = \| [\Sigma_j \Sigma_j^\top, V_{j+1}^\top U_j]U_j^\top V_{j+1} \|_F =\| \Sigma_j \Sigma_j^\top- V_{j+1}^\top U_j \Sigma_j \Sigma_j^\top U_j^\top V_{j+1} \|_F \leq 2\nu.
$$
Note that
$$
W_{j:N} W_{j:N}^\top = W_{j+1:N} U_j\Sigma_j\Sigma_j^\top U_j^\top W_{j+1:N}^\top.
$$
For matrices $A,B$, we have that $\| AB\|_F \leq \| A \|_\sigma \cdot \| B \|_F$. Therefore, for $j+1 \leq i \leq N$, we have that
\begin{eqnarray}
&& \| W_{i:N} U_{i-1} (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j} U_{i-1}^\top W_{i:N}^\top - W_{i+1:N} U_{i} (\Sigma_{i}\Sigma_{i}^\top)^{i-j+1} U_{i}^\top W_{i+1:N}^\top\|_F \nonumber\\
& =& \| W_{i+1:N}U_{i} \left(\Sigma_{i}V_{i}^\top U_{i-1} (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j} U_{i-1}^\top V_{i}\Sigma_{i}^\top - (\Sigma_{i}\Sigma_{i}^\top)^{i-j+1}  \right) U_{i}^\top W_{i+1:N}^\top \|_F \nonumber\\
&\leq&  \| W_{i+1:N} U_{i}\Sigma_{i}\|_\sigma^2 \cdot \| (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j} + [V_{i}^\top U_{i-1}, (\Sigma_{i-1}\Sigma_{i-1}^\top)^{i-j}]U_{i-1}^\top V_{i} - (\Sigma_{i}^\top\Sigma_{i})^{i-j}\|_F\nonumber\\
& \leq & \| W_{i:N}\|_\sigma^2 \left(  \| [V_{i}^\top U_{i-1}, (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j}]\|_F + \|  (\Sigma_{i-1}\Sigma_{i-1}^\top)^{i-j}  - (\Sigma_{i}^\top \Sigma_{i})^{i-j}\|_F\right)\nonumber.
\end{eqnarray}
Next, we have that
\begin{eqnarray}
 \| [V_{i}^\top U_{i-1}, (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j}]\|_F & \leq & \sum_{k=0}^{i-j-1} \| (\Sigma_{i-1} \Sigma_{i-1}^\top)^k [V_i^\top U_{i-1}, \Sigma_{i-1} \Sigma_{i-1}^\top]( \Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j-1-k}\|_F \nonumber\\
 & \leq & \sum_{k=0}^{i-j-1} \| (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j-1}\|_\sigma \cdot \| [V_i^\top U_{i-1} ,\Sigma_{i-1} \Sigma_{i-1}^\top]\|_F\nonumber\\
 & \leq & (i-j) \| W_{i-1} \|_\sigma^{2(i-j-1)} \cdot 2\nu\nonumber.
\end{eqnarray}
We now argue that $\| (\Sigma_{i-1}\Sigma_{i-1}^\top)^k - (\Sigma_i^\top \Sigma_i)^k\|_F \leq \nu \cdot kM^{2(k-1)}$. Note that $\| \Sigma_{i-1} \Sigma_{i-1}^\top - \Sigma_i^\top \Sigma_i\|_F \leq \nu$, verifying the case $k=1$. To see the general case, since square diagonal matrices commute, we have that
\begin{eqnarray}
\| (\Sigma_{i-1}\Sigma_{i-1}^\top)^k - (\Sigma_i^\top \Sigma_i)^k\|_F&=& \left\| (\Sigma_{i-1}\Sigma_{i-1}^\top - \Sigma_i^\top \Sigma_i)\cdot \left( \sum_{\ell=0}^{k-1} (\Sigma_{i-1}\Sigma_{i-1}^\top)^\ell (\Sigma_i^\top \Sigma_i)^{k-1-\ell}\right)\right\|_F\nonumber\\
& \leq & \nu \cdot \sum_{\ell=0}^{k-1} \| W_{i-1}\|_\sigma^{2\ell} \cdot \| W_i\|_\sigma^{2(k-\ell-1)}\nonumber\\
& \leq & \nu k M^{2(k-1)}\nonumber.
\end{eqnarray}
It then follows that
\begin{eqnarray}
&& \| W_{i:N} U_{i-1} (\Sigma_{i-1} \Sigma_{i-1}^\top)^{i-j} U_{i-1}^\top W_{i:N}^\top - W_{i+1:N} U_{i} (\Sigma_{i}\Sigma_{i}^\top)^{i-j+1} U_{i}^\top W_{i+1:N}^\top\|_F \nonumber\\
& \leq & \| W_{i:N}\|_\sigma^2 \cdot \left((i-j) M^{2(i-j-1)} \cdot 2\nu + \nu(i-j) M^{2(i-j-1)}\right)\nonumber\\
& = & \| W_{i:N}\|_\sigma^2 \cdot 3\nu (i-j) M^{2(i-j-1)}\nonumber.
\end{eqnarray}
By the triangle inequality, we then have that
\begin{eqnarray}
&& \| W_{j:N}W_{j:N}^\top - U_N (\Sigma_N \Sigma_N^\top)^{N-j+1} U_N^\top\|_F\nonumber\\
& \leq & \nu \sum_{i=j+1}^N \| W_{i:N}\|_\sigma^2 \cdot 3(i-j) M^{2(i-j-1)}\nonumber\\
& \leq & 3\nu \sum_{i=j+1}^N (i-j) M^{2(N-i+1)} M^{2(i-j-1)}\nonumber\\
\label{eq:diffN}
&=& 3\nu M^{2(N-j)} \sum_{i=j+1}^N (i-j) \leq \frac 32 \nu \cdot M^{2(N-j)} \cdot (N-j+1)^2.
\end{eqnarray}
By an identical argument (formally, by replacing $W_j$ with $W_{N-j+1}^\top$), we get that
\begin{equation}
\label{eq:diff1}
|| W_{1:j}^\top W_{1:j} - V_1 (\Sigma_1^\top \Sigma_1)^{j} V_1^\top\|_F \leq \frac 32\nu \cdot M^{2(j-1)} \cdot j^2.
\end{equation}
(\ref{eq:diffN}) and (\ref{eq:diff1}) verify (\ref{eq:wjnw}) and (\ref{eq:wj1w}), respectively, so it only remains to verify (\ref{eq:singvallb}).

Letting $j=1$ in (\ref{eq:diffN}), we get
\begin{equation}
  \label{eq:w1nundiff}
\| W_{1:N}W_{1:N}^\top - U_N(\Sigma_N \Sigma_N^\top)^{N} U_N^\top\|_F \leq \frac 32\nu \cdot M^{2(N-1)} \cdot N^2.
\end{equation}
Let us write the eigendecomposition of $W_{1:N}W_{1:N}^\top$ with an orthogonal eigenbasis as $W_{1:N}W_{1:N}^\top =  U  \Sigma  U^\top$, where $\Sigma$ is diagonal with its (non-negative) elements arranged in non-increasing order and $U$ is orthogonal. We can write the left hand side of (\ref{eq:w1nundiff}) as $\|  U  \Sigma  U^\top - U_N(\Sigma_N \Sigma_N^\top)^N U_N^\top\|_F = \|  \Sigma -  U^\top U_N (\Sigma_N \Sigma_N^\top)^N U_N^\top  U\|_F$.

By Lemma \ref{lem:rearrange}, we have that
\begin{equation}
\label{eq:ntildediff}
\|  \Sigma - (\Sigma_N\Sigma_N^\top)^N \|_F \leq \frac 32 \nu M^{2(N-1)} N^2.
\end{equation}
Recall that $W \in \BR^{d_N \times d_0}$. Suppose first that $d_N \leq d_0$. Let $\sigma_{min}$ denote the minimum singular value of $W_{1:N}$ (so that $\sigma_{min}^2$ is the element in the $(d_N, d_N)$ position of $ \Sigma \in \BR^{d_N \times d_N}$), and $\sigma_{N,min}$ denote the minimum singular value (i.e. diagonal element) of $\Sigma_N$, which lies in the $(d_N, d_N)$ position of $\Sigma_N$. (Note that the $(d_N, d_N)$ position of $\Sigma_N \in \BR^{d_N \times d_{N-1}}$ exists since $d_{N-1} \geq d_N$ by assumption.) Then
$$
(\sigma_{N,min}^{2N} -  \sigma_{min}^2)^2 \leq \left(\frac 32 \nu M^{2(N-1)} N^2\right)^2,
$$
so
$$
\sigma_{N,min}^{2N} \geq  \sigma_{min}^2 - {\frac 32 \nu M^{2(N-1)} N^2}.
$$

By an identical argument using (\ref{eq:diff1}), we get that, in the case that $d_0 \leq d_N$, if $\sigma_{1,min}$ denotes the minimum singular value of $\Sigma_1$, then
$$
\sigma_{1,min}^{2N} \geq  \sigma_{min}^2 - \frac{3}{2} \nu M^{2(N-1)} N^2.
$$
(Notice that we have used the fact that the nonzero eigenvalues of $W_{1:N}W_{1:N}^\top$ are the same as the nonzero eigenvalues of $W_{1:N}^\top W_{1:N}$.) This completes the proof of (\ref{eq:singvallb}).
\end{proof}

Using Lemma \ref{lem:boundcommute}, we next show in Lemma \ref{lem:boundindiv} that if {$W_1,\ldots,W_N$ are approximately balanced,}
then an upper bound on $\| W_N \cdots W_1\|_\sigma$ implies an upper bound on $\| W_j\|_\sigma$ for $1 \leq j \leq N$.
\begin{lemma}
\label{lem:boundindiv}
Suppose $\nu, C$ are real numbers satisfying $C > 0$ and $0 < \nu \leq \frac{ C^{2/N}}{30  N^2}$. Moreover suppose that the matrices $W_1, \ldots, W_N$ satisfy the following:
\begin{enumerate}
\item For $1 \leq j \leq N-1$, $\| W_{j+1}^\top W_{j+1} - W_j W_j^\top \|_F \leq \nu$.
\item $\| W_N \cdots W_1 \|_\sigma \leq C$.
\end{enumerate}
Then for $1 \leq j \leq N$, $\| W_j\|_\sigma \leq C^{1/N} \cdot 2^{1/(2N)}$.
\end{lemma}
\begin{proof}
For $1 \leq j \leq N$, let us write the singular value decomposition of $W_j$ as $W_j = U_j \Sigma_j V_j^\top$, where the singular values of $W_j$ are decreasing along the main diagonal of $\Sigma_j$. By Lemma \ref{lem:rearrange}, we have that for $1 \leq j \leq N-1$, $\| \Sigma_{j+1}^\top \Sigma_{j+1} - \Sigma_j \Sigma_j^\top \|_F \leq \nu$, which implies
that $\left|\| \Sigma_{j+1}^\top \Sigma_{j+1}\|_\sigma  -\| \Sigma_j \Sigma_j^\top \|_\sigma \right| \leq \nu$.

Write $M = \max_{1 \leq j \leq N} \| W_j\|_\sigma = \max_{1 \leq j \leq N} \| \Sigma_j \|_\sigma$. By the above we have that $\| \Sigma_j \Sigma_j^\top\|_\sigma \geq M^2 - N\nu$ for $1 \leq j \leq N$.

Let the singular value decomposition of $W_{1:N}$ be denoted by $W_{1:N} = U\Sigma V^\top$, so that $\| \Sigma \|_\sigma \leq C$. Then by (\ref{eq:wjnw}) of Lemma \ref{lem:boundcommute} and Lemma \ref{lem:rearrange} (see also (\ref{eq:ntildediff}), where the same argument was used), we have that
$$
\| \Sigma\Sigma^\top - (\Sigma_N \Sigma_N^\top)^N \|_F\leq \frac 32 \nu M^{2(N-1)}N^2.
$$
Then
\begin{equation}
  \label{eq:sigman1}
\| (\Sigma_N \Sigma_N^\top)^N\|_\sigma \leq \|\Sigma\Sigma^\top\|_\sigma + \frac 32 \nu M^{(2(N-1))}N^2 \leq \| \Sigma \Sigma^\top \|_\sigma + \frac 32 \nu \left( \| \Sigma_N \Sigma_N^\top\|_\sigma + \nu N\right)^{N-1} N^2.
\end{equation}

Now recall that $\nu$ is chosen so that $\nu \leq \frac{ C^{2/N}}{30 \cdot N^2}.$ Suppose for the purpose of contradiction that there is some $j$ such that $\| W_jW_j^\top \|_\sigma > 2^{1/N} C^{2/N}$. Then it must be the case that
\begin{equation}
  \label{eq:2nabove}
  \| \Sigma_N \Sigma_N^\top\|_\sigma > 2^{1/N} C^{2/N} - \nu \cdot N \geq (5/4)^{1/N} C^{2/N} > \nu \cdot 30N^2,
\end{equation}
where we have used that
$$
2^{1/N} - (5/4)^{1/N} \geq \frac{1}{30N}
$$
for all $N \geq 2$, which follows by considering the Laurent series $\exp(1/z) = \sum_{i=1}^\infty \frac{1}{i! z^i}$, which converges in $|z| > 0$ for $z \in \BC$.

We now rewrite inequality (\ref{eq:2nabove}) as
\begin{equation}
  \label{eq:boundepcontr}
\nu \leq \frac{\| \Sigma_N \Sigma_N^\top\|_\sigma}{30N^2}.
\end{equation}
Next, using (\ref{eq:boundepcontr}) and $(1+1/x)^x \leq e$ for all $x > 0$,
\begin{equation}
  \label{eq:sigman2}
 \frac 32 \nu \left( \| \Sigma_N \Sigma_N^\top\|_\sigma + \nu N\right)^{N-1} N^2 \leq \frac{e^{1/30}}{20} \cdot \| \Sigma_N \Sigma_N^\top \|_\sigma^{N} < \frac{e}{20} \cdot \| \Sigma_N \Sigma_N^\top \|_\sigma^{N}.
\end{equation}
Since $\| (\Sigma_N \Sigma_N^\top)^N \|_\sigma = \| \Sigma_N \Sigma_N^\top\|_\sigma^N$, we get by combining (\ref{eq:sigman1}) and (\ref{eq:sigman2}) that
$$
\| \Sigma_N \Sigma_N^\top \|_\sigma < (1-e/20)^{-1/N} \cdot \| \Sigma \Sigma^\top\|_\sigma^{1/N} \leq (1-e/20)^{-1/N} \cdot C^{2/N},
$$
and since $1-e/20 > 1/(5/4)$, it follows that $\| \Sigma_N \Sigma_N^\top\|_\sigma < (5/4)^{1/N} C^{2/N}$, which contradicts (\ref{eq:2nabove}). It follows that for all $1 \leq j \leq N$, $\| W_j W_j^\top \|_\sigma \leq 2^{1/N} C^{2/N}$. The conclusion of the lemma then follows from the fact that $\| W_j W_j^\top\|_\sigma = \| W_j\|_\sigma^2$.
\end{proof}

\subsubsection{Single-Step Descent}

Lemma~\ref{lem:boundgdupdate} below states that if certain conditions on $W_1(t),\ldots,W_N(t)$ are met, the sought-after descent~---~Equation~\eqref{eq:descent}~---~will take place at iteration~$t$.
We will later show (by induction) that the required conditions indeed hold for every~$t$, thus the descent persists throughout optimization.
The proof of Lemma~\ref{lem:boundgdupdate} is essentially a discrete, single-step analogue of the continuous proof for Lemma~\ref{lemma:descent} (covering the case of gradient flow) given in Section~\ref{sec:converge}.
\begin{lemma}
  \label{lem:boundgdupdate}
  Assume the conditions of Theorem~\ref{theorem:converge}.

  Moreover, suppose that for some $t$, the matrices $W_1(t), \ldots, W_N(t)$ and the end-to-end matrix $W(t) := W_{1:N}(t)$ satisfy the following properties:
  \begin{enumerate}
  \item $\| W_j(t)\|_\sigma \leq (4 \| \Phi\|_F)^{1/N}$ for $1 \leq j \leq N$.
  \item $\| W(t) - \Phi\|_\sigma \leq \| \Phi\|_F$.
  \item $\| W_{j+1}^\top(t) W_{j+1}(t) - W_j(t)W_j^\top(t)\|_F \leq 2\delta$ for $1 \leq j \leq N-1$.
  \item ${\sigma}_{min}:={\sigma}_{min}(W(t)) \geq c$.
  \end{enumerate}
  Then, after applying a gradient descent update (\ref{eq:gd}) we have that
  $$
  L(W(t+1)) - L(W(t)) \leq - \frac \eta 2 \sigma_{min}^{2(N-1)/N} \left\| \frac{dL}{dW}(W(t)) \right\|_F^2.
  $$
\end{lemma}
\begin{proof}

  For simplicity write $M = (4 \| \Phi\|_F)^{1/N}$ and $B = \| \Phi\|_F$. We first claim that
{\small\begin{equation}
\label{eq:assumeeta}
\eta \leq \min \left\{ \frac{1}{2M^{N-2}BN}, \frac{\sigma_{min}^{2(N-1)/N}}{24 \cdot 2 M^{3N-4} N^2B}, \frac{\sigma_{min}^{2(N-1)/N}}{24N^2 M^{4(N-1)}}, \frac{\sigma_{min}^{2(N-1)/(3N)}}{ \left(24 \cdot 4M^{6N-8}N^4B^2 \right)^{1/3}}\right\}.
\end{equation}}
Since $c \leq \sigma_{min}$, for (\ref{eq:assumeeta}) to hold it suffices to have
\begin{eqnarray}
\eta &\leq& \min \left\{ \frac{1}{8 \| \Phi\|_F^{(2N-2)/N}  N}, \frac{c^{2(N-1)/N}}{3 \cdot 2^{11}  \| \Phi\|_F^{4(N-1)/N} N^2},\frac{c^{2(N-1)/(3N)}}{3 \cdot 2^6  \left(\| \Phi\|_F^{(8N-8)/N}\right)^{1/3}  N^{4/3}}\right\}\nonumber.
\end{eqnarray}
As the minimum singular value of $\Phi$ must be at least $c$, we must have $c \leq \| \Phi\|_\sigma$. Since then $\frac{c}{\| \Phi\|_F} \leq \frac{c}{\| \Phi\|_\sigma} \leq 1$, it holds that
$$
\frac{c^{2(N-1)/N}}{\| \Phi\|_F^{4(N-1)/N}} \leq \min \left\{ \frac{1}{\| \Phi\|_F^{2(N-1)/N}}, \frac{c^{2(N-1)/(3N)}}{\| \Phi\|_F^{(8N-8)/(3N)}} \right\},
$$
meaning that it suffices to have
$$
\eta \leq \frac{ c^{2(N-1)/N}}{3 \cdot 2^{11} N^2 \| \Phi\|_F^{4(N-1)/N}},
$$
which is guaranteed by (\ref{eq:descent_eta}).

Next, we claim that
\begin{eqnarray}
\label{eq:assumeep}
  2\delta &\leq &\min \left\{ \frac{ c^{2(N-1)/N}}{8 \cdot 2^4N^3 \| \Phi\|_F^{2(N-2)/N}}, \frac{c^{2}}{6 \cdot 2^4  N^2 \| \Phi\|_F^{2(N-1)/N}}\right\}\\
  & \leq &      \min \left\{ \frac{ \sigma_{min}^{2(N-1)/N}}{8N^3M^{2(N-2)}}, \frac{\sigma_{min}^{2}}{6N^2 M^{2(N-1)}}\right\} .\nonumber
\end{eqnarray}
The second inequality above is trivial, and for the first to hold, since $c \le \| \Phi\|_F$, it suffices to take
$$
2\delta \leq \frac{c^2}{128 \cdot N^3 \cdot \| \Phi\|_F^{2(N-1)/N}},
$$
which is guaranteed by the definition of $\delta$ in Theorem \ref{theorem:converge}.

Next we continue with the rest of the proof.
It follows from (\ref{eq:wjupdate}) that\footnote{Here, for matrices $A_1, \ldots, A_K$ such that $A_K A_{K-1} \cdots A_1$ is defined, we write $\prod_{1}^{j=K} A_j := A_K A_{K-1} \cdots A_1$.
}
\begin{eqnarray}
  && W(t+1) - W(t) \nonumber\\
  &=& \prod_{1}^{j=N} \left( W_j(t) - \eta W_{j+1:N}^\top(t) \frac{dL}{dW}(W(t)) W_{1:j-1}^\top(t) \right) - W_{1:N}(t)\nonumber\\
\label{eq:wt1wt}
&=& -\eta \left(\sum_{j=1}^N W_{j+1:N}W_{j+1:N}^\top(t) \frac{dL}{dW}(W(t)) W_{1:j-1}^\top(t) W_{1:j-1}(t)\right) + (\star),
\end{eqnarray}
where $(\star)$ denotes higher order terms in $\eta$. We now bound the Frobenius norm of $(\star)$. To do this, note that since $L(W) = \frac 12 \| W - \Phi\|_F^2$, $\frac{dL}{dW}(W(t)) = W(t) - \Phi$. Then
\begin{eqnarray}
\| (\star) \|_F & \leq & \sum_{k=2}^N \eta^k \cdot M^{k(N-1) + N-k} \cdot \left\| \frac{dL}{dW}(W(t)) \right\|_F \cdot \left\| \frac{dL}{dW}(W(t)) \right\|_\sigma^{k-1} \cdot {N \choose k}\nonumber\\
& \leq & \eta M^{2N-2}N \left\| \frac{dL}{dW}(W(t)) \right\|_F \sum_{k=2}^N \left(\eta M^{N-2} BN\right)^{k-1}\nonumber\\
\label{eq:starnorm}
& \leq & \eta \cdot (2\eta M^{3N-4} N^2B) \cdot \left\| \frac{dL}{dW}(W(t)) \right\|_F,
\end{eqnarray}
where the last inequality uses $\eta M^{N-2} BN \leq 1/2$, which is a consequence of (\ref{eq:assumeeta}). Next, by Lemma \ref{lem:boundcommute} with $\nu = 2\delta$,
{\small\begin{eqnarray}
         && \left\|\sum_{j=1}^N W_{j+1:N}W_{j+1:N}^\top(t) \frac{dL}{dW}(W(t)) W_{1:j-1}^\top(t) W_{1:j-1}(t) \right.\nonumber\\
         &&- \left. \sum_{j=1}^N (W_NW_N^\top)^{N-j} \frac{dL}{dW}(W(t)) (W_1^\top W_1)^{j-1} \right\|_F\nonumber\\
& \leq & \left\|\sum_{j=1}^N (W_{j+1:N}W_{j+1:N}^\top(t) - (W_NW_N^\top)^{N-j}) \frac{dL}{dW}(W(t)) W_{1:j-1}^\top(t) W_{1:j-1}(t) \right\|_F \nonumber\\
& + & \left\| \sum_{j=1}^N (W_NW_N^\top)^{N-j} \frac{dL}{dW}(W(t)) (W_{1:j-1}^\top W_{1:j-1} - (W_1^\top W_1)^{j-1})\right\|_F\nonumber\\
& \leq & \left\| \frac{dL}{dW}(W(t)) \right\|_F \cdot \left(\sum_{j=1}^{N-1} \frac 32 2\delta \cdot M^{2(N-j)}(N-j)^2 M^{2(j-1)} + \sum_{j=2}^N \frac 32 2\delta \cdot M^{2(j-2)}(j-1)^2 M^{2(N-j)}\right)\nonumber\\
& \leq &  \left\| \frac{dL}{dW}(W(t)) \right\|_F \cdot 2\delta N^3 M^{2(N-2)}\nonumber.
\end{eqnarray}}
Next, by standard properties of tensor product, we have that
\begin{eqnarray}
&& vec\left(\sum_{j=1}^N (W_NW_N^\top)^{N-j} \frac{dL}{dW}(W(t)) (W_1^\top W_1)^{j-1} \right)\nonumber\\
&=& \sum_{j=1}^N \left( (W_1^\top W_1)^{j-1} \otimes (W_NW_N^\top)^{N-j} \right)vec \left( \frac{dL}{dW}(W(t))\right)\nonumber.
\end{eqnarray}
Let us write eigenvalue decompositions $W_1^\top W_1 = UDU^\top, W_NW_N^\top = VEV^\top$. Then
\begin{eqnarray}
&& \sum_{j=1}^N \left( (W_1^\top W_1)^{j-1} \otimes (W_NW_N^\top)^{N-j} \right)\nonumber\\
&=& \sum_{j=1}^N \left( UD^{j-1}U^\top \otimes VE^{N-j} V^\top \right)\nonumber\\
&=& (U \otimes V) \left(\sum_{j=1}^N D^{j-1} \otimes E^{N-j}\right) (U \otimes V)^\top\nonumber\\
&=& O\Lambda O^\top\nonumber,
\end{eqnarray}
with $O = U \otimes V$, and $\Lambda = \sum_{j=1}^N D^{j-1} \otimes E^{N-j}$. As $W_1 \in \BR^{d_1 \times d_0}$, and $W_N \in \BR^{d_N \times d_{N-1}}$, then $D \in \BR^{d_0 \times d_0}, E \in \BR^{d_N \times d_N}$, so $\Lambda \in \BR^{d_0d_N \times d_0d_N}$. Moreover note that $\Lambda \succeq D^0 \otimes E^{N-1} + D^{N-1} \otimes E^0 = I_{d_0} \otimes E^{N-1} + D^{N-1} \otimes I_{d_N}$. If $\lambda_D$ denotes the minimum diagonal element of $D$ and $\lambda_E$ denotes the minimum diagonal element of $E$, then the minimum diagonal element of $\Lambda$ is therefore at least $\lambda_D^{N-1} + \lambda_E^{N-1}$. But, it follows from Lemma \ref{lem:boundcommute} (with $\nu = 2\delta$) that
$$
\max\{ \lambda_D^{N}, \lambda_E^{N} \} \geq  \sigma_{min}^2 - \frac 32 2\delta M^{2(N-1)} N^2 \geq  3\sigma_{min}^2/4,
$$
where the second inequality follows from (\ref{eq:assumeep}). Hence the minimum diagonal element of $\Lambda$ is at least $( \sigma_{min}^2/(4/3))^{(N-1)/N} \geq  \sigma_{min}^{2(N-1)/N}/(4/3)$.

It follows as a result of the above inequalities that if we write $E(t) = vec(W(t+1)) - vec(W(t)) + \eta (O\Lambda O^\top) vec\left( \frac{dL}{dW}(W(t)) \right)$, then
\begin{eqnarray}
\| E(t) \|_2 & = &\left\| vec(W(t+1)) - vec(W(t)) + \eta (O\Lambda O^\top) vec\left( \frac{dL}{dW}(W(t)) \right) \right\|_2\nonumber\\
&\leq & \eta  \left\| \frac{dL}{dW}(W(t)) \right\|_F \cdot (2\eta M^{3N-4} N^2B + 2\delta N^3 M^{2(N-2)}) \nonumber.
\end{eqnarray}
Then we have
{\small\begin{eqnarray}
&& L(W(t+1)) - L(W(t)) \nonumber\\
& \leq & vec\left( \frac{d}{dW} L(W(t)) \right)^\top vec \left( W(t+1) - W(t) \right) + \frac 12 \| W(t+1) - W(t) \|_F^2\nonumber\\
         & = & \eta \left( - vec\left(\frac{d}{dW} L(W(t))\right)^\top (O\Lambda O^\top) vec \left(\frac{d}{dW} L(W(t)) \right) + \frac{1}{\eta}vec\left(\frac{d}{dW} L(W(t))\right)^\top E(t) \right) \nonumber\\
         && +  \frac 12 \| W(t+1) - W(t) \|_F^2\nonumber\\
         & \leq & \eta \left(-\left\| \frac{d}{dW} L(W(t)) \right\|_F^2 \cdot \frac{ \sigma_{min}^{2(N-1)/N}}{4/3} + \left\| \frac{d}{dW} L(W(t)) \right\|_F^2 \cdot \left(2\eta M^{3N-4} N^2B + 2\delta N^3 M^{2(N-2)}\right) \right) \nonumber\\
         && + \frac 12 \| W(t+1) - W(t) \|_F^2\nonumber,
\end{eqnarray}}
where the first inequality follows since $L(W) = \frac 12 \| W - \Phi\|_F^2$ is $1$-smooth as a function of $W$. Next, by (\ref{eq:wt1wt}) and (\ref{eq:starnorm}),
\begin{eqnarray}
&& \| W(t+1) - W(t) \|_F^2 \nonumber\\
& \leq & 2\eta^2 \cdot \left(N M^{2(N-1)} \cdot  \left\| \frac{dL}{dW}(W(t)) \right\|_F\right)^2 + 2\eta^2 \cdot (2\eta M^{3N-4} N^2B)^2 \cdot  \left\| \frac{dL}{dW}(W(t)) \right\|_F^2\nonumber\\
\label{eq:boundwtwt1}
& = &2 \eta^2  \left\| \frac{dL}{dW}(W(t)) \right\|_F^2 \cdot \left( N^2M^{4(N-1)} + (4\eta^2 M^{6N-8} N^4B^2)\right).
\end{eqnarray}
Thus
\begin{eqnarray}
&& L(W(t+1)) - L(W(t))\nonumber\\
  & \leq & \eta \cdot  \left\| \frac{dL}{dW}(W(t)) \right\|_F^2 \cdot \left( -\frac{\sigma_{min}^{2(N-1)/N}}{4/3} + 2\eta M^{3N-4} N^2B + 2\delta N^3 M^{2(N-2)} \right.\nonumber\\
  && \left.  + \eta \cdot (N^2M^{4(N-1)} + 4\eta^2 M^{6N-8}N^4B^2) \right)\nonumber.
\end{eqnarray}
By (\ref{eq:assumeeta}, \ref{eq:assumeep}), which bound $\eta, 2\delta$, respectively, we have that
{\small\begin{eqnarray}
\label{eq:descentl}
  &&L(W(t+1)) - L(W(t))\nonumber\\
  & \leq& \eta \cdot \left\| \frac{dL}{dW}(W(t)) \right\|_F^2 \cdot \left( -\frac{\sigma_{min}^{2(N-1)/N}}{4/3} + \frac{\sigma_{min}^{2(N-1)/N}}{24} + \frac{\sigma_{min}^{2(N-1)/N}}{8} +  \frac{\sigma_{min}^{2(N-1)/N}}{24} + \frac{\sigma_{min}^{2(N-1)/N}}{24} \right)\nonumber\\
  &=& - \frac 12 \sigma_{min}^{2(N-1)/N} \eta \left\| \frac{dL}{dW}(W(t)) \right\|_F^2.
\end{eqnarray}}
\end{proof}

\subsubsection{Proof of Lemma \ref{lem:remainsmooth}}

\begin{proof}[Proof of Lemma \ref{lem:remainsmooth}]

  We use induction on $t$, beginning with the base case $t=0$. Since the weights $W_1(0), \ldots, W_N(0)$ are $\delta$-balanced, we get that $\MA(0)$ holds automatically. To establish $\MB(0)$, note that since $W_{1:N}(0)$ has deficiency margin $c > 0$ with respect to $\Phi$, we must have $\| W_{1:N}(0) - \Phi \|_F \leq \sigma_{min}(\Phi) \leq \|\Phi\|_F$, meaning that $L^1(W_{1:N}(0)) \leq \frac 12 \| \Phi\|_F^2$.

Finally, by $\MB(0)$, which gives $\| W(0) - \Phi\|_F \leq \| \Phi\|_F$, we have that
\begin{equation}
  \label{eq:boundprod0}
\| W(0) \|_\sigma \leq \| W(0) \|_F \leq \| W(0) - \Phi\|_F + \| \Phi\|_F \leq 2 \| \Phi\|_F.
\end{equation}
To show that the above implies $\MC(0)$, we use condition $\MA(0)$ and Lemma \ref{lem:boundindiv} with $C = 2 \| \Phi\|_F$ and $\nu = 2\delta$. By the definition of $\delta$ in Theorem \ref{theorem:converge} and since $c \leq \| \Phi\|_F$, we have that
\begin{equation}
  \label{eq:epltc}
  2\delta \leq \frac{c^2}{128 \cdot N^3 \cdot \| \Phi\|_F^{2(N-1)/N}} = \frac{\| \Phi\|_F^{2/N}}{128N^3} \cdot \frac{c^2}{\| \Phi\|_F^2}< \frac{\| \Phi\|_F^{2/N}}{30 N^2},
\end{equation}
as required by Lemma \ref{lem:boundindiv}.
As $\MA(0)$ and (\ref{eq:boundprod0}) verify the preconditions 1.~and 2., respectively, of Lemma \ref{lem:boundindiv}, it follows that for $1 \leq j \leq N$, $\| W_j(t) \|_\sigma \leq (2 \| \Phi\|_F)^{1/N} \cdot 2^{1/(2N)} < (4 \| \Phi\|_F)^{1/N}$, verifying $\MC(0)$ and completing the proof of the base case.

The proof of Lemma \ref{lem:remainsmooth} follows directly from the following inductive claims.
\begin{enumerate}
\item $\MA(t), \MB(t), \MC(t) \Rightarrow \MB(t+1)$. To prove this, we use Lemma \ref{lem:boundgdupdate}. We verify first that the preconditions hold.
  First, $\MC(t)$ immediately gives condition 1.~of Lemma \ref{lem:boundgdupdate}. By $\MB(t)$, we have that $\| W(t) - \Phi\|_\sigma \leq \| W(t) - \Phi\|_F \leq \| \Phi\|_F$, giving condition 2.~of Lemma \ref{lem:boundgdupdate}. $\MA(t)$ immediately gives condition 3.~of Lemma \ref{lem:boundgdupdate}. Finally, by $\MB(t)$, we have that $L^N(W_1(t), \ldots, W_N(t)) \leq L^N(W_1(0), \ldots, W_N(0))$, so $\sigma_{min}(W_{1:N}(t)) \geq c$ by Claim \ref{claim:margin_interp}. This verifies condition 4.~of Lemma \ref{lem:boundgdupdate}. Then Lemma \ref{lem:boundgdupdate} gives that $L^N(W_{1}(t+1), \ldots, W_N(t+1)) \leq L^N(W_1(t), \ldots, W_N(t)) - \frac 12 \sigma_{min}(W(t))^{2(N-1)/N} \eta \left\| \frac{dL}{dW}(W(t)) \right\|_F^2$, establishing $\MB(t+1)$.

\item $\MA(0), \MA'(1), \ldots, \MA'(t), \MA(t), \MB(0), \ldots, \MB(t), \MC(t) \Rightarrow \MA(t+1), \MA'(t+1)$. To prove this, note that for $1 \leq j \leq N-1$,
\begin{eqnarray}
&& W_{j+1}^\top(t+1) W_{j+1}(t+1) - W_j(t+1) W_j^\top(t+1)\nonumber\\
  &=& \left( W_{j+1}^\top(t) - \eta W_{1:j}(t) \frac{dL}{dW}(W(t))^\top W_{j+2:N}(t) \right) \nonumber\\
  && \cdot \left( W_{j+1}(t) - \eta W_{j+2:N}^\top(t) \frac{dL}{dW}(W(t)) W_{1:j}^\top(t) \right)\nonumber\\
  &&- \left( W_j(t) - \eta W_{j+1:N}^\top(t) \frac{dL}{dW}(W(t)) W_{1:j-1}^\top(t) \right)\nonumber\\
  && \cdot \left( W_j^\top(t) - \eta W_{1:j-1}(t) \frac{dL}{dW}(W(t))^\top W_{j+1:N}(t) \right)\nonumber.
\end{eqnarray}
By $\MB(0), \ldots, \MB(t)$, $\| W_{1:N}(t) - \Phi \|_F \leq \| \Phi\|_F$. By the triangle inequality it then follows that $\| W_{1:N}(t)\|_\sigma \leq 2 \| \Phi\|_F$. Also $\MA(t)$ gives that for $1 \leq j \leq N-1$, $\| W_{j}(t) W_j^\top(t) - W_{j+1}^\top(t) W_{j+1}(t) \|_F \leq 2\delta$. By Lemma \ref{lem:boundindiv} with $C = 2 \| \Phi\|_F, \nu = 2\delta$ (so that (\ref{eq:epltc}) is satisfied),
  \begin{eqnarray}
    && \left\| W_{j+1}^\top(t+1) W_{j+1}(t+1) - W_j(t+1) W_j^\top(t+1) \right\|_F\nonumber\\
    & \leq & \| W_{j+1}^\top(t) W_{j+1}(t) - W_j(t) W_j^\top(t) \|_F + \eta^2 \left\| \frac{dL}{dW}(W(t)) \right\|_F \cdot  \left\| \frac{dL}{dW}(W(t)) \right\|_\sigma \nonumber\\
    && \cdot \left( \| W_{j+2:N}(t)\|_\sigma^2 \| W_{1:j}(t) \|_\sigma^2  + \| W_{1:j-1}\|_\sigma^2 \|W_{j+1:N} \|_\sigma^2\right)\nonumber\\
    \label{eq:wtwt1diverge}
    & \leq &  \| W_{j+1}^\top(t) W_{j+1}(t) - W_j(t) W_j^\top(t) \|_F \nonumber\\
    && + 4\eta^2 \left\| \frac{dL}{dW}(W(t)) \right\|_F  \left\| \frac{dL}{dW}(W(t)) \right\|_\sigma (2 \| \Phi\|_F)^{2(N-1)/N}.
  \end{eqnarray}
In the first inequality above, we have also used the fact that for matrices $A, B$ such that $AB$ is defined, $\| AB\|_F \leq \| A \|_\sigma \| B \|_F$. (\ref{eq:wtwt1diverge}) gives us $\MA'(t+1)$.

We next establish $\MA(t+1)$. By $\MB(i)$ for $0 \leq i \leq t$, we have that $\left\| \frac{dL}{dW}(W(i)) \right\|_F = \left\| W - \Phi \right\|_F \leq \| \Phi\|_F$. Using $\MA'(i)$ for $0 \leq i \leq t$ and summing over $i$ gives
\begin{eqnarray}
&& \| W_{j+1}^\top(t+1) W_{j+1}(t+1) - W_j(t+1) W_j^\top(t+1) \|_F\nonumber\\
\label{eq:hybridep}
  & \leq & \| W_{j+1}^\top(0) W_{j+1}(0) - W_j(0) W_j^\top(0) \|_F \nonumber\\
  && + 4  (2 \| \Phi\|_F)^{2(N-1)/N} \cdot \eta^2 \sum_{i=0}^{t}  \left\| \frac{dL}{dW}(W(i)) \right\|_F^2.
\end{eqnarray}

Next, by $\MB(0), \ldots, \MB(t)$, we have that $L(W(i)) \leq L(W(0))$ for $i \leq t$. Since $W(0)$ has deficiency margin of $c$ and by Claim \ref{claim:margin_interp}, it then follows that $\sigma_{min}(W(i)) \geq c$ for all $i \leq t$. Therefore, by summing $\MB(0), \ldots, \MB(t)$,
\begin{eqnarray}
  && \frac{1}{2} c^{2(N-1)/N} \eta\sum_{i=0}^t  \left\| \frac{dL}{dW} W(i) \right\|_F^2\nonumber\\
  & \leq & \frac 12 \eta \sum_{i=0}^t \sigma_{min}(W(i))^{2(N-1)/N} \left\| \frac{dL}{dW} (W(i)) \right\|_F^2\nonumber\\
  & \leq & L(W(0)) - L(W(t))\nonumber\\
  & \leq & L(W(0)) \leq \frac 12 \| \Phi\|_F^2\nonumber.
\end{eqnarray}
Therefore,
\begin{eqnarray}
&&4 \left( 2 \| \Phi\|_F\right)^{2(N-1)/N} \eta^2 \sum_{i=0}^t \left\| \frac{dL}{dW} W(i)\right\|_F^2 \nonumber\\
  & \leq  & 16 \| \Phi\|_F^{2(N-1)/N} \eta  \frac{\| \Phi\|_F^2}{c^{2(N-1)/N}}\nonumber\\
  \label{eq:sumetagrad}
  & \leq & 16 \| \Phi\|_F^{2(N-1)/N} \cdot \frac{1}{3 \cdot 2^{11} \cdot N^3} \cdot \frac{c^{(4N-2)/N}}{\| \Phi\|_F^{(6N-4)/N}} \cdot \frac{\| \Phi\|_F^2}{c^{2(N-1)/N}}\\
  & \leq & \frac{c^2}{256 N^3 \| \Phi\|_F^{2(N-1)/N}}\nonumber\\
  & = & \delta\nonumber,
\end{eqnarray}
where \eqref{eq:sumetagrad} follows from the definition of $\eta$ in (\ref{eq:descent_eta}), and the last equality follows from definition of $\delta$ in Theorem \ref{theorem:converge}.
By (\ref{eq:hybridep}), it follows that
$$
\| W_{j+1}^\top(t+1) W_{j+1}(t+1) - W_j(t+1)  W_j^\top(t+1) \|_F \leq 2\delta,
$$
verifying $\MA(t+1)$.

\item $ \MA(t), \MB(t) \Rightarrow \MC(t)$. We apply Lemma \ref{lem:boundindiv} with $\nu = 2\delta$ and $C = 2 \| \Phi\|_F$. First, the triangle inequality and $\MB(t)$ give
$$
\| W_{1:N}(t) \|_\sigma \leq \| \Phi\|_\sigma + \| \Phi - W_{1:N}(t) \|_\sigma \leq \| \Phi\|_F + \sqrt{2 \cdot L(W_{1:N}(t))} \leq 2 \| \Phi\|_F,
$$
verifying precondition 2.~of Lemma \ref{lem:boundindiv}. $\MA(t)$ verifies condition 1.~of Lemma \ref{lem:boundindiv}, so for $1 \leq j \leq N$, $\| W_j(t) \|_\sigma \leq (4 \| \Phi\|_F)^{1/N}$, giving $\MC(t)$.

\end{enumerate}
The proof of Lemma \ref{lem:remainsmooth} then follows by induction on $t$.
\end{proof}

\subsection{Proof of Theorem~\ref{theorem:converge_balance_init}} \label{app:proofs:converge_balance_init}

  Theorem \ref{theorem:converge_balance_init} is proven by combining Lemma \ref{lem:integrate_rad} below, which implies that the balanced initialization is likely to lead to an end-to-end matrix $W_{1:N}(0)$ with sufficiently large deficiency margin, with Theorem~\ref{theorem:converge}, which establishes convergence.

  \begin{lemma}
  \label{lem:integrate_rad}
  Let $d \in \BN, d \geq 20$; $b_2 > b_1 \geq 1$ be real numbers (possibly depending on $d$); and $\Phi \in \BR^d$ be a vector. Suppose that $\mu$ is a rotation-invariant distribution\footnote{Recall that a distribution on vectors $V \in \BR^d$ is {\it rotation-invariant} if the distribution of $V$ is the same as the distribution of $OV$, for any orthogonal $d \times d$ matrix $O$. If $V$ has a well-defined density, this is equivalent to the statement that for any $r > 0$, the distribution of $V$ conditioned on $\| V\|_2 = r$ is uniform over the sphere centered at the origin with radius $r$.} over $\BR^d$ with a well-defined density, such that, for some $0 < \ep < 1$,
  $$
\pr_{V \sim \mu} \left[ \frac{\| \Phi\|_2}{\sqrt{b_2d}} \leq \| V \|_2 \leq \frac{\| \Phi\|_2}{\sqrt{b_1d}}\right] \geq 1-\ep.
$$
Then, with probability at least $(1-\ep) \cdot \frac{3 - 4F(2/\sqrt{b_1})}{2}$, $V$ will have deficiency margin $\| \Phi\|_2 / (b_2d)$ with respect to $\Phi$.
\end{lemma}
The proof of Lemma~\ref{lem:integrate_rad} is postponed to Appendix~\ref{app:proofs:margin_init}, where Lemma~\ref{lem:integrate_rad} will be restated as Lemma~\ref{lem:integrate_rad_restate}.

One additional technique is used in the proof of Theorem \ref{theorem:converge_balance_init}, which leads to an improvement in the guaranteed convergence rate. Because the deficiency margin of $W_{1:N}(0)$ is very small, namely $\OO(\| \Phi\|_2/d_0)$ (which is necessary for the theorem to maintain constant probability), at the beginning of optimization, $\ell(t)$ will decrease very slowly. However, after a certain amount of time, the deficiency margin of $W_{1:N}(t)$ will increase to a constant, at which point the decrease of $\ell(t)$ will be much faster. To capture this acceleration, we apply Theorem~\ref{theorem:converge} a second time, using the larger deficiency margin at the new ``initialization.'' From a geometric perspective, we note that the matrices $W_1(0), \ldots, W_N(0)$ are very close to 0, and the point at which $W_j(0) = 0$ for all $j$ is a saddle. Thus, the increase in $\ell(t) - \ell(t+1)$ over time captures the fact that the iterates $(W_1(t), \ldots, W_N(t))$ escape a saddle point.

  \begin{proof}[Proof of Theorem \ref{theorem:converge_balance_init}]
    Choose some $a \geq 2$, to be specified later.
    By assumption, all entries of the end-to-end matrix at time 0, $W_{1:N}(0)$, are distributed as independent Gaussians of mean 0 and standard deviation $s \leq \|\Phi\|_2/\sqrt{ad_0^2}$. We will apply Lemma~\ref{lem:integrate_rad} to the vector $W_{1:N}(0) \in \BR^{d_0}$.
Since its distribution is obviously rotation-invariant, in remains to show that the distribution of the norm $\| W_{1:N}(0) \|_2$ is not too spread out. The following lemma~---~a direct consequence of the Chernoff bound applied to the $\chi^2$ distribution with $d_0$ degrees of freedom~---~will give us the desired result:
    \begin{lemma}[\cite{laurent_adaptive_2000}, Lemma 1]
      \label{lem:laurent}
      Suppose that $d \in \BN$ and $V \in \BR^d$ is a vector whose entries are i.i.d.~Gaussians with mean 0 and standard deviation $s$. Then, for any $k > 0$,
      \begin{eqnarray}
        \pr\left[\| V \|_2^2 \geq s^2 \left(d + 2k + 2\sqrt{kd} \right) \right] &\leq& \exp(-k)\nonumber\\
        \pr\left[\| V \|_2^2 \leq s^2 \left( d - 2\sqrt{kd} \right) \right] & \leq & \exp(-k)\nonumber.
      \end{eqnarray}
    \end{lemma}
    By Lemma \ref{lem:laurent} with $k = d_0/16$, we have that
    $$

    \pr\left[ \frac{s^2 d_0}{2} \leq \| V \|_2^2 \leq 2s^2 d_0\right] \geq 1 - 2 \exp(-d_0/16).
    $$

    We next use Lemma \ref{lem:integrate_rad}, with $b_1 = \| \Phi\|_2^2/(2s^2d_0^2), b_2 = 2\| \Phi\|_2^2/(s^2d_0^2)$; note that since $a \geq 2$, $b_1 \geq 1$, as required by the lemma. Lemma \ref{lem:integrate_rad} then implies that with probability at least
    \be
    \label{eq:expF}
    \left(1 - 2\exp(-d_0/16) \right) \frac{3-4F\left(2/\sqrt{a/2}\right)}{2},
    \ee
    $W_{1:N}(0)$ will have deficiency margin ${s^2d_0}/{2\| \Phi\|_2}$
    with respect to $\Phi$. By the definition of balanced initialization (Procedure~\ref{proc:balance_init}) $W_1(0), \ldots, W_N(0)$ are $0$-balanced. Since $2^4 \cdot 6144 < 10^5$, our assumption on $\eta$ gives
    \begin{equation}
      \label{eq:eta_single_output}
      \eta \leq \frac{(s^2d_0)^{4-2/N}}{2^4 \cdot 6144N^3 \| \Phi\|_2^{10-6/N}},
    \end{equation}
    so that Equation~(\ref{eq:descent_eta}) holds with $c = \frac{s^2d_0}{2\|\Phi\|_2}$.
    The conditions of Theorem \ref{theorem:converge} thus hold with probability at least that given in Equation~(\ref{eq:expF}). In such a constant probability event, by Theorem \ref{theorem:converge} (and the fact that a positive deficiency margin implies $L^1(W_{1:N}(0))\leq\frac{1}{2}\| \Phi\|_2^2$), if we choose
  \begin{equation}
    \label{eq:t0lb}

     t_0 \geq \eta^{-1} \left( \frac{ 2\| \Phi\|_2}{s^2d_0} \right)^{2-2/N}  \ln(4),
\end{equation}
then $L^1(W_{1:N}(t_0)) \leq \frac 18 \| \Phi\|_2^2 $, meaning that $\| W_{1:N}(t_0) - \Phi\|_2 \leq \frac 12 \| \Phi\|_2 = \| \Phi\|_2 - \frac 12 \sigma_{min}(\Phi)$. Moreover, by condition $\MA(t_0)$ of Lemma \ref{lem:remainsmooth} and the definition of $\delta$ in Theorem \ref{theorem:converge}, we have, for $1 \leq j \leq N-1$,
\begin{equation}
  \label{eq:balanced_single_output}
  \| W_{j+1}^T(t_0) W_{j+1}(t_0) - W_j(t_0) W_j^T(t_0) \|_F \leq
  \frac{2s^4d_0^2}{(2\| \Phi\|_2)^2 \cdot 256 N^3 \| \Phi\|_2^{2-2/N}} = \frac{s^4 d_0^2}{512 N^3 \| \Phi\|_2^{4-2/N}}.
\end{equation}

We now apply Theorem \ref{theorem:converge} again, verifying its conditions again, this time with the initialization $(W_1(t_0), \ldots, W_N(t_0))$. First note that the end-to-end matrix $W_{1:N}(t_0)$ has deficiency margin $c = \| \Phi\|_2/2$ as shown above. The learning rate $\eta$, by Equation~(\ref{eq:eta_single_output}), satisfies Equation~(\ref{eq:descent_eta}) with $c = \| \Phi\|_2/2$. Finally, since
$$
\frac{s^4 d_0^2}{512 N^3 \| \Phi\|_2^{4-2/N}} \leq \frac{\|\Phi\|^{2/N}}{(a^2d_0^2) \cdot 512 N^3} \leq \frac{\| \Phi\|^{2/N}(1/2)^2}{256 N^3}
$$
for $d_0 \geq 2$, by Equation~(\ref{eq:balanced_single_output}), the matrices $W_1(t_0), \ldots, W_N(t_0)$ are $\delta$-balanced with $\delta = \frac{\| \Phi\|^{2/N}(1/2)^{2}}{256 N^3}$.
Iteration~$t_0$ thus satisfies the conditions of Theorem~\ref{theorem:converge} with deficiency margin $\| \Phi\|_2/2$, meaning that for
\be
\label{eq:tmt0lb}
T - t_0 \geq \eta^{-1} \cdot 2^{2-2/N} \cdot \| \Phi\|^{2/N-2}  \ln\left( \frac{\| \Phi\|_2^2}{8\ep} \right),
\ee
we will have $\ell(T) \leq \ep$. Therefore, by Equations~(\ref{eq:t0lb}) and~(\ref{eq:tmt0lb}), to ensure that $\ell(T) \leq \ep$, we may take
$$
T \geq 4\eta^{-1} \left( \ln(4) \left(\frac{\|\Phi\|_2}{s^2d_0}\right)^{2-2/N}   + \| \Phi\|_2^{2/N-2}\ln(\| \Phi\|_2^2/(8\ep)) \right).
$$
Recall that this entire analysis holds only with the probability given in Equation~(\ref{eq:expF}). As $\lim_{d \ra \infty} (1-2\exp(-d/16)) = 1$ and $\lim_{a \ra \infty} (3 - 4F(2\sqrt{2/a}))/2 = 1/2$, for any $0 < p < 1/2$, there exist $a, d_0' > 0$ such that for $d_0 \geq d_0'$, the probability given in Equation~(\ref{eq:expF}) is at least $p$. This completes the proof.

\end{proof}

In the context of the above proof, we remark that the expressions $1- 2\exp(-d_0/16)$ and $(3 - 4F(2\sqrt{2/a}))/2$ converge to their limits of $1$ and $1/2$, respectively, as $d_0,a \ra \infty$ quite quickly. For instance, to obtain a probability of greater than $0.25$ of the initialization conditions being met, we may take $d_0 \geq 100, a \geq 100$.

\subsection{Proof of Claim~\ref{claim:balance_init}} \label{app:proofs:balance_init}

We first consider the probability of $\delta$-balancedness holding between any two layers:
\begin{lemma}
  \label{lem:abvarbound}
  Suppose $a,b,d \in \BN$ and $A \in \BR^{a \times d}, B \in \BR^{d \times b}$ are matrices whose entries are distributed as i.i.d.~Gaussians with mean 0 and standard deviation $s$. Then for $k \geq 1$,
  \begin{equation}
    \label{eq:abd}
\pr\left[ \left\|A^TA - BB^T\right\|_F \geq ks^2 \sqrt{2d(a+b)^2 + d^2(a+b)} \right] \leq 1/k^2.
\end{equation}
\end{lemma}
\begin{proof}
  Note that for $1 \leq i,j \leq d$, let $X_{ij}$ be the random variable $(A^TA - BB^T)_{ij}$, so that $$X_{ij} = (A^TA - BB^T)_{ij} = \sum_{1 \leq \ell \leq a} A_{\ell i} A_{\ell j} - \sum_{1 \leq r \leq b} B_{ir} B_{jr}.$$
  If $i \neq j$, then $$\ex[X^2] = \sum_{1 \leq \ell \leq a} \ex[A_{\ell i}^2 A_{\ell j}^2] + \sum_{1 \leq r \leq b} \ex[B_{ir}^2 B_{jr}^2]=(a+b)s^4.$$ We next note that for a normal random variable $Y$ of variance $s^2$ and mean 0, $\ex[Y^4] = 3s^4$. Then if $i = j$,
  $$
  \ex[X^2] = s^4 \cdot (3(a+b) + a(a-1) + b(b-1) {-} ab) \leq s^4((a+b)^2 + 2(a+b)).
  $$
  Thus
  \begin{eqnarray}
    \ex[\| A^TA - BB^T\|_F^2] &{\leq}& s^4(d((a+b)^2 + 2(a+b)) + d(d-1)(a+b)) \nonumber\\
    &\leq & s^4(2d(a+b)^2 + d^2(a+b))\nonumber.
  \end{eqnarray}
  Then (\ref{eq:abd}) follows from Markov's inequality.
\end{proof}

Now the proof of Claim \ref{claim:balance_init} follows from a simple union bound:
\begin{proof}[Proof of Claim \ref{claim:balance_init}]
  By (\ref{eq:abd}) of Lemma \ref{lem:abvarbound}, for each $1 \leq j \leq N-1$, $k \geq 1$,
  $$
\pr\left[ \| W_{j+1}^T W_{j+1} - W_jW_j^T \|_F {\geq} ks^2 \sqrt{10d_{max}^3} \right] \leq 1/k^2.
$$
By the union bound,
$$
\pr \left[ \forall 1 \leq j \leq N-1, \ \ \| W_{j+1}^T W_{j+1} - W_jW_j^T \|_F \leq ks^2 \sqrt{10d_{max}^3} \right] \geq 1-N/k^2,
$$
and the claim follows with $\delta = ks^2 \sqrt{10d_{max}^3}$.
  \end{proof}

\subsection{Proof of Claim~\ref{claim:margin_init}} \label{app:proofs:margin_init}

We begin by introducing some notation. Given $d \in \BN$ and $r > 0$, we let $B^d(r)$ denote the open ball of radius $r$ centered at the origin in $\BR^d$. For an open subset $U \subset \BR^d$, let $\partial U := \bar U \backslash U$ be its boundary, where $\bar U$ denotes the closure of $U$. For the special case of $U = B^d(r)$, we will denote by $S^d(r)$ the boundary of such a ball, i.e.~the sphere of radius $r$ centered at the origin in $\BR^d$. Let $S^d := S^d(1)$ and $B^d := B^d(1)$. There is a well-defined uniform (Haar) measure on $S^d(r)$ for all $d, r$, which we denote by $\sigma^{d,r}$; we assume $\sigma^{d,r}$ is normalized so that $\sigma^{d,r}(S^d(r)) = 1$.
Finally, since in the context of this claim we have~$d_N=1$, we allow ourselves to regard the end-to-end matrix $W_{1:N}\in\R^{1\times{d}_0}$ as both a matrix and a vector.

\medskip

To establish Claim \ref{claim:margin_init}, we will use the following low-degree anti-concentration result of \cite{carbery_distributional_2001} (see also \cite{lovett_elementary_2010,meka_anti-concentration_2016}):
\begin{lemma}[\cite{carbery_distributional_2001}]
  \label{lem:carbery}
  There is an absolute constant $C_{{0}}$ such that the following holds. Suppose that ${h}$ is a multilinear polynomial of $K$ variables $X_1, \ldots, X_K$ and of degree $N$. Suppose that $X_1, \ldots, X_K$ are i.i.d.~Gaussian. Then, for any $\epsilon>0$:
  $$
  \mathbb{P} \left[|{h}(X_1, \ldots, X_K)| \leq \ep \cdot \sqrt{\Var[{h}(X_1, \ldots, X_K)]}\right] \leq C_{{0}} N\ep^{1/N}.
  $$
\end{lemma}

The below lemma characterizes the norm of the end-to-end matrix~$W_{1:N}$ following zero-centered Gaussian initialization:
\begin{lemma}
  \label{lem:norm_concentrate}
  For any constant $0 < C_2 < 1$, there is an absolute constant $C_1 > 0$ such that the following holds. Let $N, d_0, \ldots, d_{N-1} \in \BN$. Set $d_N = 1$. Suppose that for $1 \leq j \leq N$, $W_j \in \BR^{d_j \times d_{j-1}}$ are matrices whose entries are i.i.d.~Gaussians of standard deviation $s$ and mean 0. Then
  $$
\pr\left[ s^{2N} d_1 \cdots d_{N-1} \left( \frac{1}{C_1N} \right)^{2N} \leq \| W_{1:N}\|_2^2 \leq C_1d_0^2 d_1 \cdots d_{N-1} s^{2N} \right] \geq C_2.
  $$
\end{lemma}
\begin{proof}
 Let $f(W_1, \ldots, W_N) = \| W_{1:N}\|_2^2$, so that $f$ is a polynomial of degree $2N$ in the entries of $W_1, \ldots, W_N$. Notice that
  $$
f(W_1, \ldots, W_N) = \sum_{i_0 = 1}^{d_0} \left( \sum_{i_1=1}^{d_1} \cdots \sum_{i_{N-1} = 1}^{d_{N-1}} (W_N)_{1,i_{N-1}} (W_{N-1})_{i_{N-1}, i_{N-2}} \cdots (W_1)_{i_1, i_0} \right)^2.
$$
For $1 \leq i_0 \leq d_0$, set
$$
g_{i_0}(W_1, \ldots, W_N) =  \sum_{i_1=1}^{d_1} \cdots \sum_{i_{N-1} = 1}^{d_{N-1}} (W_N)_{1,i_{N-1}} (W_{N-1})_{i_{N-1}, i_{N-2}} \cdots (W_1)_{i_1, i_0},
$$
so that $f = \sum_{i_0 = 1}^{d_0} g_{i_0}^2$. Since each $g_{i_0}$ is a multilinear polynomial in $W_1,\ldots,W_N$, we have that $\ex[g_{i_0}(W_1, \ldots, W_N)] = 0$ for all $1 \leq i_0 \leq d_0$. Also
\begin{eqnarray}
  \Var[g_{i_0}(W_1, \ldots, W_N)] &=& \ex[g_{i_0}(W_1, \ldots, W_N)^2]\nonumber\\
                              &=& \sum_{i_1=1}^{d_1} \cdots \sum_{i_{N-1} = 1}^{d_{N-1}} \ex\left[ (W_N)_{1,i_{N-1}}^2 (W_{N-1})_{i_{N-1}, i_{N-2}}^2 \cdots (W_1)_{i_1, i_0}^2 \right]\nonumber\\
                              &=& d_1 d_2 \cdots d_{N-1} s^{2N} \nonumber.
\end{eqnarray}
It then follows by Markov's inequality that for any $k \geq 1$, $\pr[g_{i_0}^2 \geq k s^{2N} d_1 \cdots d_{N-1}] \leq 1/k$. For any constant $B_1$ (whose exact value will be specified below), it follows that
\begin{eqnarray}
  &&\pr[f(W_1, \ldots, W_N) {\geq} B_1 d_0^2 d_1 d_2 \cdots d_{N-1} s^{2N}]\nonumber\\
  &=& \pr\left[ \sum_{i_0=1}^{d_0} g_{i_0}(W_1, \ldots, W_N)^2 {\geq} B_1 d_0^2 d_1 d_2 \cdots d_{N-2} s^{2N} \right]\nonumber\\
  & \leq & d_0 \cdot \pr[g_1(W_1, \ldots, W_N)^2 {\geq} B_1 d_0d_1 \cdots d_{N-1} s^{2N}]\nonumber\\
  \label{eq:poly_ub}
  & \leq & 1/B_1.
\end{eqnarray}
Next, by Lemma~\ref{lem:carbery}, there is an absolute constant $C_0 > 0$ such that for any $\ep > 0$, and any $1 \leq i_0 \leq d_0$,
$$
\pr\left[ |g_{i_0}(W_1, \ldots, W_N)| \leq \ep^N \sqrt{s^{2N} d_1 \cdots d_{N-1}} \right] \leq C_0 N \ep.
$$
Since $f^2 \geq g_{i_0}^2$ for each $i_0$, it follows that
\begin{equation}
  \label{eq:poly_lb}
\pr[ f(W_1, \ldots, W_N) \geq \ep^{2N} s^{2N} d_1 \cdots d_{N-1} ] \geq 1 - C_0 N \ep.
\end{equation}
Next, given $0 < C_2 < 1$, choose $\ep = (1-C_2)/(2C_0N)$, and $B_1 = 2/(1-C_2)$. Then by (\ref{eq:poly_ub}) and (\ref{eq:poly_lb}) and a union bound, we have that
$$
\pr\left[\left( \frac{1-C_2}{2C_0N}\right)^{2N} s^{2N} d_1 \cdots d_{N-1} \leq f(W_1, \ldots, W_N) \leq \frac{2}{1-C_2} {s^{2N}}d_0^2 d_1 \cdots d_{N-1} \right] \geq C_2.
$$
The result of the lemma then follows by taking $C_1 = \max\left\{ \frac{2}{1-C_2}, \frac{2C_0}{1-C_2} \right\}$.
\end{proof}

\begin{lemma}
  \label{lem:rotation_inv}
  Let $N, d_0, \ldots, d_{N-1} \in \BN$, and set $d_N = 1$. Suppose $W_j \in \BR^{d_j \times d_{j-1}}$ for $1 \leq j \leq N$, are matrices whose entries are i.i.d.~Gaussians with mean~$0$ and standard deviation~$s$. Then, the distribution of $W_{1:N}$ is rotation-invariant.
\end{lemma}
\begin{proof}
  First we remark that for any orthogonal matrix $O \in \BR^{d_0 \times d_0}$, the distribution of $W_1$ is the same as that of $W_1O$. To see this, let us denote the rows of $W_1$ by $(W_1)_1, \ldots, (W_1)_{d_1}$, and the columns of $O$ by $O^1, \ldots, O^{d_0}$. Then the $(i_1, i_0)$ entry of $W_1O$, for $1 \leq i_1 \leq d_1, 1 \leq i_0 \leq d_0$ is $\langle (W_1)_{i_1}, O^{i_0} \rangle$, which is a Gaussian with mean~$0$ and standard deviation~$s$, since $\| O^{i_0}\|_2 = 1$. Since $\langle O^{i_0}, O^{i_0'} \rangle = 0$ for $i_0 \neq i_0'$, the covariance between any two distinct entries of $W_1O$ is 0. Therefore, the entries of $W_1O$ are independent Gaussians with mean~$0$ and standard deviation~$s$, just as are the entries of $W_1$.

  But now for any matrix $O \in \BR^{d_0 \times d_0}$, the distribution of $W_{1:N}O$ is the distribution of $W_N W_{N-1} \cdots W_2 (W_1O)$, which is the same as the distribution of $W_N W_{N-1} \cdots W_2 W_1 = W_{1:N}$, since $W_1, W_2, \ldots, W_N$ are all independent.
\end{proof}

For a dimension~$d\in\N$, radius~$r>0$, and $0 < h < r$, a {\it $(d,r)$-hyperspherical cap of height~$h$} is a subset $\MC \subset B^d(r)$ of the form $\{ x \in B^d(r) : \langle x, u \rangle \geq r-h\}$, where $u$ is any $d$-dimensional unit vector. We define the {\it area of a $(d,r)$-hyperspherical cap of height $h$}~---~$\MC$~---~to be $\sigma^{d,r}(\partial \MC \cap S^d(r))$.
\begin{lemma}
  \label{lem:volcap}
  For $d \geq 20$, choose any $0 \leq h \leq 1$. Then, the area of a $(d,1)$-hyperspherical cap of height $h$ is at least
  $$
  \frac{3 -  4 F((1-h) \sqrt{d-3})}{2}.
  $$
\end{lemma}
\begin{proof}
  In \cite{chudnov_minimax_1986}, it is shown that the area of a $(d,1)$-hyperspherical cap of height $h$ is given by $\frac{1 - C_{d-2}(h)/C_{d-2}(0)}{2}$, where
  $$
C_d(h) := \int_0^{1-h} (1-t^2)^{(d-1)/2} dt.
$$
Next, by the inequality $1-t^2 \geq \exp(-2t^2)$ for $0 \leq t \leq 1/2$,
\begin{eqnarray}
  \int_0^1 (1-t^2)^{(d-3)/2} dt & \geq & \int_0^{1/2} \exp\left(2 \cdot  \frac{-t^2(d-3)}{2} \right) dt\nonumber\\
                                & = & \sqrt{\pi/(d-3)} \cdot \frac{2F(\sqrt{(d-3)/2}) - 1}{2}\nonumber\\
  \label{eq:cd0}
                                & \geq & \sqrt{\pi/(d-3)} \cdot \frac{1 - 2 \exp(-(d-3)/4)}{2},
\end{eqnarray}
where the last inequality follows from the standard estimate $F(x) \geq 1 - \exp(-x^2/2)$ for $x \geq 1$.
Also, since $1-t^2 \leq \exp(-t^2)$ for all $t$,
\begin{eqnarray}
  \int_0^{1-h} (1-t^2)^{(d-3)/2} dt & \leq &  \int_0^{1-h} \exp\left( \frac{-t^2 (d-3)}{2} \right) dt \nonumber\\
  \label{eq:cdh}
  & = & \sqrt{2\pi/(d-3)} \cdot \frac{2F((1-h) \sqrt{d-3}) - 1}{2}.
\end{eqnarray}
Therefore, for $d \geq 20$, by (\ref{eq:cd0}) and (\ref{eq:cdh}),
\begin{eqnarray}
  \frac{1-C_{d-2}(h)/C_{d-2}(0)}{2} & \geq &\frac{1 - \frac{\sqrt{2} \cdot (2F((1-h)\sqrt{d-3}) - 1)}{1 - 2 \exp(-(d-3)/4)}}{2} \nonumber\\
  & \geq & \frac{1 - \sqrt{2} \cdot (2F((1-h)\sqrt{d-3}) - 1) \cdot (1 + 4 \exp(-(d-3)/4))}{2}\nonumber\\
                            & \geq & \frac{3 - 4 F((1-h)\sqrt{d-3})}{2},\nonumber
\end{eqnarray}
where the second inequality has used $1/(1-y) \leq 1 + 2y$ for all $0 < y < 1/2$ (and where $y = 2\exp((-(d-3)/4)) < 2 \exp(-17/4) < 1/2$), and the final inequality uses $1 + 4\exp(-(d-3)/4) \leq \sqrt{2}$ for $d \geq 20$. The above chain of inequalities gives us the desired result.
\end{proof}

\begin{lemma}
  \label{lemma:haar_def}
Let $d \in \BN, d \geq 20$; $a \geq 1$ be a real number (possibly depending on $d$); and $\Phi \in \BR^d$ be some vector.
Set $r = \| \Phi\|_2 / \sqrt{ad}$, and suppose that $V \in S^d(r)$ is drawn according to the uniform measure.
Then, with probability at least $\frac{3 - 4F(2/\sqrt{a})}{2}$, $V$ will have deficiency margin $\| \Phi\|_2 / (ad)$ with respect to $\Phi$.
\end{lemma}
\begin{proof}
  By rescaling, we may assume without loss of generality that $\| \Phi\|_2 = 1$, so that $r = 1/\sqrt{ad}$. Let $\MD$ denote the intersection of $B^{d}(r)$ with the open $d$-ball of radius $1-1/(ad)$ centered at $\Phi$. Let $\MC \subset B^{d}(r)$ denote the $(d,r)$-hyperspherical cap of height $r \cdot \big( 1 - 2/(\sqrt{ad})\big) = r - 2/(ad)$ whose base is orthogonal to the line between $\mathbf{0}$ and $\Phi$ (see Figure \ref{fig:initfig}). Note that ${\sigma}^{d,r}(\partial\MD \cap S^d(r))$, the Haar measure of the portion of $\partial\MD$ intersecting $S^d(r)$, gives the probability that $V$ belongs to the boundary of $\MD$. By Lemma \ref{lem:volcap} above (along with rescaling arguments), since $d \geq 20$, $\sigma^{d,r}(\partial \MC \cap S^d(r)) \geq \frac 12 \cdot (3 - 4F(2/\sqrt{a}))$, and therefore $V \in {\partial}\MC$ with at least this probability.

We next claim that $\MC \subseteq \MD$. To see this, first let $\MT \subset \BR^d$ denote the $(d-1)$-sphere of radius $1-1/(ad)$ centered at $\Phi$ (see Figure \ref{fig:initfig}). Let $P$ be the intersection of $\MT$ with the line from $\mathbf{0}$ to $\Phi$, and $Q$ denote the intersection of this line with the unique hyperplane of codimension 1 containing $\MT \cap \partial B^{d}(r)$~---~we denote this hyperplane by $\MH$. If we can show that $\norm{P-Q}_2 \leq 1/(ad)$, then it follows that $\MC$ lies entirely on the other side of $\MH$ as $\mathbf{0}$, which will complete the proof that $\MC \subseteq \MD$.

The calculation of $\norm{P-Q}_2$ is simply an application of the law of cosines: letting $\theta$ be the angle determining the intersection of $\partial B^{d}(r)$ and $\MT$ (see Figure~\ref{fig:initfig}), note that
$$
(1-1/(ad))^2 = r^2 + 1^2 - 2r\cos \theta = 1/(ad) + 1 - 2/\sqrt{ad} \cdot \cos(\theta),
$$
so
$$
d(P,Q) = r \cos \theta - 1/(ad) = \frac 12 (1/(ad) - 1/(a^2d^2)) < 1/(ad),
$$
as desired.

Using that $\MC \subseteq \MD$, we continue with the proof. Notice the fact that $\MC\subseteq \MD$ is equivalent to $\partial \MC \cap S^d(r) \subseteq \partial \MD \cap S^d(r)$, by the structure of $\MC$ and~$\MD$. {Since the probability that $V$ lands in $\partial\MC$ is at least $\frac{3-4F(2/\sqrt{a})}{2}$, this lower bound applies to $V$ landing in $\partial\MD$ as well}. Since all $V \in \partial\MD$ have distance at most $1-1/(ad)$ from $\Phi$, and since $\sigma_{min}(\Phi) = \| \Phi\|_2 = 1$, it follows that for any $V \in \partial\MD$, $\| V - \Phi\|_2 \leq \sigma_{min}(\Phi) - 1/(ad)$. Therefore, with probability of at least $\frac{3-4F(2/\sqrt a)}{2}$, $V$ has deficiency margin $\| \Phi\|_2/(ad)$ with respect to~$\Phi$.
\end{proof}

\begin{figure}
\begin{center}
\begin{tikzpicture}
  \draw[very thin, pattern = north west lines, pattern color = gray, opacity=0.25] (2.73,-4.2) -- (2.73, 4.2) arc (56.65:-56.65:5cm) -- cycle;
  \draw[very thin, pattern = north west lines, pattern color = gray, opacity=0.25] (2.73,-4.2) -- (2.73, 4.2) arc (148:212:8cm) -- cycle;
  \draw (0,0) circle (5cm);
  \draw (0,0) -- (10,0);
  \draw (0,0) -- (2.7, 4.25);

  \draw [thick, blue] (2.7,-5) -- (2.7,5);
  \node [above] at (2.7,5) {\color{blue} $\MH$};
  \node [above] at (10, 0) {$\Phi$};
  \node [above] at (0,0) {$\mathbf{0}$};
  \node [above left] at (1.5, 0) {$P$};
  \node [above right] at (2.7, 0) {$Q$};
  \node [above right] at (0.2, 0) {$\theta$};
  \node [above] at (1.3, 2.1) {$r$};
  \node [right] at (5.5, 6.92) {\color{red} $\MT$};
  \draw [thick,red] (5.5,6.92) arc (120:240:8cm)
  \filldraw[fill opacity=0.2,fill=green] (3.8,-3.25) -- (3.8,3.25) arc (40.39:-40.39:5cm) -- cycle;
  \node [above] at (4.35,-0.9) {\color{darkspringgreen} $\MC$};
  \draw[decoration={calligraphic brace,amplitude=5pt}, decorate, line width=1.25pt] (3.85,0.05) node {} -- (4.95,0.05);
  \node [above,align=center] at (4.4, 0.1) {\small Height \\ \small of $\MC$};
\end{tikzpicture}
\end{center}
\caption{Figure for proof of Lemma \ref{lemma:haar_def}. The dashed region denotes $\MD$. Not to scale.}
\label{fig:initfig}
\end{figure}

\begin{lemma}[Lemma~\ref{lem:integrate_rad} restated]
  \label{lem:integrate_rad_restate}
  Let $d \in \BN, d \geq 20$; $b_2 > b_1 \geq 1$ be real numbers (possibly depending on $d$); and $\Phi \in \BR^d$ be a vector. Suppose that $\mu$ is a rotation-invariant distribution over $\BR^d$ with a well-defined density, such that, for some $0 < \ep < 1$,
  $$
\pr_{V \sim \mu} \left[ \frac{\| \Phi\|_2}{\sqrt{b_2d}} \leq \| V \|_2 \leq \frac{\| \Phi\|_2}{\sqrt{b_1d}}\right] \geq 1-\ep.
$$
Then, with probability at least $(1-\ep) \cdot \frac{3 - 4F(2/\sqrt{b_1})}{2}$, $V$ will have deficiency margin $\| \Phi\|_2 / (b_2d)$ with respect to $\Phi$.
\end{lemma}
\begin{proof}
  By rescaling we may assume that $\| \Phi\|_2 = 1$ without loss of generality. Then the deficiency margin of $V$ is equal to $1 - \| V- \Phi\|_2$. $\mu$~has a well-defined density, so we can set $\hat{\mu}$ to be the probability density function of $\| V\|_2$. Since $\mu$ is rotation-invariant, we can integrate over spherical coordinates, giving
  \begin{eqnarray}
    && \pr[1 - \| V - \Phi\|_2 \geq {1} / (b_2d)] \nonumber\\
    &=& \int_0^\infty \pr\big[1 - \| V - \Phi\|_2 \geq {1} / (b_2d) ~\big|~ \| V \|_2 = r \big]\hat \mu(r) dr\nonumber\\
    & \geq & \int_{1/(\sqrt{b_2d})}^{1/(\sqrt{b_1d})} \frac{3 - 4F(2r\sqrt{d} )}{2} \hat \mu(r) dr \nonumber\\
    & \geq & \frac{3 - 4F(2/\sqrt{b_1})}{2} \cdot \int_{1/(\sqrt{b_2d})}^{1/(\sqrt{b_1d})} \hat\mu(r) dr \nonumber\\
    & \geq & \frac{3 - 4F(2/\sqrt{b_1})}{2} \cdot (1-\ep)\nonumber,
  \end{eqnarray}
  where the first inequlaity used Lemma \ref{lemma:haar_def} and the fact that the distribution of $V$ conditioned on $\| V\|_2 = r$ is uniform on $S^d(r)$.
\end{proof}

Now we are ready to prove Claim \ref{claim:margin_init}:
\begin{proof}[Proof of Claim \ref{claim:margin_init}]
  We let $W \in \BR^{1 \times d_0} \simeq \BR^{d_0}$ denote the random vector $W_{1:N}$; also let $\mu$ denote the distribution of $W$, so that by Lemma \ref{lem:rotation_inv}, $\mu$ is rotation-invariant.  Let $C_1$ be the constant from Lemma~\ref{lem:norm_concentrate} for $C_2 = 999/1000$. For some $a \geq 10^5$, the standard deviation of the entries of each $W_{j}$ is given by
  \begin{equation}
    \label{eq:define_s}
s = \left( \frac{\| \Phi\|_2^2}{ad_0^3d_1 \cdots d_{N-1} C_1} \right)^{1/(2N)}.
\end{equation}
Then by Lemma \ref{lem:norm_concentrate},
$$
\pr \left[ \frac{ \| \Phi\|_2^2}{ad_0^3 C_1} \cdot \left( \frac{1}{C_1N} \right)^{2N} \leq \| W\|_2^2 \leq \frac{\| \Phi\|_2^2}{ad_0} \right] \geq {\frac{999}{1000}}.
$$
Then Lemma~\ref{lem:integrate_rad_restate}, with $d = d_0$, $b_1 = a$ and $b_2 = ad_0^2 C_1 \cdot (C_1N)^{2N}$, implies that with probability at least $\frac{999}{1000} \cdot \frac{3 - 4F(2/\sqrt{a})}{2}$, $W$ has deficiency margin $\| \Phi\|_2 / (ad_0^3 C_1^{2N+1} N^{2N})$ with respect to $\Phi$. But $a \geq 10^5$ implies that this probability is at least $0.49$, and from (\ref{eq:define_s}),
\be
\label{eq:deficiency_margin_final}
\frac{\| \Phi\|_2}{ad_0^3 C_1^{2N+1} N^{2N}} = \frac{s^{2N} d_1 \cdots d_{N-1}}{\| \Phi\|_2 (C_1N)^{2N}}.
\ee
Next recall the assumption in the hypothesis that $s\geq{C}_{1}N(c\cdot\norm{\Phi}_2/(d_1\cdots{d}_{N-1}))^{1/2N}$. Then the deficiency margin in (\ref{eq:deficiency_margin_final}) is at least
$$
\frac{\left(C_1N(c \norm{\Phi}_2 / (d_1 \cdots d_{N-1}))^{1/(2N)}\right)^{2N} d_1 \cdots d_{N-1}}{\| \Phi\|_2 (C_1N)^{2N}} = c,
$$
completing the proof.

\end{proof}

\subsection{Proof of Claim~\ref{claim:diverge_balance}} \label{app:proofs:diverge_balance}

\begin{proof}
The target matrices~$\Phi$ that will be used to prove the claim satisfy $\sigma_{min}(\Phi)=1$.
We may assume without loss of generality that $c \geq 3/4$, the reason being that if a matrix has deficiency margin $c$ with respect to $\Phi$ and $c' < c$, it certainly has deficiency margin $c'$ with respect to $\Phi$.

We first consider the case $d=1$, so that the target and all matrices are simply real numbers; we will make a slight abuse of notation in identifying $1\times1$ matrices with their unique entries. We set $\Phi = 1$. For all choices of $\eta$, we will set the initializations $W_1(0), \ldots, W_N(0)$ so that $W_{1:N}(0) = c$. Then
$$
\| W_{1:N}(0) - \Phi\|_F = |W_{1:N}(0) - \Phi| = 1-c = \sigma_{min}(\Phi) - c,
$$
so the initial end-to-end matrix $W_{1:N}(0) \in \BR^{1 \times 1}$ has deficiency margin $c$. Now fix $\eta$. Choose $A \in \BR$ with
\begin{eqnarray}
  \label{eq:define_A}
  A &=& \max \left\{\sqrt{\eta N}, \frac{2}{\eta(1-c) c^{(N-1)/N}}, 2000,20/\eta,\left( \frac{20 \cdot 10^{2N-1}}{\eta^{2N}}\right)^{1/(2N-2)} \right\}.
\end{eqnarray}
We will set:
\begin{equation}
  \label{eq:define_wj0}
  W_j(0) = \begin{cases}
    Ac^{1/N} \quad : \quad 1 \leq j \leq N/2 \\
    c^{1/N}/A \quad : \quad N/2 < j \leq N,
  \end{cases}
\end{equation}
so that $W_{1:N}(0) = c$. Then since $L^N(W_1, \ldots, W_N) = \frac 12 (1 - W_N \cdots W_1)^2$, the gradient descent updates are given by
$$
W_{j}(t+1) = W_j(t) - \eta (W_{1:N}(t) - 1) \cdot W_{1:j-1}(t) W_{j+1:N}(t),
$$
where we view $W_1(t), \ldots, W_N(t)$ as real numbers. This gives
\begin{equation}
  W_j(1) = \begin{cases}
    c^{1/N}A - \eta (c-1)c^{(N-1)/N}/A \quad : \quad 1 \leq j \leq N/2 \\
    c^{1/N}/A - \eta (c-1) c^{(N-1)/N}A \quad : \quad N/2 < j \leq N.
  \end{cases}\nonumber
\end{equation}
Since $3/4 \leq c < 1$ and $-\eta(c-1)c^{(N-1)/N}A \geq 0$, we have that $A/2 \leq 3A/4 \leq W_j(1)$ for $1 \leq j \leq N/2$. Next, since $\frac{1-c}{1-c^{1/N}} \leq N$ for $0 \leq c < 1$, we have that $A^2 \geq \eta N \geq \frac{\eta (1-c)}{1-c^{1/N}}$, which implies that $A^2 \geq c^{1/N}A^2 + \eta (1-c)$, or $c^{1/N} A + \frac{\eta(1-c)}{A} \leq A$. Thus $W_j(1) \leq A$ for $N/2 < j \leq N$.
Similarly, using the same bound $3/4 \leq c < 1$ and the fact that $\eta (1-c) c^{(N-1)/N} A \geq 2$ we get $\frac{3}{16} \eta A \leq W_j(1) \leq \eta A$ for $N/2 < j \leq N$. In particular, for all $1 \leq j \leq N$, we have that $\frac{\min\{\eta, 1\}}{10} A \leq W_j(1) \leq \max\{\eta, 1\} A$.

We prove the following lemma by induction:
\begin{lemma}
  \label{lemma:weight_explode}
  For each $t \geq 1$, the real numbers $W_1(t), \ldots, W_N(t)$ all have the same sign and this sign alternates for each integer $t$. Moreover, there are real numbers $2 \leq B(t) < C(t)$ for $t \geq 1$ such that for $1 \leq j \leq N$, $B(t) \leq |W_j(t)| \leq C(t)$ and $\eta B(t)^{2N-1} \geq 20 C(t)$.
\end{lemma}
\begin{proof}
  First we claim that we may take $B(1) = \frac{\min\{\eta, 1\}}{10} A$ and $C(1) = \max\{ \eta, 1\} A$. We have shown above that $B(1) \leq W_j(1) \leq C(1)$ for all $j$. Next we establish that $\eta B(1)^{2N-1} \geq 20 C(1)$. If $\eta \leq 1$, then
  $$
  \eta B(1)^{2N-1} =\eta^{2N} \cdot (A/10)^{2N-1} \geq 20 A = 20 C(1),
  $$
  where the inequality follows from $A \geq \left(\frac{20 \cdot 10^{2N-1}}{\eta^{2N}}\right)^{1/(2N-2)}$ by definition of $A$. If $\eta \geq 1$, then
  $$
  \eta B(1)^{2N-1} = \eta (A/10)^{2N-1} \geq 20 \eta A = 20 C(1),
  $$
  where the inequality follows from $A \geq 2000 \geq \left(20 \cdot 10^{2N-1}\right)^{1/(2N-2)}$ by definition of $A$.

  Now, suppose the statement of Lemma \ref{lemma:weight_explode} holds for some $t$. Suppose first that $W_j(t)$ are all positive for $1 \leq j \leq N$. Then for all $j$, as $B(t) \geq 2$, and $\eta B(t)^{2N-1} \geq 20 C(t)$,
  \begin{eqnarray}
    W_j(t+1) & \leq & C(t) - \eta \cdot (B(t)^N - 1) \cdot B(t)^{N-1} \nonumber\\
             & \leq & C(t) - \frac{\eta}{2} B(t)^{2N-1}\nonumber\\
             & \leq & -9 C(t)\nonumber,
  \end{eqnarray}
  which establishes that $W_j(t+1)$ is negative for all $j$.
  Moreover,
  \begin{eqnarray}
    W_j(t+1) & \geq & -\eta (C(t)^N - 1) \cdot C(t)^{N-1} \nonumber\\
             & \geq & -\eta C(t)^{2N-1}. \nonumber
  \end{eqnarray}
  Now set $B(t+1) = 9C(t)$ and $C(t+1) = \eta C(t)^{2N-1}$. Since $N \geq 2$, we have that
  $$
\eta B(t+1)^{2N-1} = \eta (9C(t))^{2N-1} \geq \eta 9^3 C(t)^{2N-1} > 20 \eta C(t)^{2N-1} = 20 C(t+1).
$$
The case that all $W_j(t)$ are negative for $1 \leq j \leq N$ is nearly identical, with the same values for $B(t+1), C(t+1)$ in terms of $B(t), C(t)$, except all $W_{j}(t+1)$ will be positive.
This establishes the inductive step and completes the proof of Lemma \ref{lemma:weight_explode}.
\end{proof}
By Lemma \ref{lemma:weight_explode}, we have that for all $t \geq 1$, $L^N(W_1(t), \ldots, W_N(t)) = \frac 12 (W_{1:N}(t) - 1)^2 \geq \frac 12 (2^N - 1)^2 > 0$, thus completing the proof of Claim \ref{claim:diverge_balance} for the case where all dimensions are equal to 1.

For the general case where $d_0 = d_1 = \cdots = d_N = d$ for some $d \geq 1$, we set $\Phi = I_d$, and given $c, \eta$, we set $W_j(0)$ to be the $d \times d$ diagonal matrix where all diagonal entries except the first one are equal to 1, and where the first diagonal entry is given by Equation~\eqref{eq:define_wj0}, where $A$ is given by Equation~\eqref{eq:define_A}. It is easily verified that all entries of $W_j(t)$, $1 \leq j \leq N$, except for the first diagonal element of each matrix, will remain constant for all $t \geq 0$, and that the first diagonal elements evolve exactly as in the 1-dimensional case presented above. Therefore the loss in the $d$-dimensional case is equal to the loss in the 1-dimensional case, which is always greater than some positive constant.
\end{proof}
We remark that the proof of Claim \ref{claim:diverge_balance} establishes that the loss $\ell(t):=L^N(W_1(t), \ldots, W_N(t))$ grows at least exponentially in $t$ for the chosen initialization. Such behavior, in which gradients and weights explode, indeed takes place in deep learning practice if initialization is not chosen with care.

\subsection{Proof of Claim~\ref{claim:diverge_margin}} \label{app:proofs:diverge_margin}

\begin{proof}
We will show that a target matrix $\Phi\in\R^{d\times{d}}$ which is symmetric with at least one negative eigenvalue, along with identity initialization ($W_j(0) = I_d$, $\forall{j}\in\{1,\ldots,N\}$), satisfy the conditions of the claim.
First, note that non-stationarity of initialization is met, as for any $1 \leq j \leq N$,
  $$
\frac{\partial L^N(W_1(0), \ldots, W_N(0))}{\partial W_j(0)} = W_{j+1:N}(0)^\top (W_{1:N}(0) - \Phi) W_{1:j-1}(0) = I_d - \Phi \neq \mathbf{0},
$$
where the last inequality follows since $\Phi$ has a negative eigenvalue.
  To analyze gradient descent we use the following result, which was established in \cite{bartlett2018gradient}:
  \begin{lemma}[\cite{bartlett2018gradient}, Lemma 6]
    \label{lemma:failure_idinit}
    If $W_1(0), \ldots, W_N(0)$ are all initialized to identity, $\Phi$ is symmetric, $\Phi = UDU^\top$ is a diagonalization of $\Phi$, and gradient descent is performed with any learning rate, then for each $t \geq 0$ there is a diagonal matrix $\hat D(t)$ such that $W_j(t) = U\hat D(t) U^\top$ for each $1 \leq j \leq N$.
  \end{lemma}
  By Lemma \ref{lemma:failure_idinit}, for any choice of learning rate $\eta$, the end-to-end matrix at time $t$ is given by $W_{1:N}(t) = U \hat D(t)^N U^\top$. As long as some diagonal element of $D$ is negative, say equal to $-\lambda < 0$, then
  $$
\ell(t)=L^N(W_1(t), \ldots, W_N(t)) = \frac 12 \| W_{1:N}(t) - \Phi\|_F^2 = \frac 12 \| \hat D(t)^L - D \|_F^2 \geq \frac 12 \lambda^2 > 0.
  $$
\end{proof}

\section{$\ell_2$ Loss over Whitened Data} \label{app:whitened}

Recall the $\ell_2$~loss of a linear predictor~$W\in\R^{d_y\times{d}_x}$ as defined in Section~\ref{sec:prelim}:
$$L(W)=\frac{1}{2m}\|WX-Y\|_F^2
\text{\,,}$$
where~$X\in\R^{d_x\times{m}}$ and~$Y\in\R^{d_y\times{m}}$.
Define $\Lambda_{xx}:=\tfrac{1}{m}XX^\top\in\R^{d_x\times{d}_x}$, $\Lambda_{yy}:=\tfrac{1}{m}YY^\top\in\R^{d_y\times{d}_y}$ and $\Lambda_{yx}:=\tfrac{1}{m}YX^\top\in\R^{d_y\times{d}_x}$.
Using the relation $\norm{A}_F^2=\Tr(AA^\top)$, we have:
\beas
L(W)&=&\tfrac{1}{2m}\Tr\big((WX-Y)(WX-Y)^\top\big) \\[1mm]
&=&\tfrac{1}{2m}\Tr(WXX^\top{W}^\top)-\tfrac{1}{m}\Tr(WXY^\top)+\tfrac{1}{2m}\Tr(YY^\top) \\[1mm]
&=&\tfrac{1}{2}\Tr(W\Lambda_{xx}{W}^\top)-\Tr(W\Lambda_{yx}^\top)+\tfrac{1}{2}\Tr(\Lambda_{yy})
\text{\,.}
\eeas
By definition, when data is whitened, $\Lambda_{xx}$~is equal to identity, yielding:
\beas
L(W)&=&\tfrac{1}{2}\Tr(W{W}^\top)-\Tr(W\Lambda_{yx}^\top)+\tfrac{1}{2}\Tr(\Lambda_{yy}) \\[1mm]
&=&\tfrac{1}{2}\Tr\big((W-\Lambda_{yx})(W-\Lambda_{yx})^\top\big)-\tfrac{1}{2}\Tr(\Lambda_{yx}\Lambda_{yx}^\top)+\tfrac{1}{2}\Tr(\Lambda_{yy}) \\[1mm]
&=&\tfrac{1}{2}\norm{W-\Lambda_{yx}}_F^2+c
\text{\,,}
\eeas
where~$c:=-\tfrac{1}{2}\Tr(\Lambda_{yx}\Lambda_{yx}^\top)+\tfrac{1}{2}\Tr(\Lambda_{yy})$ does not depend on~$W$.
Hence we arrive at Equation~\eqref{eq:loss_whitened}.

\subsubsection*{Acknowledgments}
		\else
			\section*{Acknowledgments}
		\fi
		\acknowledgments
	\fi
\fi

\section*{References}
{\small
\ifdefined\ICML
	\bibliographystyle{icml2018}
\else
	\bibliographystyle{plainnat}
\fi

\section{Conclusion} \label{sec:conc}

For deep linear neural networks, we have rigorously proven convergence of gradient descent to global minima, at a linear rate, provided that the initial weight matrices are approximately balanced and the initial end-to-end matrix has positive deficiency margin.
The result applies to networks with arbitrary depth, and any configuration of input/output/hidden dimensions that supports full rank, \ie~in which no hidden layer has dimension smaller than both the input and output.

Our assumptions on initialization~---~approximate balancedness and deficiency margin~---~are both necessary, in the sense that violating any one of them may lead to convergence failure, as we demonstrated explicitly.
Moreover, for networks with output dimension~$1$ (scalar regression), we have shown that a balanced initialization, \ie~a random choice of the end-to-end matrix followed by a balanced partition across all layers, leads assumptions to be met, and thus convergence to take place, with constant probability.
Rigorously proving efficient convergence with significant probability under customary layer-wise independent initialization remains an open problem.
The recent work of~\citet{shamir2018exponential} suggests that this may not be possible, as at least in some settings, the number of iterations required for convergence is exponential in depth with overwhelming probability.
This negative result, a theoretical manifestation of the ``vanishing gradient problem'', is circumvented by balanced initialization.
Through simple experiments we have shown that the latter can lead to favorable convergence in deep learning practice, as it does in theory.
Further investigation of balanced initialization, including development of variants for convolutional layers, is regarded as a promising direction for future research.

The analysis in this paper uncovers special properties of the optimization landscape in the vicinity of gradient descent trajectories.
We expect similar ideas to prove useful in further study of gradient descent on non-convex objectives, including training losses of deep non-linear neural networks.
\section{Convergence Analysis} \label{sec:converge}

In this section we establish convergence of gradient descent for deep linear neural networks (Equations~\eqref{eq:gd} and~\eqref{eq:lnn_obj}) by directly analyzing the trajectories taken by the algorithm.
We begin in Subsection~\ref{sec:converge:balance_margin} with a presentation of two concepts central to our analysis: \emph{approximate balancedness} and \emph{deficiency margin}.
These facilitate our main convergence theorem, delivered in Subsection~\ref{sec:converge:theorem}.
We conclude in Subsection~\ref{sec:converge:balance_init} by deriving a convergence guarantee that holds with constant probability over a random initialization.

\subsection{Approximate Balancedness and Deficiency Margin} \label{sec:converge:balance_margin}

In our context, the notion of approximate balancedness is formally defined as follows:
\vspace{1mm}
\begin{definition}
\label{def:balance}
For~$\delta\geq0$, we say that the matrices $W_j\in\R^{d_j\times{d}_{j-1}}$, $j{=}1,\ldots,N$, are $\delta$-\emph{balanced} if:
$$\norm{W_{j+1}^{\top}W_{j+1}-W_{j}W_j^\top}_F\leq\delta\quad,\,\forall{j}\in\{1,\ldots,N-1\}
\text{\,.}$$
\end{definition}
Note that in the case of $0$-balancedness, \ie~$W_{j+1}^{\top}W_{j+1}=W_{j}W_j^\top$, $\forall{j}\in\{1,\ldots,N-1\}$, all matrices~$W_j$ share the same set of non-zero singular values.
Moreover, as shown in the proof of Theorem~1 in~\citet{arora2018optimization}, this set is obtained by taking the $N$-th root of each non-zero singular value in the end-to-end matrix~$W_{1:N}$.
We will establish approximate versions of these facts for $\delta$-balancedness with $\delta>0$, and admit their usage by showing that if the weights of a linear neural network are initialized to be approximately balanced, they will remain that way throughout the iterations of gradient descent.
The condition of approximate balancedness at initialization is trivially met in the special case of linear residual networks ($d_0=\cdots=d_N=d$ and $W_1(0)=\cdots=W_N(0)=I_d$).
Moreover, as Claim~\ref{claim:balance_init} in Appendix~\ref{app:balance_margin_init} shows, for a given~$\delta>0$, the customary initialization via random Gaussian distribution with mean zero leads to approximate balancedness with high probability if the standard deviation is sufficiently small.

\medskip

The second concept we introduce~---~deficiency margin~---~refers to how far a ball around the target is from containing rank-deficient (\ie~low rank) matrices.
\vspace{1mm}
\begin{definition}
\label{def:margin}
Given a target matrix~$\Phi\in\R^{d_N\times{d}_0}$ and a constant $c>0$, we say that a matrix~$W\in\R^{d_N\times{d}_0}$ has \emph{deficiency margin~$c$ with respect to~$\Phi$} if:\note{
	Note that deficiency margin~$c>0$ with respect to~$\Phi$ implies~$\sigma_{min}(\Phi)>0$, \ie~$\Phi$ has full rank.
	Our analysis can be extended to account for rank-deficient~$\Phi$ by replacing~$\sigma_{min}(\Phi)$ in Equation~\eqref{eq:margin} with the smallest positive singular value of~$\Phi$, and by requiring that the end-to-end matrix~$W_{1:N}$ be initialized such that its left and right null spaces coincide with those of~$\Phi$.
	Relaxation of this requirement is a direction for future work.
}
\be
\norm{W-\Phi}_F\leq\sigma_{min}(\Phi)-c
\label{eq:margin}
\text{\,.}
\ee
\end{definition}
The term ``deficiency margin'' alludes to the fact that if Equation~\eqref{eq:margin} holds, every matrix~$W'$ whose distance from~$\Phi$ is no greater than that of~$W$, has singular values $c$-bounded away from zero:
\vspace{1mm}
\begin{claim}
\label{claim:margin_interp}
Suppose~$W$ has deficiency margin~$c$ with respect to~$\Phi$.
Then, any matrix~$W'$ (of same size as~$\Phi$ and~$W$) for which $\norm{W'-\Phi}_F\leq\norm{W-\Phi}_F$ satisfies~$\sigma_{min}(W')\geq{c}$.
\end{claim}
\vspace{-3mm}
\begin{proof}
Our proof relies on the inequality $\sigma_{min}(A+B)\geq\sigma_{min}(A)-\sigma_{max}(B)$~---~see Appendix~\ref{app:proofs:margin_interp}.
\end{proof}
\vspace{-2mm}
We will show that if the weights $W_1,\ldots,W_N$ are initialized such that (they are approximately balanced and) the end-to-end matrix~$W_{1:N}$ has deficiency margin~$c>0$ with respect to the target~$\Phi$, convergence of gradient descent to global minimum is guaranteed.\note{
	In fact, a deficiency margin implies that all critical points in the respective sublevel set (set of points with smaller loss value) are global minima.
	This however is far from sufficient for proving convergence, as sublevel sets are unbounded, and the loss landscape over them is non-convex and non-smooth.
	Indeed, we show in Appendix~\ref{app:fail} that deficiency margin alone is not enough to ensure convergence~---~without approximate balancedness, the lack of smoothness can cause divergence.
}
Moreover, the convergence will outpace a particular rate that gets faster when $c$ grows larger.
This suggests that from a theoretical perspective, it is advantageous to initialize a linear neural network such that the end-to-end matrix has a large deficiency margin with respect to the target.
Claim~\ref{claim:margin_init} in Appendix~\ref{app:balance_margin_init} provides information on how likely deficiency margins are in the case of a single output model (scalar regression) subject to customary zero-centered Gaussian initialization.
It shows in particular that if the standard deviation of the initialization is sufficiently small, the probability of a deficiency margin being met is close to~$0.5$; on the other hand, for this deficiency margin to have considerable magnitude, a non-negligible standard deviation is required.

\medskip

Taking into account the need for both approximate balancedness and deficiency margin at initialization, we observe a delicate trade-off under the common setting of Gaussian perturbations around zero:
if the standard deviation is small, it is likely that weights be highly balanced and a deficiency margin be met;
however overly small standard deviation will render high magnitude for the deficiency margin improbable, and therefore fast convergence is less likely to happen;
on the opposite end, large standard deviation jeopardizes both balancedness and deficiency margin, putting the entire convergence at risk.
This trade-off is reminiscent of empirical phenomena in deep learning, by which small initialization can bring forth efficient convergence, while if exceedingly small, rate of convergence may plummet (``vanishing gradient problem''), and if made large, divergence becomes inevitable (``exploding gradient problem'').
The common resolution of residual connections~\citep{he2016deep} is analogous in our context to linear residual networks, which ensure perfect balancedness, and allow large deficiency margin if the target is not too far from identity.

\subsection{Main Theorem} \label{sec:converge:theorem}

Using approximate balancedness (Definition~\ref{def:balance}) and deficiency margin (Definition~\ref{def:margin}), we present our main theorem~---~a guarantee for linear convergence to global minimum:
\vspace{1mm}
\begin{theorem}
\label{theorem:converge}
Assume that gradient descent is initialized such that the end-to-end matrix~$W_{1:N}(0)$ has deficiency margin~$c>0$ with respect to the target~$\Phi$, and the weights $W_1(0),\ldots,W_N(0)$ are $\delta$-balanced with $\delta=c^2\big/\big(256\cdot{N}^3\cdot\norm{\Phi}_{F}^{2(N-1)/N}\big)$.
Suppose also that the learning rate~$\eta$ meets:
\be
\eta\leq\frac{c^{(4N-2)/N}}{6144\cdot{N}^3\cdot\norm{\Phi}_{F}^{(6N-4)/N}}
\label{eq:descent_eta}
\text{~.}
\ee
Then, for any~$\epsilon>0$ and:
\be
T\geq\frac{1}{\eta\cdot{c}^{2(N-1)/N}}\cdot\log\left(\frac{\ell(0)}{\epsilon}\right)
\label{eq:converge_t}
\text{\,,}
\ee
the loss at iteration~$T$ of gradient descent~---~$\ell(T)$~---~is no greater than~$\epsilon$.
\end{theorem}

\subsubsection{On the Assumptions Made}

The assumptions made in Theorem~\ref{theorem:converge}~---~approximate balancedness and deficiency margin at initialization~---~are both necessary, in the sense that violating any one of them may lead to convergence failure.
We demonstrate this in Appendix~\ref{app:fail}.
In the special case of linear residual networks (uniform dimensions and identity initialization), a sufficient condition for the assumptions to be met is that the target matrix have (Frobenius) distance less than~$0.5$ from identity.
This strengthens one of the central results in~\citet{bartlett2018gradient} (see Section~\ref{sec:related}).
For a setting of random near-zero initialization, we present in Subsection~\ref{sec:converge:balance_init} a scheme that, when the output dimension is~$1$ (scalar regression), ensures assumptions are satisfied (and therefore gradient descent efficiently converges to global minimum) with constant probability.
It is an open problem to fully analyze gradient descent under the common initialization scheme of zero-centered Gaussian perturbations applied to each layer independently.
We treat this scenario in Appendix~\ref{app:balance_margin_init}, providing quantitative results concerning the likelihood of each assumption (approximate balancedness or deficiency margin) being met individually.
However the question of how likely it is that both assumptions be met simultaneously, and how that depends on the standard deviation of the Gaussian, is left for future work.

An additional point to make is that Theorem~\ref{theorem:converge} poses a structural limitation on the linear neural network.
Namely, it requires the dimension of each hidden layer~($d_i,~i=1,\ldots,N-1$) to be greater than or equal to the minimum between those of the input~($d_0$) and output~($d_N$).
Indeed, in order for the initial end-to-end matrix~$W_{1:N}(0)$ to have deficiency margin~$c>0$, it must (by Claim~\ref{claim:margin_interp}) have full rank, and this is only possible if there is no intermediate dimension~$d_i$ smaller than~$\min\{d_0,d_N\}$.
We make no other assumptions on network architecture (depth, input/output/hidden dimensions).

\subsubsection{Proof}

The cornerstone upon which Theorem~\ref{theorem:converge} rests is the following lemma, showing non-trivial descent whenever $\sigma_{min}(W_{1:N})$ is bounded away from zero:
\vspace{1mm}
\begin{lemma}
\label{lemma:descent}
Under the conditions of Theorem~\ref{theorem:converge}, we have that for every $t=0,1,2,\ldots$~:\note{
	Note that the term~$\frac{dL^1}{dW}(W_{1:N}(t))$ below stands for the gradient of~$L^1(\cdot)$~---~a convex loss over (directly parameterized) linear models (Equation~\eqref{eq:lin_obj})~---~at the point~$W_{1:N}(t)$~---~the end-to-end matrix of the network at iteration~$t$.
	It is therefore (see Equation~\eqref{eq:gd_loss}) non-zero anywhere but at a global minimum.
}
\be
\ell(t+1)\leq\ell(t)-\frac{\eta}{2}\cdot\sigma_{min}\big(W_{1:N}(t)\big)^{\frac{2(N-1)}{N}}\cdot\norm{\frac{dL^1}{dW}\big(W_{1:N}(t)\big)}_F^2
\label{eq:descent}
\text{~.}
\ee
\end{lemma}
\begin{proof}[Proof of Lemma~\ref{lemma:descent} (in idealized setting; for complete proof see Appendix~\ref{app:proofs:descent})]
\hspace{-0.5mm}We prove the lemma here for the idealized setting of perfect initial balancedness~($\delta=0$):
$$W_{j+1}^{\top}(0)W_{j+1}(0)=W_{j}(0)W_j^\top(0)\quad,~\forall{j}\in\{1,\ldots,N-1\}
\text{~,}$$
and infinitesimally small learning rate~($\eta\to0^+$)~---~\emph{gradient flow}:
$$\dot{W}_j(\tau)=-\frac{\partial{L}^N}{\partial{W}_j}\big(W_1(\tau),\ldots,W_N(\tau)\big)\quad,~j=1,\ldots,N\quad,~\tau\in[0,\infty)
\text{~,}$$
where~$\tau$ is a continuous time index, and dot symbol (in~$\dot{W}_j(\tau)$) signifies derivative with respect to time.
The complete proof, for the realistic case of approximate balancedness and discrete updates~($\delta,\eta>0$), is similar but much more involved, and appears in Appendix~\ref{app:proofs:descent}.

Recall that~$\ell(t)$~---~the objective value at iteration~$t$ of gradient descent~---~is equal to~$L^1(W_{1:N}(t))$ (see Equation~\eqref{eq:gd_loss}).
Accordingly, for the idealized setting in consideration, we would like to show:
\be
\frac{d}{d\tau}L^1\left(W_{1:N}(\tau)\right)\leq-\frac{1}{2}\sigma_{min}\big(W_{1:N}(\tau)\big)^{\frac{2(N-1)}{N}}\cdot\norm{\frac{dL^1}{dW}\big(W_{1:N}(\tau)\big)}_F^2
\label{eq:descent_ideal}
\text{\,.}
\ee
We will see that a stronger version of Equation~\eqref{eq:descent_ideal} holds, namely, one without the $1/2$ factor (which only appears due to discretization).

By (Theorem~$1$ and Claim~$1$ in)~\citet{arora2018optimization}, the weights $W_1(\tau),\ldots,W_N(\tau)$ remain balanced throughout the entire optimization, and that implies the end-to-end matrix~$W_{1:N}(\tau)$ moves according to the following differential equation:
\be
vec\left(\dot{W}_{1:N}(\tau)\right)=-{P}_{W_{1:N}(\tau)}\cdot{vec}\left(\frac{dL^1}{dW}\left(W_{1:N}(\tau)\right)\right)
\label{eq:e2e_gf}
\text{\,,}
\ee
where~$vec(A)$, for an arbitrary matrix~$A$, stands for vectorization in column-first order, and $P_{W_{1:N}(\tau)}$ is a positive semidefinite matrix whose eigenvalues are all greater than or equal to $\sigma_{min}(W_{1:N}(\tau))^{2(N-1)/N}$.
Taking the derivative of~$L^1(W_{1:N}(\tau))$ with respect to time, we obtain the sought-after Equation~\eqref{eq:descent_ideal} (with no $1/2$ factor):
\beas
\frac{d}{d\tau}L^1\left(W_{1:N}(\tau)\right)&=&\inprod{vec\left(\frac{dL^1}{dW}\big(W_{1:N}(\tau)\big)\right)}{vec\left(\dot{W}_{1:N}(\tau)\right)} \\
&=&\inprod{vec\left(\frac{dL^1}{dW}\big(W_{1:N}(\tau)\big)\right)}{-{P}_{W_{1:N}(\tau)}\cdot{vec}\left(\frac{dL^1}{dW}\left(W_{1:N}(\tau)\right)\right)} \\
&\leq&-\sigma_{min}\big(W_{1:N}(\tau)\big)^{\frac{2(N-1)}{N}}\cdot\norm{vec\left(\frac{dL^1}{dW}\big(W_{1:N}(\tau)\big)\right)}^2 \\
&=&-\sigma_{min}\big(W_{1:N}(\tau)\big)^{\frac{2(N-1)}{N}}\cdot\norm{\frac{dL^1}{dW}\big(W_{1:N}(\tau)\big)}_F^2
\text{\,.}
\eeas
The first transition here (equality) is an application of the chain rule;
the second (equality) plugs in Equation~\eqref{eq:e2e_gf};
the third (inequality) results from the fact that the eigenvalues of the symmetric matrix $P_{W_{1:N}(\tau)}$ are no smaller than $\sigma_{min}(W_{1:N}(\tau))^{2(N-1)/N}$ (recall that~$\norm{\cdot}$ stands for Euclidean norm);
and the last (equality) is trivial~---~$\norm{A}_F=\norm{vec(A)}$ for any matrix~$A$.
\end{proof}

With Lemma~\ref{lemma:descent} established, the proof of Theorem~\ref{theorem:converge} readily follows:
\begin{proof}[Proof of Theorem~\ref{theorem:converge}]
By the definition of~$L^1(\cdot)$ (Equation~\eqref{eq:lin_obj}), for any~$W\in\R^{d_N\times{d}_0}$:
$$\frac{dL^1}{dW}(W)=W-\Phi\quad\implies\quad\norm{\frac{dL^1}{dW}(W)}_F^2=2\cdot{L}^1(W)
\text{~.}$$
Plugging this into Equation~\eqref{eq:descent} while recalling that $\ell(t)=L^1(W_{1:N}(t))$ (Equation~\eqref{eq:gd_loss}), we have (by Lemma~\ref{lemma:descent}) that for every $t=0,1,2,\ldots$~:
$$L^1\big(W_{1:N}(t+1)\big)\leq{L}^1\big(W_{1:N}(t)\big)\cdot\Big(1-\eta\cdot\sigma_{min}\big(W_{1:N}(t)\big)^{\frac{2(N-1)}{N}}\Big)
\text{~.}$$
Since the coefficients $1-\eta\cdot\sigma_{min}(W_{1:N}(t))^{\frac{2(N-1)}{N}}$ are necessarily non-negative (otherwise would contradict non-negativity of $L^1(\cdot)$), we may unroll the inequalities, obtaining:
\be
L^1\big(W_{1:N}(t+1)\big)\leq{L}^1\big(W_{1:N}(0)\big)\cdot\prod\nolimits_{t'=0}^{t}\Big(1-\eta\cdot\sigma_{min}\big(W_{1:N}(t')\big)^{\frac{2(N-1)}{N}}\Big)
\text{\,.}
\label{eq:descent_concat}
\ee
Now, this in particular means that for every~$t'=0,1,2,\ldots$~:
$$L^1\big(W_{1:N}(t')\big)\leq{L}^1\big(W_{1:N}(0)\big)\quad\implies\quad\norm{W_{1:N}(t')-\Phi}_F\leq\norm{W_{1:N}(0)-\Phi}_F \text{\,.}$$
Deficiency margin~$c$ of~$W_{1:N}(0)$ along with Claim~\ref{claim:margin_interp} thus imply $\sigma_{min}\big(W_{1:N}(t')\big)\geq{c}$, which when inserted back into Equation~\eqref{eq:descent_concat} yields, for every~$t=1,2,3,\ldots$~:
\be
L^1\big(W_{1:N}(t)\big)\leq{L}^1\big(W_{1:N}(0)\big)\cdot\Big(1-\eta\cdot{c}^{\frac{2(N-1)}{N}}\Big)^t
\label{eq:descent_geo}
\text{~.}
\ee
$\eta\cdot{c}^{\frac{2(N-1)}{N}}$ is obviously non-negative, and it is also no greater than~$1$ (otherwise would contradict non-negativity of $L^1(\cdot)$).
We may therefore incorporate the inequality $1-\eta\cdot{c}^{2(N-1)/N}\leq\exp\big(-\eta\cdot{c}^{2(N-1)/N}\big)$ into Equation~\eqref{eq:descent_geo}:
$$L^1\big(W_{1:N}(t)\big)\leq{L}^1\big(W_{1:N}(0)\big)\cdot\exp\big(-\eta\cdot{c}^{2(N-1)/N}\cdot{t}\big)
\text{\,,}$$
from which it follows that $L^1(W_{1:N}(t))\leq\epsilon$ if:
$$t\geq\frac{1}{\eta\cdot{c}^{2(N-1)/N}}\cdot\log\left(\frac{L^1(W_{1:N}(0))}{\epsilon}\right)
\text{\,.}$$
Recalling again that $\ell(t)=L^1(W_{1:N}(t))$ (Equation~\eqref{eq:gd_loss}), we conclude the proof.
\end{proof}

\subsection{Balanced Initialization} \label{sec:converge:balance_init}

We define the following procedure, \emph{balanced initialization}, which assigns weights randomly while ensuring perfect balancedness:
\vspace{1mm}
\begin{procedure}[Balanced initialization]
\label{proc:balance_init}
Given $d_0, d_1, \ldots, d_N \in \N$ such that $\min\{d_1,\ldots,d_{N-1}\}\geq\min\{d_0,d_N\}$ and a distribution~$\D$ over $d_N\times{d}_0$~matrices, a \emph{balanced initialization} of~$W_j\in\R^{d_j\times{d}_{j-1}}$, $j{=}1,\ldots,N$, assigns these weights as follows:
\vspace{-1.5mm}
\begin{enumerate}[label=(\roman*)]
\item Sample $A\in\R^{d_N\times{d}_0}$ according to~$\D$.
\vspace{-1mm}
\item Take singular value decomposition~$A=U\Sigma{V}^\top$, where $U \in \R^{d_N \times \min\{d_0, d_N\}}$, $V \in \R^{d_0 \times \min\{d_0, d_N\}}$ have orthonormal columns, and $\Sigma \in \R^{\min\{d_0, d_N\} \times \min\{d_0, d_N\}}$ is diagonal and holds the singular values of $A$.
\vspace{-1mm}
\item Set $W_N\simeq{U}\Sigma^{1/N},W_{N-1}\simeq\Sigma^{1/N},\ldots,W_2\simeq\Sigma^{1/N},W_1\simeq\Sigma^{1/N}V^\top$, where the symbol~``$\simeq$'' stands for equality up to zero-valued padding.\note{
	These assignments can be accomplished since $\min\{d_1,\ldots,d_{N-1}\}\geq\min\{d_0,d_N\}$.
}
\note{
	By design $W_{1:N}=A$ and $W_{j+1}^{\top}W_{j+1}=W_{j}W_j^\top$, $\forall{j}\in\{1,\ldots,N{-}1\}$~---~these properties are actually all we need in Theorem~\ref{theorem:converge_balance_init}, and step~\emph{(iii)} in Procedure~\ref{proc:balance_init} can be replaced by any assignment that meets them.
	\label{note:balance_init_props}
}
\end{enumerate}
\end{procedure}

The concept of balanced initialization, together with Theorem~\ref{theorem:converge}, leads to a guarantee for linear convergence (applicable to output dimension~$1$~---~scalar regression) that holds with constant probability over the randomness in initialization:
\vspace{1mm}
\begin{theorem}
\label{theorem:converge_balance_init}
For any constant $0<p<1/2$, there are constants $d_0',a>0$ \note{
	As shown in the proof of the theorem (Appendix \ref{app:proofs:converge_balance_init}), $d_0',a>0$ can take on any pair of values for which:
	\emph{(i)}~$d_0'\geq20$;
	and~\emph{(ii)}~$\big(1 - 2\exp(-d_0'/16)\big)\big(3-4F(2/\sqrt{a/2})\big) \geq 2p$, where $F(\cdot)$ stands for the cumulative distribution function of the standard normal distribution.
	For example, if~$p=0.25$, it suffices to take any~$d_0'\geq100,a\geq100$.
	We note that condition~\emph{(i)} here ($d_0'\geq20$) serves solely for simplification of expressions in the theorem.
}
such that the following holds.
Assume $d_N=1,d_0\geq d_0'$, and that the weights $W_1(0),\ldots,W_N(0)$ are subject to balanced initialization (Procedure~\ref{proc:balance_init}) such that the entries in $W_{1:N}(0)$ are independent zero-centered Gaussian perturbations with standard deviation $s\leq \|\Phi\|_2/\sqrt{ad_0^2}$.
Suppose also that we run gradient descent with learning rate $  \eta \leq (s^2d_0)^{4-2/N}\big/\big(10^5 N^3 \| \Phi\|_2^{10-6/N}\big)$.
Then, with probability at least~$p$ over the random initialization, we have that for every~$\epsilon>0$ and:
$$
T \geq \frac4\eta \left( \ln(4) \left(\frac{\|\Phi\|_2}{s^2d_0}\right)^{2-2/N}   + \| \Phi\|_2^{2/N-2}\ln(\| \Phi\|_2^2/(8\ep)) \right)
\text{\,,}
$$
the loss at iteration~$T$ of gradient descent~---~$\ell(T)$~---~is no greater than~$\epsilon$.
\end{theorem}
\vspace{-2mm}
\begin{proof}
See Appendix \ref{app:proofs:converge_balance_init}.
\end{proof}

\section{Experiments} \label{sec:exper}

Balanced initialization (Procedure~\ref{proc:balance_init}) possesses theoretical advantages compared with the customary layer-wise independent scheme~---~it allowed us to derive a convergence guarantee that holds with constant probability over the randomness of initialization (Theorem~\ref{theorem:converge_balance_init}).
In this section we present empirical evidence suggesting that initializing with balancedness may be beneficial in practice as well.
For conciseness, some of the details behind our implementation are deferred to Appendix~\ref{app:impl}.

We began by experimenting in the setting covered by our analysis~---~linear neural networks trained via gradient descent minimization of $\ell_2$~loss over whitened data.
The dataset chosen for the experiment was UCI Machine Learning Repository's ``Gas Sensor Array Drift at Different Concentrations'' \citep{vergara2012chemical,rodriguez2014calibration}.
Specifically, we used the dataset's ``Ethanol'' problem~---~a scalar regression task with~$2565$ examples, each comprising~$128$ features (one of the largest numeric regression tasks in the repository).
Starting with the customary initialization of layer-wise independent random Gaussian perturbations centered at zero, we trained a three layer network ($N=3$) with hidden widths ($d_1,d_2$) set to~$32$, and measured the time (number of iterations) it takes to converge (reach training loss within $\epsilon=10^{-5}$ from optimum) under different choices of standard deviation for the initialization.
To account for the possibility of different standard deviations requiring different learning rates (values for~$\eta$), we applied, for each standard deviation independently, a grid search over learning rates, and recorded the one that led to fastest convergence.
The result of this test is presented in Figure~\ref{fig:exper}(a).
As can be seen, there is a range of standard deviations that leads to fast convergence (a few hundred iterations or less), below and above which optimization decelerates by orders of magnitude.
This accords with our discussion at the end of Subsection~\ref{sec:converge:balance_init}, by which overly small initialization ensures approximate balancedness (small~$\delta$; see Definition~\ref{def:balance}) but diminishes deficiency margin (small~$c$; see Definition~\ref{def:margin})~---~``vanishing gradient problem''~---~whereas large initialization hinders both approximate balancedness and deficiency margin~---~``exploding gradient problem''.
In that regard, as a sanity test for the validity of our analysis, in a case where approximate balancedness is met at initialization (small standard deviation), we measured its persistence throughout optimization.
As Figure~\ref{fig:exper}(c) shows, our theoretical findings manifest themselves here~---~trajectories of gradient descent indeed preserve weight balancedness.

In addition to a three layer network, we also evaluated a deeper, eight layer model (with hidden widths identical to the former~---~$N=8$, $d_1=\cdots=d_7=32$).
In particular, using the same experimental protocol as above, we measured convergence time under different choices of standard deviation for the initialization.
Figure~\ref{fig:exper}(a) displays the result of this test alongside that of the three layer model.
As the figure shows, transitioning from three layers to eight aggravated the instability with respect to initialization~---~there is now a narrow band of standard deviations that lead to convergence in reasonable time, and outside of this band convergence is extremely slow, to the point where it does not take place within the duration we allowed ($10^6$~iterations).
From the perspective of our analysis, a possible explanation for the aggravation is as follows: under layer-wise independent initialization, the magnitude of the end-to-end matrix~$W_{1:N}$ depends on the standard deviation in a manner that is exponential in depth, thus for large depths the range of standard deviations that lead to moderately sized~$W_{1:N}$ (as required for a deficiency margin) is limited, and within this range, there may not be many standard deviations small enough to ensure approximate balancedness.
The procedure of balanced initialization (Procedure~\ref{proc:balance_init}) circumvents these difficulties~---~it assigns~$W_{1:N}$ directly (no exponential dependence on depth), and distributes its content between the individual weights $W_1,\ldots,W_N$ in a perfectly balanced fashion.
Rerunning the experiment of Figure~\ref{fig:exper}(a) with this initialization replacing the customary layer-wise scheme (using same experimental protocol), we obtained the results shown in Figure~\ref{fig:exper}(b)~---~both the original three layer network, and the deeper eight layer model, converged quickly under virtually all standard deviations tried.

As a final experiment, we evaluated the effect of balanced initialization in a setting that involves non-linear activation, softmax-cross-entropy loss and stochastic optimization (factors not accounted for by our analysis).
For this purpose, we turned to the MNIST tutorial built into TensorFlow \citep{abadi2016tensorflow},\note{
	\url{https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/tutorials/mnist}
}
which comprises a fully-connected neural network with two hidden layers (width~$128$ followed by~$32$) and ReLU activation \citep{nair2010rectified}, trained through stochastic gradient descent (over softmax-cross-entropy loss) with batch size~$100$, initialized via customary layer-wise independent Gaussian perturbations centered at zero.
While keeping the learning rate at its default value $0.01$, we varied the standard deviation of initialization, and for each value measured the training loss after $10$~epochs.\note{
	As opposed to the dataset used in our experiments with linear networks, measuring the training loss with MNIST is non-trivial computationally (involves passing through $60K$~examples).
	Therefore, rather than continuously polling training loss until it reaches a certain threshold, in this experiment we chose to evaluate speed of convergence by measuring the training loss once after a predetermined number of iterations.
}
We then replaced the original (layer-wise independent) initialization with a balanced initialization based on Gaussian perturbations centered at zero (latter was implemented per Procedure~\ref{proc:balance_init}, disregarding non-linear activation), and repeated the process.
The results of this experiment are shown in Figure~\ref{fig:exper}(d).
Although our theoretical analysis does not cover non-linear activation, softmax-cross-entropy loss or stochasticity in optimization, its conclusion of balanced initialization leading to improved (faster and more stable) convergence carried over to such setting.

\begin{figure}
\vspace{-5mm}
\subfloat[]{\includegraphics[scale=0.21]{std_vs_convtime_unbalanced.png}} \
\subfloat[]{\includegraphics[scale=0.21]{std_vs_convtime_balanced.png}} \
\subfloat[]{\includegraphics[scale=0.21]{{unbalanced_shallow_balances}.png}} \
\subfloat[]{\includegraphics[scale=0.21]{mnist_std_train_loss.png}} \
\vspace{-1mm}
\caption{
Experimental results.
\textbf{(a)}~Convergence of gradient descent training deep linear neural networks (depths~$3$ and~$8$) under customary initialization of layer-wise independent Gaussian perturbations with mean~$0$ and standard deviation~$s$.
For each network, number of iterations required to reach $\epsilon=10^{-5}$ from optimal training loss is plotted as a function of~$s$ (missing values indicate no convergence within $10^6$~iterations).
Dataset in this experiment is a numeric regression task from UCI Machine Learning Repository (details in text).
Notice that fast convergence is attained only in a narrow band of values for~$s$, and that this phenomenon is more extreme with the deeper network.
\textbf{(b)}~Same setup as in~(a), but with layer-wise independent initialization replaced by balanced initialization (Procedure~\ref{proc:balance_init}) based on Gaussian perturbations with mean~$0$ and standard deviation~$s$.
Notice that this change leads to fast convergence, for both networks, under wide range of values for~$s$.
Notice also that the shallower network converges slightly faster, in line with the results of~\citet{saxe2014exact} and~\citet{arora2018optimization} for $\ell_2$~loss.
\textbf{(c)}~For the run in~(a) of a depth-$3$ network and standard deviation~$s={10^{-3}}$, this plot shows degree of balancedness (minimal~$\delta$ satisfying $\|W_{j+1}^{\top}W_{j+1}-W_{j}W_j^\top\|_F\leq\delta~,\,\forall{j}\in\{1,\ldots,N-1\}$) against magnitude of weights ($\min_{j=1,\ldots,N}\|W_{j}W_j^\top\|_F$) throughout optimization.
Notice that approximate balancedness persists under gradient descent, in line with our theoretical analysis.
\textbf{(d)}~Convergence of stochastic gradient descent training the fully-connected non-linear (ReLU) neural network of the MNIST tutorial built into TensorFlow (details in text).
Customary layer-wise independent and balanced initializations~---~both based on Gaussian perturbations centered at zero~---~are evaluated, with varying standard deviations.
For each configuration~$10$ epochs of optimization are run, followed by measurement of the training loss.
Notice that although our theoretical analysis does not cover non-linear activation, softmax-cross-entropy loss and stochastic optimization, the conclusion of balanced initialization leading to improved convergence carries over to this setting.
}
\label{fig:exper}
\vspace{-3mm}
\end{figure}

\section{Introduction} \label{sec:intro}

Deep learning builds upon the mysterious ability of gradient-based optimization methods to solve related non-convex problems.
Immense efforts are underway to mathematically analyze this phenomenon.
The prominent \emph{landscape} approach focuses on special properties of critical points (\ie~points where the gradient of the objective function vanishes) that will imply convergence to global optimum.
Several papers (\eg~\citet{ge2015escaping,lee2016gradient}) have shown that (given certain smoothness properties) it suffices for critical points to meet the following two conditions:
\emph{(i)}~\emph{no poor local minima}~---~every local minimum is close in its objective value to a global minimum;
and~\emph{(ii)}~\emph{strict saddle property}~---~every critical point that is not a local minimum has at least one negative eigenvalue to its Hessian.
While condition~\emph{(i)} does not always hold (\cf~\citet{safran2018spurious}), it has been established for various simple settings (\eg~\citet{soudry2016no,kawaguchi2016deep}).
Condition~\emph{(ii)} on the other hand seems less plausible, and is in fact provably false for models with three or more layers (\cf~\citet{kawaguchi2016deep}), \ie~for \emph{deep} networks.
It has only been established for problems involving \emph{shallow} (two layer) models, \eg~matrix factorization (\citet{ge2016matrix,du2018algorithmic}).
The landscape approach as currently construed thus suffers from inherent limitations in proving convergence to global minimum for deep networks.

A potential path to circumvent this obstacle lies in realizing that landscape properties matter only in the vicinity of trajectories that can be taken by the optimizer, which may be a negligible portion of the overall parameter space.
Several papers (\eg~\citet{saxe2014exact,arora2018optimization}) have taken this \emph{trajectory-based} approach, primarily in the context of \emph{linear neural networks}~---~fully-connected neural networks with linear activation.
Linear networks are trivial from a representational perspective, but not so in terms of optimization~---~they lead to non-convex training problems with multiple minima and saddle points.
Through a mix of theory and experiments, \citet{arora2018optimization}~argued that such non-convexities may in fact be beneficial for gradient descent, in the sense that sometimes, adding (redundant) linear layers to a classic linear prediction model can accelerate the optimization.
This phenomenon challenges the holistic landscape view, by which convex problems are always preferable to non-convex ones.

Even in the linear network setting, a rigorous proof of efficient convergence to global minimum has proved elusive.
One recent progress is the analysis of~\citet{bartlett2018gradient} for \emph{linear residual networks}~---~a particular subclass of linear neural networks in which the input, output and all hidden dimensions are equal, and all layers are initialized to be the identity matrix (\cf~\citet{hardt2016identity}).
Through a trajectory-based analysis of gradient descent minimizing $\ell_2$~loss over a whitened dataset (see Section~\ref{sec:prelim}), \citet{bartlett2018gradient}~show that convergence to global minimum at a \emph{linear rate}~---~loss is less than~$\epsilon>0$ after $\OO(\log\frac{1}{\epsilon})$~iterations~---~takes place if one of the following holds:
\emph{(i)}~the objective value at initialization is sufficiently close to a global minimum;
or \emph{(ii)}~a global minimum is attained when the product of all layers is positive definite.

The current paper carries out a trajectory-based analysis of gradient descent for general deep linear neural networks, covering the residual setting of~\citet{bartlett2018gradient}, as well as many more settings that better match practical deep learning.
Our analysis draws upon the trajectory characterization of~\citet{arora2018optimization} for gradient flow (infinitesimally small learning rate), together with significant new ideas necessitated due to discrete updates.
Ultimately, we show that when minimizing $\ell_2$~loss of a deep linear network over a whitened dataset, gradient descent converges to the global minimum, at a linear rate, provided that the following conditions hold:
\emph{(i)}~the dimensions of hidden layers are greater than or equal to the minimum between those of the input and output;
\emph{(ii)}~layers are initialized to be \emph{approximately balanced} (see Definition~\ref{def:balance})~---~this is met under commonplace near-zero, as well as residual (identity) initializations;
and \emph{(iii)}~the initial loss is smaller than any loss obtainable with rank deficiencies~---~this condition will hold with probability close to~$0.5$ if the output dimension is~$1$ (scalar regression) and standard (random) near-zero initialization is employed.
Our result applies to networks with arbitrary depth and input/output dimensions, as well as any configuration of hidden layer widths that does not force rank deficiency (\ie~that meets condition~\emph{(i)}).
The assumptions on initialization (conditions~\emph{(ii)} and~\emph{(iii)}) are necessary, in the sense that violating any one of them may lead to convergence failure.
Moreover, in the case of scalar regression, they are met with constant probability under a random initialization scheme.
We are not aware of any similarly general analysis for efficient convergence of gradient descent to global minimum in deep learning.

\medskip

The remainder of the paper is organized as follows.
In Section~\ref{sec:prelim} we present the problem of gradient descent training a deep linear neural network by minimizing the $\ell_2$~loss over a whitened dataset.
Section~\ref{sec:converge} formally states our assumptions, and presents our convergence analysis.
Key ideas brought forth by our analysis are demonstrated empirically in Section~\ref{sec:exper}.
Section~\ref{sec:related} gives a review of relevant literature, including a detailed comparison of our results against those of~\citet{bartlett2018gradient}.
Finally, Section~\ref{sec:conc} concludes.

\section{Gradient Descent for Deep Linear Neural Networks} \label{sec:prelim}

We denote by~$\norm{\vv}$ the Euclidean norm of a vector~$\vv$, and by~$\norm{A}_F$ the Frobenius norm of a matrix~$A$.

We are given a training set $\{(\x^{(i)},\y^{(i)})\}_{i=1}^{m}\subset\R^{d_x}\times\R^{d_y}$, and would like to learn a hypothesis (predictor) from a parametric family~$\HH:=\{h_\theta:\R^{d_x}\to\R^{d_y}\,|\,\theta\in\Theta\}$ by minimizing the $\ell_2$~loss:\note{
	Much of the analysis in this paper can be extended to loss types other than~$\ell_2$.
	In particular, the notion of deficiency margin (Definition~\ref{def:margin}) can be generalized to account for any convex loss, and, so long as the loss is differentiable, a convergence result analogous to Theorem~\ref{theorem:converge} will hold in the idealized setting of perfect initial balancedness and infinitesimally small learning rate (see proof of Lemma~\ref{lemma:descent}).
	We leave to future work treatment of approximate balancedness and discrete updates in this general setting.
}
$$\min_{\theta\in\Theta}~L(\theta):=\frac{1}{2m}\sum\nolimits_{i=1}^m\|h_\theta(\x^{(i)})-\y^{(i)}\|^2
\text{\,.}$$
When the parametric family in question is the class of linear predictors, \ie~$\HH=\{\x\mapsto{W}\x\,|\,W\in\R^{d_y\times{d}_x}\}$, the training loss may be written as $L(W)=\frac{1}{2m}\|WX-Y\|_F^2$, where~$X\in\R^{d_x\times{m}}$ and~$Y\in\R^{d_y\times{m}}$ are matrices whose columns hold instances and labels respectively.
Suppose now that the dataset is \emph{whitened}, \ie~has been transformed such that the empirical (uncentered) covariance matrix for instances~---~$\Lambda_{xx}:=\frac{1}{m}XX^\top \in\R^{d_x\times{d}_x}$~---~is equal to identity.
Standard calculations (see Appendix~\ref{app:whitened}) show that in this case:
\be
L(W)=\frac{1}{2}\norm{W-\Lambda_{yx}}_F^2+c
\label{eq:loss_whitened}
\text{\,,}
\ee
where~$\Lambda_{yx}:=\tfrac{1}{m}YX^\top\in\R^{d_y\times{d}_x}$ is the empirical (uncentered) cross-covariance matrix between instances and labels, and $c$ is a constant (that does not depend on $W$).
Denoting~$\Phi:=\Lambda_{yx}$ for brevity, we have that for linear models, minimizing $\ell_2$~loss over whitened data is equivalent to minimizing the squared Frobenius distance from a \emph{target matrix}~$\Phi$:
\be
\min\nolimits_{W\in\R^{d_y\times{d}_x}}~L^1(W):=\frac{1}{2}\norm{W-\Phi}_F^2
\text{\,.}
\label{eq:lin_obj}
\ee

Our interest in this work lies on \emph{linear neural networks}~---~fully-connected neural networks with linear activation.
A depth-$N$ ($N\in\N$) linear neural network with hidden widths $d_1,\ldots,d_{N-1}\in\N$ corresponds to the parametric family of hypotheses $\HH:=\{\x\mapsto{W}_{N}W_{N-1}\cdots{W}_1\x\,|\,W_j\in\R^{d_j\times{d}_{j-1}},\,j=1,\ldots,N\}$, where $d_0:=d_x,\,d_N:=d_y$.
Similarly to the case of a (directly parameterized) linear predictor (Equation~\eqref{eq:lin_obj}), with a linear neural network, minimizing $\ell_2$~loss over whitened data can be cast as squared Frobenius approximation of a target matrix $\Phi$:
\be
\min\nolimits_{W_j\in\R^{d_j\times{d}_{j-1}},\,j=1,\ldots,N}~L^N(W_1,\ldots,W_N):=\frac{1}{2}\norm{W_{N}W_{N-1}\cdots{W}_1-\Phi}_F^2
\label{eq:lnn_obj}
\text{\,.}
\ee
Note that the notation~$L^N(\cdot)$ is consistent with that of Equation~\eqref{eq:lin_obj}, as a network with depth~$N=1$ precisely reduces to a (directly parameterized) linear model.

We focus on studying the process of training a deep linear neural network by \emph{gradient descent}, \ie~of tackling the optimization problem in Equation~\eqref{eq:lnn_obj} by iteratively applying the following updates:
\be
W_j(t+1)\leftarrow{W}_j(t)-\eta\frac{\partial{L}^N}{\partial{W}_j}\big(W_1(t),\ldots,W_N(t)\big)\quad,\,j=1,\ldots,{N}\quad,\,t=0,1,2,\ldots
\label{eq:gd}
\text{~~,}
\ee
where $\eta>0$ is a configurable learning rate.
In the case of depth~$N=1$, the training problem in Equation~\eqref{eq:lnn_obj} is smooth and strongly convex, thus it is known (\cf~\citet{boyd2004convex}) that with proper choice of~$\eta$, gradient descent converges to global minimum at a linear rate.
In contrast, for any depth greater than~$1$, Equation~\eqref{eq:lnn_obj} comprises a fundamentally non-convex program, and the convergence properties of gradient descent are highly non-trivial.
Apart from the case~$N=2$ (shallow network), one cannot hope to prove convergence via landscape arguments, as the strict saddle property is provably violated (see Section~\ref{sec:intro}).
We will see in Section~\ref{sec:converge} that a direct analysis of the trajectories taken by gradient descent can succeed in this arena, providing a guarantee for linear rate convergence to global minimum.

\medskip

We close this section by introducing additional notation that will be used in our analysis.
For an arbitrary matrix~$A$, we denote by~$\sigma_{max}(A)$ and~$\sigma_{min}(A)$ its largest and smallest (respectively) singular values.\note{
	If~$A\in\R^{d\times{d'}}$, $\sigma_{min}(A)$~stands for the $\min\{d,d'\}$-th largest singular value.
	Recall that singular values are always non-negative.
}
For~$d\in\N$, we use~$I_d$ to signify the identity matrix in~$\R^{d\times{d}}$.
Given weights $W_1,\ldots,W_N$ of a linear neural network, we let~$W_{1:N}$ be the direct parameterization of the \emph{end-to-end} linear mapping realized by the network, \ie~$W_{1:N}:=W_{N}W_{N-1}\cdots{W}_1$.
Note that $L^N(W_1,\ldots,W_N)=L^1(W_{1:N})$, meaning the loss associated with a depth-$N$ network is equal to the loss of the corresponding end-to-end linear model.
In the context of gradient descent, we will oftentimes use~$\ell(t)$ as shorthand for the loss at iteration~$t$:
\be
\ell(t):=L^N(W_1(t),\ldots,W_N(t))=L^1(W_{1:N}(t))
\text{\,.}
\label{eq:gd_loss}
\ee

\section{Related Work} \label{sec:related}
\vspace{-2mm}

Theoretical study of gradient-based optimization in deep learning is a highly active area of research.
As discussed in Section~\ref{sec:intro}, a popular approach is to show that the objective landscape admits the properties of no poor local minima and strict saddle, which, by~\citet{ge2015escaping,lee2016gradient,panageas2017gradient}, ensure convergence to global minimum.
Many works, both classic (\eg~\citet{baldi1989neural}) and recent (\eg~\citet{choromanska2015loss,kawaguchi2016deep,hardt2016identity,soudry2016no,haeffele2017global,nguyen2017loss,safran2018spurious,nguyen2018loss,laurent2018deep}), have focused on the validity of these properties in different deep learning settings.
Nonetheless, to our knowledge, the success of landscape-driven analyses in formally proving convergence to global minimum for a gradient-based algorithm, has thus far been limited to shallow (two layer) models only (\eg~\citet{ge2016matrix,du2018power,du2018algorithmic}).

An alternative to the landscape approach is a direct analysis of the trajectories taken by the optimizer.
Various papers (\eg~\citet{brutzkus2017globally,li2017convergence,zhong2017recovery,tian2017analytical,brutzkus2018sgd,li2018algorithmic,du2018gradient,du2018convolutional,liao2018almost}) have recently adopted this strategy, but their analyses only apply to shallow models.
In the context of linear neural networks, deep (three or more layer) models have also been treated~---~\cf~\citet{saxe2014exact} and~\citet{arora2018optimization}, from which we draw certain technical ideas for proving Lemma~\ref{lemma:descent}.
However these treatments all apply to gradient flow (gradient descent with infinitesimally small learning rate), and thus do not formally address the question of computational efficiency.

To our knowledge, \citet{bartlett2018gradient}~is the only existing work rigorously proving convergence to global minimum for a conventional gradient-based algorithm training a deep model.
This work is similar to ours in the sense that it also treats linear neural networks trained via minimization of $\ell_2$~loss over whitened data, and proves linear convergence (to global minimum) for gradient descent.
It is more limited in that it only covers the subclass of linear residual networks, \ie~the specific setting of uniform width across all layers ($d_0=\cdots=d_N$) along with identity initialization.
We on the other hand allow the input, output and hidden dimensions to take on any configuration that avoids ``bottlenecks'' (\ie~admits $\min\{d_1,\ldots,d_{N-1}\}\geq\min\{d_0,d_N\}$), and from initialization require only approximate balancedness (Definition~\ref{def:balance}), supporting many options beyond identity.
In terms of the target matrix~$\Phi$, \citet{bartlett2018gradient}~treats two separate scenarios:\note{
	There is actually an additional third scenario being treated~---~$\Phi$~is asymmetric and positive definite~---~but since that requires a dedicated optimization algorithm, it is outside our scope.
}
\emph{(i)}~$\Phi$ is symmetric and positive definite;
and~\emph{(ii)}~$\Phi$ is within distance~$1/10e$ from identity.\note{
	$1/10e$~is the optimal (largest) distance that may be obtained (via careful choice of constants) from the proof of Theorem~$1$ in~\citet{bartlett2018gradient}.
}
Our analysis does not fully account for scenario~\emph{(i)}, which seems to be somewhat of a singularity, where all layers are equal to each other throughout optimization (see proof of Theorem~$2$ in~\citet{bartlett2018gradient}).
We do however provide a strict generalization of scenario~\emph{(ii)}~---~our assumption of deficiency margin (Definition~\ref{def:margin}), in the setting of linear residual networks, is met if the distance between target and identity is less than~$0.5$.

Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE); Machine Learning (stat.ML)