Deep Learning without Poor Local Minima
Kenji Kawaguchi
Introduction
Deep learning has been a great practical success in many fields, including the fields of computer vision, machine learning, and artificial intelligence. In addition to its practical success, theoretical results have shown that deep learning is attractive in terms of its generalization properties (Livni et al.,, 2014; Mhaskar et al.,, 2016). That is, deep learning introduces good function classes that may have a low capacity in the VC sense while being able to represent target functions of interest well. However, deep learning requires us to deal with seemingly intractable optimization problems. Typically, training of a deep model is conducted via non-convex optimization. Because finding a global minimum of a general non-convex function is an NP-complete problem (Murty & Kabadi,, 1987), a hope is that a function induced by a deep model has some structure that makes the non-convex optimization tractable. Unfortunately, it was shown in 1992 that training a very simple neural network is indeed NP-hard (Blum & Rivest,, 1992). In the past, such theoretical concerns in optimization played a major role in shrinking the field of deep learning. That is, many researchers instead favored classical machining learning models (with or without a kernel approach) that require only convex optimization. While the recent great practical successes have revived the field, we do not yet know what makes optimization in deep learning tractable in theory.
In this paper, as a step toward establishing the optimization theory for deep learning, we prove a conjecture noted in (Goodfellow et al.,, 2016) for deep linear networks, and also address an open problem announced in (Choromanska et al.,, 2015b) for deep nonlinear networks. Moreover, for both the conjecture and the open problem, we prove more general and tighter statements than those previously given (in the ways explained in each section).
Deep linear neural networks
Given the absence of a theoretical understanding of deep nonlinear neural networks, Goodfellow et al., (2016) noted that it is beneficial to theoretically analyze the loss functions of simpler models, i.e., deep linear neural networks. The function class of a linear multilayer neural network only contains functions that are linear with respect to inputs. However, their loss functions are non-convex in the weight parameters and thus nontrivial. Saxe et al., (2014) empirically showed that the optimization of deep linear models exhibits similar properties to those of the optimization of deep nonlinear models. Ultimately, for theoretical development, it is natural to start with linear models before working with nonlinear models (as noted in Baldi & Lu,, 2012), and yet even for linear models, the understanding is scarce when the models become deep.
We consider one of the most widely used loss functions, squared error loss:
where is the Frobenius norm. Note that is the usual mean squared error, for which all of our results hold as well, since multiplying by a constant in results in an equivalent optimization problem.
2 Background
Recently, Goodfellow et al., (2016) remarked that when Baldi & Hornik, (1989) proved Proposition 2.1 for shallow linear networks, they stated Conjecture 2.2 without proof for deep linear networks.
(Baldi & Hornik,, 1989: shallow linear network) Assume that (i.e., ), assume that and are invertible, assume that has distinct eigenvalues, and assume that , and (e.g., an autoencoder). Then, the loss function has the following properties:
It is convex in each matrix (or ) when the other (or ) is fixed.
(Baldi & Hornik,, 1989: deep linear network) Assume the same set of conditions as in Proposition 2.1 except for . Then, the loss function has the following properties:
For any , it is convex in each matrix when for all , is fixed.
Baldi & Lu, (2012) recently provided a proof for Conjecture 2.2 (i), leaving the proof of Conjecture 2.2 (ii) for future work. They also noted that the case of is of interest, but requires further analysis, even for a shallow network with . An informal discussion of Conjecture 2.2 can be found in (Baldi,, 1989). In Appendix D, we provide a more detailed discussion of this subject.
3 Results
We now state our main theoretical results for deep linear networks, which imply Conjecture 2.2 (ii) as well as obtain further information regarding the critical points with more generality.
(Loss surface of deep linear networks) Assume that and are of full rank with and has distinct eigenvalues. Then, for any depth and for any layer widths and any input-output dimensions (the widths can arbitrarily differ from each other and from and ), the loss function has the following properties:
Every critical point that is not a global minimum is a saddle point.
If , then the Hessian at any saddle point has at least one (strictly) negative eigenvalue.If , to be succinct, we define , with a slight abuse of notation.
(Effect of deepness on the loss surface) Assume the same set of conditions as in Theorem 2.3 and consider the loss function . For three-layer networks (i.e., ), the Hessian at any saddle point has at least one (strictly) negative eigenvalue. In contrast, for networks deeper than three layers (i.e., ), there exist saddle points at which the Hessian does not have any negative eigenvalue.
The assumptions of having full rank and distinct eigenvalues in the training data matrices in Theorem 2.3 are realistic and practically easy to satisfy, as discussed in previous work (e.g., Baldi & Hornik,, 1989). In contrast to related previous work (Baldi & Hornik,, 1989; Baldi & Lu,, 2012), we do not assume the invertibility of , , nor . In Theorem 2.3, is allowed, as well as many other relationships among the widths of the layers. Therefore, we successfully proved Conjecture 2.2 (ii) and a more general statement. Moreover, Theorem 2.3 (iv) and Corollary 2.4 provide additional information regarding the important properties of saddle points.
Theorem 2.3 presents an instance of a deep model that would be tractable to train with direct greedy optimization, such as gradient-based methods. If there are “poor” local minima with large loss values everywhere, we would have to search the entire space,Typically, we do this by assuming smoothness in the values of the loss function. the volume of which increases exponentially with the number of variables. This is a major cause of NP-hardness for non-convex optimization. In contrast, if there are no poor local minima as Theorem 2.3 (ii) states, then saddle points are the main remaining concern in terms of tractability.Other problems such as the ill-conditioning can make it difficult to obtain a fast convergence rate. Because the Hessian of is Lipschitz continuous, if the Hessian at a saddle point has a negative eigenvalue, it starts appearing as we approach the saddle point. Thus, Theorem 2.3 and Corollary 2.4 suggest that for 1-hidden layer networks, training can be done in polynomial time with a second order method or even with a modified stochastic gradient decent method, as discussed in (Ge et al.,, 2015). For deeper networks, Corollary 2.4 states that there exist “bad” saddle points in the sense that the Hessian at the point has no negative eigenvalue. However, we know exactly when this can happen from Theorem 2.3 (iv) in our deep models. We leave the development of efficient methods to deal with such a bad saddle point in general deep models as an open problem.
Deep nonlinear neural networks
Now that we have obtained a comprehensive understanding of the loss surface of deep linear models, we discuss deep nonlinear models. For a practical deep nonlinear neural network, our theoretical results so far for the deep linear models can be interpreted as the following: depending on the nonlinear activation mechanism and architecture, training would not be arbitrarily difficult. While theoretical formalization of this intuition is left to future work, we address a recently proposed open problem for deep nonlinear networks in the rest of this section.
where . In practice, we usually set to be an identity map in the last layer, in which case all our theoretical results still hold true.
2 Background
Following the work by Dauphin et al., (2014), Choromanska et al., (2015a) investigated the connection between the loss functions of deep nonlinear networks and a function well-studied via random matrix theory (i.e., the Hamiltonian of the spherical spin-glass model). They explained that their theoretical results relied on several unrealistic assumptions. Later, Choromanska et al., (2015b) suggested at the Conference on Learning Theory (COLT) 2015 that discarding these assumptions is an important open problem. The assumptions were labeled A1p, A2p, A3p, A4p, A5u, A6u, and A7p.
In this paper, we successfully discard most of these assumptions. In particular, we only use a weaker version of assumptions A1p and A5u. We refer to the part of assumption A1p (resp. A5u) that corresponds only to the model assumption as A1p-m (resp. A5u-m). Note that assumptions A1p-m and A5u-m are explicitly used in the previous work (Choromanska et al.,, 2015a) and included in A1p and A5u (i.e., we are not making new assumptions here).
Choromanska et al., (2015b) noted that A6u is unrealistic because it implies that the inputs are not shared among the paths. In addition, Assumption A5u is unrealistic because it implies that the activation of any path is independent of the input data. To understand all of the seven assumptions (A1p, A2p, A3p, A4p, A5u, A6u, and A7p), we note that Choromanska et al., (2015b, a) used these seven assumptions to reduce their loss functions of nonlinear neural networks to:
(High-level description of a main result in Choromanska et al.,, 2015a) Assume A1p (including A1p-m), A2p, A3p, A4p, A5u (including A5u-m), A6u, and A7p (Choromanska et al.,, 2015b). Furthermore, assume that . Then, the expected loss of each sample datum, , has the following property: above a certain loss value, the number of local minima diminishes exponentially as the loss value increases.
3 Results
We now state our theoretical result, which partially address the aforementioned open problem. We consider loss functions for all the data points and all possible output dimensionalities (i.e., vectored-valued output). More concretely, we consider the squared error loss with expectation, .
(Loss surface of deep nonlinear networks) Assume A1p-m and A5u-m. Let . Then, we can reduce the loss function of the deep nonlinear model to that of the deep linear model . Therefore, with the same set of conditions as in Theorem 2.3, the loss function of the deep nonlinear model has the following properties:
Every critical point that is not a global minimum is a saddle point.
The saddle points have the properties stated in Theorem 2.3 (iv) and Corollary 2.4.
Comparing Corollary 3.2 and Proposition 3.1, we can see that we successfully discarded assumptions A2p, A3p, A4p, A6u, and A7p while obtaining a tighter statement in the following sense: Corollary 3.2 states with fewer unrealistic assumptions that there is no poor local minimum, whereas Proposition 3.1 roughly asserts with more unrealistic assumptions that the number of poor local minimum may be not too large. Furthermore, our model is strictly more general than the model analyzed in (Choromanska et al.,, 2015a, b) (i.e., this paper’s model class contains the previous work’s model class but not vice versa).
Proof Idea and Important lemmas
In this section, we provide overviews of the proofs of the theoretical results. Our proof approach largely differs from those in previous work (Baldi & Hornik,, 1989; Baldi & Lu,, 2012; Choromanska et al.,, 2015a, b). In contrast to (Baldi & Hornik,, 1989; Baldi & Lu,, 2012), we need a different approach to deal with the “bad” saddle points that start appearing when the model becomes deeper (see Section 2.3), as well as to obtain more comprehensive properties of the critical points with more generality. While the previous proofs heavily rely on the first-order information, the main parts of our proofs take advantage of the second order information. In contrast, Choromanska et al., (2015a, b) used the seven assumptions to relate the loss functions of deep models to a function previously analyzed with a tool of random matrix theory. With no reshaping assumptions (A3p, A4p, and A6u), we cannot relate our loss function to such a function. Moreover, with no distributional assumptions (A2p and A6u) (except the activation), our Hessian is deterministic, and therefore, even random matrix theory itself is insufficient for our purpose. Furthermore, with no spherical constraint assumption (A7p), the number of local minima in our loss function can be uncountable.
One natural strategy to proceed toward Theorem 2.3 and Corollary 3.2 would be to use the first-order and second-order necessary conditions of local minima (e.g., the gradient is zero and the Hessian is positive semidefinite).For a non-convex and non-differentiable function, we can still have a first-order and second-order necessary condition (e.g., Rockafellar & Wets,, 2009, theorem 13.24, p. 606). However, are the first-order and second-order conditions sufficient to prove Theorem 2.3 and Corollary 3.2? Corollaries 2.4 show that the answer is negative for deep models with , while it is affirmative for shallow models with . Thus, for deep models, a simple use of the first-order and second-order information is insufficient to characterize the properties of each critical point. In addition to the complexity of the Hessian of the deep models, this suggests that we must strategically extract the second order information. Accordingly, in section 4.2, we obtain an organized representation of the Hessian in Lemma 4.3 and strategically extract the information in Lemmas 4.4 and 4.6. With the extracted information, we discuss the proofs of Theorem 2.3 and Corollary 3.2 in section 4.3.
2 Lemmas
As discussed above, we extracted the first-order and second-order conditions of local minima as the following lemmas. The lemmas provided here are also intended to be our additional theoretical results that may lead to further insights. The proofs of the lemmas are in the appendix.
(Critical point necessary and sufficient condition) is a critical point of if and only if for all ,
(Representation at critical point) If is a critical point of , then
(Block Hessian with Kronecker product) Write the entries of in a block form as
(Hessian semidefinite necessary condition) If is positive semidefinite or negative semidefinite at a critical point, then for any
If is positive semidefinite or negative semidefinite at a critical point, then for any
(Hessian positive semidefinite necessary condition) If is positive semidefinite at a critical point, then
3 Proof sketches of theorems
We now provide the proof sketch of Theorem 2.3 and Corollary 3.2. We complete the proofs in the appendix.
By case analysis, we show that any point that satisfies the necessary conditions and the definition of a local minimum is a global minimum.
Case I: and : If , Corollary 4.5 with implies the necessary condition of local minima that . If , Lemma 4.6 with and , combined with the fact that implies the necessary condition that . Therefore, we have the necessary condition of local minima, . Interpreting condition , we conclude that achieving is indeed a global minimum.
Case II: and : From Lemma 4.6, we have the necessary condition that or . If , using the exact same proof as in Case I, it is a global minimum. Suppose then that . From Lemma 4.4 with , we conclude that . Then, from Lemma 4.2, we write , which is the orthogonal projection onto the subspace spanned by the eigenvectors corresponding to the largest eigenvalues following the ordinary least square regression matrix. This is indeed the expression of a global minimum.
Case III: : We first show that if , every local minimum is a global minimum. Thus, we consider the case where and . In this case, by induction on , we prove that we can have with arbitrarily small perturbation of each entry of without changing the value of . Once this is proved, along with the results of Case I and Case II, we can immediately conclude that any point satisfying the definition of a local minimum is a global minimum.
We first prove the statement for the base case with by using an expression of that is obtained by a first-order necessary condition: for an arbitrary ,
By using Lemma 4.6 to obtain an expression of , we deduce that we can have with arbitrarily small perturbation of each entry of without changing the loss value.
For the inductive step with , from Lemma 4.4, we use the following necessary condition for the Hessian to be (positive or negative) semidefinite at a critical point: for any ,
We use the inductive hypothesis to conclude that the first condition is false, and thus the second condition must be satisfied at a candidate point of a local minimum. From the latter condition, with extra steps, we can deduce that we can have with arbitrarily small perturbation of each entry of while retaining the same loss value.
We conclude the induction, proving that we can have with arbitrarily small perturbation of each parameter without changing the value of . Upon such a perturbation, we have the case where , for which we have already proven that every local minimum is a global minimum. Summarizing the above, any point that satisfies the definition (and necessary conditions) of a local minimum is indeed a global minimum. Therefore, we conclude the proof sketch of Theorem 2.3 (ii).
3.2 Proof sketch of Theorem 2.3 (i), (iii) and (iv)
We can prove the non-convexity and non-concavity of this function simply from its Hessian (Theorem 2.3 (i)). That is, we can show that in the domain of the function, there exist points at which the Hessian becomes indefinite. Indeed, the domain contains uncountably many points at which the Hessian is indefinite.
We now consider Theorem 2.3 (iii): every critical point that is not a global minimum is a saddle point. Combined with Theorem 2.3 (ii), which is proven independently, this is equivalent to the statement that there are no local maxima. We first show that if , the loss function always has some strictly increasing direction with respect to , and hence there is no local maximum. If , we show that at a critical point, if the Hessian is negative semidefinite (i.e., a necessary condition of local maxima), we can have with arbitrarily small perturbation without changing the loss value. We can prove this by induction on , similar to the induction in the proof of Theorem 2.3 (ii). This means that there is no local maximum.
Theorem 2.3 (iv) follows Theorem 2.3 (ii)-(iii) and the analyses for Case I and Case II in the proof of Theorem 2.3 (ii); when , if at a critical point, is a global minimum.
3.3 Proof sketch of Corollary 3.2
Since the activations are assumed to be random and independent, the effect of nonlinear activations disappear by taking expectation. As a result, the loss function is reduced to .
Conclusion
In this paper, we addressed some open problems, pushing forward the theoretical foundations of deep learning and non-convex optimization. For deep linear neural networks, we proved the aforementioned conjecture and more detailed statements with more generality. For deep nonlinear neural networks, when compared with the previous work, we proved a tighter statement (in the way explained in section 3) with more generality ( can vary) and with strictly weaker model assumptions (only two assumptions out of seven). However, our theory does not yet directly apply to the practical situation. To fill the gap between theory and practice, future work would further discard the remaining two out of the seven assumptions made in previous work. Our new understanding of the deep linear models at least provides the following theoretical fact: the bad local minima would arise in a deep nonlinear model but only as an effect of adding nonlinear activations to the corresponding deep linear model. Thus, depending on the nonlinear activation mechanism and architecture, we would be able to efficiently train deep models.
The author would like to thank Prof. Leslie Kaelbling, Quynh Nguyen, Li Huan and Anirbit Mukherjee for their thoughtful comments on the paper. We gratefully acknowledge support from NSF grant 1420927, from ONR grant N00014-14-1-0486, and from ARO grant W911NF1410433.
References
Appendix A Proofs of lemmas and corollary in Section 4.2
We complete the proofs of the lemmas and corollary in Section 4.2.
Since ,
By setting for all , we obtain the statement of Lemma 4.1. For the boundary cases (i.e., or ), it can be seen from the second to the third lines that we obtain the desired results with the definition, (i.e., and ).
A.2 Proof of Lemma 4.2
From the critical point condition with respect to (Lemma 4.1),
which is true if and only if . By expanding , . By solving for ,
for an arbitrary matrix . Due to the property of any generalized inverse (Zhang,, 2006, p. 41), we have that . Thus,
A.3 Proof of Lemma 4.3
For the diagonal blocks: the entries of diagonal blocks are obtained simply using the result of Lemma 4.1 as
Using the formula of computed in the proof of of Lemma 4.1 yields the desired result.
For the off-diagonal blocks with :
The first term above is reduced to the first term of the statement in the same way as the diagonal blocks. For the second term,
where and . The third line follows the fact that . In the last line, we have the desired result by rewriting .
For the off-diagonal blocks with : The first term in the statement is obtained in the same way as above (for the off-diagonal blocks with ). For the second term, notice that where is the -th row vector of or the vector corresponding to the -th output component. That is, it is conveniently organized as the blocks, each of which corresponds to each output component (or rather we chose instead of for this reason, among others). Also,
For each block entry in the above, similarly to the case of ,
Here, we have the desired result by rewriting .
A.4 Proof of Lemma 4.4
Note that a similarity transformation preserves the eigenvalues of a matrix. For each , we take a similarity transform of (whose entries are organized as in Lemma 4.3) as
Here, the first implication follows the necessary condition with any principal submatrix and the second implication follows the necessary condition with the Schur complement (Zhang,, 2006, theorem 1.20, p. 44).
Note that (Zhang,, 2006, p. 41). Thus, by plugging in the formulas of and that are derived in Lemma 4.3,
where and . Here, we can replace by (see Appendix A.7). Thus, can be replaced by . Accordingly, the first term is reduced to zero as
since (Zhang,, 2006, p. 41). Thus, with the second term remained, the condition is reduced to
which concludes the proof for the positive semidefinite case. For the necessary condition of the negative semidefinite case, we obtain the same condition since
A.5 Proof of Corollary 4.5
From the first condition in the statement of Lemma 4.4,
The first implication follows the fact that the rank of a product of matrices is at most the minimum of the ranks of the matrices, and the fact that the column space of is subspace of the column space of .
A.6 Proof of Lemma 4.6
For the condition: Let be the principal submatrix as defined in the proof of Lemma 4.4 (the principal submatrix of that consists of the first four blocks of it). Let . Let . Using Lemma 4.3 for the blocks corresponding to and ,
where . Then, by the necessary condition with the Schur complement (Zhang,, 2006, theorem 1.20, p. 44), implies
where the second line follows the fact that can be replaced by (see Appendix A.7). The third line follows the fact that because . In the fourth line, we expanded and used the definition of the Kronecker product. It implies
Here, if , we have obtained the statement of the lemma. Thus, from now on, we focus on the case where and to obtain the other condition, .
For the condition: By using another necessary condition of a matrix being positive semidefinite with the Schur complement (Zhang,, 2006, theorem 1.20, p. 44), implies that
Since we can replace by (see Appendix A.7), the second term in the left hand side is simplified as
In the third line, the crossed terms – and its transpose – are vanished to 0 because of the following. From Lemma 4.1, at any critical point. Thus, The forth line follows
where the last line is due to the fact that is a scalar and the fact that for any matrix , .
From equations 3 and A.6,
In the following, we simplify equation 5 by first showing that and then simplifying and .
Showing that (following the proof in Baldi & Hornik,, 1989): Let be the projection operator on . We first show that .
where the first line follows Lemma 4.2, the second line is due to Lemma 4.1 with (i.e., ), the third line follows Lemma 4.2, and the fourth line uses the definition of . Since is symmetric, is also symmetric and hence . Thus, . Note that as . Thus,
which implies that . Since the eigenvalues () are distinct, this implies that is a diagonal matrix (otherwise, implies for , resulting in contradiction). Because is the orthogonal projector of rank (as ), this implies that is a diagonal matrix with its diagonal entries being ones ( times) and zeros ( times). Thus,
for some index set . This means that .
where and the last line follows the facts:
and similarly, .
Simplifying : In the proof of Lemma 4.2, by using Lemma 4.1 with , we obtained that . Also, from Lemma 4.4, we have that or . If , we got the statement of the lemma, and so we consider the case of . Therefore,
Since ,
Simplifying : From Lemma 4.4, (again since we are done if ). Thus, . As discussed above, we write . Thus,
Putting results together: We use the simplified formulas of , and in equation 5, obtaining
Due to Sylvester’s law of inertia (Zhang,, 2006, theorem 1.5, p. 27), with a nonsingular matrix (it is nonsingular because each of and is nonsingular), the necessary condition is reduced to
which implies that for all , . In other words, the index set must select the largest eigenvalues whatever is. Since (which is obtained above), we have that in this case.
Summarizing the above case analysis, if at a critical point, or .
A.7 Generalized inverse of Kronecker product
is a generalized inverse of .
For a matrix , the definition of a generalized inverse, , is . Setting , we check if satisfies the definition: as desired.
Here, we are not claiming that is the unique generalized inverse of . Notice that the necessary condition that we have in our proof (where we need a generalized inverse of ) is for any generalized inverse of . Thus, replacing it by one of any generalized inverse suffices to obtain a necessary condition. Indeed, choosing MoorePenrose pseudoinverse suffices here, with which we know . But, to give a simpler argument later, we keep more generality by choosing as a generalized inverse of .
Appendix B Proof of Theorem 2.3
We complete the proofs of Theorem 2.3. Since we heavily rely on the necessary conditions of local minima, we remind the reader of the elementary logic: for a point to be a local minimum, it must satisfy all the necessary conditions of local minima, but a point satisfying the necessary conditions can be a point that is not a local minimum (in contrast, a point satisfying the sufficient condition of local minimum is a local minimum).
By case analysis, we show that any point that satisfies the necessary conditions and the definition of a local minimum is a global minimum. When we write a statement in the proof, we often mean that a necessary condition of local minima implies the statement as it should be clear (i.e., we are not claiming that the statement must hold true unless the point is the candidate of local minima.).
Case I: and : Assume that . We first obtain a necessary condition of the Hessian being positive semidefinite at a critical point, , and then interpret the condition. If , Corollary 4.5 with implies the necessary condition that . This is because the other condition is false.
If , Lemma 4.6 with implies the necessary condition that or . Suppose that . Then, we have that . That is, .
From Corollary 4.5 with implies the necessary condition that
Suppose the latter: . Since and , the left null space of contains only zero. Thus,
Suppose the former: . Because , , and as shown in the proof of Lemma 4.6, we have that .
where the last equality follows the fact that since and thereby the projection of onto the range of is . Therefore, we have the condition, when .
Thus, we have proved that when and , if at a critical point, it is a global minimum.
Case II: and : We first obtain a necessary condition of the Hessian being positive semidefinite at a critical point and then interpret the condition. From Lemma 4.6, we have that or . If , with the exact same proof as in the case of , it is a global minimum. Suppose that . Combined with Lemma 4.2, we have a necessary condition:
From Lemma 4.4 with , , which implies that (since ). Thus, we can rewrite the above equation as , which is the orthogonal projection on to subspace spanned by the eigenvectors corresponding to the largest eigenvalues following the ordinary least square regression matrix. This is indeed the expression of a global minimum (Baldi & Hornik,, 1989; Baldi & Lu,, 2012).
Thus, we have proved that when , if at a critical point, it is a global minimum.
Case III: : Suppose that . Let . Then, if , every local minimum is a global minimum because of the following. If , and thereby we have the case of (since we have that where the first inequality follows the definition of ). For this case, we have already proven the desired statement above. On the other hand, if , we have . Thus, , which is a global minimum. We can see this in various ways. For example, , which means that it is a global minimum as discussed above.
Thus, in the following, we consider the remaining case where and . In this case, we show that we can have with arbitrarily small perturbations of each entry of , without changing the loss value. In order to show this, by induction on , we prove that we can have with arbitrarily small perturbation of each entry of without changing the value of .
We start with the base case with . For convenience, we reprint a necessary condition of local minima that is represented by equation 2 in the proof of Lemmas 4.2: for an arbitrary ,
Again, note that the set of all generalized inverse of is as follows (Zhang,, 2006, p. 41):
Since equation 6 must necessarily hold for any generalized inverse in order for a point to be a local minimum, we choose a generalized inverse with to have a weaker yet simpler necessary condition. That is,
By plugging this into equation 6, we obtain the following necessary condition of local minima: for an arbitrary ,
This means that changing the values of the last () rows of (i.e., ) does not change the value of . Thus, we consider the perturbation of each entry of as follows:
Thus, we have shown that we can have with arbitrarily small perturbation of each entry of with the loss value being unchanged. This concludes the proof for the base case of the induction with .
For the inductive stepThe boundary cases with and as well pose no problem during the proof for the inductive step: remember our notational definition, if . with , we have the inductive hypothesis that we can have with arbitrarily small perturbations of each entry of without changing the loss value. Here, we want to show that if , we can have with arbitrarily small perturbation of each entry of without changing the value of . Accordingly, suppose that . From Lemma 4.4, we have the following necessary condition for the Hessian to be (positive or negative) semidefinite at a critical point: for any ,
where the first condition is shown to imply in Corollary 4.5. If the former condition is true, , which is false in the case being analyzed (i.e., the case where . If this is not the case, we can immediately conclude the desired statement as it has been already proven for the case where ). Thus, we suppose that the latter condition is true. Let . Then, for an arbitrary ,
where the last two equalities follow Lemmas 4.2 and 4.6 (since if , we immediately obtain the desired result as discussed above). Taking transpose,
Since is full rank with (i.e., ), there exists a left inverse and the solution of the above linear system is unique as , yielding,
In other words, .
and plugging this into the condition in equation B.1: for an arbitrary ,
which means that changing the values of the last () rows does not change the value of .
We consider the perturbation of each entry of as follows. From equation 9, all the possible solutions of can be written as: for an arbitrary and ,
where and is the the Moore–Penrose pseudoinverse of . We perturb as
where . Then,
Thus, we conclude the induction, proving that we can have with arbitrarily small perturbation of each parameter without changing the value of . Since , upon such a perturbation, we have the case where , for which we have already proven that a critical point is not a local minimum unless it is a global minimum. This concludes the proof of the case where .
Summarizing the above, any point that satisfies the definition (and necessary conditions) of a local minimum is a global minimum, concluding the proof of Theorem 2.3 (ii).
B.2 Proof of Theorem 2.3 (i)
We can prove the non-convexity and non-concavity from its Hessian (Theorem 2.3 (i)). First, consider . For example, from Corollary 4.5 with , it is necessary for the Hessian to be positive or negative semidefinite at a critical point that or . The instances of unsatisfying this condition at critical points form some uncountable set. As an example, consider a uncountable set that consists of the points with and with any . Then, every point in the set defines a critical point from Lemma 4.1. Also, as . So, it does not satisfy the first semidefinite condition. On the other hand, with any instance of such that , we have that . So, it does not satisfy the second semidefinite condition as well. Thus, we have proven that in the domain of the loss function, there exist points, at which the Hessian becomes indefinite. This implies Theorem 2.3 (i): the functions are non-convex and non-concave.
B.3 Proof of Theorem 2.3 (iii)
We now prove Theorem 2.3 (iii): every critical point that is not a global minimum is a saddle point. Here, we want to show that if the Hessian is negative semidefinite at a critical point, then there is a increasing direction so that there is no local maximum. From Lemma 4.3 with
The positive semidefiniteness follows the fact that and are positive semidefinite. Since is full rank, if has at least one strictly positive eigenvalue, has at least one strictly positive eigenvalue (by the spectrum property of Kronecker product). Thus, with other variables being fixed, if , with respect to at any critical point, there exists some increasing direction that corresponds to the strictly positive eigenvalue. This means that there is no local maximum if .
If , we claim that at a critical point, if the Hessian is negative semidefinite (i.e., a necessary condition of local maxima), we can make with arbitrarily small perturbation of each parameter without changing the loss value. We can prove this by using the similar proof procedure to that used for Theorem 2.3 (ii) in the case of . Suppose that and thus . By induction on , we prove that we can have with arbitrarily small perturbation of each entry of without changing the loss value.
We start with the base case with . From Lemma 4.4, we have a following necessary condition for the Hessian to be (positive or negative) semidefinite at a critical point: for any ,
where the first condition is shown to imply in Corollary 4.5. Let . From the condition with , we have that or . The former condition is false since . From the latter condition, for an arbitrary ,
where the last follows the critical point condition (Lemma 4.2). Then, similarly to the proof of Theorem 2.3 (ii),
In other words, .
Suppose that . Then, since , we have that , which is false (or else the desired statement). Thus, , which implies that . Then, since with , we can have without changing the loss value with arbitrarily small perturbation of .
For the inductive step with , we have the inductive hypothesis that we can have with arbitrarily small perturbation of each parameter without changing the loss value. Accordingly, suppose that . Again, from Lemma 4.4, for any ,
If the former is true, , which is false (or the desired statement). If the latter is true, for an arbitrary ,
where the last follows the critical point condition (Lemma 4.2). Then, similarly to the above,
In other words, .
Suppose that . Then, since , we have that , which is false (or the desired statement). Thus, , which implies that . Then, since with , we can have without changing the loss value with arbitrarily small perturbation of each parameter.
Thus, we conclude the induction, proving that if , with arbitrarily small perturbation of each parameter without changing the value of , we can have . Thus, at any candidate point for local maximum, the loss function has some strictly increasing direction in an arbitrarily small neighborhood. This means that there is no local maximum. Thus, we obtained the statement of Theorem 2.3 (iii).
B.4 Proof of Theorem 2.3 (iv)
In the proof of Theorem 2.3 (ii), the case analysis with the case, , revealed that when , if at a critical point, is a global minimum. Thus, when , if is not a global minimum at a critical point, its Hessian is not positive semidefinite, containing some negative eigenvalue. From Theorem 2.3 (ii), if it is not a global minimum, it is not a local minimum. From Theorem 2.3 (iii), it is a saddle point. Thus, if , the Hessian at any saddle point has some negative eigenvalue, which is the statement of Theorem 2.3 (iv).
Appendix C Proofs of Corollaries 2.4 and 3.2
We complete the proofs of Corollaries 2.4 and 3.2.
For example, with ,
where we can see that there exists a strictly lower value of than the loss value with , which is (since and ).
Thus, these are not global minima, and thereby these are saddle points by Theorem 2.3 (ii) and (iii). On the other hand, from the proof of Lemma 4.3, every diagonal and off-diagonal element of the Hessian is zero if . Thus, the Hessian is simply a zero matrix, which has no negative eigenvalue.
C.2 Proof of Corollary 3.2 and discussion of the assumptions used in the previous work
Since , .
The previous work also assumes the use of “independent random” loss functions. Consider the hinge loss, . By modeling the max operator as a Bernoulli random variable , we can then write . A1p then assumes that for all and , the are Bernoulli random variables with equal probabilities of success. Furthermore, A5u assumes that the independence of , and . Finally, A6u assumes that for all and are independent. In section 3.2, we discuss the effect of all of the seven previous assumptions to see why these are unrealistic.
Appendix D Discussion of the 1989 conjecture
The 1989 conjecture is based on the result for a 1-hidden layer network with (e.g., an autoencoder). That is, the previous work considered with the same loss function as ours with the additional assumption . The previous work denotes and .
The conjecture was expressed by Baldi & Hornik, (1989) as
Our results, and in particular the main features of the landscape of , hold true in the case of linear networks with several hidden layers.
Here, the “main features of the landscape of ” refers to the following features, among other minor technical facts: 1) the function is convex in each matrix (or ) when fixing other (or ), and 2) every local minimum is a global minimum. No proof was provided in this work for this conjecture.
In 2012, the proof for the conjecture corresponding to the first feature (convexity in each matrix (or ) when fixing other (or )) was provided in (Baldi & Lu,, 2012) for both real-valued and complex-valued cases, while the proof for the conjecture for the second feature (every local minimum being a global minimum) was left for future work.
In (Baldi,, 1989), there is an informal discussion regarding the conjecture. Let be an index of a layer with the smallest width . That is, . We write
Now that we have proved the incompleteness of this reasoning, we discuss where the reasoning actually breaks down in a more concrete example. From Lemmas 4.1 and 4.2, if , we have the following representation at critical points:
where and . In contrast, from Lemmas 4.1 and 4.2, if is arbitrary,
where and as discussed above, and . Note that by using other critical point conditions from Lemmas 4.1, we cannot obtain an expression such that in the above expression unless . Therefore, even though what and can represent is the same, the critical condition becomes different (and similarly, the conditions from the Hessian). Because the proof in the previous work with heavily relies on the fact that , the same proof does not apply for deeper models (we may continue providing more evidence as to why the same proof does not work for deeper models, but one such example suffices for the purpose here).
In this respect, we have completed the proof of the conjecture and also provided a complete analytical proof for more general and detailed statements; that is, we did not assume that , and we also proved saddle point properties with negative eigenvalue information.