TL;DR
This paper introduces a novel double convex-concave backtracking method for inertial Bregman proximal gradient algorithms, enabling adaptive step size and extrapolation control, with proven convergence and improved performance in non-convex problems.
Contribution
It proposes a new double backtracking procedure controlling both step size and extrapolation in inertial Bregman proximal methods, with theoretical convergence guarantees.
Findings
Convergence to critical points is proven for the proposed algorithms.
Numerical experiments show improved performance in image processing and machine learning tasks.
The method effectively balances inertial step and backtracking for non-convex optimization.
Abstract
Backtracking line-search is an old yet powerful strategy for finding a better step sizes to be used in proximal gradient algorithms. The main principle is to locally find a simple convex upper bound of the objective function, which in turn controls the step size that is used. In case of inertial proximal gradient algorithms, the situation becomes much more difficult and usually leads to very restrictive rules on the extrapolation parameter. In this paper, we show that the extrapolation parameter can be controlled by locally finding also a simple concave lower bound of the objective function. This gives rise to a double convex-concave backtracking procedure which allows for an adaptive choice of both the step size and extrapolation parameters. We apply this procedure to the class of inertial Bregman proximal gradient methods, and prove that any sequence generated by these algorithms…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Convex-Concave Backtracking for Inertial Bregman Proximal Gradient Algorithms in Non-Convex Optimization
Mahesh Chandra Mukkamala111Faculty of Mathematics and Computer Science, Saarland University, 66123 Saarbrücken, Germany, E-mail: [email protected]
Peter Ochs222Faculty of Mathematics and Computer Science, Saarland University, 66123 Saarbrücken, Germany, E-mail: [email protected]
Thomas Pock333Institute of Computer Graphics and Vision, Graz University of Technology, 8010 Graz, Austria. E-mail: [email protected]
Shoham Sabach444Faculty of Industrial Engineering, The Technion, Haifa, 3200003, Israel. E-mail: [email protected].
Abstract
Backtracking line-search is an old yet powerful strategy for finding a better step sizes to be used in proximal gradient algorithms. The main principle is to locally find a simple convex upper bound of the objective function, which in turn controls the step size that is used. In case of inertial proximal gradient algorithms, the situation becomes much more difficult and usually leads to very restrictive rules on the extrapolation parameter. In this paper, we show that the extrapolation parameter can be controlled by locally finding also a simple concave lower bound of the objective function. This gives rise to a double convex-concave backtracking procedure which allows for an adaptive choice of both the step size and extrapolation parameters. We apply this procedure to the class of inertial Bregman proximal gradient methods, and prove that any sequence generated by these algorithms converges globally to a critical point of the function at hand. Numerical experiments on a number of challenging non-convex problems in image processing and machine learning were conducted and show the power of combining inertial step and double backtracking strategy in achieving improved performances.
2010 Mathematics Subject Classification: 90C25, 26B25, 49M27, 52A41, 65K05.
Keywords: Composite minimization, proximal gradient algorithms, inertial methods, convex-concave backtracking, non-Euclidean distances, Bregman distance, global convergence, Kurdyka-Łojasiewicz property.
1 Introduction
In this work we are interested in tackling non-convex additive composite minimization problems, which include the sum of two extended-valued functions: a non-smooth function denoted by (possibly non-convex) and a smooth function denoted by (possibly non-convex). More precisely, we consider problems of the following form
[TABLE]
where is a nonempty, closed and convex set in d. We will give a more precise statement in Section 2 about the involved functions and set. There is a tremendous number of applications in machine learning, computer vision, statistics, and many more, that can be formulated in this framework.
Motivated by challenging applications as illustrated in Section 6, we consider here an instance of problem , where the smooth function has a gradient that is not necessarily globally Lipschitz continuous. The restrictive assumption of having Lipschitz continuous gradient can be replaced with a certain convexity condition, which was proposed and developed first in [5] for problems with convex functions, and recently extended to the non-convex setting in [12]. More details on these recent developments will be given below in Section 2.
This convexity condition easily yields an approximation of the objective function at hand by a convex function from above (majorant) and a concave function from below (minorant). In the traditional setting, where the gradient of the smooth function is Lipschitz continuous, the majorant and the minorant are quadratic functions. In this case, it is well-known that the tightness of the quadratic approximations is directly related to restrictions on the step size to be used in the algorithm. The same relation is true for the convexity condition. In addition to their global existence, these approximations can be locally improved by backtracking (line search) strategies and it is well-known that tight approximations are advantageous, as we explain below in more detail.
Interestingly, while the step size is usually restricted by the quality of the majorant, the extrapolation (also known as inertia or over-relaxation) parameter is also affected by the quality of the minorant. This observation suggests to adapt the majorant and the minorant independently. In this paper we propose an efficient backtracking strategy that locally determines a tight majorant and minorant to exploit as much information as possible from the objective function, to be used in the proposed algorithm. This leads to a highly efficient algorithm, which is able to detect “the degree of local convexity” of the objective function (see Section 3 for details). As the backtracking procedure seeks for tight convex majorants and concave minorants, our idea is to combine it with an inertial step. We propose an inertial version of the Bregman Proximal Gradient (BPG) algorithm, which uses a convex-concave backtracking procedure to dynamically adjust the step size and the extrapolation parameter. Therefore, we call our algorithm Convex-Concave Inertial BPG (CoCaIn BPG in short). We prove a global convergence result of this algorithm (see Section 3.2 for an overview of the results and Section 5 for the details) to critical points of the objective function. The efficiency, which we demonstrate on several practical applications, comes from combining the inertial step with the novel convex-concave backtracking strategy, which fully exploits the power of tight local approximations in achieving large step sizes and large extrapolation parameters that can be used at the same time.
Before concluding this section, we would like to give the reader a first intuition about the convex-concave backtracking strategy on a simple instance of problem .
A simple illustrative example.
In the following, we consider the following particular instance of problem : , and the gradient of is -Lipschitz continuous. Even in this simpler setting, the convex-concave backtracking strategy is novel.
In this smooth and non-convex setting, an update step of a classical inertial based gradient method, starting with some , reads as follows
[TABLE]
where , , is an extrapolation parameter and . If is convex and the extrapolation parameter is carefully chosen, this recovers the popular Nesterov Accelerated Gradient method [32] (for , again in the convex setting, see [7]). It is well-known that the gradient step above, can be equivalently written as follows
[TABLE]
For a proper , the function to be minimized above is a convex quadratic majorant of the function (due to the classical Descent Lemma), which is a property that is also crucial for the convergence analysis of the algorithm. Classically, , , is a sufficient condition to guarantee the existence of a quadratic majorant. However, locally, i.e., between the points and , the parameter may be significantly smaller than the global Lipschitz constant (which will immediately affect the step size of the algorithm). More precisely, note that the Descent Lemma,
[TABLE]
actually guarantees the existence of a quadratic minorant and a quadratic majorant that are determined by the same (global) parameter . However, only the majorant limits the step size that is used in the algorithm. As shown in Figure 1, tighter approximations can be computed if the parameters of the minorant and the majorant are allowed to differ:
[TABLE]
i.e., the minorant parameter could be different from the majorant parameter .
While the step size of the algorithm only depends on the majorant parameter , the extrapolation parameter also depends on the minorant parameter . When and , for all , it was established in [46] that for any , when
[TABLE]
the generated sequence converges linearly (under certain error bound condition).
If the minorant parameter is close to [math], which means that the function is “locally convex”, the extrapolation parameter can be taken close to , which makes the algorithm we present “similar” to an Accelerated Gradient method in the non-convex setting.
Below, we will show that using the minorant and the majorant in a local fashion (instead of their global counterparts) is very useful in developing the inertial Bregman Proximal Gradient method.
Notation. We use standard notation and concepts which, unless otherwise specified, can all be found in [41].
2 The Bregman Framework
In this section we will first recall the definition of Bregman distance, which stands at the heart of our developments. It was introduced in [14] and popularized by [17]. Based on that we will shortly review the recent concept of smooth adaptable functions, which in some sense extends and generalizes the class of smooth functions with globally Lipschitz continuous gradient. Then, we will provide the basic and essential ingredients to deal with the Bregman Proximal Gradient method.
We begin with the notion of kernel generating distance functions, which was recently stated in [12] (in this respect see also [4]).
Definition 2.1**.**
(Kernel Generating Distance) Let be a nonempty, convex and open subset of d. Associated with , a function is called a kernel generating distance if it satisfies the following:
is proper, lower semicontinuous and convex, with and .
is on .
We denote the class of kernel generating distances by .
Given , the Bregman distance that is associated to , is a proximity measure which is defined by
[TABLE]
This object is not a distance according to the classical definition (for example, it is not symmetric in general). However, the Bregman distance between two points is nonnegative if and only if the function is convex. If is known to be strictly convex, we have that if and only if . The classic example of a Bregman distance is the squared Euclidean distance, which is generated by . For more examples, results and applications of Bregman distances, see [18, 43, 22, 6, 44] and references therein.
An important property that is always crucial when dealing with Bregman distances is the well-known three-points identity [20, Lemma 3.1]: for any and ,
[TABLE]
We conclude this part by restating our optimization model
[TABLE]
and making the first connection to the Bregman framework. One important feature of using Bregman distances in optimization algorithms is the ability of relate the constraint set to a certain kernel generating distances function . From now on, we make the following assumption.
Assumption A**.**
with .
is a proper and lower semicontinuous function (possibly non-convex) with .
is a proper and lower semicontinuous function (possibly non-convex) with , which is continuously differentiable on .
.
2.1 Smooth Adaptable Functions
One goal of this work is to deal with the non-convex optimization model () where the gradient of the smooth function is not globally Lipschitz. Recently, Bauschke, Bolte and Teboulle [5], observed that the property of having a Lipschitz continuous gradient can be interpreted equivalently as a certain convexity condition on the function itself. This opens the gate for generalizing known results in the convex setting. It was extended to the non-convex setting in [12] with the concept of smooth adaptable functions given below.
Definition 2.2** (L-smooth Adaptable).**
A pair is called -smooth adaptable (-smad) on if there exists such that and are convex on .
The convexity requirement of can be written with respect to a different parameter , which is key to the proposed double backtracking procedure to be developed in Section 3.1. In this section, for the sake of simplicity, we use .
The optimization model () appears with a smooth term in the objective function which is very common in many fields of applications. A crucial pillar in designing and analyzing algorithms for tackling this model, is usually based on the fact that the smooth part in the objective function has a Lipschitz continuous gradient. This property, via the well-known Descent Lemma, guarantees us that a lower and an upper quadratic approximation exist. For -smooth adaptable functions, we will use the following extended version of the Descent Lemma (see [12, Lemma 2.1, p. 2134]).
Lemma 2.1** (Extended Descent Lemma).**
The pair of functions is -smooth adaptable on if and only if:
[TABLE]
Remark 2.1** (Invariance to Strong Convexity).**
We would like to note that the -smooth adaptable property is invariant when is additionaly assumed to be -strongly convex. Indeed, as described in [12], since convexity of is not needed, we can define , and then for any , we have
[TABLE]
namely, the new pair satisfies the L-smad property on .
2.2 The Bregman Proximal Gradient Algorithm
In this section we review the basic notations and results needed to study Bregman based optimization methods. We first recall the definition of the Bregman proximal mapping [43], which is associated with a proper and lower semi-continuous function , and is defined by
[TABLE]
With , the above boils down to the classical set-valued Moreau proximal mapping introduced in [29]. We refer the reader to the recent survey paper [44], and references therein. Here, we will focus on the Bregman proximal gradient mapping, which will take a central role in the algorithm to be developed in the next section. Given and a step size parameter , the Bregman proximal gradient mapping is defined by
[TABLE]
where the second equality follows from the fact that . Note that here with , the above recovers the classical proximal gradient mapping. Since could be non-convex, the mapping is not, in general, single-valued. This mapping emerges from the usual approach, which consists of linearizing the differentiable function around a point and regularizing it with a proximal distance from that point. Similar to [12], the following assumption guarantees that the Bregman proximal gradient mapping is well-defined.
Assumption B**.**
The function is supercoercive for all , that is,
[TABLE]
For all , we have .
Assumption B(i) is a standard coercivity condition, which is for instance automatically satisfied when is compact. On the other hand, Assumption B(ii) can be shown to hold under a classical constraint qualification condition. It also holds automatically when is convex or when . The following result from [12], ensures that the Bregman proximal gradient mapping is well-defined.
Lemma 2.2** (Well-Posedness of ).**
Suppose that Assumptions A and B hold, and let . Then, the set is a nonempty and compact subset of .
3 The Inertial Bregman Proximal Gradient Method
Our proposed algorithm belongs to the class of inertial based optimization methods. The most well-known method in this class is the so-called Heavy-ball method, which was introduced by Polyak [40] to minimize convex and smooth functions. A popular variant of the method, when applied to the additive composite model () with , takes the following form. Start with any , and generate iteratively a sequence via
[TABLE]
where is an extrapolation parameter and is a step size paramter. In [37], an inertial proximal gradient algorithm, called iPiano, was proposed555With a small modification that the proximity term is centered around the extrapolated point , while the gradient of is evaluated at .. It was shown that under Assumption A, if is convex and has a globally Lipschitz continuous gradient, the sequence converges globally to a critical point (in this setting, under additional error-bound condition, a linear rate of convergence was proved in [46]). The case where also the function is not necessarily convex was treated in [13, 34]. Two years later, in [39] a block version of the method, called iPALM was proposed and analyzed in the fully non-convex setting, i.e., both and are non-convex. In this case, a global convergence result to critical points was also established. A unified analysis was presented in [36].
In this work we propose a Bregman variant of the method mentioned above (see steps (3.1) and (3.2)), which also handle the two involved parameter and , , in a dynamic fashion. To this end we incorporate into our basic steps two routines aiming at controlling and updating these parameters.
3.1 The Convex-Concave Backtracking Procedure
As we already illustrated on a simple example in the introduction, the origin of this procedure comes from the fact that for smooth adaptable functions we can build lower and upper approximations as given in Lemma 2.1:
[TABLE]
Even though the existence of the parameters and could be globally guaranteed, in practice it is often difficult or computationally expensive to evaluate them. In such cases it is recommended to apply a backtracking procedure that can locally verify the validity of the inequalities given in (3.3). However, in most cases only the upper approximation and the corresponding parameter are used. Here, we will develop a double backtracking procedure that locally verifies both the lower and the upper approximations, in order to better control and update the extrapolation parameter and the step size parameter at each iteration . To the best of our knowledge, this is the first attempt to use the lower approximation in algorithms for tackling non-convex problems. It should be noted that in the case that is convex we have by definition , or even a convex quadratic lower approximation can be found when is strongly convex (see [44] for a discussion and references about a strong convexity property with respect to a Bregman distance). Based on the concepts described above, we will make the following additional assumptions on the involved functions.
Assumption C**.**
The function is -strongly convex on .
The pair of functions is -smooth adaptable on .
There exists such that is convex666Such functions are called semi-convex with modulus (see [34, 35])..
A few comments on the assumption above are now in order. The first item is related to Remark 2.1, which says that the smooth adaptable property is invariant to strongly convex kernel generating distance functions . The third assumption allows us to deal with non-convex functions since could be negative. See Section 6 for examples of functions that satisfy all these assumptions. Now we are ready to present our algorithm, which is called Convex-Concave Inertial (CoCaIn) Bregman Proximal Gradient.
The two input parameters and are free to be chosen by the user. As we will see later the parameter measures the descent to be achieved at each iteration of the algorithm.
The steps (3.4) and (3.7) are the classical steps of the inertial proximal gradient method, while here since we are dealing with the Bregman variant, it must be guaranteed that the auxiliary vector as defined in (3.4) belongs to . Otherwise the Bregman proximal gradient step (3.7) is not defined (see Section 2.2). Even though, in general, it is not easy to guarantee that, in our case this will not be an issue. Indeed, in order to derive global convergence results of Bregman based algorithms in the non-convex setting an essential assumption seems to be that the kernel generating distance function has a full domain, i.e., (see, for instance, [12] for more details about this limitation). The steps (3.6) and (3.8) implement the double backtracking procedure (see Section 5.4). The step (3.5) is designed to control the extrapolation parameter , , and should be validated at each iteration. However, a natural question would be if such a parameter always exists? We postpone the positive answer to this question, to Section 4, and conclude this section with a list of our theoretical contributions.
3.2 Summary of the Convergence Results
Before we proceed with the well-posedness of CoCaIn BPG and the convergence analysis, we provide here a brief summary of our results.
- •
We show the well-posedness of CoCaIn BPG, in the sense that, one can always find such that (3.5) is satisfied for all (see Lemma 4.1). Moreover, we show that it suffices to know the Bregman symmetric coefficient (Definition 4.1), in order to estimate the extrapolation parameter , .
- •
In the Euclidean setting, i.e., when , we provide an explicit formula for the maximal extrapolation parameter
[TABLE]
which uses the majorant parameter from the previous iterate, which is a key for the efficient implementation of the proposed convex-concave backtracking procedure. When , we easily recover that .
- •
Stability and convergence of the objective function values of CoCaIn BPG, which relies on finding an appropriate sequence of Lyapunov functions that enjoys a sufficient descent property (see Proposition 5.1).
- •
Global convergence of a sequence generated by the CoCaIn BPG method to critical points of the objective function (see Theorem 5.2). This result relies on the concept of Gradient-like Descent Sequences (see Definition 5.1 below).
4 Well-Posedness of CoCaIn BPG
Now, we would like to verify the well-posedness of the CoCaIn BPG algorithm. An important tool in achieving our goal is the recently introduced symmetry coefficient of a Bregman distance, which measures the lack of symmetry in , see [5].
Definition 4.1** (Symmetry Coefficient).**
Given , its symmetry coefficient is defined by
[TABLE]
An important and immediate consequence of this definition is the fact that for all we have
[TABLE]
where we have adopted the convention that and for all . Clearly, the closer is to , the more symmetric is with perfect symmetry when (which holds if and only if ).
To this end, we need to convince the reader about the existence of , , which satisfies (3.5), i.e., that
[TABLE]
holds true. The following result provides a positive answer to the existences question and information on the relevant extrapolation parameters that satisfy this inequality.
Lemma 4.1** (General Extrapolation Behavior).**
Given with . Let and with . Then, for a given , there exists such that
[TABLE]
Proof.
From the three points identity (see (2.1)) we have
[TABLE]
Now, from (4.1), we obtain that
[TABLE]
On the other hand, since , we can use the fact that , for a fixed , is a convex function and therefore
[TABLE]
where the last inequality follows from (4.1). By combining the last two inequalities we derive that
[TABLE]
and, by re-arranging we have
[TABLE]
First, it is easy to verify that for , the denominator is positive. In addition, to find such that
[TABLE]
we will use simple algebraic manipulations. Indeed, by re-arranging we have
[TABLE]
Since , it follows that . We also have that , and thus there exists a positive root denoted by . Therefore, for any , the desired result follows. ∎
Remark 4.1**.**
Note that in the above lemma, depends only on the symmetry coefficient . Therefore, for the Euclidean distance with , this implies that,
[TABLE]
However, for the Euclidean distance, the expression in (4.2), can be simplified significantly. Indeed, since we take , then using the fact that we obtain that . In the case of CoCaIn BPG, we have the following restriction on the maximal extrapolation parameter that can be used
[TABLE]
A related bound also appeared in [46] as we discussed in the introduction. When, the values of and are almost equal and , then it is possible to choose the inertial parameter such that . We discuss more about bounds of , , in Section 5.3.
5 Convergence Analysis of CoCaIn BPG
Before we proceed to the convergence analysis, we need the following technical lemma.
Lemma 5.1** (Function Descent Property).**
Let be a sequence generated by CoCaIn BPG. Then, for all , we have
[TABLE]
Proof.
Fix . From the convexity of , which holds thanks to Assumption C(iii), we obtain from the sub-gradient inequality [41, Example 8.8 and Proposition 8.12] that
[TABLE]
where . By rearranging the inequality we obtain
[TABLE]
From the optimality condition of step (3.7), we have that
[TABLE]
which combined with (5.1) yields that
[TABLE]
where the last equality follows from the three-points identity (see (2.1)). On the other hand, using the lower approximation given in (3.6) and the upper approximation given in (3.8), we have that
[TABLE]
Combining the last two inequalities and using the fact that , implies that
[TABLE]
which completes the proof. ∎
Since we are dealing with inertial based methods, which belong to the class of non-descent methods, we can not expect to use classical convergence techniques for non-convex problems (see below for more information about it). In order to overcome the lack of descent, we will use the Lyapunov technique, which involves the construction of a sequence of new functions, which will be used to “better” measure the progress of the algorithm, where by progress we mean a decrement in the Lyapunov function values. In several cases a trivial Lyapunov function would be to use the function itself, however in the case of non-descent methods, it is not a good choice, since it does not capture well the behavior of the iterates. The behavior of two subsequent iterates must be taken into consideration along with the function, as observed in [37, 42].
5.1 Lyapunov Function Descent Property of CoCaIn BPG
Let be a sequence generated by CoCaIn BPG. We define, at iterate , the following Lyapunov function
[TABLE]
This Lyapunov function involves two terms: (i) the term , which measures the progress in original function values with respect to the global optimal value of problem () and (ii) the term given by , which ensures that the iterates stay close enough, with respect to the Bregman distance. Before we motivate further the usage of this Lyapunov function, we show its descent property.
Proposition 5.1**.**
Let be a sequence generated by CoCaIn BPG. Then, for all , we have
[TABLE]
Proof.
Multiplying (5.1) with , we obtain
[TABLE]
By the definition of the Lyapunov function and the fact that we have
[TABLE]
With and the strong convexity of , that follows from Assumption C(i), we obtain
[TABLE]
where the last inequality holds, since and . Next, we observe that
[TABLE]
where the first inequality is due to the step (3.5) of the algorithm and the second inequality is due to fact that . By rearranging we obtain,
[TABLE]
thus completing the proof. ∎
Proposition 5.2**.**
Let be a sequence generated by CoCaIn BPG. Then, the following assertions hold:
The sequence is nonincreasing.
, and hence the sequence converges to zero.
.
Proof.
This follows trivially from Proposition 5.1, since .
Let be a positive integer. Summing (5.4) from to we get
[TABLE]
since . Taking the limit as , we obtain the first desired assertion, from which we immediately deduce that converges to zero.
From (5.5) we also obtain,
[TABLE]
which after division by yields the desired result.
∎
In order to proceed with the global convergence analysis of CoCaIn BPG, we will need throughout the rest of this section, to additionally assume the following.
Assumption D**.**
.
and are Lipschitz continuous on any bounded subset of d.
5.2 Global Convergence for CoCaIn BPG
In this subsection we show the global convergence result of CoCaIn BPG. The goal is to show that the whole sequence , that is generated by CoCaIn BPG, converges to a critical point. To this end, we denote the set of critical points by
[TABLE]
Note that, such a set is well-defined due to Fermat’s rule [41, Theorem 10.1, p. 422] and due to the concept of limiting subdifferential.
From now on we will make the following assumption regarding the sequence of majorant parameters : there exists an integer such that for all ( can be as large as the user wishes). It should be noted that thanks to Assumption C(ii) and Lemma 2.1, there exists a global majorant parameter such that (3.8) holds true for all . On the other hand, since in anyway we require that the parameters do not decrease between two successive iterations, it makes sense that at some point we will stop changing them and continue with a fixed value. However, it is very important not using the global parameter right from the beginning since in practice the parameter determined by (3.8) might be much smaller (especially in early stages of the algorithm).
In the second phase of the algorithm, i.e., when , it also makes sense to assume that for all where . This immediately suggests that our Lyapunov function can also be simplified. More precisely, we define the following new Lyapunov function:
[TABLE]
where .
The global convergence result is based on showing that CoCaIn BPG generates a gradient-like descent sequence according to Definition 5.1 (see below). This involves three properties which need to be verified: “sufficient descent condition”, “relative error condition” and “continuity condition”. Such a convergence analysis is based on a recent technique, which was initiated by Attouch and Bolte [1], and later on was simplified and unified in [11]. A more general framework was proposed in [36].
The main tool that stands behind this technique is the Kurdyka-Łojasiewicz (KL) property [26, 27] (see [8] for the non-smooth case), which is properly defined in the appendix. This property has been used in several recent works that deal with non-convex optimization problems (see [1, 3, 11] for early foundational works). For more details and information on the KL property, we refer the reader to the following papers [8, 1, 10, 2, 3, 11, 36] and references therein.
Verifying that a given function satisfies the KL property could be difficult, however in their seminal work [8], Bolte, Daniilidis and Lewis prove that any proper, lower semicontinuous and semi-algebraic function satisfies the KL property on its domain. This important result makes this proof technique very powerful, since we are familiar with many semi-algebraic functions that appear very often in applications. In fact, the same result holds for (possibly non-smooth) functions that are definable in an o-minimal structure [8, 9]. For examples and more details about the relations between KL and other important notions, see [8, 10] and references therein.
In order to derive the global convergence of our algorithm we follow this proof technique that we shortly recall now. For the interested readers we refer to [12, Appendix 6, p. 2147], where a short and self-contained summary of this proof methodology can be found. It should be noted again that here we consider a modification, which fits non-descent methods like CoCaIn BPG.
Definition 5.1** (Gradient-like Descent Sequence).**
A sequence is called a gradient-like descent sequence for minimizing if the following three conditions hold:
Sufficient decrease condition. There exists a positive scalar such that
[TABLE]
Relative error condition. There exist an integer and a positive scalar such that
[TABLE]
Continuity condition. Let be a limit point of a subsequence , then .
Based on Definition 5.1 and the KL property, the following global convergence result holds true. We provide its proof in the appendix.
Theorem 5.1** (Global Convergence).**
Let be a bounded gradient-like descent sequence for minimizing . If satisfies the KL property, then the sequence has finite length, i.e., and it converges to .
Now, in a sequence of lemmas, we prove that CoCaIn BPG generates a gradient-like descent sequence for minimizing . In order to prove condition (C1), we first note that Proposition 5.2 is also valid for the new Lyapunov function as recorded now (for the sake of simplicity we omit the exact details of the proof, which is almost identical to the proof above).
Proposition 5.3**.**
Let be a sequence generated by CoCaIn BPG. Then, the following assertions hold:
The sequence is nonincreasing, converging and condition (C1) of Definition 5.1 holds true.
, and hence the sequence converges to zero.
* where (by Assumption A(iv)).*
Now we can prove the following result, which means that condition (C2) holds true.
Proposition 5.4**.**
Let be a bounded sequence generated by CoCaIn BPG. Then, there exist and a positive scalar such that
[TABLE]
Proof.
Fix . By the definition of the Lyapunov function we obtain that
[TABLE]
Writing the optimality condition of the optimization problem which defines (see (3.7) and recall that for , we have that ) yields that
[TABLE]
Therefore
[TABLE]
and by defining
[TABLE]
and we obviously obtain that where . Since is a bounded sequence and both and are Lipschitz continuous on bounded subsets of d (see Assumption D(ii)), there exists such that
[TABLE]
where the last inequality follows also from the fact that , since is Lipschitz continuous on bounded subsets of d. Using step (3.4) we obtain that
[TABLE]
where we have used the fact that , . Since, we also have that
[TABLE]
the desired result is proved and condition (C2) also holds true. ∎
Now we are left with showing that CoCaIn BPG generates a sequence that satisfies condition (C3).
Proposition 5.5**.**
Let be a bounded sequence generated by CoCaIn BPG. Let be a limit point of a subsequence , then .
Proof.
Consider a subsequence which converges to (there exists such a subsequence since the sequence is assumed to be bounded). Using Proposition 5.3(ii) and the strong convexity of , we obtain that . Therefore, the sequence also converges to . From the definition of , see (3.4), it also follows that also converges to . In addition, since is continuously differentiable on d we have that . Now, from (3.7), it follows (after some simplifications), for all , that
[TABLE]
Substituting by and letting , we obtain from the fact that is continuously differentiable on d, that
[TABLE]
Using this, and recalling that here is continuous, we obtain that , where . ∎
The global convergence of CoCaIn BPG now easily follows from our general result on gradient-like descent sequences (see Theorem 5.1)
Theorem 5.2** (Global Convergence of CoCaIn BPG).**
Let be a bounded sequence generated by CoCaIn BPG. If and satisfy the KL property, then the sequence has finite length, i.e., and it converges to .
Before we conclude this section, we provide a simplified variant of CoCaIn BPG.
5.3 CoCaIn BPG Without Backtracking
Note that CoCaIn BPG uses a local estimate of the minorant and majorant parameters and , , determined by the backtracking steps (3.6) and (3.8), respectively. However, when the global parameter is known (guaranteed in Assumption C(ii)), we can skip the backtracking steps, and provide a simplified variant of CoCaIn BPG.
For the inertial step (5.9), when we can obtain that
[TABLE]
with . Using Remark 4.1, if , one could choose the extrapolation parameter as follows . However, in general, the closed form expression for is difficult to obtain, for which backtracking line-search strategy can be used. Recently, in [31] the authors showed a technique to obtain closed form inertia for general Bregman distances. We use their technique later in the context of Quadratic inverse problems to propose a new variant of CoCaIn BPG with closed form inertia.
5.4 Implementing the Double Backtracking Procedure
The update steps of CoCaIn BPG are based on the double backtracking strategy (see steps (3.6) and (3.8)). Here, we describe some implementation details of these two steps. Note that the inner loops for finding the minorant and the majorant parameters and , , are implemented in a sequential fashion. By this, we mean that at iteration we first execute the steps (3.4), (3.5) and (3.6) in order to compute an appropriate , only then we proceed to steps (3.7) and (3.8) in order to compute . Note that the fact that the sequence does not decrease is crucial in order to decouple the steps (3.4) and (3.7). More precisely, we now describe the backtracking procedure to find . Let be a scaling parameter and arbitrarily initialize . Then, we find the smallest that satisfies (3.6) and such that satisfies
[TABLE]
We can now describe the procedure to find . Let and initialize , then we take the smallest that satisfies (3.8). Therefore, is monotonically non-decreasing. Note, however, we do not require any monotonicity of the sequence .
The double backtracking strategy preserves the sign of , however, only is required. Changing the sign of when the function is locally strongly convex might lead to additional acceleration. However, we leave this kind of adaptation for future work.
6 Numerical Experiments
Our goal in this section is to illustrate the performance of CoCaIn BPG in various situations. We start with minimization of univariate functions, which emphasizes the power of incorporating inertial terms into the BPG algorithm and using the double backtracking procedure. Then we provide some insights on the following practical applications: Quadratic Inverse Problems in Phase Retrieval and Non-convex Robust Denoising with Non-convex Total Variation Regularization. More recently, the efficiency of CoCaIn BPG is also demonstrated in related work for Matrix Factorization [30] and Deep Linear Neural Networks [31].
6.1 Finding Global Minima of Univariate Functions
We begin with two examples of minimizing univariate non-convex functions, which shed some light on the two main features of our algorithm: (i) inertial term and (ii) double backtracking procedure. We consider unconstrained minimization of functions , with Lipschitz continuous gradient, i.e., model () with , and . The two functions are: and . We compare three methods: CoCaIn BPG with and refer to it as CoCaIn with Euclidean distance, classical Gradient Descent (GD) method with backtracking (which is actually CoCaIn with Euclidean distance and with for all ), and iPiano777In this particular case, the method coincides with the Heavy-ball method [40]. [37] (with the inertial parameter set to ). When using a backtracking procedure in GD and iPiano methods, we mean that only the majorant parameter is varied. We use the same initialization for all the algorithms and report the performance in Figure 2.
In the second experiment, we illustrate the robustness of CoCaIn BPG to local minima and critical points. We consider the non-smooth and non-convex function , with many critical points as shown in the center plot of Figure 3, and set and (which is obviously a non-convex function with Lipschitz continuous gradient). Here again we take . In order to apply CoCaIn BPG, the main computational step is of the following form:
[TABLE]
which results in the following update step
[TABLE]
We compare CoCaIn BPG with Euclidean distance to the classical Proximal Gradient (PG) method with backtracking (CoCaIn BPG with Euclidean distance and , ), and iPiano. As mentioned in the first experiment, when using a backtracking procedure in PG and iPiano methods we mean that only the majorant parameter is varied.
As shown in Figure 3, CoCaIn BPG achieves the global minimum, whereas the PG with backtracking gets stuck in a local minimum. We performed the same experiment starting at equidistant points sampled from the interval . The average final function value for CoCaIn was , whereas for PG method with backtracking it was and for the iPiano it was . This means that CoCaIn BPG reaches the global minimum from points, PG method with backtracking achieves the global minimum only from points and iPiano from points. Hence, the behavior illustrated in Figure 3 is not due to the choice of initialization, but rather due to additional features of the CoCaIn BPG algorithm. This illustrates the great power of using double backtracking procedure in minimizing univariate non-convex functions.
6.2 Escaping Spurious Stationary Points
Here, we provide evidence that CoCaIn BPG can escape spurious stationary points in minimizing non-convex functions of two variables. Let , , be samples of a noisy signal with additive Gaussian noise. A very common task in signal processing is to recover the true data. However, due to the noise, data can be prone to several outliers. In such cases, a robust loss [23] is used. Moreover, prior information about the data, can be embedded through a regularizing term (for instance, a sparsity promoting regularizer). Given , we consider minimization of
[TABLE]
with
[TABLE]
The function is a non-convex sparsity promoting regularizer (also known as the log-sum penalty term [16, 33]) and the function is a robust loss. For illustration purposes, we consider a simple instance of problem (6.3) where , and . For minimizing this function we set and to be used in the CoCaIn BPG method.
Before presenting the numerical results, we would like to note that in this example, the function is convex for any and is convex for all . Each iteration of CoCaIn BPG would require to compute the Bregman proximal gradient mapping, which in this case reduces to the classical proximal gradient mapping (due to the choice of ). Note that due to the separability of the functions and , the needed minimization problem can be split into two individual minimizations with respect to and . These two optimization problems (after simple manipulations) reduces to computation of the proximal mapping of the univariate function . A closed form formula can be found in [24] and reads as follows:
[TABLE]
where
[TABLE]
with .
Now we can apply CoCaIn BPG method and the function behavior is described in Figure 4.
The performance of CoCaIn BPG is illustrated in Figure 5, which shows that CoCaIn BPG can indeed escape spurious critical points to reach the global minimum.
6.3 Quadratic Inverse Problems in Phase retrieval
Phase retrieval has been an active research topic for several years [15, 45, 21, 28]. It gained a lot of attention from the optimization community, due to resulting hard non-convex problems [12, 21, 19]. The phase retrieval problem can be described as follows. Given sampling vectors , , and measurements , we seek to find a vector such that the following system of quadratic equations is approximately satisfied,
[TABLE]
One typical way to tackle this system is by solving an optimization problem that seeks to minimize a certain error/noise measure in accomodating the equations. The objective function also depends on the type of noise [19] in the system (for instance, Gaussian or Poisson noise). We assume additive Gaussian noise and the squared error measure
[TABLE]
with
[TABLE]
The function acts as a regularizing term and is used to incorporate certain prior information on the wished solution. We conduct experiments with two options of regularizing functions: (i) squared -norm, and (ii) -norm, . When applying here the CoCaIn BPG method we use the following kernel generating distance function
[TABLE]
We obviously have that and we record below a result [12, Lemma 5.1, p. 2143], which shows that the pair satisfies the L-smad property (see Definition 2.2).
Lemma 6.1**.**
Let and be as defined above. Then, for any satisfying
[TABLE]
the function is convex on d.
By the design of CoCaIn BPG algorithm, the inertial parameter must satisfy (3.5). However, this involves backtracking over , which can computationally expensive for high dimensional problems. To this regard, following [31], we propose closed form expression for which satisfies (3.5). We also illustrate with our numerical experiments, that CoCaIn BPG variant with closed form inertia is competitive to our main algorithm CoCaIn BPG.
Lemma 6.2** (Closed form inertia).**
For defined in (6.6), we obtain the following gradient
[TABLE]
and for any , we have
[TABLE]
Proof.
Consider the expansion at till second order terms, we thus have
[TABLE]
The first order terms result in (6.7) and we also have
[TABLE]
where the inequality follows due to Cauchy-Schwarz inequality. ∎
Lemma 6.3** ([31]).**
Let be twice continuously differentiable on . Then, the following identity holds
[TABLE]
Proposition 6.1**.**
Denote , for any the following holds
[TABLE]
Proof.
We use the strategy from [31, Lemma 15]. From Lemma 6.3, we have
[TABLE]
where in the last step we used the upper bound (6.8) from Lemma 6.2. Also, we used the following inequality
[TABLE]
where in the last step we used and . With the result follows.∎
Therefore, in this case, Assumptions A, B, C and D are valid. We now discuss the update step of CoCaIn BPG, which requires the solution of the following subproblem
[TABLE]
Following [12], we provide closed form formulas for these optimization problems when is either the squared -norm or the -norm.
-norm.
Here we use the following closed form solution, derived in [12, Proposition 5.1, p. 2145]. First, we define the soft-thresholding operator with respect to the parameter , as follows
[TABLE]
where all operations are applied coordinate-wise. Then the closed form solution of problem (6.9) is given by
[TABLE]
where is the unique positive real root of the following cubic equation
[TABLE]
Squared -norm.
Using similar arguments as of [12, Proposition 5.1, p. 2145], we can easily derive that the solution of problem (6.9) is given by
[TABLE]
where is the unique real root of the following cubic equation
[TABLE]
We illustrate, in Figure 6, the performance of CoCaIn BPG and CoCaIn BPG with closed form inertia (CoCaIn BPG CFI), compared with two other algorithms: (i) the Bregman Proximal Gradient Method with backtracking (denoted by BPG-WB) using the same kernel generating distance function (which is exactly CoCaIn BPG with for all ) and (ii) the Inexact Bregman Proximal Minimization Line Search Algorithm (denoted by IBPM-LS) of [38]. We also compare with the Bregman Proximal Gradient (BPG) method of [12] without backtracking and with the parameter as derived in Lemma 6.1.
6.4 Non-convex Robust Denoising with Non-convex TV Regularization
We consider the problem of image denoising of a given image , where . The goal is to obtain the true image, denoted by . However, in real world applications, it is possible that the measurements are noisy with outliers. The standard routine to deal with outliers is to use robust loss function. The basic idea is to heavily penalize small errors and reasonably penalize large errors. This is done to ensure that the predicted data , is not influenced significantly by outliers. We consider a fully non-convex formulation of the problem, which includes a non-convex loss function along with a non-convex regularization.
We need the following technical details to provide the full problem statement. The spatial finite difference operator is given by
[TABLE]
where and . The horizontal spatial finite differences are given by for all and [math] otherwise. The vertical spatial finite differences are given by for all and [math] otherwise.
The problem involves the following functions
[TABLE]
where . The function is non-smooth non-convex and is smooth non-convex. The function is a non-convex variant of the popular Total Variation (TV) regularizer, which is used to prefer smooth signals while preserving sharp changes in the signal (such as edges of images). For an overview on non-convex regularizations we refer the reader to [33, 47]. Consider . It is easy to prove the convexity of , by checking that its right derivative is monotonically increasing [25, Theorem 6.4], for all . The function is convex for .
Due to separability of the function , we can split the computation of the corresponding Bregman Proximal Gradient mapping, into the following separable subproblems
[TABLE]
which as discussed in Section 6.2, can be reduced to the computation of the proximal mapping of the function .
We consider two additional experimental settings apart from our main setting given by (6.12) and (6.13). Firstly, we use the -norm based data term with the same regularization as in (6.13). Secondly, we use the squared -norm based data term with regularization as in (6.13). We use the good image given in Figure 7(a) and add severe noise randomly of magnitude. We illustrate the robustness of the model given by (6.12) and (6.13) to such outliers. The reconstructed image from -norm based data penalty term is given in Figure 7(c) and the reconstructed image from -norm based data penalty term is given in Figure 7(d), after applying CoCaIn BPG. Clearly the -norm based data penalty is better than -norm based data penalty term, which is due to the robustness properties of -norm. However, even using -norm is not enough in the presence of severe outliers, the robustness properties are not so significant. This is mitigated by our setting, where the reconstructed image is given in Figure 7(e). In our setting, the data term in (6.12) is very robust to outliers. In all the settings, we used and . The convergence plots for the experiments with (6.12) and (6.13) are given in Figure 7(f) and 7(g). Note that CoCaIn BPG CFI uses the closed form inertia with Euclidean distance. BPG-WB and BPG are same as in earlier experiments. IBPM-LS is a general purpose line-search algorithm for nonconvex nonsmooth problems proposed in [38]. Even though, IBPM-LS is general, BPG based methods are much faster. The comparisons also illustrate that CoCaIn BPG is better in terms of convergence with respect to iterations and competitive with respect to time. CoCaIn BPG CFI performs very similar to CoCaIn BPG and as anticipated the time plots illustrate that CoCaIn BPG CFI is slightly faster than CoCaIn BPG.
7 Acknowledgments
Mahesh Chandra Mukkamala and Peter Ochs acknowledge funding by the German Research Foundation (DFG Grant OC 150/1-1). Thomas Pock acknowledges support by the ERC starting grant HOMOVIS, No. 640156.
8 Appendix: Proof of Theorem 5.1
The set of all limit points of is defined by
[TABLE]
We first prove the following result.
Lemma 8.1**.**
Let be a bounded gradient-like descent sequence for minimizing . Then, is a nonempty and compact subset of , and we have
[TABLE]
In addition, the objective function is finite and constant on .
Proof.
Since is bounded there is and a subsequence such that as and hence is nonempty. Moreover, the set is compact since it can be viewed as an intersection of compact sets. Now, from conditions (C1) and (C3), and the lower semicontinuity of (which follows from the lower semi-continuity of and , see Assumption A), we obtain
[TABLE]
and therefore
[TABLE]
On the other hand, from conditions (C1) and (C2), we know that there is , , such that as . The closedness property of implies thus that . This proves that is a critical point of , and hence (8.1) is valid.
To complete the proof, let . Then converges to and from (8.2) we have . Hence the restriction of to equals . ∎
We recall now the definition of the Kurdyka-Łojasiewicz (KL) property [26, 27] and [8] (for the non-smooth case). Denote . Let , and set
[TABLE]
Definition 8.1** (The Non-smooth KL Property).**
A proper and lower semicontinuous function has the Kurdyka-Łojasiewicz (KL) property locally at if there exist , , and a neighborhood such that
[TABLE]
for all .
Our last ingredient is a key uniformization of the KL property proven in [11, Lemma 6, p. 478], which we record below.
Lemma 8.2** (Uniformized KL Property).**
Let be a compact set and let be a proper and lower semicontinuous function. Assume that is constant on and satisfies the KL property at each point of . Then, there exist , and such that for all in one has,
[TABLE]
for all .
We can now restate and prove Theorem 5.1.
Theorem 8.1**.**
Let be a bounded gradient-like descent sequence for minimizing . If and satisfy the KL property, then the sequence has finite length, i.e., and it converges to .
Proof.
Since is bounded there exists a subsequence such that as . In a similar way as in Lemma 8.1 we get that
[TABLE]
If there exists an integer for which then condition (C1) would imply that . A trivial induction show then that the sequence is stationary and the announced results are obvious. Since is a nonincreasing sequence, it is clear from (8.4) that for all . Again from (8.4) for any there exists a nonnegative integer such that for all . From Lemma 8.1 we know that . This means that for any there exists a positive integer such that for all .
From Lemma 8.1 applied to , we know that is nonempty and compact and that the function is finite and constant on . Hence, we can apply the Uniformization Lemma 8.2 applied to , which satisfies the KL property since and do, with . Therefore, for any , we have
[TABLE]
This makes sense since we know that for any . Combining (8.5) with condition (C2), see Proposition 5.4, we get that
[TABLE]
For convenience, we define for all and the following quantity
[TABLE]
From the concavity of we get that
[TABLE]
Combining condition (C1) with (8.6) and (8.7) yields, for any , that
[TABLE]
Using the fact that for all , we infer from the later inequality that
[TABLE]
and thus
[TABLE]
Summing up (8.8) for yields
[TABLE]
where the last equality follows from the fact that for all . Since , recalling the definition of , we thus have for any that
[TABLE]
which implies that , i.e., is a Cauchy sequence and hence together with Lemma 8.1, we obtain the global convergence to a critical point. ∎
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] H. Attouch and J. Bolte. On the convergence of the proximal algorithm for nonsmooth functions involving analytic features. Mathematical Programming , 116(1-2):5–16, 2009.
- 2[2] H. Attouch, J. Bolte, P. Redont, and A. Soubeyran. Proximal alternating minimization and projection methods for nonconvex problems: an approach based on the Kurdyka-Łojasiewicz inequality. Mathematics of Operations Research , 35(2):438–457, 2010.
- 3[3] H. Attouch, J. Bolte, and B. F. Svaiter. Convergence of descent methods for semi-algebraic and tame problems: proximal algorithms, forward–backward splitting, and regularized Gauss–Seidel methods. Mathematical Programming , 137(1-2):91–129, 2013.
- 4[4] A. Auslender and M. Teboulle. Interior gradient and proximal methods for convex and conic optimization. SIAM Journal on Optimization , 16(3):697–725, 2006.
- 5[5] H. H. Bauschke, J. Bolte, and M. Teboulle. A descent lemma beyond Lipschitz gradient continuity: first-order methods revisited and applications. Mathematics of Operations Research , 42(2):330–348, 2017.
- 6[6] H. H. Bauschke and J. M. Borwein. Legendre functions and the method of random Bregman projections. Journal of Convex Analysis , 4(1):27–67, 1997.
- 7[7] A. Beck and M. Teboulle. A fast iterative shrinkage-thresholding algorithm for linear inverse problems. SIAM Journal on Imaging Sciences , 2(1):183–202, 2009.
- 8[8] J. Bolte, A. Daniilidis, and A. Lewis. The Łojasiewicz inequality for nonsmooth subanalytic functions with applications to subgradient dynamical systems. SIAM Journal on Optimization , 17:1205–1223, 2006.
