Convergence of Learning Dynamics in Stackelberg Games
Tanner Fiez, Benjamin Chasnov, Lillian J. Ratliff
Introduction
Tools from game theory now play a prominent role in machine learning. The emerging coupling between the fields can be credited to the formulation of learning problems as interactions between competing algorithms and the desire to characterize the limiting behaviors of such strategic interactions. Indeed, game theory provides a systematic framework to model the strategic interactions found in modern machine learning problems.
A significant portion of the game theory literature concerns games of simultaneous play and equilibrium analysis. In simultaneous play games, each player reveals the strategy they have selected concurrently. The solution concept often adopted in non-cooperative simultaneous play games is the Nash equilibrium. In a Nash equilibrium, the strategy of each player is a best response to the joint strategy of the competitors so that no player can benefit from unilaterally deviating from this strategy.
The study of equilibrium gives rise to the question of when and why the observed play in a game can be expected to correspond to an equilibrium. A common explanation is that an equilibrium emerges as the long run outcome of a process in which players repeatedly play a game and compete for optimality over time . Consequently, an important topic in the study of learning in games is the convergence behavior of learning algorithms reflecting the underlying game dynamics. Adopting this viewpoint and analyzing so-called ‘natural’ dynamics often provides deep insights into the structure of a game. Moreover, a firm understanding of the structure of a game can inform how to design learning algorithms strictly for computing equilibria. Seeking equilibria via computationally efficient learning algorithms is an equally important perspective on equilibrium analysis .
The classic objectives of learning in games are now being widely embraced in the machine learning community. While not encompassing, the prevailing research areas epitomizing this phenomenon are adversarial training and multi-agent learning. A considerable amount of this work has focused on generative adversarial networks (GANs) . Finding Nash equilibria in GANs is challenging owing to the complex optimization landscape that arises when each player in the game is parameterized by a neural network. Consequently, significant effort has been spent lately on developing principled learning dynamics for this application . In general, this line of work has analyzed learning dynamics designed to mitigate rotational dynamics and converge faster to stable fixed points or to avoid spurious stable points of the dynamics and reach equilibria almost surely. In our work, we draw connections to this literature and believe that the problem we study gives an under-explored perspective that may provide valuable insights moving forward.
Characterizing the outcomes of competitive interactions and seeking equilibria in multi-agent learning gained prominence much earlier than adversarial training. However, following initial works on this topic , scrutiny was given to the solution concepts being considered and the field cooled . Owing to the arising applications with interacting agents, problems of this form are being studied extensively again. There has been a shift toward analyzing gradient-based learning rules, in part due to their scalability and success in single-agent reinforcement learning, and rigorous convergence analysis .
The progress analyzing learning dynamics and seeking equilibria in games is promising, but the work has been narrowly focused on simultaneous play games and the Nash equilibrium solution concept. There are many problems exhibiting a hierarchical order of play between agents in a diverse set of fields such as human-robot collaboration and interacting autonomous systems in artificial intelligence , mechanism design and control , and organizational structures in economics . In game theory, this type of game is known as a Stackelberg game and the solution concept studied is called a Stackelberg equilibrium.
In the simplest formulation of a Stackelberg game, there is a leader and a follower that interact in a hierarchical structure. The sequential order of play is such that the leader is endowed with the power to select an action with the knowledge that the follower will then play a best-response. Specifically, the leader uses this knowledge to its advantage when selecting a strategy.
In this paper, we study the convergence of learning dynamics in Stackelberg games. Our motivation stems from the emergence of problems in which there is a distinct order of play between interacting learning agents and the lack of existing theoretical convergence guarantees in this domain. The dynamics analyzed in this work reflect the underlying game structure and characterize the expected outcomes of hierarchical game play. The rigorous study of the learning dynamics in Stackelberg games we provide also has implications for simultaneous play games relevant to adversarial training.
We formulate and study a novel set of gradient-based learning rules in continuous, general-sum games that emulate the natural structure of a Stackelberg game. Building on work characterizing a local Nash equilibrium in continuous games , we define the differential Stackelberg equilibrium solution concept (Definition 4), which is a local notion of a Stackelberg equilibrium amenable to computation. An analogous local minimax equilibrium concept was developed concurrently with this work, but strictly for zero-sum games . Importantly, the equilibrium notion we present generalizes the local minimax equilibrium concept to general-sum games. In our work, we draw several connections between Nash and Stackelberg equilibria for the class of zero-sum games, which can be summarized as follows:
We show in Proposition 2 that stable Nash equilibria are differential Stackelberg equilibria in zero-sum games. Concurrent with our work, Jin et al. equivalently show that local Nash equilibria are local minimax equilibria. This result indicates learning dynamics seeking Nash equilibria are simultaneously seeking Stackelberg equilibria.
We reveal that there exist stable attractors of simultaneous gradient play that are Stackelberg equilibria and not Nash equilibria. Moreover, in Propositions 3 and 4 we give necessary and sufficient conditions under which the simultaneous gradient play dynamics can avoid Nash equilibria and converge to Stackelberg equilibria. To demonstrate the relevancy to deep learning applications, Propositions 5 and 6 specialize the general necessary and sufficient conditions from Propositions 3 and 4 to GANs satisfying the realizable assumption , which presumes the generator is able to create the underlying data distribution. This set of results has implications for the optimization landscape in GANs as we explore in our numerical experiments.
Our primary contributions concern the convergence behavior of the gradient-based learning rules we formulate that mirror the Stackelberg game structure. These contributions can be summarized as follows:
We demonstrate in Proposition 1 that the only stable critical points of the Stackelberg gradient dynamics are Stackelberg equilibria in zero-sum games. This is in contrast to the simultaneous gradient play dynamics, which can be attracted to non-Nash critical points in zero-sum games. This insight allows us to define a gradient-based learning rule for the leader while the follower plays a best response for which each attracting critical point is a Stackelberg equilibria in zero-sum games. As a result, the learning rule provably converges to an equilibria given an initialization in the region of attraction of a stable critical point. A formal exposition of this set of dynamics and results is provided in Section 3.1.
Leveraging the Stackelberg game structure, for general-sum games, we formulate a gradient-based learning rule in which the leader and follower have an unbiased estimator of their gradient so that updates are stochastic.
In Section 3.2, we consider a formulation in which the follower uses a gradient-play update rule instead of an exact best response strategy and propose a two-timescale algorithm to learn Stackelberg equilibria. We show almost sure asymptotic convergence to Stackelberg equilibria in zero-sum games and to stable attractors in general-sum games; a finite-time high probability bound for local convergence to a neighborhood of a stable Stackelberg equilibrium in general-sum games is also given.
We present this paper with a single leader and a single follower, but this is only for ease of presentation. The extension to followers that play in a staggered hierarchical structure or simultaneously is in Appendix F; equivalent results hold with some additional assumptions.
Finally, we present several numerical experiments in Section 4, which we now detail:
We present a location game on a torus and a Stackelberg duopoly game. The examples are general-sum games with equilibrium that can be solved for directly, allowing us to numerically validate our theory. The games demonstrate the advantage the leader gains from the hierarchical order of play compared to the simultaneous play versions of the games.
We evaluate the Stackelberg learning dynamics as a GAN training algorithm. In doing so, we find that the leader update removes rotational dynamics and prevents the type of cycling behavior that plagues simultaneous gradient play. Moreover, we discover that the simultaneous gradient dynamics can empirically converge to non-Nash attractors that are Stackelberg equilibria in GANs. The generator and the discriminator exhibit desirable performance at such points, indicating that Stackelberg equilibria can be as desirable as Nash equilibria. Lastly, the Stackelberg learning dynamics often converge to non-Nash attractors and reach a satisfying solution quickly using learning rates that can cause the simultaneous gradient descent dynamics to cycle.
The perspective we explore on analyzing games in which there is an order of play or hierarchical decision making structure has been generally ignored in the modern learning literature. However, this viewpoint has long been researched in the control literature on games . Similarly, work on bilevel optimization adopts this perspective.
The select few recent works in the machine learning literature on learning in games considering a hierarchical decision-making structure exclusively focus on zero-sum games , unlike our work, which extends to general-sum games. A noteworthy paper in the line of work in the zero-sum setting adopting a min-max perspective was the introduction of unrolled GANs . The authors consider a timescale separation between the generator and discriminator, giving the generator the advantage as the slower player. This work used the Schur complement structure presented in Danskin to define a minimax solution of a zero-sum game abstraction of an adversarial training objective. Essentially the discriminator is allowed to perform a finite roll-out in an inner loop of the algorithm with multiple updates; this process is referred to as ‘unrolling’. It is (informally) suggested that, using the results of Danskin , as the roll-out horizon approaches infinity, the discriminator approaches a critical point of the cost function along the discriminators axis given a fixed generator parameter configuration.
The unrolling procedure has the same effect as a deterministic timescale separation between players. Formal convergence guarantees to minimax equilibria in zero-sum games characterizing the limiting behavior of simultaneous individual gradient descent with timescale separation were recently obtained . While related, simultaneous individual gradient play with time-scale separation is a distinct set of dynamics that departs from the dynamics we propose that reflect the Stackelberg game structure.
It is also worth pointing out that the multi-agent learning papers of Foerster et al. and Letcher et al. do in some sense seek to give a player an advantage, but nevertheless focus on the Nash equilibrium concept in any analysis that is provided.
In Section 2, we formalize the problem we study and provide background material on Stackelberg games. We then draw connections between learning in Stackelberg games and existing work in zero-sum and general sum-games relevant to GANs and multi-agent learning, respectively. In Section 3, we give a rigorous convergence analysis of learning in Stackelberg games. Numerical examples are provided in Section 4 and we conclude in Section 5.
Preliminaries
We leverage the rich theory of continuous games and dynamical systems in order to analyze algorithms implemented by agents interacting in a hierarchical game. In particular, each agent has an objective they want to selfishly optimize that depends on not only their own actions but also on the actions of their competitor. However, there is an order of play in the sense that one player is the leader and the other player is the followerWhile we present the work for a single leader and a single follower, the theory extends to the multi-follower case (we discuss this in Appendix F) and to the case where the single leader abstracts multiple cooperating agents.. The leader optimizes its objective with the knowledge that the follower will respond by selecting a best response. We refer to algorithms for learning in this setting as hierarchical learning algorithms. We specifically consider a class of learning algorithms in which the agents act myopically with respect to their given objective and role in the underlying hierarchical game by following the gradient of their objective with respect to their choice variable.
The leader aims to solve the optimization problem given by
and the follower aims to solve the optimization problem . As noted above, the learning algorithms we study are such that the agents follow myopic update rules which take steps in the direction of steepest descent with respect to the above two optimizations problems, the former for the leader and the latter for the follower.
Before formalizing these updates, let us first discuss the equilibrium concept studied for simultaneous play games and contrast it with that which is studied in the hierarchical play counterpart. The typical equilibrium notion in continuous games is the pure strategy Nash equilibrium in simultaneous play games and the Stackelberg equilibrium in hierarchical play games. Each notion of equilibria can be characterized as the intersection points of the reaction curves of the players .
The joint strategy is a Nash equilibrium if for each ,
The strategy is a local Nash equilibrium on if for each ,
In a two-player game with player 1 as the leader, a strategy is called a Stackelberg equilibrium strategy for the leader if
where is the rational reaction set of .
This definition naturally extends to the -follower setting when is replaced with the set of Nash equilibria , given that player 1 is playing so that the follower’s reaction set is a Nash equilibrium.
We denote as the derivative of with respect to , as the partial derivative of with respect to , and as the total derivativeFor example, given a function , .. Denote by the vector of individual gradients for simultaneous play and as the equivalent for hierarchical play where is the total derivative of with respect to and is implicitly a function of , which captures the fact that the leader operates under the assumption that the follower will play a best response to its choice of .
It is possible to characterize a local Nash equilibrium using sufficient conditions for Definition 1.
The joint strategy is a differential Nash equilibrium if and for each .
Analogous sufficient conditions can be stated to characterize a local Stackelberg equilibrium strategy for the leader using first and second order conditions on the leader’s optimization problem. Indeed, if and is positive definite, then is a local Stackelberg equilibrium strategy for the leader. We use these sufficient conditions to define the following refinement of the Stackelberg equilibrium concept.
The pair with , where is implicitly defined by , is a differential Stackelberg equilibrium for the game with player 1 as the leader if , and is positive definite..
Before moving on, let us make a few remarks about similar, and in some cases analogous, equilibrium definitions. For zero-sum games, the differential Stackelberg equilibrium notion is the same as a local min-max equilibrium for a sufficiently smooth cost function . This is a well-known concept in optimization (see, e.g., , among others), and it has recently been introduced in the learning literature . The benefit of the Stackelberg perspective is that it generalizes from zero-sum games to general-sum games, while the min-max equilibrium notion does not. A number of adversarial learning formulations are in fact general-sum, often as a result of regularization and well-performing heuristics that augment the cost functions of the generator or the discriminator.
We utilize these local characterizations in terms of first and second order conditions to formulate the myopic hierarchical learning algorithms we study. Following the preceding discussion, consider the learning rule for each player to be given by
recalling that and the notation indicates the entry of corresponding to the –th player. Moreover, the sequence of learning rates and is the noise process for player , both of which satisfy the usual assumptions from theory of stochastic approximation provided in detail in Section 3. We note that the component of the update captures the case in which each agent does not have oracle access to , but instead has an unbiased estimator for it. The given update formalizes the class of learning algorithms we study in this paper.
We require a timescale separation between the leader and the follower: the leader is assumed to be learning at a slower rate than the follower so that . The reason for this timescale separation is that the leader’s update is formulated using the reaction curve of the follower. In the gradient-based learning setting considered, the reaction curve can be characterized by the set of critical points of that have a local positive definite structure in the direction of , which is
This set can be characterized in terms of an implicit map , defined by the leader’s belief that the follower is playing a best response to its choice at each iteration, which would imply . Moreover, under sufficient regularity conditions, the implicit mapping theorem gives rise to the implicit map on a neighborhood of . Formalized in Section 3, we note that when is defined uniformly in on the domain for which convergence is being assessed, the update in (1) is well-defined in the sense that the component of the derivative corresponding to the implicit dependence of the follower’s action on via is well-defined and locally consistent. In particular, for a given point such that with an isomorphism, the implicit function theorem implies there exists an open set such that there exists a unique continuously differentiable function such that and for all . Moreover,
on . Thus, in the limit of the two-timescale setting, the leader sees the follower as having equilibriated (meaning ) so that
The map is an implicit representation of the follower’s reaction curve.
The following describes the general approach to studying the hierarchical learning dynamics in (1). The purpose of this overview is to provide the reader with the high-level architecture of the analysis approach.
The analysis techniques we employ combine tools from dynamical systems theory with the theory of stochastic approximation. In particular, we leverage the limiting continuous time dynamical systems derived from (1) to characterize concentration bounds for iterates or samples generated by (1). We note that the hierarchical learning update in (1) with timescale separation has a limiting dynamical system that takes the form of a singularly perturbed dynamical system given by
which, in the limit as , approximates (1).
The limiting dynamical system has known convergence properties (asymptotic convergence in a region of attraction for a locally asymptotically stable attractor). Such convergence properties can be translated in some sense to the discrete time system by comparing pseudo-trajectories—in this case, linear interpolations between sample points of the update process—generated by sample points of (1) and the limiting system flow for initializations containing the set of sample points of (1). Indeed, the limiting dynamical system is then used to generate flows initialized from the sample points generated by (1). Creating pseudo-trajectories, we then bound the probability that the pseudo-trajectories deviate by some small amount from the limiting dynamical system flow over each continuous time interval between the sample points. A concentration bound can be constructed by taking a union bound over each time interval after a finite time; following this we can guarantee the sample path has entered the region of attraction, on which we can produce a Lyapunov function for the continuous time dynamical system. The analysis in this paper is based on the high-level ideas outlined in this section.
2 Connections and Implications
Before presenting convergence analysis of the update in (1), we draw some connections to application domains—including adversarial learning, where zero-sum game abstractions have been touted for finding robust parameter configurations for neural networks, and opponent shaping in multi-agent learning—and equilibrium concepts commonly used in these domains. Let us first remind the reader of some common definitions from dynamical systems theory.
Further, is said to be asymptotically stable if is additionally attractive—that is, for all , there exists such that
A critical point is said to be non-degenerate if the determinant of the Jacobian of the dynamics at the critical point is non-zero. For a non-degenerate critical point, the Hartman-Grobman theorem enables us to check the eigenvalues of the Jacobian to determine asymptotic stability. In particular, at a non-degenerate critical point, if the eigenvalues of the Jacobian are in the open left-half complex plane, then the critical point is asymptotically stable. The dynamical systems we study in this paper are of the form for some vector field determined by the gradient based update rules employed by the agents. Hence, to determine if a critical point is stable, we simply need to check that the spectrum of the Jacobian of is in the open right-half complex plane.
Zero-sum games are a very special class since there is a strong connection between Nash equilibria and Stackelberg equilibria. Let us first show that for zero-sum games, attracting critical points of are differential Stackelberg equilibria.
Proof. Consider an arbitrary sufficiently smooth zero-sum game on continuous strategy spaces. The Jacobian of the Stackelberg limiting dynamics at a stable critical point is
The structure of the Jacobian follows from the fact that
The eigenvalues of a lower triangular block matrix are the union of the eigenvalues in each of the block diagonal components. This implies that if , then necessarily and . Consequently, any stable critical point of the Stackelberg limiting dynamics must be a differential Stackelberg equilibrium by definition. The result of Proposition 1 implies that with appropriately chosen stepsizes the only attracting critical points of the update rule in (1) will be Stackelberg equilibria and thus, unlike simultaneous play individual gradient descent (known as gradient-play in the game theory literature), will not converge to spurious locally asymptotically stable attractors of the dynamics that are not relevant to the underlying game.
In a recent work on GANs , hierarchical learning of a similar nature proposed in this paper is studied in the context of zero-sum games. In the author’s formulation, the generator is deemed the leader and the discriminator as the follower. The idea is to allow the discriminator to take individual gradient steps to update its parameters, while the parameters of the generator are held fixed. The effect of ‘unrolling’ the discriminator update for steps is that a surrogate objective of arises for the generator, meaning that the timescale-separation between the discriminator and the follower induces an update reminiscent of that given for the leader in (2). In particular, when the follower converges to a local optimum as a function of the generator’s parameters so that . As a result, the critical points coincide with the Stackelberg dynamics we study, indicating that unrolled GANs are converging only to Stackelberg equilibria. Empirically, GANs learned with such timescale separation procedures seem to outperform gradient descent with uniform stepsizes , providing evidence Stackelberg equilibria can be sufficient in GANs.
This begs a further question of if attractors of the dynamics are Stackelberg equilibria. We begin to answer this inquiry by showing that stable differential Nash are differential Stackelberg equilibria.
Proof. Consider an arbitrary sufficiently smooth zero-sum game on continuous strategy spaces. Suppose is a stable differential Nash equilibrium so that by definition , , and
Then, the Schur complement of is also positive definite:
Hence, is a differential Stackelberg equilibrium since the Schur complement of is exactly the derivative at critical points and since is a differential Nash equilibrium.
In the zero-sum setting, the fact that Nash equilibria are a subset of Stackelberg equilibria (or minimax equilibria) for finite games is well-known . We show the result for the notion of differential Stackelberg equilibria for continuous action space games that we introduce. Similar to our work and concurrently, Jin et al. also show that local Nash equilibria are local minmax solutions for continuous zero-sum games. It is interesting to point out that for a subclass of zero-sum continuous games with a convex-concave structure for the leader’s cost the set of (differential) Nash and (differential) Stackelberg equilibria coincide. Indeed, at critical points for convex-concave games, so that if is a differential Stackelberg equilibrium, it is also a Nash equilibrium.
This result indicates that recent works seeking Nash equilibria in GANs are seeking Stackelberg equilibria concurrently. Given that it is well-known simultaneous gradient play can converge to attracting critical points that do not satisfy the conditions of a Nash equilibria, it remains to determine when such spurious non-Nash attractors of the dynamics will be an attractor of the Stackelberg dynamics .
Let player be the leader who aims to minimize with respect to taking into consideration that player 2 (follower) aims to minimize with respect to . In Fig. 1, we show the trajectories for various initializations for this game with ; it can be seen that for several initializations, simultaneous gradient play leads to non-Nash attractors which are differential Stackelberg equilibria.
We now proceed to provide necessary and sufficient conditions for the phenenom demonstrated in Example 1. Attracting critical points of the dynamics that are not Nash equilibria are such that either or are not positive definite. Without loss of generality, considering player 1 to be the leader, an attractor of the Stackelberg dynamics requires both and to be positive definite. Hence, if is not positive definite at a non-Nash attractor of , then will also not be an attractor of . We focus on non-Nash attractors with and seek to determine when the Schur complement is positive definite, so that is an attractor of .
and define .
Consider a non-Nash attracting critical point of the gradient dynamics such that . Given such that , if , then and for all .
This means that does not satisfy the conditions for a differential Stackelberg, however, the point does satisfy necessary conditions for a local Stackelberg equilibrium and the point is a marginally stable attractor of the dynamics.
While the results depend on conditions that are difficult to check a priori without knowledge of , certain classes of games for which these conditions hold everywhere and not just at the equilibrium can be constructed. For instance, alternative conditions can be given: if the function which defines the zero-sum game is such that it is concave in and there exists a such that
where Functions such that derivative of is Lipschitz will satisfy this condition. and with again a (not necessarily positive) diagonal matrix, then the results of Proposition 4 hold. From a control point of view, one can think about the leader’s update as having a feedback term with the follower’s input. On the other hand, the results are useful for the synthesis of games, such as in reward shaping or incentive design, where the goal is to drive agents to particular desirable behavior.
We remark that the fact that the eigenvalues of are in the open-right-half complex plane is not used in proving this result. We believe that further investigation could lead to a less restrictive sufficient condition. Empirically, by randomly generating the different block matrices, it is quite difficult to find examples such that has positive eigenvalues, , and the Schur complement is not positive definite. In fact, for games on scalar action spaces, it turns out that non-Nash attracting critical points of the simultaneous gradient play dynamics at which must be differential Stackelberg equilibria and attractors of the Stackelberg limiting dynamics.
The fact that the real components of the eigenvalues of the Jacobian are positive implies that and since the determinant and the trace of the Jacobian must be positive. Using this information, it directly follows that the Schur complement of is positive definite:
As a result, is a differential Stackelberg equilibrium and an attractor of since the Schur complement of is the derivative and was given. We suspect that using the notion of quadratic numerical range , which is a super set of the spectrum of a block operator matrix, along with the fact that the Jacobian of the simultaneous gradient play dynamics has its spectrum in the open right-half complex plane, may lead to an extension of the result to arbitrary dimensions.
The results of Propositions 3 and 4, Corollary 1, and Example 1 imply that some of the non-Nash attractors of are in fact Stackelberg equilibria. This is a meaningful insight since recent works have proposed schemes to avoid non-Nash attractors of the dynamics as they have been classified or viewed as lacking game-theoretic meaning . Moreover, some recent empirical results show that a number of successful approaches to training GANs are not converging to Nash equilibria, but rather to non-Nash attractors of the dynamics . It would be interesting to characterize whether or not the attractors satisfy the conditions we propose, and if such conditions could provide insights into how to improve GAN training. It also further suggests that the Stackelberg equilibria may be a suitable solution concept for GANs.
One of the common assumptions in some of the recent GANs literature is that the discriminator network is zero in a neighborhood of an equilibrium parameter configuration (see, e.g., ). This assumption limits the theory to the ‘realizable’ case; the work by provides relaxed assumptions for the non-realizable case. In both cases, the Jacobian for the dynamics is such that .
Consider a GAN satisfying the realizable assumption—that is, the discriminator network is zero in a neighborhood of any equilibrium. Then, an attracting critical point for the simultaneous gradient dynamics at which is positive semi-definite satisfies necessary conditions for a local Stackelberg equilibrium, and it will be a marginally stable point of the Stackelberg dynamics .
Proof. Consider an attracting critical point of such that . Note that the realizable assumption implies that the Jacobian of is
(see, e.g., ). Hence, since ,
Since is an attractor, and so that
Consequently, the necessary conditions for a local Stackelberg equilibrium are satisfied. Moreover, since both and the Schur complement , the Jacobian of is positive semi-definite so that the point is marginally stable.
Now, simply satisfying the necessary conditions is not enough to guarantee that attractors of the simultaneous play gradient dynamics will be a local Stackelberg equilibrium. We can state sufficient conditions by examining Proposition 4.
Consider a GAN satisfying the realizable assumption—that is, the discriminator network is zero in a neighborhood of any equilibrium—and an attractor for the simultaneous gradient dynamics at which is positive definite. Suppose that there exists a diagonal matrix with non-zero entries such that where are the orthonormal eigenvectors of . Then, is a differential Stackelberg equilibrium and an attractor of .
The proof follows directly from Proposition 4 and Proposition 5. It is not directly clear how restrictive these sufficient conditions are for GANs. We leave this for future inquiry.
2.2 Connections to Opponent Shaping
Beyond the work in zero-sum games and applications to GANs, there has also been recent work, which we will refer to as ‘opponent shaping’, where one or more players takes into account its opponents’ response to their action . The initial work of Foerster et al. bears the most resemblance to the learning algorithms studied in this paper. The update rule (LOLA) considered there (in the deterministic setting with constant stepsizes) takes the following form:
The attractors of these dynamics are not necessarily Nash equilibria nor are they Stackelberg equilibria as can be seen by looking at the critical points of the dynamics. Indeed, the LOLA dynamics lead only to Nash or non-Nash stable attractors of the limiting dynamics. The effect of the additional ‘look-ahead’ term is simply that it changes the vector field and region of attraction for stable critical points. In the zero-sum case, however, the critical points of the above are the same as those of simultaneous play individual gradient updates, yet the Jacobian is not the same and it is still possible to converge to a non-Nash attractor.
With a few modifications, the above update rule can be massaged into a form which more closely resembles the hierarchical learning rules we study in this paper. In particular, if instead of , player 2 employed a Newton stepsize of , then the update would look like
which resembles a deterministic version of (1). The critical points of this update coincide with the critical points of a Stackelberg game . With appropriately chosen stepsizes and with an initialization in a region on which the implicit map, which defines the component of the update, is well-defined uniformly in , the above dynamics will converge to Stackelberg equilibria. In this paper, we provide an in-depth convergence analysis and for the stochastic settingIn , the authors do not provide convergence analysis; they do in their extension, yet only for constant and uniform stepsizes and for a learning rule that is different than the one studied in this paper as all players are conjecturing about the behavior of their opponents. This distinguishes the present work from their setting. of the above update.
2.3 Comparing Nash and Stackelberg Equilibrium Cost
We have alluded to the idea that the ability to act first gives the leader a distinct advantage over the follower in a hierarchical game. We now formalize this statement with a known result that compares the cost of the leader at Nash and Stackelberg equilibrium.
([4, Proposition 4.4]). Consider an arbitrary sufficiently smooth two-player general-sum game on continuous strategy spaces. Let denote the infimum of all Nash equilibrium costs for player 1 and denote an arbitrary Stackelberg equilibrium cost for player 1. Then, if is a singleton for every , .
This result says that the leader never favors the simultaneous play game over the hierarchical play game in two-player general-sum games with unique follower responses. On the other hand, the follower may or may not prefer the simultaneous play game over the hierarchical play game.
The fact that under certain conditions the leader can obtain lower cost under a Stackelberg equilibrium compared to any of the Nash equilibrium may provide further explanation for the success of the methods in . Commonly, the discriminator can overpower the generator when training a GAN and giving the generator an advantage may mitigate this problem. In the context of multi-agent learning, the advantage of the leader in hierarchical games leads to the question of how the roles of each player in a game are decided. While we do not focus on this question, it is worth noting that when each player mutually benefits from the leadership of a player the solution is called concurrent and when each player prefers to be the leader the solution is called non-concurrent. We believe that exploring classes of games in which each solution concept arises is an interesting direction of future work.
Convergence Analysis
Following the preceding discussion, consider the learning rule for each player to be given by
where recall that . Moreover, for each , is the sequence of learning rates and is the noise process for player . As before, suppose player 1 is the leader and conjectures that player 2 updates its action in each round via . This setting captures the scenario in which players do not have oracle access to their gradients, but do have an unbiased estimator. As an example, players could be performing policy gradient reinforcement learning or alternative gradient-based learning schemes. Let for each and .
For each , the learning rates satisfy , .
A nonempty invariant set for is said to be internally chain transitive if for any and , , there exists a finite sequence with and , , such that , .
Suppose that the leader (player 1) operates under the assumption that the follower (player 2) is playing a local optimum in each round. That is, given , for which is a first-order local optimality condition. If, for a given , is invertible and , then the implicit function theorem implies that there exists neighborhoods and and a smooth map such that .
where is defined via the map defined implicitly in a neighborhood of .
Suppose that for each , is non-degenerate and Assumption 1 holds for . Then, converges almost surely to an (possibly sample path dependent) equilibrium point which is a local Stackelberg solution for the leader. Moreover, if Assumption 1 holds for and Assumption 2 holds, then so that is a differential Stackelberg equilibrium.
Proof. This proof follows primarily from using known stochastic approximation results. The update rule in (7) is a stochastic approximation of and consequently is expected to track this ODE asymptotically. The main idea behind the analysis is to construct a continuous interpolated trajectory for and show it asymptotically almost surely approaches the solution set to the ODE. Under Assumptions 1–3, results from [11, §2.1] imply that the sequence generated from (7) converges almost surely to a compact internally chain transitive set of . Furthermore, it can be observed that the only internally chain transitive invariant sets of the dynamics are differential Stackelberg equilibria since at any stable attractor of the dynamics and from assumption . Finally, from [11, §2.2], we can conclude that the update from (7) almost surely converges to a possibly sample path dependent equilibrium point since the only internally chain transitive invariant sets for are equilibria. The final claim that is guaranteed since is Lipschitz and .
The above result can be stated with a relaxed version of Assumption 2.
Given a differential Stackelberg equilibrium , let for some on which is non-degenerate. Suppose that Assumption 1 holds for and that . Then, converges almost surely to . Moreover, if Assumption 1 holds for , is a locally asymptotically stable equilibrium uniformly in on the ball , and , then .
The proof follows the same arguments as the proof of Proposition 8.
2 Learning Stackelberg Equilibria: Two-Timescale Analysis
where . Suppose that faster than so that in the limit , the above approximates the singularly perturbed system defined by
The learning rates can be seen as stepsizes in a discretization scheme for solving the above dynamics. The condition that induces a timescale separation in which evolves on a faster timescale than . That is, the fast transient player is the follower and the slow component is the leader since implies that from the perspective of the follower, appears quasi-static and from the perspective of the leader, appears to have equilibriated, meaning given . From this point of view, the learning dynamics (8)–(9) approximate the dynamics in the preceding section. Moreover, stable attractors of the dynamics are such that the leader is at a local optima for , not just along its coordinate axis but in both coordinates constrained to the manifold ; this is to make a distinction between differential Nash equilibria in agents are at local optima aligned with their individual coordinate axes.
The following two results are fairly classical results in stochastic approximation. They are leveraged here to making conclusions about convergence to Stackelberg equilibria in hierarchical learning settings.
While we do not need the following assumption for all the results in this section, it is required for asymptotic convergence of the two-timescale process in (8)–(9).
The dynamics have a globally asymptotically stable equilibrium.
Under Assumption 1–3, and the assumption that , classical results imply that the dynamics (8)–(9) converge almost surely to a compact internally chain transitive set of (10); see, e.g., [11, §6.1-2], [10, §3.3]. Furthermore, it is straightforward to see that stable differential Nash equilibria are internally chain transitive sets since they are stable attractors of the dynamics from (10).
There are two important points to remark on at this juncture. First, the flow of the dynamics (10) is not necessarily a gradient flow, meaning that the dynamics may admit non-equilibrium attractors such as periodic orbits. The dynamics correspond to a gradient vector field if and only if , meaning when the dynamics admit a potential function. Equilibria may also not be isolated unless the Jacobian of , say , is non-degenerate at the points. Second, except in the case of zero-sum settings in which , non-Stackelberg locally asymptotically stable equilibria are attractors. That is, convergence does not imply that the players have settled on a Stackelberg equilibrium, and this can occur even if the dynamics admit a potential.
Let be the (continuous) time accumulated after samples of the slow component . Define to be the flow of starting at time from intialization .
Suppose that Assumptions 1 and 2 hold. Then, conditioning on the event , for any integer , almost surely.
Now, by Assumption 1, is Lipschitz and bounded (in fact, independent of A1a., since , , it is locally Lipschtiz and, on the event , it is bounded). In turn, it induces a continuous globally integrable vector field, and therefore satisfies the assumptions of Benaïm [6, Prop. 4.1]. Moreover, under Assumptions A1b. and A1c., the assumptions of Benaïm [6, Prop. 4.2] are satisfied, which gives the desired result.
Under Assumption 3 and the assumptions of Proposition 9, almost surely conditioned on the event . That is, the learning dynamics (8)–(9) converge to stable attractors of (10), the set of which includes the stable differential Stackelberg equilibria.
Proof. Continuing with the conclusion of the proof of Proposition 9, on intervals the norm difference between interpolates of the sample path and the trajectories of vanish asymptotically; applying Lemma 4 (Appendix A) gives the result. Leveraging the results in Section 2.2.1, the convergence guarantees are stronger since in zero-sum settings all attractors are Stackelberg; this contrasts with the Nash equilibrium concept.
Consider a zero-sum setting . Under the assumptions of Proposition 9 and Assumption 3, conditioning on the event , the learning dynamics (8)–(9) converge to a differential Stackelberg equilibria almost surely.
The proof of this corollary follows the above analysis and invokes Proposition 1. As with Corollary 2, we can relax Assumption 2 and 3 to local asymptomatic stability assumptions and obtain similarity convergence guarantees.
Given a differential Stackelberg equilibrium where , let for some on which is non-degenerate. Suppose that Assumption 1 holds for each player, is a locally asymptotically stable attractor uniformly in on the ball for the dynamics , and there exists a locally asymptotically stable attractor on for the dynamics . Then, given an initialization and , it follows that almost surely.
2.2 Finite-Time High-Probability Guarantees
While asymptotic guarantees of the proceeding section are useful, high-probability finite-time guarantees can be leveraged more directly in analysis and synthesis, e.g., of mechanisms to coordinate otherwise autonomous agents. In this section, we aim to provide concentration bounds for the purpose of deriving convergence rate and error bounds in support of this objective. The results in this section follow the very recent work by Borkar and Pattathil . We highlight key differences and, in particular, where the analysis may lead to insights relevant for learning in hierarchical decision problems between non-cooperative agents.
The basic idea of the proof is to leverage Alekseev’s formula (Thm. 3, Appendix A) to bound the difference between the asymptotic pseudo-trajectories and the flow of the corresponding limiting differential equation on each continuous time interval between each of the successive iterates and by sequences of constants that decay asymptotically. Then, a union bound is used over all time intervals after defined for in order to construct a concentration bound. This is done first for the follower, showing that tracks the leader’s ’conjecture’ or belief about the follower’s reaction, and then for the leader.
where is constant (since ), , and
In addition, for , satisfies linear system
with and and where the Jacobian of . We provide more detail on this derivation in Appendix B.
Given that is a stable differential Stackelberg equilibrium, is positive definite. Hence, as in [57, Lem. 5.3], we can find , such that for , , ; this result follows from standard results on stability of linear systems (see, e.g., Callier and Desoer [14, §7.2, Thm. 33]) along with a bound on
Now, an interesting point worth making is that this analysis leads to a very nice result for the leader-follower setting. In particular, through the use of the auxiliary variable , we can show that the follower’s sample path ‘tracks’ the leader’s conjectured sample path. Indeed, consider , that is, where . Then, using a Taylor expansion of the implicitly defined conjecture , we get where is the error from the remainder terms. Plugging in ,
The terms after are , and hence asymptotically negligible, so that this sequence tracks dynamics as . We show that with high probability, they asymptotically contract, leading to the conclusion that the follower’s dynamics track the leader’s conjecture.
Towards this end, we first bound the normed difference between and . Define constants
and let .
For any , there exists such that conditioned on ,
Suppose that Assumptions 1, 2, and 3 hold and let . Given a stable differential Stackelberg equilibrium , the follower’s sample path generated by (9) with asymptotically track the leader’s conjecture and, given , will get ‘locked in’ to a –neighborhood with high probability conditioned on reaching by iteration . That is, letting , for some ,
with .
The key technique in proving the above theorem (which is done in detail in Borkar and Pattathil using results from Thoppe and Borkar ), is taking a union bound of the errors over all the continuous time intervals defined for .
The above theorem can be restated to give a guarantee on getting locked-in to an -neighborhood of a stable differenital Stackelberg equilibria if the learning processes are initialized in .
Given that the follower’s action tracks , we can also show that gets locked into an –neighborhood of after a finite time with high probability. First, a similar bound as in Lemma 1 can be constructed for .
Define the event where for each , is a linear interpolates between the samples , , and . Then as above, Alekseev’s formula can again be applied to get
and is the solution to a linear system with dynamics , the Jacobian of , and with initial data . This linear system, as above, has bound for some . Define .
with \eta_{n}=\max_{n_{0}\leq k\leq n-1}\big{(}e^{-\kappa_{1}(\sum_{i=k+1}^{n-1}\gamma_{1,i})}\gamma_{1,k}\big{)}.
An analogous corollary to Corollary 6 can be stated for with .
Numerical Examples
In this section, we present and extensive set of numerical examples to validate our theory and demonstrate that the learning dynamics in this paper can effectively train GANsCode is available at github.com/fiezt/Stackelberg-Code..
In Cournot’s duopoly model a single good is produced by two firms so that the industry is a duopoly. The cost for firm for producing units of the good is given by where is the unit cost. The total output of the firms is . The market price is when and when . We can assume that for . The profit of each firm is . Moreover, the unique Nash equilibrium in the game is so that the market price is and each firm obtains a profit of .
In the Stackelberg duopoly model with two firms, there is a leader and a follower. The leader moves and then the follower produces a best response to the action of the leader. Knowing this, the leader seeks to maximize profit taking advantage of the power to move before the follower. The unique Stackelberg equilibrium in the game is , . In equilibrium the market price is , the profit of the leader is , and the profit of the follower is .
The key point we want to highlight is that in this game, firm 1’s (leader) profit is always higher in the hierarchical play game than the simultaneous play game. We also use it as a simple validation example for our theory. For this problem, we simulate the Nash gradient dynamics and our two-timescale algorithm for learning Stackelberg equilibria to illustrate the distinctions between the Cournot and Stackelberg duopoly models. In this simulation, we select a decaying step-size of for each player in the Nash gradient dynamics. The decaying step-size is chosen to be for the leader and for the follower in the Stackelberg two-timescale algorithm so that the leader moves on a slower timescale than the follower as required. The noise at each update step is drawn as for each firm. The parameters of the example are selected to be . In Figure 2 we show the results of the simulation. Figure 2a shows the production path of each firm and Figure 2b shows the profit path of each firm. Under the Nash gradient dynamics, the firms converge to the unique Nash equilibrium of that gives profit of . The Stacklberg procedure converges to the unique Stackelberg equilibrium of that gives profit of . Hence as expected the two-timescale procedure converges to the Stackelberg equilibrium and gives the leader higher profit than under the Nash equilibrium.
2 Location Game on Torus
In this section, we examine a two-player game in which each player is selecting a position on a torus. Precisely, each player has a choice variable that can be chosen in the interval . The cost for each player is defined as , where each and are constants. The cost function is such that each player must trade-off being close to and far from . For the simulation of this game, we select the parameters and . There are multiple Nash and Stackelberg equilibria under these parameters. Each equilibrium is a stable equilibrium in this example. The Nash equilbria are and , and the costs are each and . The Stackelberg equilbria are and , and the costs are each . Hence, the ability to play before the follower gives the leader a smaller cost at any equilibrium. The equilibrium the dynamics will converge to depends on the initialization as we demonstrate. For this simulation, we select a decaying step-size of for each player in the Nash gradient dynamics. The decaying step-size is chosen to be for the leader and for the follower in the Stackelberg two-timescale dynamics. The noise at each update step is drawn as for each player. In Figure 3 we show the results of our simulation. The Nash and Stackelberg dynamics converge to an equilibrium as expected. In Figures 3a and 3b, we visualize multiple sample learning paths for the Nash and Stackelberg dynamics, respectively. The black lines depict for Nash and for Stackelberg and demonstrate how the order of play warps the first-order conditions for the leader and consequently produces equilibria which move away from the Nash equilibria. In Figure 3c we give a detailed look at the convergence to an equilibrium for a sample path. Finally, in Figure 3d, we present the evolution of the cost while learning and demonstrate the benefit of being the leader and the disadvantage of being the follower.
3 Generative Adversarial Networks
We now present a set of illustrative experiments showing the role of Stackelberg equilibria in the optimization landscape of GANs and the empirical benefits of training GANs using the Stackelberg learning dynamics compared to the simultaneous gradient descent dynamics. We find that the leader update empirically cancels out rotational dynamics and prevents cycling behavior. Moreover, we discover that the simultaneous gradient dynamics can empirically converge to non-Nash stable attractors that are Stackelberg equilibria in GANs. The generator and the discriminator exhibit desirable performance at such points, indicating that Stackelberg equilibria can be as desirable as Nash equilibria. We also find that the Stackelberg learning dynamics often converge to non-Nash stable attractors and reach a satisfying solution quickly using learning rates that can cause the simultaneous gradient descent dynamics to cycle. We provide details on our implementation of the Stackelberg leader update and the techniques to compute relevant eigenvalues of games in Appendix E. More details for specific hyperparameters can be found in Appendix D.
We compare the deterministic gradient update for Stackelberg learning dynamics and simultaneous gradient descent, and analyze the distance from equilibrium as a function of time. We plot for the generator’s performance and for the discriminator’s performance in Fig. 4 for varying dimensions with learning rate where and fixed regularization terms . We observe that Stackelberg learning converges to an equilibrium in fewer iterations than simultaneous gradient descent. For zero-sum games, our theory provides reasoning for this behavior since at any critical point the eigenvalues of the game Jacobian are purely real. This is in contrast to the game Jacobian for the simultaneous gradient descent, which can admit imaginary eigenvalue components that are know to cause rotational forces in the dynamics. This example provides empirical evidence that the Stackelberg dynamics cancel out rotations in general-sum games.
Example 2: Mixture of Gaussian (Diamond). We also train a GAN to learn a mixture of Gaussian distributions, where the generator is the leader and the discriminator is the follower. The generator network has two hidden layers and the discriminator has one hidden layer; each hidden layer has neurons. We train using a batch size of , a latent dimension of , and the default ADAM optimizer configuration in PyTorch version 1. Since the updates are stochastic, we decay the learning rates to satisfy our timescale separation assumption and regularize the implicit map of the follower using the parameter . We derive the regularized leader update in Appendix C.
The underlying data distribution for this problem consists of Gaussian distributions with means given by for and each with covariance where . Each sample of real data given to the discriminator is selected uniformly at random from the set of Gaussian distributions. We train each learning rule using learning rates that begin at . Moreover, in this example, the activation following the hidden layers in each network is the tanh function.
We train this experiment using the saturating GAN objective . In Fig. 5a–5b and Fig. 5g–5h we show a sample of the generator and the discriminator for simultaneous gradient descent and the Stackelberg dynamics after 40,000 training batches. Each learning rule converges so that the generator can create a distribution that is close to the ground truth and the discriminator is nearly at the optimal probability throughout the input space. In Fig. 5c–5f and Fig. 5i–5l, we show eigenvalues from the game that allow us to get a deeper view of the convergence behavior. We observe that the simultaneous gradient dynamics appear be in a neighborhood of a non-Nash equilibrium since the individual Hessian for the leader is indefinite, the individual Hessian for the follower is positive definite, and the Schur complement is positive definite. Moreover, the eigenvalues of the leader individual Hessian are nearly zero, which would reflect the realizable assumption from Section 2. The Stackelberg learning dynamics converge to a point with similar eigenvalues, which would be a non-Nash Stackelberg equilibrium. This example demonstrates that standard GAN training can converge to non-Nash attractors that are Stackelberg equilibria and the Stackelberg equilibria can produce good generator and discriminator performance. This indicates that it may not be necessary to look only for Nash equilibria and instead it may be easier to find Stackelberg equilibria and the performance could be as desirable.
Example 3: Mixure of Gaussian (Circle). The underlying data distribution for this problem consists of Gaussian distributions with means given by for and each with covariance where , sampled in the similar manner as the previous example. We train each learning rule using learning rates that begin at . Moreover, in this example, the activation following the hidden layers in each network is the ReLU function.
We train the GAN with the non-saturating objective . We show the the performance in Fig. 6 along the learning path for the simultaneous gradient descent dynamics and the Stackelberg learning dynamics. The simultaneous gradient descent dynamics cycle and perform poorly until the learning rates have decayed enough to stabilize the training process. The Stackelberg learning dynamics converge quickly to a solution that nearly matches the ground truth distribution. In a similar fashion as in the covariance example, the leader update is able to cancel out rotations and converge to a desirable solution with a learning rate that destabilizes the training process for standard training techniques. We show the eigenvalues after training and see that for this configuration the simultaneous gradient dynamics converge to a Nash equilibrium and the Stackelberg learning dynamics converge again to a non-Nash Stackelberg equilibrium. This provides further evidence that Stackelberg equilibria may be easier to reach and can provide suitable generator performance.
Example 4: MNIST dataset. To demonstrate that the Stackelberg learning dynamics can scale to high dimensional problems, we train a GAN on the MNIST dataset using the DCGAN architecture adapted to handle images. We train on an MNIST dataset consisting of only the digits 0 and 1 from the training images and on an MNIST dataset containing the entire set of training images. We train using a batch size of , a latent dimension of , and the ADAM optimizer with the default parameters for the DCGAN network. We regularize the implicit map of the follower as detailed in Appendix C using the parameter . If we view the regularization as a linear function of the number of parameters in the discriminator, then this selection of regularization is nearly equal to that from the mixture of Gaussian experiments.
We show the results in Fig. 7 after 2900 batches. For each dataset we show a sample of 16 digits to get a clear view of the generator performance and a sample of 256 digits to get a broader view of the generator output. The Stackelberg dynamics are able to converge to a solution that generates realistic handwritten digits. The primary purpose of this example is to show that the learning dynamics including second order information and an inverse is not an insurmountable problem for training large scale networks with millions of parameters. We believe the tools we develop for our implementation can be helpful to researchers working on GANs since a number of theoretical works on this topic require second order information to strengthen the convergence guarantees.
Conclusion
We study the convergence of learning dynamics in Stackelberg games. This class of games broadly pertains to any application in which there is an order of play between the players in the game. However, the problem has not been extensively analyzed in the way the learning dynamics of simultaneous play games have been. Consequently, we are able to give novel convergence results and draw connections to existing work focused on learning Nash equilibria.
References
A Mathematical Preliminaries
In this appendix, we show some preliminary results on linear algebra and recall some definitions and results from dynamical systems theory that are needed to state and prove the results in the main paper.
The results in this subsection follow from the theory of block operator matrices and indefinite linear algebra .
The following lemma is a very well-known result in linear algebra and can be found in nearly any advanced linear algebra text such as .
We can now use the above Lemma to prove Proposition 3. The proof follows the main arguments in the proof of Lemma 3.2 in the work by Berger et al. with some minor changes due to the nature of our problem.
By assumption, we have that so that, since ,
Thus, . Now, . Hence, for any non-trivial vector , so that we have
Note that the inequality in (13) holds because the vector is in the non-positive eigenspace of and the second term is clearly non-positive. Thus, cannot be positive definite, which gives a contradiction so that .
Claim: is necessary. Let the maps denote the eigenvalues of its argument arranged in non-increasing order. Then, by the Weyl theorem for Hermitian matrices , we have that
We can now combine this inequality with Lemma 3. Indeed, we have that
Since we have shown both the necessary conditions, this concludes the proof.
Now, let us prove Proposition 4 which gives sufficient conditions for when a stable non-Nash attractor of is a differential Stackelberg equilibrium. Then, combining this with Proposition 1, we have a sufficient condition under which stable non-Nash attractors are in fact stable attractors of .
Hence, to understand the eigenstructure of the Schur complement, we simply need to compare the all negative eigenvalues of in increasing order with the most positive eigenvalues of in decreasing order. Indeed, by assumption, and for each . Thus,
since it is a symmetric matrix. Combining this with the fact that , is a differential Stackelberg equilibrium. Hence, by Proposition 1 it is an attractor of .
A.2 Dynamical Systems Theory Primer
Given , , if there exists an increasing sequence of times with and for each and solutions , of with initialization such that for some bounded, measurable , the we call a –perturbation.
Given , , there exists such that for all , every –perturbation of converges to an –neighborhood of the global attractor set for .
A key tool used in the finite-time two-timescale analysis is the nonlinear variation of constants formula of Alekseev , .
with , the –dimensional identity matrix.
Typical two-timescale analysis has historically leveraged the discrete Bellman-Grownwall lemma [11, Chap. 6]. Recent application of Alekseev’s formula has lead to tighter bounds, and is thus becoming commonplace in such analysis.
B Extended Analysis
The results in Section 3.2.2 leverage classical results from stochastic approximation including recent advances in that same domain . Here we provide more detail on the derivation of the bounds presented in Section 3.2.2 in order to provide insight into what the constants are in the concentration bounds in Theorems 1 and 2. Moreover, the presentation here is somewhat distilled and the aim is to help the reader through the analysis in Borkar and Pattathil and Thoppe and Borkar as it pertains to the setting we consider. We refer the reader to each of these papers and references therein for even more detail.
For , let . Then, there is also and such that for ,
We can express the asymptotic pseudo-trajectories for any as
Then, by the nonlinear variation of constants formula (Alekseev’s formula), we have
where is constant (since ) and . Moreover, for , satisfies linear system
with initial data and and where the Jacobian of .
Given that is a stable differential Stackelberg equilibrium, is positive definite. Hence, as in [57, Lem. 5.3], we can find , such that for , ,
Analogously we can define linear interpolates or asymptotic pseudo-trajectories for . Indeed,
are the linear interpolated points between the samples where , and . Then, as above, Alekseev’s formula can again be applied to get
where (again, since ) and the following hold:
Moreover, is the solution to a linear system with dynamics , the Jacobian of , and with initial data . This linear system, as above, has bound
Now, in addition to the linear iterpolates for and , we define an auxiliary sequence representing the leader’s conjecture about the follower with the goal of bounding the normed difference between follower’s response and this auxiliary sequence. Indeed, using a Taylor expansion of the implicitly defined map , we get
where are the remainder terms which satisfy by assumption. Plugging in ,
The terms after are , and hence asymptotically negligible, so that this sequence tracks dynamics as . Using similar techniques as above, we can express linear interpolates of the leader’s belief regarding the follower’s reaction as
where the ’s are defined as follows:
with . Once again, Alekseev’s formula can be applied where and is the same as in the application of Alekseev’s to . Indeed, this gives us
Applying the linear system stability results, we get that
Each of the terms (a)–(d) can be bound as in Lemma III.1–5 in . The bounds are fairly straightforward using (16).
Applying Lemma 5.8 , conditioned on , we get there exists some constant such that
Using the bound on the linear system , this exactly leads to the bound
Thus, leveraging Lemma III.1–5 , we obtain the result of Lemma 1 in the main body of the paper, and stated here for easy access.
For any , there exists such that conditioned on ,
Lastly, in a similar fashion we can obtain a bound for the leader’s sample path .
for some . This gives the result of Theorem 1 in the main body with , , , . An exactly analogous analysis holds for obtaining the concentration bound in Theorem 2.
C Regularizing the Follower’s Implicit Map
The derivative of the implicit function used in the leader’s update requires the follower’s Hessian to be an isomorphism. In practice, this may not always be true along the learning path. Consider the modified update
in which we regularize the inverse of term. This update can be derived from the following perspective. Suppose player 1 views player 2 as optimizing a linearized version of its cost with a regularization term which captures the leader’s lack of confidence in the local linearization holding globally:
The first-order optimality conditions for this problem are
Hence, if the leader views the follower as updating along the gradient direction determined by these first order conditions, then the follower’s response map is given by
Ignoring higher order terms in the derivative of the response map, the approximate Stackelberg update is given by
In our GAN experiments, we use the regularized update since it is quite common for the discriminator’s Hessian to be ill-conditioned if not degenerate. Similarly, the Schur complement we present the eigenvalues for in the experiments includes the regularized individual Hessian for the follower.
A point such that the first order conditions and hold, and such that and is a differential Stackelberg equilibrium with respect to the regularized dynamics.
A differential Stackelberg equilibrium of the regularized dynamics satisfies and hold, and and .
This result can be seen by examining first and second order sufficient conditions for the leader’s optimization problem given the regularized conjecture about the follower’s update, i.e.
and for the problem follower is actually solving with its update .
D Experiment Details
This section includes complete details on the training process and hyper-parameters selected in the mixture of Gaussian and MNIST experiments.
The underlying data distribution for the diamond experiment consists of Gaussian distributions with means given by for and each with covariance where . Each sample of real data given to the discriminator is selected uniformly at random from the set of Gaussian distributions. The underlying data distribution for the circle experiment consists of Gaussian distributions with means given by for and each with covariance where . Each sample of real data given to the discriminator is selected uniformly at random from the set of Gaussian distributions.
D.2 MNIST
E Computing the Stackelberg Update and Schur Complement
The learning rule for the leader involves computing an inverse-Hessian-vector product for the inverse term and Jacobian-vector product for the term. These operations can be done efficiently in Python by utilizing Jacobian-vector products in auto-differentiation libraries combined with the sparse.LinearOperator class in scipy. These objects can also be used to compute their eigenvalues, inverses, or the Schur complement of the game dynamics using the scipy.sparse.linalg package. We found that the conjugate gradient method cg can compute the regularized inverse-Hessian-vector products for the leader update accurately with 5 iterations and a warm start.
The operators required for the leader update can be obtained by the following. Consider the Jacobian of the simultaneous gradient descent learning dynamics at a critical point for the general sum game :
Its block components consist of four operators that can be computed using forward-mode or reverse-mode Jacobian-vector products. Instantiating these operators as a linear operator in scipy allows us to compute the eigenvalues of the two player’s individual Hessians. Properties such as the real eigenvalues of a Hermitian matrix or complex eigenvalues of a square matrix can be computed using eigsh or eigs respectively. Selecting to compute the smallest or largest eigenvalues—sorted by either magnitude, real or imaginary values—allows one to examine the positive-definiteness of the operators.
Operators can be combined to compute other operators relatively efficiently for large scale problems without requiring to compute their full matrix representation. For an example, take the Schur complement of the Jacobian above at fixed network parameters , We create an operator that maps a vector to by performing the following four operations: , , , and . Each of the operations can be computed using a single backward pass through the network except for computing , since the inverse-Hessian requires an iterative method which can be computationally expensive. It solves the linear equation and there are various available methods: we tested (bi)conjugate gradient methods, residual-based methods, or least-squares methods, and each of them provide varying amounts of error when compared with the exact solution. Particularly, when the Hessian is poorly conditioned, some methods may fail to converge. More investigation is required to determine which method is best suited for specific uses. For example, a fixed iteration method with warm start might be appropriate for computing the leader update online, while a residual-based method might be better for computing the the eigenvalues of the Schur complement. Specifically, for our mixture of gaussians and MNIST GANs, we found that computing the leader update using the conjugate gradient method with maximum of 5 iterations and warm-start works well. We compared using the true Hessian for smaller scale problems and found the estimate to be within numerical precision.
F N𝑁N–Follower Setting
In this section, we show that the results extend to the setting where there is a single leader, but non-cooperative followers.
𝑁1N+1 Staggered Learners, All with Non-Uniform Learning Rates Note that if there is a layered hierarchy in which each, for example, the first follower is a leader for the second follower, the second follower a leader for the third follower and so on, then the results in Section 3 apply under additional assumptions on the learning rates.
For instance, consider a three player setting where and so that player 1 is the slowest player (hence, the ‘leader’), player 2 the second slowest, and player 3 the fastest, the ‘leader’. Then similar asymptotic analysis can be applied with the following assumptions. Consider
where we will explicitly define shortly. Let and .
There exists a Lipschitz continuous function such that for any , solutions of (20) asymptotically converge to given initial data .
There exists a Lipschitz continuous function such that for any , solutions of (21) asymptotically converge to given initial data .
Now, define for notation simplicity. Let and where the notation indicates the total derivative with respect to arguments up to .
Under Assumptions 4 and 5 and Assumption 1 from the main paper,
Of course the framework naturally extends to -followers; a similar framework can be found for reinforcement learning algorithms in normal form games .
F.2 N𝑁N Simultaneously Play Followers
On the other hand, consider a setting in which the followers play a Nash equilibrium in a simultaneous play game and are assumed to have the same learning rate. That is, where all followers use the learning rate and the leader uses the learning rate . The results for this section assume that the follower game has a unique differential Nash equilibrium uniformly in .
has a globally asymptotically stable differential Nash equilibrium uniformly in with a –Lipschitz function.
All the results in Section 3 of the main body hold replacing Assumption 2 with the above assumption. This is a somewhat strong assumption, however, -player convex games that are diagonally strictly convex admit unique Nash equilibria which are attracting .