Gradient Methods for Submodular Maximization
Hamed Hassani, Mahdi Soltanolkotabi, Amin Karbasi
Introduction
Submodular set functions exhibit a natural diminishing returns property, resembling concave functions in continuous domains. At the same time, they can be minimized exactly in polynomial time (while can only be maximized approximately), which makes them similar to convex functions. They have found numerous applications in machine learning, including viral marketing , dictionary learning network monitoring , sensor placement , product recommendation , document and corpus summarization data summarization , crowd teaching , and probabilistic models . However, submodularity is in general a property that goes beyond set functions and can be defined for continuous functions. In this paper, we consider the following stochastic continuous submodular optimization problem:
where is a bounded convex body, is generally an unknown distribution, and ’s are are continuous submodular functions for every . Such a setting has recently been introduced in . We also denote the optimum value as . We note that the function is itself also continuous submodular as a non-negative combination of submodular functions are still submodular . The formulation covers popular instances of submodular optimization. For instance, when puts all the probability mass on a single function, (1.1) reduces to deterministic continuous submodular optimization. Another common objective is the finite-sum continuous submodular optimization where is uniformly distributed over instances, i.e., .
A natural approach to solving problems of the form (1.1) is to use projected stochastic methods. As we shall see in Section 5, these local search heuristics are surprisingly effective. However, the reasons for this empirical success is completely unclear. The main challenge is that maximizing corresponds to a nonconvex optimization problem (as the function is not concave), and a priori it is not clear why gradient methods should yield a reliable solution. This leads us to the main challenge of this paper
Do projected gradient methods lead to provably good solutions for continuous submodular maximization with general convex constraints?
We answer the above question in the affirmative, proving that projected gradient methods produce a competitive solution with respect to the optimum. More specifically, given a general bounded convex body and a continuous function that is monotone, smooth, and (weakly) DR-submodular we show that
All stationary points of a DR-submodular function over provide a approximation to the global maximum. Thus, projected gradient methods with sufficiently small step sizes (a.k.a. gradient flows) always lead to a solutions with approximation guarantees.
More generally, for weakly continuous DR-submodular functions with parameter (define in (2.6)) we prove the above results with approximation guarantee.
Our result have some important implications. First, they show that projected gradient methods are an efficient way of maximizing the multilinear extension of (weakly) submodular set functions for any submodularity ratio (note that corresponds to submodular functions) . Second, in contrast to conditional gradient methods for submodular maximization that should always start from the origin , projected gradient methods can start from any initial point in the constraint set and still produce a competitive solution. Third, such conditional gradient methods, when applied to the stochastic setting (with a fixed batch size), perform poorly and can produce arbitrarily bad solutions when applied to continuous submodular functions (see Appendix B for an example and further discussion on why conditional gradient methods do not easily admit stochastic variants). In contrast, stochastic projected gradient methods are stable by design and provide a solution with guarantee in expectation. Finally, our work provides a unifying approach for solving the stochastic submodular maximization problem
Continuous submodular maximization
One can easily verify that for a differentiable DR-submodular functions the gradient is an antitone mapping, i.e., for all such that we have . When twice differentiable, DR-submodularity is equivalent to
The above twice differentiable functions are sometimes called smooth submodular functions in the literature . However, in this paper, we say a differentiable submodular function is -smooth w.r.t a norm (and its dual norm ) if for all we have
We note that for set functions, DR-submodularity (i.e., Eq. 2.4) and submodularity (i.e., Eq. 2.1) are equivalent. However, this is not true for the general submodular functions defined on integer lattices or product of sub-intervals .
The focus of this paper is on continuous submodular maximization defined in Problem (1.1). More specifically, we assume that is a a general bounded convex set (not necessarily down-closed as considered in ) with diameter . Moreover, we consider ’s to be monotone (weakly) DR-submodular functions with parameter .
Background and related work
Generalization of submodular set functions has lately received a lot of attention. For instance, a line of recent work considered DR-submodular function maximization over an integer lattice . Interestingly, Ene and Nguyen provided an efficient reduction from an integer-lattice DR-submodular to a submodular set function, thus suggesting a simple way to solve integer-lattice DR-submodular maximization. Note that such reductions cannot be applied to the optimization problem (1.1) as expressing general convex body constraints may require solving a continuous optimization problem.
Algorithms and main results
In this section we discuss our algorithms together with the corresponding theoretical guarantees. In what follows, we assume that is a weakly DR-submodular function with parameter .
We begin with the definition of a stationary point.
Stationary points are of interest because they characterize the fixed points of the Gradient Ascent (GA) method. Furthermore, (projected) gradient ascent with a sufficiently small step size is known to converge to a stationary point for smooth functions . To gain some intuition regarding this connection, let us consider the GA procedure. Roughly speaking, at any iteration of the GA procedure, the value of increases (to the first order) by . Hence, the progress at time is at most . If at any time we have , then the GA procedure will not make any progress and it will be stuck once it falls into a stationary point.
The next natural question is how small can the value of be at a stationary point compared to the global maximum? The following lemma relates the value of at a stationary point to OPT.
If is a stationary point of in , then .
Furthermore, if is -smooth, gradient ascent with a step size smaller than will converge to a stationary point.
The theorem above guarantees that all fixed points of the GA method yield a solution whose function value is at least OPT. Thus, all fixed point of GA provide a factor approximation ratio. The particular case of , i.e., when is DR-submodular, asserts that at any stationary point is at least . This lower bound is in fact tight. In Appendix A we provide a simple instance of a differentiable DR-Submodular function that attains at a stationary point that is also a local maximum.
2 (Stochastic) gradient methods
We now discuss our first algorithmic approach. For simplicity we focus our exposition on the DR submodular case, i.e., , and discuss how this extends to the more general case in the proofs (Section 7.4). A simple approach to maximizing DR submodular functions is to use the (projected) Gradient Ascent (GA) method. Starting from an initial estimate obeying the constraints, GA iteratively applies the following update
Here, is the learning rate and denotes the Euclidean projection of onto the set . However, in many problems of practical interest we do not have direct access to the gradient of . In these cases it is natural to use a stochastic estimate of the gradient in lieu of the actual gradient. This leads to the Stochastic Gradient Method (SGM). Starting from an initial estimate obeying the constraints, SGM iteratively applies the following updates
Specifically, at every iteration , the current iterate is updated by adding , where is an unbiased estimate of the gradient and is the learning rate. The result is then projected onto the set . We note that when , i.e., when there is no randomness in the updates, then the SGM updates (4.2) reduce to the GA updates (4.1). We detail the SGM method in Algorithm 1.
As we shall see in our experiments detained in Section 5, the SGM method is surprisingly effective for maximizing monotone DR-submodular functions. However, the reasons for this empirical success was previously unclear. The main challenge is that maximizing corresponds to a nonconvex optimization problem (as the function is not concave), and a priori it is not clear why gradient methods should yield a competitive ratio. Thus, studying gradient methods for such nonconvex problems poses new challenges:
Do (stochastic) gradient methods converge to a stationary point?
We run stochastic gradient updates of the form (4.2) with . Let be a random variable taking values in with equal probability. Then,
We would like to note that if we pick to be a random variable taking values in with probability and and each with probability then
The above results roughly state that iterations of the stochastic gradient method from any initial point, yields a solution whose objective value is at least . Stated differently, iterations of the stochastic gradient method provides in expectation a value that exceeds approximation ratio for DR-submodular maximization. As explained in Section 4.1, it is not possible to go beyond the factor approximation ratio using gradient ascent from an arbitrary initialization.
An important aspect of the above result is that it only requires an unbiased estimate of the gradient. This flexibility is crucial for many DR-submodular maximization problems (see, (1.1)) as in many cases calculating the function and its derivative is not feasible. However, it is possible to provide a good un-biased estimator for these quantities.
We would like to point out that our results are similar in nature to known results about stochastic methods for convex optimization. Indeed, this result interpolates between the for stochastic smooth optimization, and the for deterministic smooth optimization. The special case of which corresponds to Gradient Ascent deserves particular attention. In this case, and under the assumptions of Theorem 4.3, it is possible to show that , without the need for a randomized choice of .
Finally, we would like to note that while the first term in (4.4) decreases as , the pre-factor could be rather large in many applications. For instance, this quantity may depend on the dimension of the input (see Section C in the Appendix). Thus, the number of iterations for reaching a desirable accuracy may be very large. Such a large computational load causes (stochastic) gradient methods infeasible in some application domains. We will overcome this deficiency in the next section by using stochastic mirror methods.
3 Stochastic mirror method
is strictly convex and differentiable.
We define the Bregman divergence associated to a mirror map as
We also define the projection onto a set with respect to a mapping via
Finally, we define the diameter as follows
Let be a mirror map that is -strongly convex on with respect to the norm . Assume that is -smooth with respect to the norm and is a monotone, continuous submodular function. Furthermore, assume that we have access to a stochastic oracle obeying
We start from and run the mirror ascent updates of the form
with . Let be a random variable taking values in with equal probability. Then,
We would like to note that if we pick to be a random variable taking values in with probability and and each with probability then
Experiments
Therefore, by running the stochastic versions of projected gradient methods, we can find a solution in the continuous domain that is at least approximation to the optimal value. By rounding that fractional solution (for instance via randomized Pipage rounding ) we obtain a set whose utility is at least of the optimum solution set of size . We note that randomized Pipage rounding does not need access to the value of . We also remark that projection onto can be done very efficiently in time (see ). Therefore, such approach easily scales to big data scenarios where the size of the data set (e.g. number of users) or the number of items (e.g. number of movies) are very large.
In our experiments, we consider the following baselines:
Stochastic Gradient Ascent (SG): with the step size and batch size . The details for computing an unbiased estimation for the gradient of are given in Appendix D.
Stochastic Mirror Ascent (SM): with the step size and batch size .
Frank-Wolfe (FW) variant of : with parameter for the total number of iterations and batch size (we further let , see Algorithm 1 in for more details).
Batch-mode Greedy (Greedy): by running greedy algorithm over the empirical objective function with samples.
To run the experiments we use the MovieLens data set. It consists of 1 million ratings (from 1 to 5) by users for movies. Let denote the rating of user for movie (if such a rating does not exist we assign to 0). In our experiments, we consider two well motivated objective functions. The first one is the facility location where the valuation function by user is defined as . In words, the way user evaluates a set is by picking the highest rated movie in . For simplicity, we also assume that the distribution is uniform. Thus, the objective function is .
In our second experiment, we consider a different user-specific valuation function which is a concave function composed with a modular function, i.e., Again, by considering the uniform distribution over the set of users, we obtain Note that the multilinear extensions of and are neither concave nor convex.
Figure 1 depicts the performance of different algorithms for the two proposed objective functions. As Figures 1(a) and 1(c) show, the FW algorithm needs a much higher batch size to be comparable in performance w.r.t. to our stochastic gradient methods. With the same batch size and number of iterations SG, SM, and FW have similar computational complexity. Therefore, a smaller batch size leads to less computational effort. Figure 1(b) shows that after a few hundred iterations both SG and SM with obtain almost the same utility as Greedy with a large batch size (). Finally, Figure 1(d) shows the performance of the algorithms with respect to the number of times the single functions (’s) are evaluated. This further shows that gradient based methods have comparable complexity w.r.t. the Greedy algorithm in the discrete domain.
Conclusion
In this paper we studied gradient methods for submodular maximization. Despite the lack of convexity of the objective function we demonstrated that local search heuristics are effective at finding approximately optimal solutions. In particular, we showed that all fixed point of projected gradient ascent provide a factor approximation to the global maxima. We also demonstrated that stochastic gradient and mirror methods achieve an objective value of in iterations. We further demonstrated the effectiveness of our methods with experiments on real data.
While in this paper we have focused on convex constraints, our framework may allow non-convex constraints as well. For instance it may be possible to combine our framework with recent results in to deal with general nonconvex constraints. Furthermore, in some cases projection onto the constraint set may be computationally intensive or even intractable but calculating an approximate projection may be possible with significantly less effort. One of the advantages of gradient descent-based proofs is that they continue to work even when some perturbations are introduced in the updates. Therefore, we believe that our framework can deal with approximate projections and we hope to pursue this in future work.
Proofs
To prove part (i) we first prove that for any two vectors we have
To this aim note that for all s.t. , by using (7.1), we have
Now, from (7.1) we deduce that for any :
From these two inequalities we immediately obtain
and we obtain (7.2) by noting that and .
Part (i) of the theorem follows from (7.2) by letting to be a stationary point and .
To prove part (ii) note that by the smoothness of the function (more specifically the quadratic upper bound) we have
Now note that and thus using the properties of convex projections we have
Plugging this into the latter inequality we conclude that for
By definition of projection the latter implies that . A well known result in convex analysis (e.g., see [33, Lemma 6.4] or [34, Lemma 7.11]) implies , concluding the proof.
2 Proof of (stochastic) mirror method (Proof of Theorem 4.7)
We begin by stating some lemmas about mirror descent together with some useful preliminary lemmas in the next section.
We begin with two lemmas about mirror descent adapted from .
Let and , then
Also we need the following well-known identity about Bregman divergences which will be useful several times in our proofs.
We next state a lemma due to Chekuri, Vondrak, and Zenkluser.
[36, Lemma 3.2] Assume is a monotone and submodular function. Then, for any two points
Consider one iteration of the mirror descent update
Using , we conclude that
where the last equality follows from Lemma 7.2.
Consider the setting of Theorem 4.7 and let be the Bregman divergence corresponding to the mirror map . Let be a nonnegative scalar with obeying
Proof Using smoothness of the function we have the following chain of inequalities
where (a) follows from the fact that by Young’s inequality and (b) follows from strong convexity of the mirror map . Rearranging the above inequality we arrive at the following chain of inequalities
where (a) follows from Lemma 7.4 and (b) from the choice .
Using Lemma 7.5 with (Global optimum) and we have
Using \big{\langle}\bm{g}_{t},\bm{x}_{t}-\bm{x}^{*}\big{\rangle}=\big{\langle}\nabla F(\bm{x}_{t}),\bm{x}_{t}-\bm{x}^{*}\big{\rangle}+\big{\langle}\bm{g}_{t}-\nabla F(\bm{x}_{t}),\bm{x}_{t}-\bm{x}^{*}\big{\rangle} we conclude that
Using Lemma 7.3 with and in the above inequality we conclude that
Taking expectation of both sides we arrive at
Summing both sides from to we conclude that
Here, (a) follows from the fact that . Now using and we arrive at
Here, (a) follows from the fact that . Thus
The proof now follows from the fact that . Note also that from (7.6) we have
3 Proof of (stochastic) gradient method (Proof of Theorem 4.3)
4 Extensions to weakly submodular functions
In this section we shall show that Theorem 4.7 extends to weakly submodular functions with the new guarantee given by
To this aim using Lemma 7.5 with (Global optimum) and we have
Using \big{\langle}\bm{g}_{t},\bm{x}_{t}-\bm{x}^{*}\big{\rangle}=\big{\langle}\nabla F(\bm{x}_{t}),\bm{x}_{t}-\bm{x}^{*}\big{\rangle}+\big{\langle}\bm{g}_{t}-\nabla F(\bm{x}_{t}),\bm{x}_{t}-\bm{x}^{*}\big{\rangle} we conclude that
Using condition (7.2) with and in the above inequality we conclude that
Taking expectation of both sides we arrive at
Summing both sides from to we conclude that
Here, (a) follows from the fact that . Now using and we arrive at
Here, (a) follows from the fact that . Thus
Dividing both sides by concludes the proof.
Acknowledgements
This work was done while the authors were visiting the Simon’s Institute for the Theory of Computing. The authors would like to thank Jeff Bilmes, Volkan Cevher, Maryam Fazel, Mohammad-Reza Karimi, Andreas Krause, Mario Lucic, and Andrea Montanari for helpful discussions.
References
Appendix A A DR-Submodular Function that Attains OPT/2+ϵOPT2italic-ϵ\rm{OPT}/2+\epsilon on a local maximum
Now, define . We claim that is a local maximum. To see this, we have . As a result, for any :
As a result, is a stationary point. It remains to show that in a sufficiently small neighborhood of inside , becomes the maximizer of . Note that . Consider a point
It is easy to see that . We have
Thus, by choosing , we conclude that any with form as in (A.1) has a lower function value than . This proves that is a local maximum. Now, consider the vector . We have . As a result, by considering a large enough , the value of at the local maximum becomes .
Appendix B An Example for Deficiency of the Frank-Wolfe Type Algorithm of [16] in the Stochastic Setting
Assume we want to maximize a DR-Submodular function over a convex set . Assume further that . The Frank-Wolfe Type algorithm discussed in can be briefly stated as follows (note that for simplicity we let and , see Algorithm 1 in ): Fix a (large) number as the total number of iterations, let and for do:
Assume now that instead of we have access to an unbiased estimator where . Note that . As a result, for we obtain , where is the vector that has at position and elsewhere. Interestingly for this example, the stochastic Frank-Wolf algorithm never makes any progress on the coordinate and will always take on the -th coordinate. As a result, it is easy to see that for large the algorithm will end up at . However, we have and which can become arbitrarily small with .
Let us briefly explain why conditional gradient methods do not easily admit stochastic variants. The main bottleneck is in the update step of the continuous greedy algorithm (FW). As stated above, in each iteration, FW finds a point in the constraint set which has the highest inner product with the gradient and then uses this vector in order to update the current position. However, this step is not very robust to the noise. More precisely, if instead of the gradient of we plug into the a noisy (and unbiased) version of the gradient, the outcome may be far from . In other words, expectation and are not interchangeable. It is easy to see that the above example extends to FW with any fixed natch size (i.e. when gradient is approximated by averaging a fixed number of i.i.d. samples).
Before proving the lemma, let us remark that in many practical applications, the value of is not so large (see for example the movie recommendation setting of Section 5 where is less than the maximum possible rating).
Proof At any point , the Hessian of , denoted by , has the following property (see ):
Appendix D How to Construct an Unbiased Estimator of the Gradient in Multilinear Extensions
where for example by we mean a vector which has value on its -th coordinate and is equal to elsewhere. To create an unbiased estimator for at a point we can simply sample a set by including each element in it independently with probability and use as an unbiased estimator for the -th partial derivative. We can sample one single set and use the above trick for all the coordinates. This involves function computations for . Having a batch size we can repeat this procedure times and then average.