May 01, 2023
Performative prediction, as introduced by [1], is a framework for studying social prediction in which the data distribution itself changes in response to the deployment of a model. Existing work in this field usually hinges on three assumptions that are easily violated in practice: that the performative risk is convex over the deployed model, that the mapping from the model to the data distribution is known to the model designer in advance, and the first-order information of the performative risk is available. In this paper, we initiate the study of performative prediction problems that do not require these assumptions. Specifically, we develop a reparameterization framework that reparametrizes the performative prediction objective as a function of the induced data distribution. We then develop a two-level zeroth-order optimization procedure, where the first level performs iterative optimization on the distribution parameter space, and the second level learns the model that induces a particular target distribution at each iteration. Under mild conditions, this reparameterization allows us to transform the non-convex objective into a convex one and achieve provable regret guarantees. In particular, we provide a regret bound that is sublinear in the total number of performative samples taken and is only polynomial in the dimension of the model parameter.
Performative prediction, as introduced by [1], provides a framework for studying prediction and risk minimization when the data distribution itself changes in response to the deployment of a model. Such phenomena, usually referred to as "performativity," are prevalent in various social prediction contexts, including education, recommendation systems, and criminal prediction, among others [1]–[5]. For instance, consider a college admission process that places significant importance on standardized test scores. This process can incentivize students to invest more effort in test preparation, ultimately leading to a pool of applicants with much higher test scores than initially expected. This phenomenon is also prevalent in real-world applications, particularly in large-scale online recommendation systems, where the high frequency of updates to the recommendation algorithm can reshape users’ future behavior. For example, video platforms such as TikTok, Netflix, and YouTube provide personalized recommendations that can influence users’ future preferences and lead to shifts in the user-advertiser interaction patterns, thereby creating a dynamic and evolving data distribution.
More formally, consider the standard empirical risk minimization (ERM) problem defined by a loss function \(\ell\), a model parameter space \(\Theta \subset \mathbb{R}^{d_\Theta}\) where \(d_\Theta\in\mathbb{Z}_{>0}\), an instance space \(Z = X \times Y\), and a fixed data distribution \(\mathcal{D}\) over \(Z\). The task is to find a model that minimizes the empirical risk defined as: \(\mathsf{ER}(\theta, \mathcal{D}):= \mathbb{E}_{z \sim \mathcal{D}} [\ell(z;\theta)].\) Performative prediction extends this learning task by positing that the data distribution \(\mathcal{D}\) is not fixed but is instead a function of the model parameter \(\theta \in \Theta\). Here, we refer to \(\mathcal{D}(\cdot)\) as a distribution map, and \(\mathcal{D}(\theta)\) as the data distribution induced by the model \(\theta\). The objective is then to minimize the performative risk, defined as \[\begin{align} \mathsf{PR}(\theta, \mathcal{D}(\theta)) := \mathbb{E}_{z \sim \mathcal{D}(\theta)} [\ell(z;\theta)]~. \end{align}\] Intuitively, the performative prediction risk evaluates the performance of the model \(\theta\) on the resulting distribution \(\mathcal{D}(\theta)\) via the loss function \(\ell\). When it is clear from the context, we also use \(\mathsf{PR}(\theta)\) to shorthand the performative risk.
Optimizing the performative risk is generally challenging. In standard ERM, a convex loss function \(\ell\) implies a convex empirical risk. However, as [1] observed, the performative risk \(\mathsf{PR}(\cdot)\) may be non-convex even when the loss \(\ell\) itself is convex. For this reason, earlier works [1], [6]–[8] then focus on computing a performative stable solution instead, which is easier to achieve using standard optimization tools like repeated risk minimization. A performative stable model is loss-minimizing on the data distribution it induces, though other models may incur smaller losses on their respective induced distributions. However, as recent works [9], [10] point out, such stable solutions may be highly suboptimal and, worse yet, may not exist in certain settings.
One major challenge in performative risk minimization is the unknown distribution map between the model parameter \(\theta\) and the distribution \(\mathcal{D}(\theta)\) without making any structural assumption. For example, one can hardly anticipate the click-through rate of an ad without putting out the ad first. In the language of performative prediction, only by deploying a model \(\theta\) can the learner observe data samples that are i.i.d realized from the induced data distribution \(\mathcal{D}(\theta)\). Due to this inherent uncertainty about \(\mathcal{D}(\theta)\), it is impossible to compute the gradient of \(\mathsf{PR}(\theta)\) w.r.t \(\theta\), not to mention finding a model with the lowest performative risk offline. Instead, the learner must interact with the environment and deploy models \(\theta\) to explore the induced distributions \(\mathcal{D}(\theta)\), which involves deploying “imperfect” models on decision subjects.
In this paper, we propose to measure the loss incurred by deploying a sequence of models \(\theta_1, \ldots, \theta_{T_{\textsf{total}}}\) by evaluating the following regret measured with respect to the total number of samples deployed during the process: \[\begin{align} \label{eqn:regret-def} \mathcal{R}_N(\mathcal{A}, \mathsf{PR}) = \sum_{\tau=1}^{T_{\textsf{total}}} \sum_{i=1}^{n_{\tau}} \ell(z_{\tau}^{(i)}; \theta_{\tau})- N \cdot \mathsf{PR}(\theta_{\texttt{OPT}}) \end{align}\tag{1}\] where \(N:=\sum_{\tau=1}^{T_{\textsf{total}}} n_{\tau}\) denotes the total number of observed data samples throughout the process, \(\mathcal{A}\) corresponds to the particular algorithm, and \(\mathsf{PR}\) represents the objective function. This regret measures the suboptimality of the deployed sequence of models relative to a performative optimum \(\theta_{\texttt{OPT}} \in \operatorname{argmin}_\theta \mathrm{PR}(\theta)\) in terms of how much loss they incur on the population with \(N\) decision subjects.
In contrast to earlier studies that primarily assess the final model’s performance based on optimality rather than the cumulative loss incurred throughout the process, we argue that this constitutes a more practical evaluation metric in predictive scenarios involving multiple rounds of human feedback. In particular, since the process of finding the optimal performative model involves deploying sub-optimal models on human agents in the process, it is more appropriate to define regret on the total number of agents that are subjected to the “imperfect’’ algorithmic system rather than only caring about whether the final model is optimal. We believe this provides a unique evaluation metric suitable for performative prediction.
Later in 4.3, we compare our proposed regret definition with the standard regret measured in \(T_{\textsf{total}}\) in more detail and show that our algorithm is, in fact, also sublinear in the total deployment steps \(T_{\textsf{total}}\). This, combined with the fact that sublinear regret implies model convergence (1), also means that our algorithm can guarantee to output a model arbitrarily close to the performative optimal model \(\theta_{\texttt{OPT}}\).
Our main contributions are a two-level zeroth order optimization algorithm that achieves a sublinear regret bound measured using the total number of samples and a novel reparametrization framework attempting to tackle a particular non-convex performative prediction problem.
Departing from previous work, we allow \(\mathsf{PR}(\theta)\) to be non-convex in the model parameter \(\theta\), but suppose it is convex in the data distribution parameter \(\phi \equiv \varphi(\theta)\). Informally, under mild conditions, we show that non-convex \(\mathsf{PR}(\theta)\) can be reparameterized as a new (convex) function \(\mathsf{PR}^\dagger(\phi)\) over the induced data distribution parameter \(\phi\). We discuss detailed parametrization procedure in 3.
Given the parametrization framework proposed above, we propose a two-level zeroth-order optimization procedure, which, to our knowledge, is novel in performative prediction. We believe our method enjoys the following benefits:
\(\bullet\)
No Requirement for Gradient Information Unlike the traditional gradient-based optimization procedure, our method does not require the explicit calculation of gradients that may be complex or unavailable.
Black-Box Models Our method can still be effective when dealing with models or systems that are treated as black boxes, where the internal mechanisms are not well understood (such as complicated economic systems) since it doesn’t require knowledge of the underlying model structure.
Robustness to Noise In many real-world applications, objective function evaluations may be noisy or subject to uncertainty, such as modeling consumer behavior. Our method can handle noisy evaluations and make decisions that are robust to noise.
Our main results can be summarized as follows:
thm There exists an algorithm that, under appropriate conditions, incurs regret \(\widetilde{O}((d_\Theta+ d_\Phi) \cdot N_{\mathsf{KL}}^{1/6} \cdot N^{5/6})\)1 after \(N\) performative samples2 with probability at least \(1-p\), where \(N_{\mathsf{KL}}\) depends on the sample efficiency of an off-the-shelf estimator for KL divergence, and \(d_\Theta\) and \(d_\Phi\) denote the dimension of the model and distribution parameter space, respectively.
The \(N_\mathsf{KL}\) term in our regret depends on the sample efficiency of the estimator for KL divergence. The detailed discussion is provided in 4.2.
Our work most closely relates to performative prediction and zeroth-order optimization. Due to page limit, we include additional related work in 12, including detailed comparisons of our work to three closely related jobs [9], [11], [12], and more recent developments of performative prediction.
Performative Prediction, first explored in [1], has recently received many follow-up works, including but not limited to [6]–[11], [13], [14] and [15]. These works mostly focus on the performative stability and the performative optimality, including developing an algorithmic procedure that converges to performatively stable or optimal points. Similar to this line of research [9]–[11], [13], our work also focuses on performative optimality.
Our algorithms and techniques are based on the line of work on zeroth-order optimization (also known as bandit optimization) initiated by [16], which studies how to optimize an unknown convex function \(f\) using only function value query access to \(f\). [17] and [18] later extend the technique that allows multiple points query and show that two points suffice to guarantee that the regret bounds that closely resemble the regret bounds for the full information case. The reparameterization approach proposed in our paper mirrors the intuition behind the algorithms proposed for learning from revealed feedback or preferences (see, e.g., [4], [19], [20]), which consider a Stackelberg game involving a utility-maximizing learner and strategic agent. Our work, focusing on performative prediction with an environment response exogenously characterized by a distribution map \(\mathcal{D}(\cdot)\), differs from theirs in problem consideration.
Let \(d_\Theta\in \mathbb{Z}_{>0}\) denote the dimension of the model parameter \(\theta\), and let \(D_\Theta:= \sup\{\|\theta-\theta'\|, \forall \theta,\theta' \in \Theta\}\) denote the diameter of the model parameter space \(\Theta\). The data distribution \(\mathcal{D}(\theta)\) has a parametric continuously differentiable density \(p(z; \varphi(\theta))\) where \(\varphi(\theta)\) denote the distribution parameter for \(\mathcal{D}(\theta)\). We use \(\varphi(\cdot)\) to denote the distribution parameter mapping while \(\phi\) to denote a given distribution parameter. Let \(d_\Phi\in \mathbb{Z}_{>0}\) denote the dimension of the model parameter \(\phi\), and let \(D_\Phi:= \sup\{\|\phi-\phi'\| ~|, \forall \phi,\phi' \in \Phi\}\) denote the diameter of the model parameter space \(\Phi\). When it is clear from the content, we use \(\varphi(\theta)\) to represent \(\mathcal{D}(\theta)\) the distribution \(\theta\) induces. Let \(\vartheta^*(\phi)\) denote the optimal model parameter that induces a specific target distribution parameter \(\phi\) – in case of having multiple model parameters that potentially induce the same distribution parameter \(\phi\), \(\vartheta^*(\phi)\) is the one that achieves the minimum performative prediction risk.
The rest of the paper is organized as follows: In 2, we introduce the problem formulation and provide a warm-up setting when \(\mathsf{PR}(\theta)\) is convex over the model parameter \(\theta\). Using this simple setting, we introduce the zeroth-order optimization technique we use, which will serve as the building block to solve for a more complicated setting (i.e., when \(\mathsf{PR}(\theta)\) is not convex over \(\theta\)). We also present a fundamental fact in convex optimization that sublinear regret implies model convergence (1), which unifies the goal of regret minimization and model optimality in our setting. In 3, we provide an overview of our proposed solution. In 4.1, we describe the outer algorithm, and 4.2 describes the inner algorithm called \(\operatorname{\mathsf{LearnModel}}\), which is used to solve a subroutine problem using black-box oracle. 4.3 contains the overall regret analysis. Lastly, in 5, we discuss the limitations and potential future work. All omitted proofs can be found in the Appendix.
We begin by formally defining our problem.
The objective of performative prediction is to minimize the performative risk defined as \(\mathsf{PR}(\theta):= \mathbb{E}_{z \sim \mathcal{D}(\theta)} [\ell(z;\theta)]~.\) A model \(\theta_{\texttt{OPT}} \in \Theta\) is said to be performatively optimal if \(\mathsf{PR}(\theta_{\texttt{OPT}}) = \min_{\theta \in \Theta} \mathsf{PR}(\theta)\). To find the performatively optimal model, one usually needs to have the full knowledge of the underlying distribution map \(\mathcal{D}: \Theta \rightarrow \Phi\). In this work, we consider a more practical scenario where the distribution map \(\mathcal{D}\) is not known in advance, and to learn the performatively optimal model, the learner has to adaptively deploy models to gradually learn the underlying distribution map.
Formally, we consider the following repeated interaction between the learner and the environment consisting of decision subjects where we can only query through samples. The interaction proceeds for \(T_{\textsf{total}}\) steps, at each time step \(\tau = 1, \ldots, T_{\textsf{total}}\): (1) the learner deploys a model \(\theta_{\tau}\in\Theta\); (2) the learner observes \(n_\tau\) data samples \(\{z_{\tau}^{(i)}\}_{i\in[n_\tau]}\) where each \(z_{\tau}^{(i)}\overset{\text{iid}}{\sim} \mathcal{D}(\theta_{\tau})\); (3) the learner incurs empirical loss \(\ell(z_{\tau}^{(i)}; \theta_{\tau})\) for each sample.
The goal of the learner is to design an online model deployment policy \(\mathcal{A}\) such that it minimizes her cumulative empirical risk over all observed data samples: \[\begin{align} \mathcal{R}_N(\mathcal{A}, \mathsf{PR}) = \sum_{\tau=1}^{T_{\textsf{total}}} \sum_{i=1}^{n_{\tau}} \ell(z_{\tau}^{(i)}; \theta_{\tau})- N \cdot \mathsf{PR}(\theta_{\texttt{OPT}}) \end{align}\] where \(N:=\sum_{\tau=1}^{T_{\textsf{total}}} n_{\tau}\) denotes the total number of observed data samples throughout the process. The reason we introduce \(T_{\textsf{total}}\) instead of \(N\) directly is that each step (\(\tau\)) of our algorithm performs different tasks, where we would impose different requirements of samples to be collected. This shall become clear later when we present our algorithm in the following sections.
In this section, we analyze a simple scenario when we assume that the performative risk \(\mathsf{PR}(\theta)\) is convex over the model parameter \(\theta\). The technique we use to solve this simple case will be the building block to solve the later more challenging problem where \(\mathsf{PR}(\theta)\) is not convex over the model parameter \(\theta\).
Recall that when the learner deploys a model \(\theta\), she observes a set of data samples which are i.i.d drawn from the underlying data distribution \(\mathcal{D}(\theta)\). This enables us to compute an unbiased estimate \(\widetilde{\mathsf{PR}}(\theta)\) for the performative risk \(\mathsf{PR}(\theta)\) of the deployed model \(\theta\): \[\begin{align} \widetilde{\mathsf{PR}}(\theta) = \frac{1}{n_\tau}\sum_{i=1}^{n_\tau} \ell(z_\tau^{(i)}; \theta), ~~ \text{and}~~ \mathbb{E}[\widetilde{\mathsf{PR}}(\theta)] = \mathsf{PR}(\theta), \forall \theta \in \Theta \end{align}\] where the expectation is over the randomness of the observed samples. Since \(\mathsf{PR}(\theta)\) is convex over the model parameter \(\theta\), one can use an off-the-shelf zeroth-order convex optimization technique [17] to solve this problem and get the following regret guarantee:
lemconvexregretbound When \(\mathsf{PR}(\theta)\) is convex, \(L\)-Lipschitz w.r.t. the deployed model parameter \(\theta\), there exists an algorithm (4) achieving \(\mathcal{R}_N(\mathcal{A}_{\ref{algorithm:minimize-convex-function}}, \mathsf{PR}) = O(\sqrt{d_\Theta N\log \frac{1}{p}})\) with probability at least \(1-p\), where \(N\) is the total number of samples deployed during the process.
We defer the proof and the details of 4 to 7. In particular, 4 deploys two models at each time step, in doing so, one can show that the regret bounds closely resemble bounds for the full information case where the learner knows the distribution map \(\mathcal{D}(\cdot)\). The proof of the above result builds on the main result of [17], and also incorporates an improved analysis of the gradient estimate due to [18].
A folklore fact in online and zeroth-order optimization is that if a function \(f\) is convex and we wish to converge to an approximately optimal point, it suffices to show a query algorithm that achieves \(o(n)\) regret after \(n\) queries. In particular, we have the following proposition:
Proposition 1 (Sublinear Regret Implies Convergence). Let \(f: X \to \mathbb{R}\) be convex, and let \(\mathcal{A}\) be an algorithm for minimizing \(f\) whose regret after \(n\) queries is sublinear in \(n\), i.e. \(\mathcal{R}_n(\mathcal{A},f) = o(n)\). Then we can compute an \(\epsilon\)-suboptimal point for \(f\) in \(\mathcal{R}_n(\mathcal{A},f)/\epsilon\) queries of \(f\).
This proposition establishes a strong link between achieving sublinear regret and the convergence toward an optimal model. It implies that if our proposed algorithm attains a sublinear regret as defined in Equation 1 , this automatically suggests that we can obtain an almost optimal model, which is exceptionally close to the truly optimal model, denoted as \(\theta_{\texttt{OPT}}\). This closeness is achieved simply by averaging the models \(\theta_1, \ldots, \theta_{T_{\textsf{total}}}\) throughout the deployment process. This helps us unify the goal of regret minimization and finding the optimal model.
When \(\mathsf{PR}(\theta)\) is not convex over the model parameter \(\theta\), the zeroth-order convex optimization technique used in 2.2 is not directly applicable. Instead, we leverage the structure of \(\mathsf{PR}(\theta)\) and reparameterize it as a function of the induced data distribution \(\mathcal{D}(\theta)\). In particular, we consider the setting where the data distribution \(\mathcal{D}(\theta)\) has a parametric continuously differentiable density \(p(z; \varphi(\theta))\), and the functional form \(p(z; \phi)\) is known to the learner but the distribution parameter \(\phi\) remains unknown. Under mild conditions, we show that the performative risk \(\mathsf{PR}(\theta)\) can be reformulated as a function of the induced distribution distribution parameter \(\phi \equiv \varphi(\theta)\), namely, \[\begin{align} \mathsf{PR}(\theta) = \mathsf{PR}^\dagger(\varphi(\theta)) \equiv\mathsf{PR}(\vartheta^*(\phi))~, \end{align}\] and \(\mathsf{PR}^\dagger(\phi)\) is convex over the distribution parameter \(\phi\) (See more details in 4.1).
Here we provide two real-life settings to justify our model:
Example 1. (Biased coin flip). Consider the task of predicting the outcome of a biased coin flip similar to [1], where the bias of the coin depends on a feature \(X\) and the assigned score \(f_\theta(X)\). In particular, define \(\mathcal{D}(\theta)\) in the following way: \(X\) is a 1-dimensional feature supported on \([0, 1]\) and \(Y \sim \operatorname{Bernoulli}(\varphi(\theta))\). Assume that the class of predictors consists of linear models of the form \(f_\theta(x)= \theta x\) and that the objective is to minimize the squared loss: \(\ell(x, y; \theta)=\left(y-f_\theta(x)\right)^2\). When the probability of the coin landing on heads \(\varphi(\theta) = \theta^2\), we can verify that \(\mathsf{PR}(\theta)\) is convex in \(\varphi(\theta)\), not in \(\theta\) (by similar argument provided below in 3).
Example 2. (Expected revenue of goods). Let \(\theta\in \mathbb{R}^d\) denote a vector of prices for various goods the distributor sets. A vector \(z\) denotes a customer’s demand for each good. The distributor’s goal is to maximize the expected revenue \(\mathsf{PR}(\theta) = \mathbb{E}_{z\sim \mathcal{D}(\theta)}\left[\theta^{\top} z\right]\). In other words, the loss function is \(\ell(z ; \theta)=-\theta^{\top} z\). When \(\mathcal{D}(\theta)= N(\varphi(\theta), \Sigma^2)\) with \(\varphi(\theta) = \sqrt{\theta}\) and a fixed \(\Sigma^2\), we can verify that \(\mathsf{PR}(\theta)\) is not convex in \(\theta\) but is convex in \(\varphi(\theta) = \sqrt{\theta}\).
With this reparameterization, one can operate on the space of distribution parameters and hopefully apply the zeroth-order convex optimization technique. However, one notable challenge is in zeroth-order convex optimization, the learner is usually assumed to have direct query access to the unknown convex function \(f\). Namely, when querying point \(x\), the learner is able to immediately obtain the information about the (noisy) value \(f(x)\). In our setting, such direct access is, unfortunately, not available since the mapping \(\varphi(\cdot)\) is not known to the learner. Indeed, the learner can only deploy a model \(\theta\) to observe the empirical performative risk \(\widetilde{\mathsf{PR}}(\theta)\) which is evaluated over the observed data samples drawn from the induced data distribution \(\mathcal{D}(\theta)\). Hence, to evaluate the value \(\mathsf{PR}^\dagger(\phi)\) on a target data distribution with the parameter \(\phi\), we use another algorithm called \(\operatorname{\mathsf{LearnModel}}\) as a subroutine to find a model \(\bar{\theta}\) such that \(\varphi(\bar{\theta}) \approx \phi\) (See 4.2).
Intuitively, the outer loop optimizes the objective function \(\mathsf{PR}\) in the distribution parameter space \(\phi\in \Phi\) iteratively and tries to find the optimal data parameter \(\phi *\), while the inner loop (\(\operatorname{\mathsf{LearnModel}}\)) tries to find a model parameter to induce the particular data parameter that the outer loop is currently iterating on.3 A graph illustration of our algorithm procedure in given in 1.
We first provide three examples in which our condition (\(\mathsf{PR}\) loss is convex in the induced distribution parameter \(\varphi_\theta:= \phi(\theta)\), but non-convex in the model parameter \(\theta\)) holds. See more examples and derivation details in 8.
Example 3. (Bernoulli distribution) Consider the following one-dimension linear model with the squared loss \(\ell(\theta;(x,y)) = - (\theta x - y)^2\). Assuming a model \(\theta\in \Theta = [0,1]\) induces a Bernoulli distribution over the labels with the distribution parameter \(\varphi(\theta):= \theta^2\), i.e., \(y \sim \mathsf{Bern}(\varphi(\theta))\). Then its \(\mathsf{PR}\) loss is convex in its data parameter \(\varphi_\theta\) but not convex in its model parameter \(\theta\).
Notice that the example provided can be generalized to any distribution map \(\phi(\theta)\) that satisfies \(\varphi(\theta) = \theta^\alpha\) for any \(\alpha> 1\), and any \(\ell_\beta\) loss for even \(\beta\) value. In addition, Example 1 can also be any generalized monotone polynomial function \(\varphi(\theta)\).
Example 4 (Gaussian distribution). For a random variable \(x\) following a one-dimensional Gaussian distribution with fixed variance, i.e., \(\mathcal{D}(\theta) = N(\varphi(\theta), \sigma^2)\), and let the loss as \(\ell(x;\theta) = (\theta x)^2\), then we have \(\mathsf{PR}(\theta) = \theta^2(\sigma^2 + \varphi(\theta)^2)\). With \(\varphi(\theta) = \sqrt{\theta}\), we verify that \(\mathsf{PR}(\theta)\) is not convex in the model parameter \(\theta\) but is convex in the distribution parameter \(\varphi(\theta) = \sqrt{\theta}\).
Example 5 (Uniform distribution). Fix a random variable \(x\). Let \(y\) follow a uniform distribution with parameter \(\varphi(\theta)\), e.g., \(y \sim \text{Uniform}[0, \varphi(\theta)]\), and with the loss being \(\ell(\theta; x, y) = -(\theta x - y)^2\), we have \(\mathsf{PR}(\theta) = \theta^2 x^2 - \theta x \varphi(\theta) + \frac{1}{3} \varphi(\theta)^2\). Setting \(\varphi(\theta) = \theta^2\), we verify that it’s convex in the distribution parameter \(\varphi(\theta)\) but not in the model parameter \(\theta\).
Remark 1. Earlier work [9] posits the “mixture dominance assumption”, under which the performative prediction risk turns out to be convex in the model parameter \(\theta\). In particular, the assumption requires that for any triple \(\theta, \theta', \theta_0\in\Theta\), the following condition holds: \(\mathbb{E}_{z \sim \mathcal{D}(\alpha\theta + (1-\alpha)\theta')} [\ell(\theta_0;z)] \le \mathbb{E}_{z \sim \alpha \mathcal{D}(\theta) + (1-\alpha) \mathcal{D}(\theta')} [\ell(\theta_0; z)]\). The primary distinction between our condition and theirs is that our condition only needs to be valid for each individual data parameter. This is in contrast to [9]’s condition, which must be met for any combination of \(\theta, \theta'\), and \(\theta_0\in \Theta\). We believe our approach has greater versatility and is more likely to be fulfilled in various scenarios 4.
In this section, we provide the details of our proposed algorithm, and also the associated performance analysis.
As we mentioned, in this work, we study the scenario where \(\mathsf{PR}(\theta)\) is not convex over the model parameter. The high-level idea is that we can reparameterize the performative risk \(\mathsf{PR}(\theta)\) as a function \(\mathsf{PR}^\dagger(\phi)\) over the data distribution parameter \(\phi\). In particular, we first reformulate the learner’s loss function so that it can be expressed as a function only in the induced data distribution. For each data distribution \(\phi \in \Phi\), assume the set of learner’s actions (deployed model parameters) that induce \(\phi\) is \(\Theta^*(\phi) = \{\theta \in \Theta | \varphi(\theta) = \phi \}\) Among all of the learner’s actions that induce \(\phi\), the optimal one that achieves the minimal \(\mathsf{PR}\) loss across the whole population is: \[\begin{align} \vartheta^*(\phi) = \mathop{\mathrm{\arg\!\min}}_{\theta\in \Theta^*(\phi )} \mathsf{PR}(\theta ) \end{align}\] where ties are broken arbitrarily. Now we can rewrite learner’s objective function as a function of \(\phi\) \[\begin{align} \label{eq:reformulate-objective-function} \mathsf{PR}^\dagger(\phi) = \mathsf{PR}(\vartheta^*(\phi )) \end{align}\tag{2}\]
To make the problem tractable, we consider following generic class of \(\mathsf{PR}^\dagger(\cdot)\) that is convex and Lipchitz continuous.
Assumption 1. \(\mathsf{PR}^\dagger(\phi)\) is convex and \(L^\dagger\)-Lipschitz over the data distribution parameter \(\phi\in \Phi\).
With reparameterizing \(\mathsf{PR}(\theta)\) as a function \(\mathsf{PR}^\dagger(\phi)\) over the induced data distribution parameter \(\phi\), we now wish to minimize a bounded, \(L^\dagger\)-Lipschitz function \(\mathsf{PR}^\dagger(\cdot): \Phi \to \mathbb{R}\), where \(\Phi \subset \mathbb{R}^{d_\Phi}\) has bounded diameter \(D_\Phi\), by operating on the distribution parameter space \(\Phi\).
Instead of having immediate query access in zeroth-order convex optimization algorithm, in our setting, we cannot directly evaluate the (noisy) value \(\mathsf{PR}^\dagger(\phi)\) for a particular data distribution parameter, but may query the following oracles:
\(\bullet\)
A noisy function oracle \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\), which takes \(\theta \in \Theta\) as input and returns an unbiased estimate \(\widetilde{\mathsf{PR}}\) such that \(\mathbb{E}[\widetilde{\mathsf{PR}}(\theta)] = \mathsf{PR}(\theta)\). This noisy oracle can be implemented by simply making a prediction and observing the loss as defined in 2.2 5.
A noisy reparameterization oracle \(\operatorname{\mathsf{LearnModel}}(\phi, \epsilon_\mathsf{LM}, p_\mathsf{LM})\), which takes \(\phi \in \Phi\), \(\epsilon_\mathsf{LM}, p_\mathsf{LM}> 0\) as input and returns \(\theta \in \Theta\) such that \(\Pr(\|\varphi(\theta) - \phi\| \geq \epsilon_\mathsf{LM}) \leq p_\mathsf{LM}\). We will specify \(\operatorname{\mathsf{LearnModel}}\) in 4.2.
2 achieves this task. Specifically, it returns both \(\bar{\theta} \in \Theta\) and \(\bar{\phi} \in \Phi\) such that with probability at least \(1-p\), \(|\mathsf{PR}(\bar{\theta}) - \mathsf{PR}(\theta_\texttt{OPT})| \leq \epsilon\) and \(|\mathsf{PR}^\dagger(\bar{\phi}) - \mathsf{PR}^\dagger(\varphi(\theta_\texttt{OPT}))| \leq \epsilon\).
For analysis purpose, we also define regret in \(T\), the total number of steps \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) has to go through in order to get an \(\epsilon\)-suboptimal model parameter w.r.t the \(\mathsf{PR}\) objective function: \[\begin{align} &\mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})\\ =& \sum_{t=1}^T \left[ \operatorname{\mathsf{Estimate}\mathsf{PR}}(\hat{\theta}_t^+) + \operatorname{\mathsf{Estimate}\mathsf{PR}}(\hat{\theta}_t^-) - 2 \mathsf{PR}(\theta_\texttt{OPT}) \right] \end{align}\]
We demonstrate the following regret bound for this algorithm:
thmouteralgorithmregret When 2 is called with arguments \(\epsilon_\mathsf{LM}\) and \(p_\mathsf{LM}\), we have for every \(p > 0\) that \[\begin{align} \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}},\mathsf{PR}) = ~ O\left( \sqrt{d_\Phi T} + \sqrt{\epsilon_\mathsf{LM}d_\Phi} \cdot T + \sqrt{T \log\frac{1}{p}} \right) \end{align}\] with probability at least \(1 - p - 2Tp_\mathsf{LM}\).
The above [theorem:regret-bound-for-indirectly-convex-functions] requires that the output of \(\operatorname{\mathsf{LearnModel}}\) is \(\epsilon_\mathsf{LM}\)-close to the target distribution parameter \(\phi\) with probability at least \(1 - p_\mathsf{LM}\). Later in 4.2, we show how we achieve this by developing an zeroth-order convex optimization algorithm with the objective of minimizing the \(\mathsf{KL}\) divergence of two distributions.
In this section, we show how to solve the sub-problem \(\operatorname{\mathsf{LearnModel}}\) mentioned in 2: given a target distribution with the parameter \(\phi\in \Phi\), find a model \(\theta\in \Theta\) whose corresponding distribution parameter \(\varphi(\theta)\) is close to \(\phi\).
To this end, we consider minimizing the \(\mathsf{KL}\) divergence between \(\phi\) and \(\varphi(\theta)\): 6 \[\begin{align} \label{eqn:kl-divergence} \mathsf{KL}(\phi|| \varphi(\theta)) := \int_z p(z; \phi) \log \frac{p(z; \phi)}{p(z;\varphi(\theta))} dz \end{align}\tag{3}\] where \(p(z;\phi)\) denotes the pdf for the target distribution \(\phi\), and \(p(z;\varphi(\theta))\) denotes the pdf for the distribution induced by deploying \(\theta\).
In general, \({\mathsf{KL}}(\phi|| \varphi(\theta))\) measures how much a distribution with the parameter \(\varphi(\theta)\) is away from the target distribution with the parameter \(\phi\): if the two distributions \(\phi_1,\phi_2 \in \Phi\) satisfy \(\phi_1 = \phi_2\), then \({\mathsf{KL}} (\phi_1||\phi_2) = 0\), otherwise \({\mathsf{KL}} (\phi_1||\phi_2) >0\). Intuitively, the lower the value \({\mathsf{KL}} (\phi_1||\phi_2)\) is, the better we have matched the target distribution with our approximate distribution induced by the chosen model. However, \(\mathsf{KL}(\phi|| \cdot )\) is generally not convex nor Lipschitz. Hence, to make the problem tractable, we will make several assumptions. We view these assumptions as comparatively mild, and provide examples shortly after stating the assumptions we need.
Assumption 2. The function \(\mathsf{KL}(\phi|| \varphi(\cdot))\), the data distribution \(\mathcal{D}(\theta)\), and its parameter mapping \(\varphi(\cdot)\) satisfies the following properties.
\(\mathsf{KL}(\phi|| \varphi(\cdot))\) is convex in the model parameter \(\theta\in \Theta\);
The data distribution \(\mathcal{D}(\theta)\) with the parameter \(\varphi(\theta)\) is \((\ell_2, K)\)-Lipschitz continuous in the model parameter \(\theta \in \Theta\) with constant \(K(z), \forall z\in Z\) 7;
Let \(\mathcal{D}_1, \mathcal{D}_2\) be two data distributions with the parameter \(\phi_1, \phi_2 \in \Phi\), and \(d_\mathsf{TV}(\mathcal{D}_1, \mathcal{D}_2)\) be the total variation distance. Then \(\|\phi_1 - \phi_2\| \leq L_\mathsf{TV}\cdot d_\mathsf{TV}(\mathcal{D}_1, \mathcal{D}_2)\) for some constant \(L_\mathsf{TV}>0\).
Here, we provide examples to demonstrate that the above assumptions are comparatively mild. The following is an example showing the convexity of \(\mathsf{KL}(\phi|| \varphi(\cdot))\).
examklconvexexample Consider the density function \(p(z; \varphi(\theta))\) of the data distribution \(\mathcal{D}(\theta)\) satisfying \(p(z; \varphi(\theta)) = \mathrm{Unif}(\exp(c\varphi(\theta)))\) for some constant \(c > 0\) and for any convex function \(\varphi(\theta)\), then \(\mathsf{KL}(\phi|| \varphi(\cdot))\) is convex over \(\theta\).
In the above Assumption 2b, we assume a family of distribution called the \((\ell_2, K)\)-Lipschitz continuous. This Lipschitz continuity over the parametrization of probability distributions allows us to have the following Lipschitz condition of the function \(\mathsf{KL}(\phi||\varphi(\cdot))\) over the model parameter \(\theta\):
lemkllipschitzcondition Given two \(\left(\ell_{2}, K\right)\)-Lipschitz continuous distributions \(\mathcal{D}_{1}=p\left(\cdot \mid \varphi(\theta_{1})\right)\) and \(\mathcal{D}_{2}=p\left(\cdot \mid \varphi(\theta_{2})\right)\), and a target distribution parameter \(\phi\in \Phi\), we have \(\left|\mathsf{KL}\left(\phi||\varphi(\theta_{1})\right) - \mathsf{KL}\left(\phi|| \varphi(\theta_{2})\right) \right|\leq {L_\mathsf{KL}}\left\|\theta_{1}-{\theta}_{2}\right\|\) with a constant \(L_\mathsf{KL}>0\).
The above Assumption 2c is about the continuity on the distribution parameter \(\phi\in \Phi\). Intuitively, this assumption ensures that if the parameters of two distribution are close, then their total variation distance is close as well. With this assumption, we can show that the distance between two distribution parameters \(\|\phi_1 - \phi_2\|\) can be bounded by the KL divergence between the corresponding data distributions.
lemphiboundbykl With Assumption 2c, we have \(\|\phi_1 - \phi_2\|\leq L_\phi \sqrt{\mathsf{KL}(\phi_1||\phi_2)}\) for some constant \(L_\phi>0\).
Intuitively, the above result ensures that given a target distribution parameter \(\phi\), as long as a model \(\theta\) whose corresponding data distribution is close (i.e., \(\mathsf{KL}(\phi || \varphi(\theta))\) is small) to the distribution with the parameter \(\phi\), then \(\varphi(\theta)\) is close to \(\phi\). We will use [lemma:phi-bound-by-KL] in the proof of our main theorem in 4.3.
When \(\mathsf{KL}(\phi|| \varphi(\cdot))\) is convex and Lipschitz over the model \(\theta\), its minimizer can be computed using algorithms similar to 2. In our problem, given a target data distribution with the parameter \(\phi\), we can use the observed data samples to approximately compute the \(\mathsf{KL}(\phi|| \varphi(\theta))\) when deploying a model \(\theta\). Indeed, we assume an existence of an oracle \(\operatorname{\mathsf{EstimateKL}}(\phi, (z_t^{(i)})_{i\in[n_t]})\) which takes the observed samples \((z_t^{(i)})_{i\in[n_t]}\) realized from the induced data distribution \(\mathcal{D}(\theta)\) and the target data distribution parameter \(\phi\) as input to approximate the value \(\mathsf{KL}(\phi|| \varphi(\theta))\). We remark that such oracle has been widely used in the literature on KL divergence estimation [22].
Definition 1 (Oracle \(\operatorname{\mathsf{EstimateKL}}\)). There exists an oracle \(\operatorname{\mathsf{EstimateKL}}\) that given any target parameter \(\phi\in \Phi\), error tolerance \(\epsilon_\mathsf{KL}> 0\) and error probability \(p_\mathsf{KL}> 0\), and \(N_\mathsf{KL}(\epsilon_\mathsf{KL}, p_\mathsf{KL})\) samples \(z_1,\ldots,z_{N_\mathsf{KL}(\epsilon_\mathsf{KL}, p_\mathsf{KL})}\) from a distribution with parameter \(\phi'\), returns an estimated \(\mathsf{KL}\) divergence \(\widetilde{\mathsf{KL}}(\phi||\phi')\) satisfying \(\big\|\widetilde{\mathsf{KL}}(\phi||\phi') - \mathsf{KL}(\phi||\phi')\big\|\leq \epsilon_\mathsf{KL}\) with probability at least \(1 - p_\mathsf{KL}\).
With the oracle \(\operatorname{\mathsf{EstimateKL}}\) to approximately compute the KL divergence, we are now ready to present our inner algorithm \(\operatorname{\mathsf{LearnModel}}\) (see 3).
Similar to before, for analysis purpose, we also define regret of \(\operatorname{\mathsf{LearnModel}}\) in \(S\), the total number of rounds \(\operatorname{\mathsf{LearnModel}}\) has to go through in order to output a \(\epsilon_\mathsf{LM}\)-suboptimal model parameter w.r.t the \(\mathsf{KL}\) objective function: \[\begin{align} & \mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL}) \\ = ~ & \sum_{s=1}^S \left[ \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^+)) + \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^-)) - 2\mathsf{KL}(\phi||\vartheta^*(\phi)) \right] \end{align}\] where \(\vartheta^*(\phi)\) is the model that can induce the target distribution \(\phi\). Using the similar arguments in [theorem:regret-bound-for-indirectly-convex-functions], we first show the following regret guarantee for \(\operatorname{\mathsf{LearnModel}}\):
thmlearnmodelregret When \(\operatorname{\mathsf{LearnModel}}\) is run for \(S\) steps and invokes \(\operatorname{\mathsf{EstimateKL}}\) with arguments \(\epsilon_\mathsf{KL}> 0\) and \(p_\mathsf{KL}> 0\), we have \(\forall p > 0\) \[\begin{align} \mathcal{R}_S(\operatorname{\mathsf{LearnModel}},\mathsf{KL}) = ~ O\left( \sqrt{d_\Phi S} + \sqrt{\epsilon_\mathsf{KL}d_\Phi} \cdot S + \sqrt{S \log\frac{1}{p}} \right) \end{align}\] with probability at least \(1 - p - 2S p_\mathsf{KL}>0\).
[theorem:regret-bound-for-learnmodel] characterizes the regret as a function of the total number of deployments of the procedure in \(\operatorname{\mathsf{LearnModel}}\). Together with regret characterization of the outer algorithm in [theorem:regret-bound-for-indirectly-convex-functions], we can get the final regret bound.
As shown in the previous section, both the outer algorithm (\(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) – in 4.1) and inner algorithm (\(\operatorname{\mathsf{LearnModel}}\) – in 4.2) achieve a sublinear regret w.r.t the total number of steps (\(T\) and \(S\)) when outputting an \(\epsilon\)-optimal solutions. In this section, we combine the results in 4.1 and 4.2 to conclude the analysis for \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) (2) for convex \(\mathsf{PR}^\dagger(\phi)\). The main result of this section is summarized as follows:
thmtotalregret Under Assumption 2, and given access an oracle \(\operatorname{\mathsf{EstimateKL}}\), there exists a choice of \(\epsilon_\mathsf{KL}, p_\mathsf{KL}> 0\) in 3 such that for every \(p > 0\), \[\begin{align} &\mathcal{R}_N(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})\\ = ~ &\widetilde{O}\left( (d_\Theta + d_\Phi) N_\mathsf{KL}(\epsilon_\mathsf{KL},p_\mathsf{KL})^{1/6} N^{5/6} \sqrt{\log\frac{1}{p}} \right) \end{align}\] with probability at least \(1-p\).
Proof Sketch of [theorem:total-regret]. Let \(T\) be the number of steps executed by the outer algorithm \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\), and \(S\) the number of steps in \(\operatorname{\mathsf{LearnModel}}\). Let \(N_\mathsf{KL}(\epsilon_\mathsf{KL}, p_\mathsf{KL})\) (or \(N_\mathsf{KL}\) for short) denote the number of samples used by \(\operatorname{\mathsf{EstimateKL}}\). Since \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) calls \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) and \(\operatorname{\mathsf{LearnModel}}\) \(2T\) times, and \(\operatorname{\mathsf{LearnModel}}\) calls \(\operatorname{\mathsf{EstimateKL}}\) \(2S\) times, the overall number of samples involved in the whole process is \(N = 2(2N_\mathsf{KL}S + 1)T\). Following the regret definition, we can break down the regret into the regret from calling \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) in the outer algorithm and the regret from calling \(\operatorname{\mathsf{EstimateKL}}\) in \(\operatorname{\mathsf{LearnModel}}\). Using the fact that \(\mathsf{PR}^\dagger\) is Lipschitz in the distribution parameter \(\phi\) and the distance between any two distribution parameters can be bounded by the KL divergence between the corresponding data distributions ([lemma:phi-bound-by-KL]), we show that the total regret in \(N\) can be expressed as: \[\begin{align} &\mathcal{R}_N(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})\\ =& O\left( \sqrt{N} + N_\mathsf{KL}T \cdot \sqrt{S \cdot \mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL})}\right.\\ & \quad \quad \left. + (N_\mathsf{KL}S + 1) \cdot \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})\right) \end{align}\] where \(\mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})\) and \(\mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL})\) are obtained from [theorem:regret-bound-for-indirectly-convex-functions] and [theorem:regret-bound-for-learnmodel] as functions of \(\epsilon_\mathsf{LM}, \epsilon_\mathsf{KL}, S, T\) and \(D_\Theta\) and \(D_\Phi\). Then by balancing the terms and setting \(\epsilon_\mathsf{LM}\) and \(\epsilon_\mathsf{KL}\) according to the convergence analysis for both \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) and \(\operatorname{\mathsf{LearnModel}}\) (9 and 10), we can get an express of the total regret. ◻
[theorem:total-regret] show that our procedure is sublinear in the \(N\), the total number of samples we deploy during the process. Notice that this also implies that our method is sublinear w.r.t. the total number of deployments \(T_{\textsf{total}}= S \times T\). To see this, recall that the total number of samples \(N\) required throughout the process is \(N =2(2N_{{kl}} \times S + 1) T\); if we measure the regret w.r.t. the total number of deployments, \(N_{{kl}}\) will be constant, and thus the regret will still be sublinear w.r.t. \(T_{\textsf{total}}\). In addition, recall from 1, the lowest regret optimal classifier implies the optimal classifier up to some addictive error, which means that by having a sublinear regret in \(N\) (and \(T_{\textsf{total}}\)), we also get a model that is arbitrarily close to the performative optimal model.
We provide empirical results using a toy example to demonstrate the efficiency of our method. In particular, we compare our proposed method (which minimizes \(\text{PR}\) as a function of the distribution parameter \(\phi\) after reparametrization) with the baseline method (which directly minimizes \(\mathsf{PR}\) as a model parameter \(\theta\)). We observe that under
different settings, both methods converge. However, our proposed method (shown in orange) is more efficient: it demonstrates a much faster convergence rate on average over multiple runs, indicating that our reparametrization method is effective when
dealing with distributions that are non-convex in \(\theta\) but convex in \(\phi\) (as per Assumption 1). The plot can be found in 13. The details for
reproducing our experimental results can be found at https://github.com/UCSC-REAL/PP-bandit-feedback.
Since our method uses a double-loop zeroth order optimization method, the convergence is likely to be slow in practice. Thus, it requires extra consideration regarding the convergence rate, and efficiency can vary based on particular settings. In particular, the success of our method depends on the fast deployment of the frequently updated models. One potential way to speed up the deployment process may be performing parallelization, which accelerates the optimization process and reduces the time required to find a suitable solution (see, e.g., [23] for a detailed reference). In our algorithm, parallelization can be applied to several places, e.g., the two-point estimations can be computed in parallel and potentially speed up the process.
Y. Chen and Y. Liu are partially supported by the National Science Foundation (NSF) under grants IIS-2143895 and IIS-2040800.
Since the process of finding the optimal performative model involves deploying sub-optimal models on human agents, sample efficiency is important. Thus sample complexity needs to be taken into consideration when choosing the particular KL divergence oracle used in \(\operatorname{\mathsf{LearnModel}}\). Additionally, the concept of performativity highlights a significant broader impact of our work: the importance of recognizing scenarios where predictions can modify the very environment they’re meant to predict. Take, for example, an online advertising platform utilizing machine learning models to tailor ads for users. These models, by analyzing user behaviors and traits to serve personalized ads, might inadvertently influence both user and advertiser actions, thereby establishing a feedback loop. This dynamic underscores the need for a mindful approach in deploying predictive models, especially in settings sensitive to the outcomes of such predictions.
We arrange the appendix as follows:
\(\bullet\)
7 provides omitted algorithm and proofs for 2.
8 provides more examples that satisfies \(\mathsf{PR}\) being convex in \(\phi\) not in \(\theta\).
9 provides omitted example and proofs for 4.1.
10 provides omitted proofs for 4.2.
11 provides omitted proof for 4.3.
12 provides an additional literature review on performative prediction and related literature.
13 provides empirical verification using toy example generated by 3 to demonstrate the efficiency of our proposed method.
Proof. Let \(x_1,\ldots,x_n\) be the first \(n\) points queried by \(\mathcal{A}\). By the convexity of \(f\), the average of these points \(\bar{x} = \frac{1}{n} \sum_i x_i\) satisfies \[\begin{align} f(\bar{x}) - f(x^*) \leq \frac{1}{n} \sum_{i=1}^n \left[ f(x_i) - f(x^*) \right] = \frac{\mathcal{R}_n(\mathcal{A},f)}{n} \end{align}\] Thus if \(\mathcal{R}_n(\mathcal{A},f) = o(n)\), then after \(n = \mathcal{R}_n(\mathcal{A},f)/\epsilon\) queries, \(\bar{x}\) satisfies \(f(\bar{x}) - f(x^*) \leq \epsilon\) as required. ◻
4 is a straightforward generalization of the algorithm introduced by [17], while we generalize their setting where the function can be evaluated exactly to the setting where noisy evaluation is allowed.
To prove [lemma:regret-bound-for-model-convex-performative-risk], we first provide a series of lemmas and claims that will be useful later.
Claim 1 (Regret from estimating \(\mathsf{PR}\)). For any \(p>0\), with probability at least \(1-p\), \[\begin{align} \sum_{t=1}^T \left[ \widetilde{\mathsf{PR}}(\theta_t^+) - f(\theta_t^+) \right] \leq F\sqrt{T \log\frac{1}{p}} \quad \mathrm{and} \quad \sum_{t=1}^T \left[ \widetilde{\mathsf{PR}}(\theta_t^-) - f(\theta_t^-) \right] \leq F\sqrt{T \log\frac{1}{p}} \end{align}\]
Proof. The claim follows from Hoeffding’s inequality, since \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) is unbiased and bounded by \([0,F]\). ◻
Claim 2 (Regret from smoothing over the sphere or ball). For any \(\theta \in \Theta\), \(u \in \mathbb{S}\), and \(\delta > 0\), all of the following are at most \(\delta L\): \[\begin{align} |\mathsf{PR}(\theta + \delta u) - \mathsf{PR}(\theta)|, \quad |\mathsf{PR}(\theta - \delta u) - \mathsf{PR}(\theta)|, \quad\\ \left| \frac{1}{2}[ \mathsf{PR}(\theta + \delta u) + \mathsf{PR}(\theta - \delta u) ] - \mathsf{PR}(\theta) \right|, \quad \text{and} \quad |\widehat{\mathsf{PR}}(\theta) - \mathsf{PR}(\theta)|. \end{align}\]
Proof sketch. Lipschitzness of \(\mathsf{PR}\). ◻
Claim 3 (Deviation of smoothed function). For any \(p > 0\), with probability at least \(1-p\), \[\begin{align} \sum_{t=1}^T \widehat{\mathsf{PR}}(\theta_t) - \mathbb{E}_T \left[ \sum_{t=1}^T \widehat{\mathsf{PR}}(\theta_t) \right] \leq F \sqrt{T \log\frac{1}{p}} \end{align}\]
Proof sketch. The left-hand side is the sum of a martingale difference sequence. The Azuma-Hoeffding inequality yields the result. ◻
Claim 4 (Gradient estimate is unbiased and bounded). There exists a constant \(c>0\) such that for all \(t \in [T]\), \(\mathbb{E}_t[g_t] = \nabla \widehat{\mathsf{PR}}(\theta_t)\) and \(\|g_t\|_2^2 \leq cd_\theta L^2\).
Proof. Proved in [18] (see Lemma 10, noting that the \(\ell_2\) norm is its own dual). ◻
lem Let \(\theta \in \Theta\), and let \(\theta_1,\ldots,\theta_t \in \Theta\) be a sequence of iterates given by the update rule \(\theta_{t+1} = \Pi_{(1-\delta)\theta} (\theta_t - \eta g_t) - \theta\) for some sequence of gradient estimates \(g_t \in \mathbb{R}^{d_\Theta}\). Then \[\begin{align} \mathbb{E}_T \left[\sum_{t=1}^T \widehat{\mathsf{PR}}(\theta_t)\right] - \sum_{t=1}^T \widehat{\mathsf{PR}}(\theta) \leq \frac{D_\Theta^2}{\eta} + \eta cd_\theta L^2 T \end{align}\]
Proof of [lemma:expected-suboptimality-under-smoothing]. Observe that \[\begin{align} \mathbb{E}_T \left[\sum_{t=1}^T \widehat{\mathsf{PR}}(\theta_t)\right] - \sum_{t=1}^T \widehat{\mathsf{PR}}(\theta) &= \sum_{t=1}^T \mathbb{E}_t \left[ \widehat{\mathsf{PR}}(\theta_t) - \widehat{\mathsf{PR}}(\theta) \right] \\ &\leq \sum_{t=1}^T \mathbb{E}_t \left[ \nabla \widehat{\mathsf{PR}}(\theta_t)^\top (\theta_t - \theta) \right] }} \\ &= \sum_{t=1}^T \mathbb{E}_t \left[ g_t^\top (\theta_t - \theta) \right] } \end{align}\] To decompose \(g_t^\top (\theta_t - \theta)\), note that \[\begin{align} \|\theta_{t+1} - \theta\|^2 &= \|\Pi_{(1-\delta)\theta} (\theta_t - \eta g_t) - x\|^2 \\ &\leq \|\theta_t - \eta g_t - \theta\|^2 \\ &= \|\theta_t - \theta\|^2 + \eta^2 \|g_t\|^2 - 2\eta \cdot g_t^\top (\theta_t - \theta) \end{align}\] Therefore \[\begin{align} g_t^\top(\theta_t - x) &\leq \frac{\|\theta_t - \theta\|^2 - \|\theta_{t+1} - \theta\|^2 + \eta^2 \|g_t\|^2}{2\eta} \\ \sum_{t=1}^T \mathbb{E}_t \left[ g_t^\top (\theta_t - \theta) \right] &\leq \sum_{t=1}^T \mathbb{E}_t \left[ \frac{\|\theta_t - \theta\|^2 - \|\theta_{t+1} - \theta\|^2 + \eta^2 \|g_t\|^2}{2\eta} \right] \\ &\leq \frac{1}{2\eta} \mathbb{E}_t \left[ \|\theta_1 - \theta\|^2 + \eta^2 cd_\Theta L^2T \right] } \\ &\leq \frac{D_\Theta^2}{2\eta} + \frac{\eta cd_\Theta L^2 T}{2} \end{align}\] as required. ◻
Claim 5 (Regret from projection). For any \(\theta \in \Theta\), \(\mathsf{PR}(\theta_\delta) - \mathsf{PR}(\theta) \leq \delta D_\Theta L\).
Proof. Since \(\mathsf{PR}\) is \(L\)-Lipschitz and \(\Pi_{(1-\delta)\Theta}\) projects from a set of diameter \(D_\Theta\) to a set of diameter \((1-\delta)D_\Theta\), we have \(\mathsf{PR}(\theta_\delta) - \mathsf{PR}(\theta) \leq L\|\theta_\delta - \theta\| \leq \delta D_\Theta L\). ◻
Claim 6 (Optimality of projected parameters). Since \(\mathsf{PR}\) is convex in \(\theta\), \(\mathsf{PR}\left(\Pi_{(1-\delta)\Theta}(\theta_{\texttt{OPT}})\right) = \mathop{\mathrm{\arg\!\min}}_{\theta \in (1-\delta)\Theta} \mathsf{PR}(\theta)\).
We can now complete our regret bound for [lemma:regret-bound-for-model-convex-performative-risk]. Recall the lemma statement:
Proof of [lemma:regret-bound-for-model-convex-performative-risk]. We have \[\begin{align} \mathcal{R}_T(\mathcal{A}_{\ref{algorithm:minimize-convex-function}},f) &= \sum_{t=1}^T \left[ \operatorname{\mathsf{Estimate}\mathsf{PR}}(\theta_t^+) + \operatorname{\mathsf{Estimate}\mathsf{PR}}(\theta_t^-) - 2 \mathsf{PR}(\theta_{\texttt{OPT}}) \right] \\ &= \underbrace{\sum_{t=1}^T \left[ \widetilde{\mathsf{PR}}(\theta_t^+) + \widetilde{\mathsf{PR}}(\theta_t^-) - \mathsf{PR}(\theta_t^+) - \mathsf{PR}(\theta_t^-) \right]}_{\text{(I)}} + \underbrace{\sum_{t=1}^T \left[ \mathsf{PR}(\theta_t^+) + \mathsf{PR}(\theta_t^-) - 2\widehat{\mathsf{PR}}(\theta_t) \right]}_{\text{(II)}} \\ &\qquad + \underbrace{2 \sum_{t=1}^T \left[ \widehat{\mathsf{PR}}(\theta_t) - \mathbb{E}_t [\widehat{\mathsf{PR}}(\theta_t)] \right]}_{\text{(III)}} + \underbrace{2 \sum_{t=1}^T \left[ \mathbb{E}_t [\widehat{\mathsf{PR}}(\theta_t)] - \hat{f}(\theta^*_\delta) \right]}_{\text{(IV)}}\\ &\qquad + \underbrace{2 \sum_{t=1}^T \left[ \widehat{\mathsf{PR}}(\theta^*_\delta) - f(\theta^*_\delta) \right]}_{\text{(V)}} + \underbrace{2 \sum_{t=1}^T \left[ \mathsf{PR}(\theta^*_\delta) - \mathsf{PR}(\theta_\texttt{OPT}) \right]}_{\text{(VI)}} \\ &\leq \underbrace{ 2F\sqrt{T \log\frac{1}{p_1}} }_{\substack{ \text{(I), w.p. 1-2p_1} \\ \text{(\Cref{claim:regret-from-estimating-f})} }} + \underbrace{ 4\delta LT }_{\substack{ \text{(II), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} + \underbrace{ 2F\sqrt{T \log\frac{1}{p_2}} }_{\substack{ \text{(III), w.p. 1-2p_2} \\ \text{(\Cref{claim:deviation-of-smoothed-function})} }} + \underbrace{ \frac{2 D_\Theta^2}{\eta} + 2\eta cd_\theta L^2 T }_{\substack{ \text{(IV), w.p. 1} \\ \text{(\Cref{lemma:expected-suboptimality-under-smoothing})} }} + \underbrace{ 2\delta LT }_{\substack{ \text{(V), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} + \underbrace{ 2\delta D_\Theta LT }_{\substack{ \text{(V), w.p. 1} \\ \text{(\Cref{claim:regret-from-projection})} }} \end{align}\]
Thus for any \(p>0\), a choice of \(p_1 = p_2 = p/4\), along with \(\eta = 1/\sqrt{d_\theta T}\) and any \(\delta \leq \sqrt{d_\theta/T}\), yields \(\mathcal{R}_T(\mathcal{A}_{\ref{algorithm:minimize-convex-function}},\mathsf{PR}) = O(\sqrt{d_\theta T\log \frac{1}{p}})\) with probability at least \(1-p\). Finally, since \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) is queried twice per step, \(n = 2T\), which gives us \(\mathcal{R}_n(\mathcal{A}_{\ref{algorithm:minimize-convex-function}},\mathsf{PR}) = \mathcal{R}_T(\mathcal{A}_{\ref{algorithm:minimize-convex-function}},\mathsf{PR}) = O(\sqrt{d_\theta n\log \frac{1}{p}})\), completing the proof. ◻
Since \(\phi\) is strictly increasing in \([0, 1]\), the inverse mapping \(\phi^{-1}\) is well-defined, and we can reformulate the performative risk \(\mathsf{PR}(\theta)\) as a function of \(\varphi_\theta\), denoted \(\mathsf{PR}^\dagger(\varphi_\theta)\), as follows: \[\begin{align} \mathsf{PR}(\theta;x) &= \mathbb{E}_{y \sim \mathsf{Bern}(\varphi_\theta)}[\ell(\theta;x,y)] \\ &= \varphi_\theta \ell(\theta;x,1) + (1- \varphi_\theta) \ell(\theta;x,0) \\ &= \varphi_\theta \ell\left(\phi^{-1}(\varphi_\theta);x,1\right) + (1- \varphi_\theta) \ell\left(\phi^{-1}(\varphi_\theta);x,0\right) \\ &=: \mathsf{PR}^\dagger(\varphi_\theta;x) \end{align}\]
Plugging in \(\ell\), we have \[\begin{align} \mathsf{PR}^\dagger(\varphi_\theta;x) &= -\varphi_\theta \cdot \left(\phi^{-1}(\varphi_\theta)x - 1\right)^2 - (1- \varphi_\theta) \cdot \left(\phi^{-1}(\varphi_\theta)x\right)^2 \\ &= -\varphi_\theta \cdot \left(\sqrt{\varphi_\theta}x - 1\right)^2 - (1- \varphi_\theta)\varphi_\theta x^2 (\varphi_\theta) = \sqrt{\varphi_\theta}} \end{align}\] Note that for all \(x \in [0,1]\), \(\mathsf{PR}^\dagger(\varphi_\theta;x) = \mathsf{PR}(\theta;x)\) is convex in \(\varphi_\theta\) over \([0,1]\). In contrast, \[\begin{align} \mathsf{PR}(\theta;x) &= \theta^2 \cdot \ell(\theta;x,1) + (1-\theta^2) \cdot \ell(\theta;x,0) \\ &= -\theta^2\cdot (\theta x - 1)^2 - (1 - \theta^2) \cdot (\theta x)^2 \end{align}\] which is non-convex in \(\theta\) over \([0,1]\) for all \(x \in [0,1]\).


Figure 5: An example showing that our assumption is weaker than the mixture dominance assumption in [9]. In the left figure, the blue curve represents the function \(\mathsf{PR}^\dagger(\varphi_\theta)\) which is convex w.r.t the data distribution parameter \(\varphi_\theta\); while the red curve represents the function \(\mathsf{PR}(\theta)\), which is not a convex function with respect to \(\theta\). In the right two figures, we compare \(\mathsf{PR}\) as a function of the model parameter \(\theta\) and as a function of the distribution parameter \(\phi\)..
Notice that 3 can be generalized to any distribution map \(\phi(\theta)\) that satisfies \(\phi(\theta) = \theta^\alpha\) for any \(\alpha> 1\), and any \(\ell_\beta\) loss for even \(\beta\) value. Below in 6, we provide the plot for for \(\phi(\theta) = \theta^4\) with \(\ell_4\) norm loss (\(L_4\) norm is defined as \(L_4(x, y)=\left(\sum_{i=1}^d\left|x_i-y_i\right|^4\right)^{\frac{1}{4}}\) where \(d\) is the dimension of \(x\) and \(y\)). The original \(\mathsf{PR}\) loss \(\mathsf{PR}(\theta)\) is in red, which is non-convex), and the reformulated PR loss \(\mathsf{PR}^\dagger (\theta)\) is in blue via reparameterization, which is convex).
In addition, 3 can also be a generalized monotone polynomial function \(\phi(\theta)\). For example, \(\phi(\theta) = c \theta^2 + d \theta\), for \(d\geq 0, d \geq -2c\).
\[\begin{align} \mathsf{PR}(\theta) &= \mathbb{E}_{x\sim \mathcal{D}(\theta)} \ell(x;\theta)\\ &= \mathbb{E}_{x\sim N(\varphi(\theta), \sigma^2)} (\theta x)^2\\ &= \theta^2 (\sigma^2 + \varphi(\theta)^2) \end{align}\] when \(\varphi(\theta) = \sqrt{\theta}\), \(\mathsf{PR}(\theta)\) is not convex in \(\theta\). To see this: \[\begin{align} \mathsf{PR}(\theta) = \theta^2(\sigma^2 + \theta) \end{align}\] On the other hand, since \(\phi = \varphi(\theta) = \sqrt{\theta}\), we have \[\begin{align} \mathsf{PR}(\phi) = \phi (\theta^2 + \phi) \end{align}\] which is convex in \(\phi\).
\[\begin{align} \mathsf{PR}(\theta) &= \mathbb{E}_{y\sim \mathcal{D}(\theta)} (\ell(\theta; x, y))\\ &= - \mathbb{E}_{\text{Uniform}[0, \varphi(\theta)]} (\theta x - y)^2\\ &= \theta^2 x^2 - \theta x \varphi(\theta) + \frac{1}{3} \varphi(\theta)^2 \end{align}\] when \(\varphi(\theta) = \theta^2\), we have \[\begin{align} \mathsf{PR}(\theta) = \theta^2 x^2 - \theta^3 x + \frac{1}{3}\theta^4 \end{align}\] which is non-convex in \(\theta\). On the other hand, denote \(\phi = \varphi(\theta)\), we have \(\theta = \sqrt{\phi}\), plug it into \(\mathsf{PR}\), we have: \[\begin{align} \mathsf{PR}(\phi) = \phi x^2 - \phi \sqrt{\phi} + \frac{1}{3} \phi^2 \end{align}\] which is convex in \(\phi\).
We present a series of lemmas and claims that are helpful for proving [theorem:regret-bound-for-indirectly-convex-functions].
Claim 7 (Deviation of \(\mathsf{PR}^\dagger\) due to error of \(\operatorname{\mathsf{LearnModel}}\)). If \(\mathsf{PR}^\dagger\) is \(L^\dagger\)-Lipschitz, then for any \(\phi \in \Phi\), the value \(\hat{\theta} \in \Theta\) returned by \(\operatorname{\mathsf{LearnModel}}(\phi,\epsilon_\mathsf{LM},p_\mathsf{LM})\) satisfies \(|\mathsf{PR}^\dagger(\phi) - \mathsf{PR}(\hat{\theta})| \leq L^\dagger \epsilon_\mathsf{LM}\) with probability at least \(1-p_\mathsf{LM}\).
Proof. We have \[\begin{align} \left|\mathsf{PR}^\dagger(\phi) - \mathsf{PR}(\hat{\theta})\right| &= \left|\mathsf{PR}^\dagger(\phi) - \mathsf{PR}^\dagger(\varphi(\hat{\theta}))\right| \\ &\leq L^\dagger \left\|\phi - \varphi(\hat{\theta})\right\| ^\dagger} \\ &\leq L^\dagger \epsilon_\mathsf{LM} }} \end{align}\] where the last inequality holds with probability at least \(1-p_\mathsf{LM}\). ◻
Claim 8 (Deviation of gradient estimate due to error of \(\operatorname{\mathsf{LearnModel}}\) and \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\)). Define \[\begin{align} \tilde{g}_t := \frac{d_\Phi}{\delta} \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) u_t \qquad \text{and} \qquad g_t := \frac{d_\Phi}{\delta} \mathsf{PR}^\dagger(\phi_t^+) u_t \end{align}\] For any \(t \in [T]\), \[\begin{align} g_t - \tilde{g}_t \leq \frac{d_\Phi}{\delta} \left[ \mathsf{PR}(\hat{\theta}_t^+) - \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) + \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] u_t . \end{align}\]
Proof. We have \[\begin{align} g_t &= \frac{d_\Phi}{\delta} \mathsf{PR}^\dagger(\phi_t^+) u_t \\ &= \frac{d_\Phi}{\delta} \left[ \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) - \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^+) - \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}^\dagger(\phi_t^+) \right] u_t \\ &= \tilde{g}_t + \frac{d_\Phi}{\delta} \left[ \mathsf{PR}(\hat{\theta}_t^+) - \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) + \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] u_t _t} \end{align}\] ◻
lem For any \(\phi \in \Phi\), with probability at least \(1 - Tp_\mathsf{LM}\) over the calls to \(\operatorname{\mathsf{LearnModel}}\), \[\begin{align} \mathbb{E}_T \left[ \sum_{t=1}^T \widehat{\mathsf{PR}}^\dagger(\phi_t) \right] - \sum_{t=1}^T \widehat{\mathsf{PR}}^\dagger(\phi) \leq \frac{D_\Phi^2}{\eta} + \eta cd_\Phi L^2 T + \frac{D_\Phi L^\dagger \epsilon_\mathsf{LM}d_\Phi T}{\delta} \end{align}\]
Proof of [lemma:high-probability-expected-suboptimality-under-smoothing]. For any \(\phi \in \Phi\), we have \[\begin{align} & \mathbb{E}\left[ \sum_{t=1}^T \widehat{\mathsf{PR}}^\dagger(\phi_t) \right] - \sum_{t=1}^T \widehat{\mathsf{PR}}^\dagger(\phi) \\ =& \sum_{t=1}^T \mathbb{E}\left[ \widehat{\mathsf{PR}}^\dagger(\phi_t) - \widehat{\mathsf{PR}}^\dagger(\phi) \right] \\ \leq& \sum_{t=1}^T \mathbb{E}\left[ \nabla \widehat{\mathsf{PR}}^\dagger(\phi_t)^\top(\phi_t - \phi) \right] }^\dagger}\\ =& \sum_{t=1}^T \mathbb{E}\left[ g_t^\top (\phi_t - \phi) \right] } \\ =& \sum_{t=1}^T \mathbb{E}\left[ \left( \tilde{g}_t + \frac{d_Y}{\delta} \left[ \mathsf{PR}(\hat{\theta}_t^+) - \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) + \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] \cdot u_t \right)^\top(\phi_t - \phi) \right] } \\ =& \sum_{t=1}^T \mathbb{E}\left[ \left( \tilde{g}_t + \frac{d_Y}{\delta} \left[ \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] \cdot u_t \right)^\top(\phi_t - \phi) \right] [\widetilde{\mathsf{PR}}(\cdot)] = \mathsf{PR}(\cdot) since \operatorname{\mathsf{Estimate}\mathsf{PR}} is unbiased} \\ =& \sum_{t=1}^T \mathbb{E}\left[ \tilde{g}_t^\top (\phi_t - \phi) \right] + \frac{d_Y}{\delta} \sum_{t=1}^T \mathbb{E}\left[ \left( \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right) u_t^\top (\phi_t - \phi) \right] \\ \leq& \sum_{t=1}^T \mathbb{E}\left[ \tilde{g}_t^\top (\phi_t - \phi) \right] + \frac{d_Y}{\delta} \sum_{t=1}^T \mathbb{E}\left[ \left| \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right| \cdot \|u_t\| \cdot \|\phi_t - \phi\| \right] \\ \leq& \sum_{t=1}^T \mathbb{E}\left[ \tilde{g}_t^\top (\phi_t - \phi) \right] + \frac{d_Y}{\delta} \sum_{t=1}^T \mathbb{E}\left[ L^\dagger \epsilon_h \cdot D_Y \right] , w.p. 1-Tp_h} \\ \leq& \frac{D_Y^2}{\eta} + \eta cd_Y L^2 T + \frac{d_Y}{\delta}L^\dagger \epsilon_h D_Y T } \end{align}\] ◻
We can now complete our regret bound for \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) (2). We recall the theorem statement for [theorem:regret-bound-for-indirectly-convex-functions]:
Proof of [theorem:regret-bound-for-indirectly-convex-functions]. We have \[\begin{align} & \;\mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}},\mathsf{PR}) \\ &= \sum_{t=1}^T \left[ \operatorname{\mathsf{Estimate}\mathsf{PR}}(\hat{\theta}_t^+) + \operatorname{\mathsf{Estimate}\mathsf{PR}}(\hat{\theta}_t^-) - 2 \mathsf{PR}(\theta_\texttt{OPT}) \right] \\ &= \underbrace{\sum_{t=1}^T \left[ \widetilde{\mathsf{PR}}(\hat{\theta}_t^+) + \widetilde{\mathsf{PR}}(\hat{\theta}_t^-) - \mathsf{PR}(\hat{\theta}_t^+) - \mathsf{PR}(\hat{\theta}_t^-) \right]}_{\text{(I)}} + \underbrace{\sum_{t=1}^T \left[ \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-) - \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}^\dagger(\phi_t^-) \right]}_{\text{(II)}} \\ &\qquad + \underbrace{\sum_{t=1}^T \left[ \mathsf{PR}^\dagger(\phi_t^+) + \mathsf{PR}^\dagger(\phi_t^-) - 2\widehat{\mathsf{PR}}^\dagger(\phi_t) \right]}_{\text{(III)}} + \underbrace{2 \sum_{t=1}^T \left[ \widehat{\mathsf{PR}}^\dagger(\phi_t) - \mathbb{E}_t [\widehat{\mathsf{PR}}^\dagger(\phi_t)] \right]}_{\text{(IV)}} \\ &\qquad + \underbrace{2 \sum_{t=1}^T \left[ \mathbb{E}_t [\widehat{\mathsf{PR}}^\dagger(\phi_t)] - \widehat{\mathsf{PR}}^\dagger(\phi^*_\delta) \right]}_{\text{(V)}} + \underbrace{2 \sum_{t=1}^T \left[ \widehat{\mathsf{PR}}^\dagger(\phi^*_\delta) - \mathsf{PR}^\dagger(\phi^*_\delta) \right]}_{\text{(VI)}} + \underbrace{2 \sum_{t=1}^T \left[ \mathsf{PR}^\dagger(\phi^*_\delta) - \mathsf{PR}^\dagger(\phi_\texttt{OPT}) \right]}_{\text{(VII)}} \\ &\leq \underbrace{ 2F\sqrt{T \log\frac{1}{p_1}} }_{\substack{ \text{(I), w.p. 1-2p_1} \\ \text{(\Cref{claim:regret-from-estimating-f})} }} \qquad + \underbrace{ 2L^\dagger \epsilon_\mathsf{LM}T }_{\substack{ \text{(II), w.p. 1-2Tp_\mathsf{LM}} \\ \text{(\Cref{claim:deviation-of-PRdagger-due-to-error-of-learnmodel})} }} \qquad + \underbrace{ 4\delta LT }_{\substack{ \text{(III), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} \qquad + \underbrace{ 2F\sqrt{T \log\frac{1}{p_2}} }_{\substack{ \text{(IV), w.p. 1-2p_2} \\ \text{(\Cref{claim:deviation-of-smoothed-function})} }} \\ &\qquad \qquad \qquad + \underbrace{ \frac{2D_\Phi^2}{\eta} + 2 \eta cD_\Phi L^2 T + \frac{2 D_\Phi L^\dagger \epsilon_\mathsf{LM}D_\Phi T}{\delta} }_{\substack{ \text{(V), w.p. 1-2Tp_\mathsf{LM}} \\ \text{(\Cref{lemma:high-probability-expected-suboptimality-under-smoothing})} }} \qquad + \underbrace{ 2\delta L^\dagger T }_{\substack{ \text{(VI), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} \qquad + \underbrace{ 2\delta D_\Phi L^\dagger T }_{\substack{ \text{(VII), w.p. 1} \\ \text{(\Cref{claim:regret-from-projection})} }} \end{align}\]
Recall that in 2, we set \(\delta = \sqrt{\epsilon_\mathsf{LM}D_\Phi}\) and \(\eta = 1/\sqrt{D_\Phi T}\). Thus for any \(p'>0\), a choice of \(p_1 = p_2 = p'/4\) yields \[\begin{align} \mathcal{R}_T(\mathcal{A}_{\ref{algorithm:minimize-indirectly-convex-function}},\mathsf{PR}) = O\left( \sqrt{D_\Phi T} + \sqrt{\epsilon_\mathsf{LM}D_\Phi} \cdot T + \sqrt{T \log\frac{1}{p'}} \right) \end{align}\] with probability at least \(1 - p' - 2Tp_\mathsf{LM}\) as required. ◻
We first provide a proof for [lemma:lip-phi-in-KL]. Recall the lemma statement:
Proof of [lemma:lip-phi-in-KL]. \[\begin{align} & \left| \mathsf{KL}(\phi||\varphi(\theta_1))-\mathsf{KL}(\phi||\varphi(\theta_2)) \right|\\ =& \left| \int_z p(z|\phi) \log \frac{p(z|\phi)}{p(z|\varphi(\theta_1))}dz - \int_z p(z|\phi) \log \frac{p(z|\phi)}{p(z|\varphi(\theta_2))}dz \right|\\ =& \left| \int_z p(z|\phi) (\log p(z|\varphi(\theta_1)) - \log p(z|\varphi(\theta_2)))dz \right|\\ \leq & \int_z p(z|\phi) \left| \log p(z|\varphi(\theta_1)) - \log p(z|\varphi(\theta_2)) \right|dz \\ \leq & \int_z p(z|\phi) L_\mathsf{KL}\|\theta_1 - \theta_2\| dz _{1} and \mathcal{P}_{2} are lipschitzness continuous, Theorem 3 of \cite{Honorio2012lipschitz}}\\ = & L_\mathsf{KL}\|\theta_1 - \theta_2\| \underbrace{\int_z p(z|\phi) dz}_{=1}\\ =& L_\mathsf{KL}\|\theta_1 - \theta_2\| \end{align}\] ◻
Next, we provide the proof for [lemma:phi-bound-by-KL]. Recall the lemma statement:
Proof of [lemma:phi-bound-by-KL]. \[\begin{align} \|\phi_1 - \phi_2\|_2 \leq L_\mathsf{TV}d_\mathsf{TV}(\phi_1, \phi_2) \leq L_\mathsf{TV}\sqrt{\frac{1}{2} \mathsf{KL}(\phi_1, \phi_2)} \triangleq L_{\phi} \sqrt{\mathsf{KL}(\phi_1, \phi_2)} \end{align}\] The second inequality is due to Pinsker’s inequality. ◻
We then show the example provide by [example:convex-kl] is convex in \(\theta\). Recall the example:
Below we provide proof for it being convex in \(\theta\):
Proof for [example:convex-kl] being convex in \(\theta\). Under condition 1, we have \(p(z;\phi) = \frac{1}{\exp(c\varphi(\theta))}\). We can rewrite the \(\mathsf{KL}(\phi||\varphi(\theta))\) divergence as: \[\begin{align} \mathsf{KL}(\phi||\varphi(\theta)) =& \int_z p(z;\phi)\log \frac{p(z;\phi)}{p(z;\varphi(\theta))} dz\\ =& \int_z \frac{1}{\exp(c\phi)} \log \frac{\exp(c\varphi(\theta))}{ \exp(c\phi)} dz\\ =& \frac{\exp(c\varphi(\theta))}{\exp(c\phi)} \log \frac{\exp(c\varphi(\theta)}{\exp(c\phi)}\\ =& \exp(c(\varphi(\theta) - \phi)) c(\varphi(\theta) - \phi) \end{align}\]
Denote \(\mathsf{KL}(\phi||\varphi(\theta)) = f(g(\theta))\) where \(f(x) = cx\exp(cx)\) and \(g(\theta) = \varphi(\theta) - \phi\).
To show 3 is convex in \(\theta\), it suffices to show f(x) is convex non-decreasing in x, and \(g(\theta)\) is convex in \(\theta\). First, \(g(\theta)\) is convex in \(\theta\) due to condition 2.
For \(f(x)\), take the first and second derivative and find conditions to make them both non negative: \[\begin{align} \frac{\partial f(x)}{\partial x} &= c\exp(cx)+ cx^2 \exp(cx)\\ &=
c\exp(cx)(1 + cx)\geq 0\\ \frac{\partial^2 f(x)}{\partial x^2} &= c^2\exp(cx)(2 + cx)\geq 0
\end{align}\]
It suffices to set \((2+cx)\geq 0\) and \(c(1+cx)\geq 0\) which suffices to set \(c\geq \frac{2}{\max |\varphi(\theta) - \phi|}\). ◻
We can now complete our regret bound for \(\operatorname{\mathsf{LearnModel}}\) (3). Recall the theorem statement:
Proof of [theorem:regret-bound-for-learnmodel]. \[\begin{align} &\mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL}) \\ =& \sum_{s=1}^S \left[ \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^+)) + \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^-)) - 2\underbrace{\mathsf{KL}(\phi||\varphi(\vartheta^*(\phi)))}_{=0, \varphi(\vartheta^*(\phi))) = \phi } \right]\\ =& \underbrace{\sum_{s=1}^S \left[ \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^+)) - {\mathsf{KL}}(\phi||\varphi(\theta_{s}^+)) + \widetilde{\mathsf{KL}}(\phi||\varphi(\theta_{s}^+)) - \mathsf{KL}(\phi||\varphi(\theta_{s}^-)) \right]}_{\text{(I)}}\\ &\qquad + \underbrace{\sum_{s=1}^S \left[ \mathsf{KL}(\phi||\varphi(\theta_s^+)) + \mathsf{KL}(\phi||\varphi(\theta_s^-)) - 2\widehat{\mathsf{KL}}(\phi||\varphi(\theta_s)) \right]}_{\text{(II)}} \\ &\qquad + \underbrace{2 \sum_{s=1}^S \left[ \widehat{\mathsf{KL}}(\phi||\varphi(\theta_s)) - \mathbb{E}_s [\widehat{\mathsf{KL}}(\phi||\varphi(\theta_s))] \right]}_{\text{(III)}} + \underbrace{2 \sum_{s=1}^S \left[ \mathbb{E}_s [\widehat{\mathsf{KL}}(\phi||\varphi(\theta_s))] - \widehat{\mathsf{KL}}(\phi||\varphi(\theta^*_\delta)) \right]}_{\text{(IV)}}\\ &\qquad + \underbrace{2 \sum_{s=1}^S \left[ \widehat{\mathsf{KL}}(\phi||\varphi(\theta^*_\delta)) - {\mathsf{KL}}(\phi||\varphi(\theta^*_\delta)) \right]}_{\text{(V)}} + \underbrace{2 \sum_{s=1}^S \left[ {\mathsf{KL}}(\phi||\varphi(\theta^*_\delta)) - {\mathsf{KL}}(\phi||\varphi(\theta^*)) \right]}_{\text{(VI)}} \\ &\leq \underbrace{ 2\epsilon_\mathsf{KL}S }_{\substack{ \text{(I), w.p. 1-2 S p_\mathsf{KL}} \\ \text{(Assumption \ref{ass:kl-oracle})} }} \qquad + \underbrace{ 4\delta L_\mathsf{KL}S }_{\substack{ \text{(II), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} \qquad + \underbrace{ 2F_\mathsf{KL}\sqrt{S \log\frac{1}{p_2}} }_{\substack{ \text{(III), w.p. 1-2p_2} \\ \text{(\Cref{claim:deviation-of-smoothed-function})} }} \\ &\qquad + \underbrace{ \frac{2D_\Theta^2}{\eta_\mathsf{LM}} + 2 \eta_\mathsf{LM}d_\Theta L_\mathsf{KL}^2 S + \frac{2 D_\Theta L_\mathsf{KL}\epsilon_\mathsf{KL}d_\Theta S}{\delta_\mathsf{LM}} }_{\substack{ \text{(IV), w.p. 1-2 S p_\mathsf{KL}} \\ \text{(Similar argument as \Cref{lemma:high-probability-expected-suboptimality-under-smoothing})} }} \qquad + \underbrace{ 2\delta_\mathsf{LM}L_\mathsf{KL}S }_{\substack{ \text{(V), w.p. 1} \\ \text{(\Cref{claim:regret-from-smoothing-over-sphere-or-ball})} }} \qquad + \underbrace{ 2\delta_{\mathsf{LM}} D_\Theta L_\mathsf{KL}S }_{\substack{ \text{(VI), w.p. 1} \\ \text{(\Cref{claim:regret-from-projection})} }} \end{align}\] Similar to 2, we set \(\delta_\mathsf{LM}= \sqrt{\epsilon_\mathsf{KL}d_\Theta}\), \(\eta_\mathsf{LM}= 1/\sqrt{d_\Theta S}\). For any \(p_2 = p'/2 >0\), it yields \[\begin{align} R_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL}) = O\left( \sqrt{d_\Theta S} + \sqrt{\epsilon_\mathsf{KL}d_\theta}S + \sqrt{S \log\frac{1}{p}} \right) \end{align}\] with probability \(1 - p' - 2S p_\mathsf{KL}\) > 0. ◻
We start with leveraging [theorem:regret-bound-for-indirectly-convex-functions] to show the following convergence guarantee for \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) (2).
Claim 9 (Convergence of \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\)). Given any \(\epsilon, p > 0\), \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) outputs an \(\epsilon\)-suboptimal solution for \(\mathsf{PR}(\theta)\) with probability at least \(1-p\). Moreover, \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) runs for \(T = O(d_\Phi/\epsilon^2)\) steps and performs \(O(d_\Phi/\epsilon^2)\) queries to \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\), as well as \(O(d_\Phi/\epsilon^2)\) queries to \(\operatorname{\mathsf{LearnModel}}\) with \(\epsilon_\mathsf{LM}= O(\epsilon^2)\) and \(p_\mathsf{LM}= O(\epsilon^2 p/d_\Phi)\).
Proof of 9. Choosing \(\epsilon_\mathsf{LM}= 1/T\), \(p_\mathsf{LM}= p/2T\), and \(p' = p/2\), [theorem:regret-bound-for-indirectly-convex-functions] shows that \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) satisfies \[\begin{align} \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}},\mathsf{PR}) = O\left(\sqrt{d_\Phi T}\right) \end{align}\] with probability \(1 - p\), using \(2T\) queries to \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) and \(2T\) queries to \(\operatorname{\mathsf{LearnModel}}\). By 1, \(T = O(d_\Phi / \epsilon^2)\) steps suffice to output a model that is \(\epsilon\)-suboptimal with respect to \(\mathsf{PR}\). Plugging in this bound on \(T\) into the expressions for \(\epsilon_\mathsf{LM}\) and \(p_\mathsf{LM}\) above yields the result. ◻
Similarly, we have the convergence guarantee for \(\operatorname{\mathsf{LearnModel}}\) as well:
Claim 10 (Convergence of \(\operatorname{\mathsf{LearnModel}}\)). Given any \(\phi \in \Phi\) and \(\epsilon_\mathsf{LM}, p_\mathsf{LM}> 0\), \(\operatorname{\mathsf{LearnModel}}\) outputs an \(\epsilon_\mathsf{LM}\)-suboptimal model for 3 with probability at least \(1 - p_\mathsf{LM}\). Moreover, \(\operatorname{\mathsf{LearnModel}}\) runs for \(S = O(d_\Theta/\epsilon_\mathsf{LM}^2)\) steps and performs two queries to \(\operatorname{\mathsf{EstimateKL}}\) per step with \(N_\mathsf{KL}( \frac{\epsilon^2_\mathsf{LM}}{d_\theta},\frac{\epsilon^2_\mathsf{LM}p_\mathsf{LM}}{4d_\theta})\) samples per query.
Proof of 10. Choosing \(\epsilon_\mathsf{KL}= 1/S\), \(p_\mathsf{KL}= p_\mathsf{LM}/4S\) and \(p' = p_\mathsf{LM}/2\), [theorem:regret-bound-for-learnmodel] shows that \(\operatorname{\mathsf{LearnModel}}\) satisfies \[\begin{align} \mathcal{R}_S(\operatorname{\mathsf{LearnModel}},\mathsf{KL}) = O\left( \sqrt{d_\Phi S} \right) \end{align}\] By 1, \(S = O(d_\Theta / \epsilon_\mathsf{LM}^2)\) steps suffice to output a model that is \(\epsilon_\mathsf{LM}\)-suboptimal with respect to \(\mathsf{KL}\); thus we have \(\epsilon_\mathsf{KL}= \frac{\epsilon_\mathsf{LM}^2}{d_\Theta}\), \(p_\mathsf{KL}= \frac{1}{4Sp_\mathsf{LM}}\). In total, \(\operatorname{\mathsf{LearnModel}}\) makes \(2S\) queries to \(\operatorname{\mathsf{EstimateKL}}\) with \(N_\mathsf{KL}(\frac{\epsilon_\mathsf{LM}^2}{d_\Theta}, \frac{\epsilon^2_\mathsf{LM}p_\mathsf{LM}}{4d_\theta})\) samples per query. ◻
Now are are ready to prove [theorem:total-regret]. Recall the theorem statement:
Proof of [theorem:total-regret]. Let \(T\) be the number of steps executed by \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\), and \(S\) the number of steps in \(\operatorname{\mathsf{LearnModel}}\). Let \(N_\mathsf{KL}(\epsilon_\mathsf{KL}, p_\mathsf{KL})\) (or \(N_\mathsf{KL}\) for short) denote the number of samples used by \(\operatorname{\mathsf{EstimateKL}}(\cdot, \cdots, \epsilon_\mathsf{KL}, p_\mathsf{KL})\). Since \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\) calls \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) and \(\operatorname{\mathsf{LearnModel}}\) \(2T\) times, and \(\operatorname{\mathsf{LearnModel}}\) calls \(\operatorname{\mathsf{EstimateKL}}\) \(2S\) times, the overall number of samples is \(N = 2(2N_\mathsf{KL}S + 1)T\).
Let \(\theta_{t,s}^+, \theta_{t,s}^-\) denote the models deployed by \(\operatorname{\mathsf{EstimateKL}}\) in the \(s\)-th step of \(\operatorname{\mathsf{LearnModel}}\) within the \(t\)-th step of \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\), obtaining samples \(z_{t,s,1}^+,\ldots,z_{t,s,N_\mathsf{KL}}^+\) and \(z_{t,s,1}^-,\ldots,z_{t,s,N_\mathsf{KL}}^-\), respectively. Similarly, let \(\hat{\theta}_t^+, \hat{\theta}_t^-\) denote the models deployed by \(\operatorname{\mathsf{Estimate}\mathsf{PR}}\) in the \(t\)-th step of \(\operatorname{\mathsf{Minimize}\mathsf{PR}}\), obtaining samples \(\hat{z}_t^+, \hat{z}_t^-\).
The total regret can be written as \[\begin{align} &\quad \mathcal{R}_N(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) \\ &= \sum_{t=1}^T \left( \ell(\hat{z}_t^+; \hat{\theta}_t^+) + \ell(\hat{z}_t^-; \hat{\theta}_t^-) - 2\mathsf{PR}(\theta^*) + \sum_{s=1}^S \sum_{i=1}^{N_\mathsf{KL}} \left[ \ell(z_{t,s,i}^+; \theta_{t,s}^+) + \ell(z_{t,s,i}^-; \theta_{t,s}^-) - 2\mathsf{PR}(\theta^*) \right] \right) \\ &= \underbrace{ \sum_{t=1}^T \left( \ell(\hat{z}_t^+; \hat{\theta}_t^+) - \mathsf{PR}(\hat{\theta}_t^+) + \ell(\hat{z}_t^-; \hat{\theta}_t^-) - \mathsf{PR}(\hat{\theta}_t^-) + \sum_{s=1}^S \sum_{i=1}^{N_\mathsf{KL}} \left[ \ell(z_{t,s,i}^+; \theta_{t,s}^+) - \mathsf{PR}(\theta_{t,s}^+) + \ell(z_{t,s,i}^-; \theta_{t,s}^-) - \mathsf{PR}(\theta_{t,s}^-) \right] \right) }_{n \text{ difference terms with expectation zero}} \\ &\qquad \qquad + \sum_{t=1}^T \left( \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-) - 2\mathsf{PR}(\theta^*) + \sum_{s=1}^S \sum_{i=1}^{N_\mathsf{KL}} \left[ \mathsf{PR}(\theta_{t,s}^+) + \mathsf{PR}(\theta_{t,s}^-) - 2\mathsf{PR}(\theta^*) \right] \right) \\ &= O\left(\sqrt{N}\right) + \sum_{t=1}^T \left( \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-) - 2\mathsf{PR}(\theta^*) + \sum_{s=1}^S \sum_{i=1}^{N_\mathsf{KL}} \left[ \mathsf{PR}(\theta_{t,s}^+) + \mathsf{PR}(\theta_{t,s}^-) - 2\mathsf{PR}(\theta^*) \right] \right)\\ &= O\left(\sqrt{N}\right) + \sum_{t=1}^T \left[ \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-) - 2\mathsf{PR}(\theta^*) \right] \\ &\qquad \qquad + N_\mathsf{KL}\cdot \sum_{t=1}^T \sum_{s=1}^S \left[ (\mathsf{PR}(\theta_{t,s}^+) + \mathsf{PR}(\theta_{t,s}^-)) - (\mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-)) + (\mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-)) - 2\mathsf{PR}(\theta^*) \right] \\ &= O\left(\sqrt{N}\right) + (N_\mathsf{KL}S + 1) \sum_{t=1}^T \left[ \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\hat{\theta}_t^-) - 2\mathsf{PR}(\theta^*) \right]\\ &\qquad \qquad + N_\mathsf{KL}\cdot \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}(\theta_{t,s}^+) - \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\theta_{t,s}^-) - \mathsf{PR}(\hat{\theta}_t^-) \right] \\ &= O\left(\sqrt{N}\right) + (N_\mathsf{KL}S + 1) \cdot \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) + N_\mathsf{KL}\cdot \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}(\theta_{t,s}^+) - \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}(\theta_{t,s}^-) - \mathsf{PR}(\hat{\theta}_t^-) \right] \\ &= O\left(\sqrt{N}\right) + (N_\mathsf{KL}S + 1) \cdot \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) + N_\mathsf{KL}\cdot \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\varphi(\theta_{t,s}^+)) - \mathsf{PR}(\hat{\theta}_t^+) + \mathsf{PR}^\dagger(\varphi(\theta_{t,s}^-)) - \mathsf{PR}(\hat{\theta}_t^-) \right] \\ &= O\left(\sqrt{N}\right) + (N_\mathsf{KL}S + 1) \cdot \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) \\ &\qquad \qquad + N_\mathsf{KL}\cdot \underbrace{ \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\varphi(\theta_{t,s}^+)) - \mathsf{PR}^\dagger(\phi_t^+) \right] }_{\text{(I)}} + N_\mathsf{KL}\cdot \underbrace{ \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] }_{\text{(II)}} \\ &\qquad \qquad + N_\mathsf{KL}\cdot \underbrace{ \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\varphi(\theta_{t,s}^-)) - \mathsf{PR}^\dagger(\phi_t^-) \right] }_{\text{(III)}} + N_\mathsf{KL}\cdot \underbrace{ \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\phi_t^-) - \mathsf{PR}(\hat{\theta}_t^-) \right] }_{\text{(IV)}} \end{align}\]
Term (I) is: \[\begin{align} \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\varphi(\theta_{t,s}^+)) - \mathsf{PR}^\dagger(\phi_t^+) \right] &\leq L^\dagger \cdot \sum_{t=1}^T \sum_{s=1}^S \left\|\varphi(\theta_{t,s}^+) - \phi_t^+\right\| ^\dagger} \\ &\leq L^\dagger \cdot \sum_{t=1}^T \sqrt{ S \sum_{s=1}^S \left(\left\|\varphi(\theta_{t,s}^+) - \phi_t^+\right\|^2 \right)}\\ &= L^\dagger T \sqrt{S \sum_{s=1}^S L_{\theta}^2 {\mathsf{KL}(\phi_t^+ || \varphi(\theta_{t,s}^+))}} } \\ &\leq L^\dagger L_\theta T \cdot \sqrt{S \cdot \mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL}) } \end{align}\]
and term (III) is analogous. Term (II) is \[\begin{align} \sum_{t=1}^T \sum_{s=1}^S \left[ \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}(\hat{\theta}_t^+) \right] &= S \cdot \sum_{t=1}^T \left[ \mathsf{PR}^\dagger(\phi_t^+) - \mathsf{PR}^\dagger(\varphi(\hat{\theta}_t^+)) \right] \\ &\leq L^\dagger S \cdot \sum_{t=1}^T \left\|\phi_t^+ - \varphi(\hat{\theta}_t^+)\right\| ^\dagger} \\ &\leq L^\dagger S \cdot \sum_{t=1}^T L_\theta \sqrt{\mathsf{KL}(\phi_t^+ ||\varphi(\hat{\theta}_t^+))}}\\ &\leq L^\dagger L_\theta \cdot S \cdot \sum_{t=1}^T \sqrt{ \frac{1}{S} \sum_{s=1}^S \mathsf{KL}(\phi_t^+||\varphi(\theta_{t,s}^+)) } _t^+ := \frac{1}{S}\sum_{s=1}^S \theta_{t,s}^+, convexity of \mathsf{KL}(\phi^+_t||\varphi(\theta))} \\ &\leq L^\dagger L_\phi T S \sqrt{\frac{1}{S}\mathcal{R}_S(\operatorname{\mathsf{LearnModel}},\mathsf{KL}) }\\ &= L^\dagger L_\phi T\sqrt{S\cdot\mathcal{R}_S(\operatorname{\mathsf{LearnModel}},\mathsf{KL})} \end{align}\]
and term (IV) is analogous. In total we have \[\begin{align} & \mathcal{R}_N(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) \\ &= O\left( \sqrt{N} + N_\mathsf{KL}T \cdot \sqrt{S \cdot \mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL})} + (N_\mathsf{KL}S + 1) \cdot \mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) \right) \\ &= N \cdot O\left( \frac{1}{\sqrt{N}} + \sqrt{\frac{\mathcal{R}_S(\operatorname{\mathsf{LearnModel}}, \mathsf{KL})}{S}} + \frac{\mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR})}{T} \right) 2S + 1)T} \\ &= N \cdot O\left( \frac{1}{\sqrt{N}} + \sqrt{ \sqrt{\frac{d_\Theta \log\frac{1}{p'}}{S}} + \sqrt{\epsilon_\mathsf{KL}d_\Theta} } + \sqrt{\frac{d_\Phi \log\frac{1}{p''}}{T}} + \sqrt{\epsilon_\mathsf{LM}d_\Phi} \right) ,\Cref{theorem:regret-bound-for-learnmodel}, w.p. to be analyzed later} \\ &= N \cdot O\left( \left(\frac{d_\Theta}{S}\log\frac{1}{p'}\right)^{1/4} + (\epsilon_\mathsf{KL}d_\Theta)^{1/4} + \left(\frac{d_\Phi}{T}\log\frac{1}{p''}\right)^{1/2} + (\epsilon_\mathsf{LM}d_\Phi)^{1/2} \right)\leq \sqrt{a} + \sqrt{b}; \frac{1}{\sqrt{n}} \leq \sqrt{\frac{d_\Phi}{T}}} \\ &\leq N \cdot \left(1 + \left(\log\frac{1}{p'}\right)^{1/4} + \left(\log\frac{1}{p''}\right)^{1/2}\right) \cdot O\left( \left(\frac{d_\Theta}{S}\right)^{1/4} + (\epsilon_\mathsf{KL}d_\Theta)^{1/4} + \left(\frac{d_\Phi}{T}\right)^{1/2} + (\epsilon_\mathsf{LM}d_\Phi)^{1/2} \right) \\ &= N \cdot \left(1 + \left(\log\frac{1}{p'}\right)^{1/4} + \left(\log\frac{1}{p''}\right)^{1/2}\right) \cdot O\left( \left(\frac{d_\Theta}{S}\right)^{1/4} + (\epsilon_\mathsf{KL}d_\Theta)^{1/4} + \left(\frac{d_\Phi N_\mathsf{KL}S}{N}\right)^{1/2} + (\epsilon_\mathsf{LM}d_\Phi)^{1/2} \right) {N_\mathsf{KL}S + 1}} \end{align}\]
Choose \(\epsilon_\mathsf{LM}= \left(\frac{N_\mathsf{KL}}{N}\right)^{1/3}\) and \(\epsilon_\mathsf{KL}= \frac{1}{4d_\Theta}\left(\frac{N_\mathsf{KL}}{N}\right)^{2/3}\).
To balance the terms, set the number of steps for the outer algorithm to be \(T = \frac{d_\Phi}{(\epsilon - \sqrt{\epsilon_\mathsf{LM}d_\Phi})^2}\), and the number of steps in \(\operatorname{\mathsf{LearnModel}}\) to be \[\begin{align} S = \frac{d_\Theta}{\left( \epsilon_\mathsf{LM} - \sqrt{\epsilon_\mathsf{KL}d_\Theta} \right)^2} = 4d_\Theta \left(\frac{N}{N_\mathsf{KL}}\right)^{2/3} \end{align}\]
Plugging these expressions for \(\epsilon_\mathsf{KL}\), \(\epsilon_\mathsf{LM}\), and \(S\) in above, we have \[\begin{align} \mathcal{R}_n(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) &= N \cdot \left(1 + \left(\log\frac{1}{p'}\right)^{1/4} + \left(\log\frac{1}{p''}\right)^{1/2}\right) \cdot O\left( (d_\Theta d_\Phi)^{1/2} \left(\frac{N_\mathsf{KL}}{N}\right)^{1/6} \right) \\ &= O\left( \left(1 + \left(\log\frac{1}{p'}\right)^{1/4} + \left(\log\frac{1}{p''}\right)^{1/2}\right) (d_\Theta + d_\Phi) N_\mathsf{KL}^{1/6} N^{5/6} \right) \end{align}\]
We would like to ensure that this bound holds with probability \(p>0\). To that end, observe that the probabilistic terms are the high-probability bounds on \(\mathcal{R}_S(\operatorname{\mathsf{LearnModel}},\mathsf{KL})\) and \(\mathcal{R}_T(\operatorname{\mathsf{Minimize}\mathsf{PR}},\mathsf{PR})\). By recalling [theorem:regret-bound-for-indirectly-convex-functions] and [theorem:regret-bound-for-learnmodel], the probability that any of these bounds fails is at most \[\begin{align} p' + Tp_\mathsf{LM} = p' + T(p'' + S p_\mathsf{KL}) = p' + Tp'' + STp_\mathsf{KL} \end{align}\] for any \(p', p'' > 0\). For a choice of \(p' = p/3\), \(p'' = p/3T\), and \(p_\mathsf{KL}= \frac{p N_\mathsf{KL}}{3n}\), this is at most \(p\) as required. Finally, plugging these choices into the above regret bound yields \[\begin{align} \mathcal{R}_n(\operatorname{\mathsf{Minimize}\mathsf{PR}}, \mathsf{PR}) &= O\left( \left(1 + \left(\log\frac{1}{p'}\right)^{1/4} + \left(\log\frac{1}{p''}\right)^{1/2}\right) (d_\Theta + d_\Phi) N_\mathsf{KL}^{1/6} N^{5/6} \right) \\ &= O\left( \left(1 + \left(\log\frac{1}{p}\right)^{1/4} + \left(\log\frac{T}{p}\right)^{1/2}\right) (d_\Theta + d_\Phi) N_\mathsf{KL}^{1/6} N^{5/6} \right) \\ &= O\left( \left(1 + \sqrt{\log\frac{1}{p}}\right) (d_\Theta + d_\Phi) N_\mathsf{KL}^{1/6} N^{5/6} \sqrt{\log N} \right) \end{align}\] with probability at most \(p\) as required. ◻
In this section, we provide additional related work in performative prediction and a detailed comparison of our work and some closely related work.
Performative prediction is a new type of supervised learning problem in which the underlying data distribution shifts in response to the deployed model [1], [7], [8], [10], [12], [24]–[26]. It is also called the decision-dependent risk minimization problem [12], [27], [28]. In particular, [1] first propose the notion of the performative risk defined as \(\mathsf{PR}(\theta):= \mathbb{E}_{z \sim \mathcal{D}(\theta)}[\ell(\theta;z)]\) where \(\theta\) is the model parameter, and \(\mathcal{D}(\theta)\) is the induced distribution due to the deployment of \(\theta\).
One of the major focuses of performative prediction is to find the optimal model \(\theta_{\textsf{OPT}}\) which achieves the minimum performative prediction risk: \(\theta_{\textsf{OPT}}:= \mathop{\mathrm{\arg\!\min}}_{\theta\in \Theta} \mathsf{PR}(\theta)\), or performative stable model \(\theta_{\textsf{ST}}\), which is optimal under its own induced distribution: \(\theta_{\textsf{ST}}:= \mathop{\mathrm{\arg\!\min}}_{\theta\in \Theta}\mathbb{E}_{z \sim \mathcal{D}(\theta_{\textsf{ST}})}[\ell(\theta;z)]\) . In particular, one way to find a performative stable model \(\theta_{\textsf{ST}}\) is to perform repeated retraining [1].
In order to get meaningful theoretical guarantees on any proposed algorithms, works in this field generally require particular assumptions on the mapping between the model parameter and its induced distribution (e.g., the smoothness of the mapping), or require multiple rounds of deployments and observing the corresponding induced distributions, which can be costly in practice [6], [11]. A few recent works are on finding performative optimal solutions without explicitly making the convexity assumption. For example, [13] does not explicitly convexity assumption, but they focus on optimization heuristics that are not guaranteed to minimize performative regret. below, we will provide the discussions for three of them.
In addition, minimizing the performative risk often requires knowing a specific model for the distribution map \(\mathcal{D}(\cdot)\) that can be fit. To ensure performative risk minimization is tractable, one also requires imposing structural assumptions on the distribution map. For example, [10] makes parametric assumptions on \(\mathcal{D}(\theta)\) and assumes that \(\mathcal{D}(\theta)\) has a continuously differentiable density \(p(z; \varphi(\theta))\), where \(\varphi(\cdot): \Theta \rightarrow \Phi\) represents the mapping from the model parameter space \(\Theta\) to the data distribution parameter space \(\Theta\). [9] assume the underlying data distribution follows a location family distribution, and then impose a mixture dominance assumption on the distribution map \(\mathcal{D}(\cdot)\) from which it follows that \(\mathsf{PR}(\theta)\) is convex; this again leads to a gradient-based optimization algorithm. Similar work include [6], [7], [10], [14], to name a few.
[9] identifies mixture dominance condition for any particular model parameter pairs under which the performative risk is convexity. In particular, they posit a simple distribution map in which \(\varphi(\theta) = \phi_0 + M\theta\), where \(M \in \mathbb{R}^{d_\Phi \times d_\Theta}\) and \(\phi_0 \in \Phi\) is some “base” distribution parameter; in other words, they assume that the data population reacts to a model by shifting each of their features according to some linear transformation of the model parameter. Their algorithm for this special case works in two stages: first estimating \(\varphi_0\) and \(M\) by deploying random models; then, once this distribution map has been accurately estimated, the performative loss is convex in \(\theta\), and can be optimized offline. The distribution map estimation takes \(O(d_\Theta/\epsilon)\) samples to obtain an \(\epsilon\)-suboptimal model.
For example, closely related is a recent paper that proposes using the Lipschitz bandit approach to solve the performative prediction problem [11]. The major differences between this work and their work are: first, we define the regret w.r.t \(N\) rather than w.r.t \(T\), which is a more realistic measure in the performative prediction setting; second, their regret has exponential dependency on the “zooming dimension” \(d\) (which is roughly the model parameter \(d_\Theta\)), while our dependency on the model and distribution dimensions are both linear.
Another closely related work is [12] uses zeroth-order methods for the convex-concave minimax problem. Specifically, they proposed to formulate the performative prediction problem as the Wasserstein distributionally robust learning with decision-dependent data problem, and further reduce it to a constrained finite-dimensional smooth convex-concave min-max problem, and propose a zeroth-order random reshuffling-based algorithm to solve the problem without assuming any other structure on the curvature of the min-max loss. Similar to ours, they also use the zeroth-order method to perform their optimization procedure; different from ours, they approach the performative prediction problem through the angle of robustness, which accounts for model misspecification in their analysis.
Also related are the recently developed lines of work on the multiplayer version of the performative prediction problem [15], [27], [29], [30]. While existing strategic classification and performative prediction problems focus primarily on the interplay between a single learner and the population that reacts to the learner’s actions, this line of work takes into account competition from multi-learners, and develop performatively stable equilibria and Nash equilibria of the game. Similarly, [28] confront multiple interactive models in some dynamic environments. Another line of work is the economic aspects of performative prediction [31], [32]. From the optimization aspect, [33] focuses on the optimization aspect of finding the performative optimal point and offers an online stochastic primal-dual algorithm for tracking equilibrium trajectories. Also related is the recent development of the concept called induced domain adaptation [2], whose primary focus is to study the transferability of a particular model trained primarily on the source distribution and provide theoretical bounds on its performance on its induced distribution, which is helpful in estimating the effect of a given classifier when repeated retraining is unavailable.
When the problem reduces to a convex, differentiable Lipschitz case, our algorithm will be reduced to the convex case provided in 2.2 (the warm-up setting), which achieves a \(\tilde{O}(\sqrt{d N})\) regret bound. This implies that our algorithm achieves a \(\Delta\)-suboptimal model with \(O(d/\Delta^2)\) samples (see [lemma:regret-bound-for-model-convex-performative-risk]).
Here, we compare the three papers that the reviewer mentioned:
[10] focus on a single-distribution Gaussian distribution with a fixed variance setting while we cover a boarder range of settings. Their theoretical guarantee shows that the proposed method converges to a performative optimal point as the number of iterations \(T\approx \sigma^{-4/5}\) where \(\sigma\) bounds the output of PR from the PR of the optimal performative point. The sample required at each iteration is \(O(1/\sigma^2\log T)\).
[9] show that when the distribution maps \(\mathcal{D}(\cdot)\) form a location-scale family and when the model dimension is \(O(d)\), computing a \(\Delta\)-suboptimal classifier requires \(O(d/\delta)\) samples. We do not require these assumptions.
[1] focus on achieving a performative stable point while we focus on attaining a performative optimal point.
Figure 7: Empirical results comparing baseline method (zeroth-order optimization without reparametrization, orange curve) vs. our method (zeroth order optimization after reparametrization, (blue curve) based on 3.. a — \(\theta\in[0.2, 0.4]\), b — \(\theta\in[0.4, 0.8]\), c — \(\theta\in[0.4, 0.8]\)
\(\widetilde{O}(\cdot)\) suppresses polylogarithmic factors in \(N\) and the failure probability \(1/p\).↩︎
Samples that the learner deploys along the way of finding the performative optimal model.↩︎
One may wonder how to find the optimal \(\vartheta^*(\phi)\) when there are two model parameters \(\theta\) and \(\theta^{\prime}\) that realize the same \(\phi\) (i.e., \(\varphi(\theta)=\varphi\left(\theta^{\prime}\right)\) and \(\mathsf{PR}\left(\theta^{\prime}\right) \geq \mathsf{PR}(\theta))\). Recall that the objective function for \(\operatorname{\mathsf{LearnModel}}\) is to find any model \(\theta\) that leads to the particular target data parameter \(\phi\) such that \(\varphi(\theta)=\phi\). It is quite possible that multiple models can induce the same target data parameter; however, since the goal is to find any one of them, having multiple model parameters won’t be an issue – in fact, it can only help speed up the process↩︎
We can also verify that 3 does not satisfy the mixture dominance assumption.↩︎
The sample required for each round of estimate EstimatePR is \(\mathcal{O}(1)\); this is because EstimatePR itself is an unbiased estimator, so even with one sample, in expectation, the estimation will be unbiased.↩︎
For notation simplicity, here, we use \(\mathsf{KL}(\phi_1|| \phi_2)\) to represent \(\mathsf{KL}(\mathcal{D}_1|| \mathcal{D}_2)\) where the data distribution \(\mathcal{D}_1\) and \(\mathcal{D}_2\) has the parameter \(\phi_1\) and \(\phi_2\), respectively.↩︎
A distribution \(\mathcal{D}(\theta)\) with the density function \(p(\cdot | \varphi(\theta))\) parameterized by \(\theta \in \Theta\) is called \((\ell_2, K)\)-Lipschitz continuous [21] if for all \(z\) in the sample space, the log-likelihood \(f(\theta) = \log p(z|\varphi(\theta))\) is Lipschitz continuous with respect to the \(\ell_2\) norm of \(\theta\) with constant \(K(z)\).↩︎