This paper presents probably approximately correct (PAC)-Bayesian bounds on variational Bayesian (VB) approximations of fractional or tempered posterior distributions for Markov data generation models. Exact computation of either standard or tempered posterior distributions is a hard problem that has, broadly speaking, spawned two classes of computational methods. The first, Markov chain Monte Carlo (MCMC), constructs ergodic Markov chains to approximately sample from the posterior distribution. MCMC is known to suffer from high variance and complex diagnostics, leading to the development of variational Bayesian (VB) [1] methods as an alternative in recent years. VB methods pose posterior computation as a variational optimization problem, approximating the posterior distribution of interest by the 'closest' element of an appropriately defined class of 'simple' probability measures. Typically, the measure of closeness used by VB methods is the Kullback-Leibler (KL) divergence. Excellent introductions to this so-called KL-VB method can be found in [2][3][4]. More recently, there has also been interest in alternative divergence measures, particularly the α-Rényi divergence [5][6][7], though in this paper, we focus on the KL-VB setting.
Theoretical properties of VB approximations, and in particular asymptotic frequentist consistency, have been studied extensively under the assumption of an independent and identically distributed (i.i.d.) data generation model [4,8,9]. On the other hand, the common setting where data sets display temporal dependencies presents unique challenges. In this paper, we focus on homogeneous Markov chains with parameterized transition kernels, representing a parsimonious class of data generation models with a wide range of applications. We work in the Bayesian framework, focusing on the posterior distribution over the unknown parameters of the transition kernel. Our theory develops PAC bounds that link the ergodic and mixing properties of the data generating Markov chain to the Bayes risk associated with approximate posterior distributions.
Frequentist consistency of Bayesian methods, in the sense of concentration of the posterior distribution around neighborhoods of the 'true' data generating distribution, have been established in significant generality, in both the i.i.d. [10][11][12] and in the non-i.i.d. data generation setting [13,14]. More recent work [14][15][16] has studied fractional or tempered posteriors, a class of generalized Bayesian posteriors obtained by combining the likelihood function raised to a fractional power with an appropriate prior distribution using Bayes' theorem. Tempered posteriors are known to be robust against model misspecification: in the Markov setting we consider, the associated stationary distribution as well as mixing properties are sensitive to model parameterization. Further, tempered posteriors are known to be much simpler to analyze theoretically [14,16]. Therefore, following [14][15][16] we focus on tempered posterior distributions on the transition kernel parameters, and study the rate of concentration of variational approximations to the tempered posterior. Equivalently, as shown in [16] and discussed in Section 1.1, our results also apply to so-called α-variational approximations to standard posterior distributions over kernel parameters. The latter are modifications of the standard KL-VB algorithm to address the well-known problem of overconfident posterior approximations.
While there have been a number of recent papers studying the consistency of approximate variational posteriors [5,8,15] in the large sample limit, rates of convergence have received less attention. Exceptions include [9,15,17], where an i.i.d. data generation model is assumed. [15] establishes PAC-Bayes bounds on the convergence of a variational tempered posterior with fractional powers in the range [0, 1), while [9] considers the standard variational posterior case (where the fractional power equals 1). [17], on the other hand, establishes PAC-Bayes bounds for risk-sensitive Bayesian decision making problems in the standard variational posterior setting. The setting in [15] allows for model misspecification and the analysis is generally more straightforward than that in [9,17]. Our work extends [15] to the setting of a discrete-time Markov data generation model.
Our first results in Theorem 1 and Corollary 1 of Section 2 establish PAC-Bayes bounds for sequences with arbitrary temporal dependence. Our resultsgeneralize [15], [Theorem 2.4] to the non-i.i.d. data setting in a straightforward manner. Note that Theorem 1 also recovers ( [16], [Theorem 3.3]), which is established under different 'existence of test' conditions. Our objective in this paper is to explicate how the ergodic and mixing properties of the Markov data generating process influences the PAC-Bayes bound. The sufficient conditions of our theorem, bounding the mean and variance of the log-likelihood ratio of the data, allows for developing this understanding, without the technicalities of proving the existence of test conditions intruding on the insights.
In Section 3, we study the setting where the data generating model is a stationary α-mixing Markov chain. Stationarity means that the Markov chain is initialized with the invariant distribution corresponding to the parameterized transition kernel, implying all subsequent states also follow this marginal distribution. The α-mixing condition ensures that the variance of the likelihood ratio of the Markov data does not grow faster than linear in the sample size. Our main results in this setting are applicable when the state space of the Markov chain is either continuous or discrete. The primary requirement on the class of data generating Markov models is for the log-likelihood ratio of the parameterized transition kernel and invariant distribution to satisfy a Lipschitz property. This condition implies a decoupling between the model parameters and the random samples, affording a straightforward verification of the mean and variance bounds. We highlight this main result by demonstrating that it is satisfied by a finite state Markov chain, a birth-death Markov chain on the positive integers, and a one-dimensional Gaussian linear model.
In practice, the assumption that the data generating model is stationary is unlikely to be satisfied. Typically, the initial distribution is arbitrary, with the state distribution of the Markov sequence converging weakly to the stationary distribution. In this setting, we must further assume that the class of data generating Markov chains are geometrically ergodic.
We show that this implies the boundedness of the mean and variance of the log-likelihood ratio of the data generating Markov chain. Alternatively, in Theorem 4 we directly impose a drift condition on random variables that bound the log-likelihood ratio. Again, in this more general nonstationary setting, we illustrate the main results by showing that the PAC-Bayes bound is satisfied by a finite state Markov chain, a birth-death Markov chain on the positive integers, and a one-dimensional Gaussian linear model.
In preparation for our main technical results starting in Section 2 we first note relevant notations and definitions in the next section.
We broadly adopt the notation in [15]. Let the sequence of random variables
, where θ 0 ∈ Θ ⊆ R d is the 'true' parameter underlying the data generation process. We assume the state space S ⊆ R m of the random variables X i is either discrete-valued or continuous, and write {x 0 , . . . , x n } for a realization of the dataset. We also adopt the convention that 0 log(0/0) = 0.
For each θ ∈ Θ, we will write p
θ as the probability density of
with respect to some measure Q (n) , i.e., p
, where Q (n) is either Lebesgue measure or the counting measure. Unless stated otherwise, all probabilities, expectations and variances, which we represent as P, E[X] and Var[X], are with respect to the true distribution
Let π(θ) be a prior distribution with support Θ. The α te -fractional posterior is defined as
where, for θ 0 , θ ∈ Θ, r n (θ, θ 0 )(•) := log
, is the log-likelihood ratio of the corresponding density functions, and α te ∈ (0, ∞) is a tempering coefficient. Setting α te = 1 recovers the standard Bayesian posterior. Note that we will use superscripts to distinguish different quantities that are referred to just as α in the literature. The Kullback-Leibler (KL) divergence between distributions P, Q is defined as
where p, q are the densities corresponding to P, Q on some sample space X . In particular, the KL divergence between the distributions parameterized by θ 0 and θ is
The α re -Rényi divergence D α re (P
where α re ∈ (0, 1). As α re → 1, the α re -Rényi divergence recovers the KL divergence. Let F be some class of distributions with support in R d and such that any distribution P in F is absolutely continuous with respect to the tempered posterior: P ≪ π n,α te |X n .
Many choices of F exist; for instance (see also [15]), F can be the set of Gaussian measures, denoted F Φ id :
where P.D. references the class of positive definite matrices. Alternately, F can be the family of mean-field or factored distributions where the components θ i of θ are independent of each other. Let πn,α te |X n be the variational approximation to the tempered posterior, defined as πn,α te |X n := arg min
It is easy to see that finding πn,α te |X n in Equation ( 5) is equivalent to the following optimization problem:
Setting α te = 1 again recovers the usual variational solution that seeks to approximate the posterior distribution with the closest element of F (the right-hand side above is called the evidence lower bound (ELBO)). Other settings of α te constitute α te -variational inference [16], which seeks to regularize the 'overconfident' approximate posteriors that standard variational methods tend to produce. Our results in this paper focus on parametrized Markov chains. We term a Markov chain as 'parameterized' if the transition kernel p θ (•|•) is parametrized by some θ ∈ Θ ⊆ R d . Let q (0) (•) be the initial density (defined with respect to the Lebesgue measure over R m ) or initial probability mass function. Then, the joint density is p
θ (x 0 , . . . , x n ) corresponds to the walk probability of a time-homogeneous Markov chain. We assume that corresponding to each transition kernel p θ , θ ∈ Θ, there exists an invariant distribution q (∞) θ ≡ q θ that satisfies
We will also use q θ to designate the density of the invariant measure (as before, this is with respect to the Lebesgue or counting measure for continuous or discrete state spaces, respectively). A Markov chain is stationary if its initial distribution is the invariant probability distribution, that is, X 0 ∼ q θ .
Our results in the ensuing sections will be established under strong mixing conditions [18] on the Markov chain. Specifically, recall the definition of the α-mixing coefficients of a Markov chain {X n }: Definition 1 (α-mixing coefficient). Let M j i denote the σ-field generated by the Markov chain {X k : i ≤ k ≤ j} parameterized by θ ∈ Θ. Then, the α-mixing coefficient is defined as
Informally speaking, the α-mixing coefficients {α k } measure the dependence between any two events A (in the 'history' σ-algebra) and B (in the 'future' σ-algebra) with a time lag k. We note that we do not use superscripts to identify these α parameters, since they are the only ones with subscripts, and can be identified through this.
The object of analysis in what follows is the probability measure πn,α te |X n (θ), the variational approximation to the tempered posterior. Our main result establishes a bound on the Bayes risk of this distribution; in particular, given a sequence of loss functions ℓ n (θ, θ 0 ), we bound ℓ n (θ, θ 0 ) πn,α te |X n (θ)dθ. Following recent work in both the i.i.d. and dependent sequence settings [14][15][16], we will use ℓ n (θ,
θ and P (n) θ 0 as our loss function. Unlike loss functions like Euclidean distance, Rényi divergence compares θ and θ 0 through their effect on observed sequences, so that issues like parameter identifiability no longer arise. Our first result generalizes [15], [Theorem 2.1] to a general non-i.i.d. data setting.
Proposition 1. Let F be a subset of all probability distributions on Θ. For any α re ∈ (0, 1), ǫ ∈ (0, 1) and n ≥ 1, the following probabilistic uniform upper bound on the expected α re -Rényi divergence holds:
The proof of Proposition 1 follows easily from [15], and we include it in Appendix B.1.1 for completeness. Mirroring the comments in [15], when ρ = πn,α te this result is precisely [14,Theorem 3.4]. We also note from [14] that ∀ α re , β ∈ (0, 1] α re -Rényi divergences are all equivalent through the following inequality
Hence, for the subsequent results, we simplify by assuming that α te = α re . This probabilistic bound implies the following PAC-Bayesian concentration bound on the model risk computed with respect to the fractional variational posterior: Theorem 1. Let F be a subset of all probability distributions parameterized by Θ, and assume there exist ǫ n > 0 and ρ n ∈ F such that i.
K(P
Var(r n (θ, θ 0 ))ρ n (dθ) ≤ nǫ n , and iii.
K(ρ n , π) ≤ nǫ n .
Then, for any α re ∈ (0, 1) and (ǫ, η) ∈ (0, 1) × (0, 1),
The proof of Theorem 1 is a generalization of [15] (Theorem 2.4) to the non-i.i.d. setting, and a special case of [16] (Theorem 3.1), where the problem setting includes latent variables. We include a proof for completeness. As noted in [15], the sufficient conditions follow closely from [13] and we will show that they hold for a variety of Markov chain models.
A direct corollary of Theorem 1 follows by setting η = 1 nǫ n , ǫ = e -nǫ n and using the fact that e -nǫ n ≥ 1 nǫ n . Note that Equation ( 9) is vacuous if η + ǫ > 1. Therefore, without loss of generality, we restrict ourselves to the condition 2 nǫ n < 1.
Corollary 1. Assume ∃ ǫ n > 0, ρ n ∈ F such that the following conditions hold:
i. K(P
Var(r n (θ, θ 0 ))ρ n (dθ) ≤ nǫ n , and iii.
K(ρ n , π) ≤ nǫ n .
Then, for any α re ∈ (0, 1),
We observe that Theorem 1 and Corollary 1 place no assumptions on the nature of the statistical dependence between data points. However, verification of the sufficient conditions is quite hard, in general. One of our key contributions is to verify that under reasonable assumptions on the smoothness of the transition kernel, the sufficient conditions of Theorem 1 and Corollary 1 are satisfied by ergodic Markov chains.
Observe that the first two conditions in Corollary 1 ensure that the distribution ρ n concentrates on parameters θ ∈ Θ around the true parameter θ 0 , while the third condition requires that ρ n not diverge from the prior π rapidly as a function of the sample size n. In general, verifying the first and third conditions is relatively straightforward. The second condition, on the other hand, is significantly more complicated in the current setting of dependent data, as the variance of r n (θ, θ 0 ) includes correlations between the observations {X 0 , . . . , X n }. In the next section, we will make assumptions on the transition kernels (and corresponding invariant densities) that 'decouple' the temporal correlations and the model parameters in the setting of strongly mixing and ergodic Markov chain models, and allow for the verification of the conditions in Corollary 1. Towards this, Propositions 2 and 3 below characterize the expectation and variance of the log-likelihood ratio r n (•, •) in terms of the one-step transition kernels of the Markov chain. First, consider the expectation of r n (•, •) in condition (i).
Proposition 2. Fix θ 1 , θ 2 ∈ Θ and consider the parameterized Markov transition kernels p θ 1 and p θ 2 , and initial distributions q
be the corresponding joint probability densities; that is, p
for j ∈ {1, 2}. Then, for any n ≥ 1, the log-likelihood ratio r n (θ 2 , θ 1 ) satisfies
where Z 0 := log
. The expectation in the first term is with respect to the joint density
where the marginal density satisfies
If the Markov chain is also stationary under θ 1 , then Equation (12) simplifies to
Notice that E θ 1 [r n (θ 2 , θ 1 )] is precisely the KL divergence, K(P
θ 2 ). Next, the following proposition uses [19] (Lemma 1.3) to upper bound the variance of the loglikelihood ratio. Proposition 3. Fix θ 1 , θ 2 ∈ Θ and consider parameterized Markov transition kernels p θ 1 and p θ 2 , with initial distributions q (0) θ 1 and q
be the corresponding joint probability densities of the sequence (x 0 , . . . , x n ), and q (i) θ j the marginal density for i ∈ {1, . . . , n} and j ∈ {1, 2}. Fix δ > 0 and, for each i ∈ {1, . . . , n}, define
Similarly, define Z 0 := log
, and D 1,2 := E θ 1 |Z 0 | 2+δ . Suppose the Markov chain corresponding to θ 1 is α-mixing with coefficients {α k }. Then,
Note that this result holds for any parameterized Markov chain. In particular, when the Markov chain is stationary,
θ 1 ,θ 2 ∀ i and ∀θ ∈ Θ, and Equation ( 14) simplifies to
If the sum ∑ k≥0 α δ/(2+δ) k is infinite, the bound is trivially true. For it to be finite, of course, the coefficients α k must decay to zero sufficiently quickly. For instance, Theorem A.1.2 shows that if the Markov chain is geometrically ergodic, then the α-mixing coefficients are geometrically decreasing. We will use this fact when the Markov chain is non-stationary, as in Section 4. In the next section, however, we first consider the simpler stationary Markov chain setting where geometric ergodic conditions are not explicitly imposed. We also note that unless only a finite number of α k are nonzero, the sum ∑ k≥0 α δ/(2+δ) k is infinite when δ = 0, and our results will typically require δ > 0.
Observe that the PAC-Bayesian concentration bound in Corollary 1 specifically requires bounding the mean and variance of the log-likelihood ratio r n (θ, θ 0 ). We ensure this by imposing regularity conditions on the log-ratio of the one-step transition kernels and the corresponding invariant densities. Specifically, we assume the following conditions that decouple the model parameters from the random samples, allowing us to verify the bounds in Corollary 1.
(1)
. . , m} such that for any parameters θ 1 , θ 2 ∈ Θ, the log of the ratio of one-step transition kernels and the log of the ratio of the invariant distributions satisfy, respectively,
We further assume that for some δ > 0, the functions f
k and M
(1) k satisfy the following:
i. there exist constants C
for t ∈ {1, 2}, n ≥ 1 and k ∈ {1, 2, . . . , m}, and ii.
there exists a constant B such that M
(1)
The following examples illustrate Equations ( 17) and ( 18) for discrete and continuous state Markov chains.
Example 1. Suppose {X 0 , . . . , X n } is generated by the birth-death chain with parameterized transition probability mass function,
In this example, the parameter θ denotes the probability of birth. We shall see that, m = 3:
θ . The derivation of these terms and that they satisfy the conditions of Assumption 1 is provided in the proof of Proposition 6.
Example 2. Suppose {X 0 , . . . , X n } is generated by the 'simple linear' Gauss-Markov model
where {W n } is a sequence of i.i.d. standard Gaussian random variables. Then, m = 2, with M
2 (X) = 0. Cor- responding to these, we have f
2 (θ 0 , θ 0 ) = 0. The derivation of these quantities and that these satisfy the conditions of Assumption 1 under appropriate choice of ρ n is shown in the proof of Proposition 10.
(1)
k involves no loss of generality, since these functions can be set to 0. Both Equations ( 17) and ( 18) can be viewed as generalized Lipschitz-smoothness conditions, recovering the usual Lipschitz-smoothness when m = 1 and when f (t) k is Euclidean distance. Our generalized conditions are useful for distributions like the Gaussian, where Lipschitz smoothness does not apply. From Jensen's inequality we have | f
and Assumption 1(i) above implies that for some constant C > 0 and k ∈ {1, 2, . . . , m}, t ∈ {1, 2},
Assumption 1(i) is satisfied in a variety of scenarios, for example, under mild assumptions on the partial derivatives of the functions f
To illustrate this, we present the following proposition.
Proposition 4. Let f (θ, θ 0 ) be a function on a bounded domain with bounded partial derivatives with f (θ 0 , θ 0 ) = 0. Let {ρ n (•)} be a sequence of probability densities on θ such that
n for some σ > 0. Then, for some C > 0,
∂θ as the partial derivative of the function f . By the mean value theorem,
Since the partial derivatives are bounded, there exists
without loss of generality we can use Jensen's inequality to conclude that, for all 0
We can now state the main theorem of this section.
Theorem 2. Let {X 0 , . . . , X n } be generated by a stationary, α-mixing Markov chain parametrized by θ 0 ∈ Θ. Suppose that Assumption 1 holds and that the α-mixing coefficients satisfy
Then, the conditions of Corollary 1 are satisfied with
Theorem 2 is satisfied by a large class of Markov chains, including chains with countable and continuous state spaces. In particular, if the Markov chain is geometrically ergodic, then it follows from Equation (A4) (in the appendix) that ∑ k≥1 α δ/(2+δ) k < +∞. Observe that in order to achieve O( 1 √ n ) convergence, we need δ ≤ 1. Key to the proof of Theorem 2 is the fact that the variance of the log-likelihood ratio can be controlled via the application of Assumption 1 and Proposition 3. Note also that as δ decreases, satisfying the condition
requires the Markov chain to be faster mixing. We now illustrate Theorem 2 for a number of Markov chain models. First, consider a birth-death Markov chain on a finite state space.
Proposition 5. Suppose the data-generating process is a birth-death Markov chain, with onestep transition kernel parametrized by the birth probability θ 0 ∈ Θ. Let F be the set of all Beta distributions. We choose the prior to be a Beta distribution. Then, the conditions of Theorem 2 are satisfied and
Proof. The proof of Proposition 5 follows from the more general Proposition 8, by fixing the initial distribution to the invariant distribution under θ 0 . Therefore it has been omitted. We simply refer to the proof of Proposition 8 under a more general setup in Appendix B.3.
The birth-death chain on the finite state space is, of course, geometrically ergodic and the α-mixing coefficients α k decay geometrically. Note that the invariant distribution of this Markov chain is uniform over the state space, and consequently this is a particularly simple example. A more complicated and more realistic example is a birth-death Markov chain on the nonnegative integers. We note that if the probability of birth θ in a birth-death Markov chain on positive integers is greater than 0.5, then the Markov chain is transient, and consequently, not ergodic. Hence, our prior should be chosen to have support within (0, 0.5). For that purpose, we define the class of scaled beta distributions. Definition 2 (Scaled Beta). If X is a beta distribution on with parameters a and b, then Y is said to be a scaled beta distribution with same parameters on the interval (c, m + c) if
and in that case, the pdf of Y is obtained as
. For the birth-death chain, we set m = 0.5 and c = 0 giving it support on (0, 1 2 ). Setting m = 2 and c = -1 gives a beta distribution rescaled to have support on (-1, 1). Proposition 6. Suppose the data-generating process is a positive recurrent birth-death Markov chain on the positive integers parameterized by the birth probability θ 0 ∈ (0, 1 2 ). Further let F be the set of all Beta distributions rescaled to have support (0, 1 2 ). We choose the prior to be a scaled Beta distribution on (0, 1/2) with parameters a and b. Then, the conditions of Theorem 2 are satisfied with
Proof. The proof of Proposition 6 (for the stationary case) follows from the more general Proposition 9 (the nonstationary case) by fixing the initial distribution to the invariant distribution under θ 0 . We omit the proof and simply refer to the proof of Proposition 9 under a more general setup in Appendix B.3.
Unlike with the finite state-space, the invariant distribution now depends on the parameter θ ∈ Θ, and verification of the conditions of the proposition is more involved. In Appendix A.2, we prove that the class of scaled beta distributions satisfy the condition K(ρ n , π) ≤ nǫ n when the prior π is a beta or an uniform distribution. This fact will allow us to prove the above propositions.
Both Proposition 5 and Proposition 6 assume a discrete state space. The next example considers a strictly stationary simple linear model (as defined in Example 2), which has a continuous, unbounded state space. Proposition 7. Suppose the data-generating model is a stationary simple linear model:
where {W n } are i.i.d. standard Gaussian random variables and |θ 0 | < 1. Suppose that F is the class of all beta distributions rescaled to have the support (-1, 1). Then, the conditions of Theorem 2 are satisfied with
Proof. This is a special case of the more general non-stationary simple linear model which is detailed in Proposition 10. Therefore, the proof of the fact that the simple linear model satisfies Assumption 1 when starting from stationarity is deferred to the proof of Proposition 10. The simple linear model with |θ 0 | < 1 has geometrically decreasing (and therefore summable) α-mixing coefficients as a consequence of [20] (eq. (15.49)) and Theorem A.1.2. Combining these two facts, it follows that the conditions of Theorem 2 are satisfied.
Observe that Theorem 1 (and Corollary 1) are general, and hold for any dependent data-generating process. Therefore, there can be Markov chains that satisfy these, but do not satisfy Assumption 1 which entails some loss of generality. However, as our examples demonstrate, common Markov chain models do indeed satisfy the latter assumption.
We call a time-homogeneous Markov chain non-stationary if the initial distribution q (0) is not the invariant distribution. There are two sets of results in this setting: in Theorem 3 and Theorem 4 we explicitly impose the α-mixing condition, while in Theorem 5 we impose a f -geometric ergodicity condition (Definition A.1.2 in the appendix). As seen in Equation (A4) (in the appendix) if the Markov chain is also geometrically ergodic, then ∀ δ > 0, ∑ α δ/(2+δ) k < ∞. This condition can be relaxed, albeit at the risk of more complicated calculations that, nonetheless, mirror those in the geometrically ergodic setting. A common thread through these results is that we must impose some integrability or regularity conditions on the functions M k functions in Assumption 1 are uniformly bounded and that the α-mixing condition is satisfied. This result holds for both discrete and continuous state space settings. Theorem 3. Let {X 0 , . . . , X n } be generated by an α-mixing Markov chain parametrized by θ 0 ∈ Θ with transition probabilities satisfying Assumption 1 and with known initial distribution q (0) . Let {α k } be the α-mixing coefficients under θ 0 , and assume that ∑ k≥1 α δ/(2+δ) k < +∞. Suppose that there exists B ∈ R such that sup x,y |M (1) k (x, y)| < B for all k ∈ {1, 2, . . . , m} in Assumption 1. Furthermore, assume that there exists
k (X 0 )| 2 < +∞ for all k ∈ {1, 2, . . . , m}, then the conditions of Corollary 1 are satisfied with
The following result in Proposition 8 illustrates Theorem 3 in the setting of a finite state birth-death Markov chain. Proposition 8. Suppose the data-generating process is a finite state birth-death Markov chain, with one-step transition kernel parametrized by the birth probability θ 0 . Let F be the set of all Beta distributions. We choose the prior on θ 0 to be a Beta distribution. Then, the conditions of Theorem 3 are satisfied with ǫ n = O 1 √ n for any initial distribution q (0) . Theorem 3 also applies to data generated by Markov chains with countably infinite state spaces, so long as the class of data-generating Markov chains is strongly ergodic and the initial distribution has finite second moments. The following example demonstrates this in the setting of a birth-death Markov chain on the positive integers, where the initial distribution is assumed to have finite second moments. Proposition 9. Suppose the data-generating process is a birth-death Markov chain on the nonnegative integers, parameterized by the probability of birth θ 0 ∈ (0, 1 2 ). Further let F be the set of all Beta distributions rescaled upon the support (0, 1 2 ). Let q (0) be a probability mass function on non-negative integers such that ∑ ∞ i=1 i 2 q (0) (i) < +∞. We choose the prior to be a scaled Beta distribution on (0, 1/2) with parameters a and b. Then, the conditions of Theorem 3 are satisfied with
Since continuous functions on a compact domain are bounded, we have the following (easy) corollary (stated without proof).
Corollary 2. Let {X 0 , . . . , X n } be generated by an α-mixing Markov chain parametrized by θ 0 ∈ Θ on a compact state space, and with initial distribution q (0) . Suppose the α-mixing coefficients satisfy ∑ k≥1 α δ/(2+δ) k < +∞, and that Assumption 1 holds with continuous functions M (1) k (•, •), k ∈ {1, 2, . . . , m}. Furthermore, assume that there exists ρ n such that K(ρ n , π) ≤ √ nC for some constant C. Then, Theorem 3 is satisfied with
In general, the M
k functions will not be uniformly bounded (consider the case of the Gauss-Markov simple linear model in Example 2), and stronger conditions must be imposed on the data-generating Markov chain itself. The following assumption imposes a 'drift' condition from [21]. Specifically, [21] (Theorem 2.3) shows that under the conditions of Assumption 2, the moment generating function of an aperiodic Markov chain {X n } can be upper bounded by a function of the moment generating function of X 0 . Together with the α-mixing condition, Assumption 2 implies that this Markov data generating process satisfies Corollary 1.
k , for each k = 1, . . . , m 1 , are defined in Assumption 1. For each k = 1, . . . , m, assume the process {M k n } satisfies the following conditions:
Under this drift condition, the next theorem shows that Corollary 1 is satisfied.
Theorem 4. Let {X 0 , . . . , X n } be generated by an aperiodic α-mixing Markov chain parametrized by θ 0 ∈ Θ and initial distribution q (0) . Suppose that Assumption 1 and Assumption 2 hold, and that the α-mixing coefficients satisfy
Verifying the conditions in Theorem 4 can be quite challenging. Instead, we suggest a different approach that requires f -geometric ergodicity. Unlike the drift condition in Assumption 2, f -geometric ergodicity additionally requires the existence of a petite set. As noted before, geometric ergodicity implies α-mixing with geometrically decaying mixing coefficients. As with Theorem 4, we assume for simplicity that the Markov chain is aperiodic.
Theorem 5. Let {X 0 , . . . , X n } be generated by an aperiodic Markov chain parametrized by θ 0 ∈ Θ with known initial distribution q (0) , and assumed to be V-geometrically ergodic for some V : R m → [1, ∞). Suppose that Assumption 1 holds and M
(1) k (y, x) 2+δ p θ 0 (y|x)dy < V(x) ∀ k, x and some δ > 0. Furthermore, assume that K(ρ n , π) ≤ √ nC for some constant C > 0. If the initial distribution q (0) satisfies E q (0) [V(X 0 )] < +∞, then the conditions of Corollary
The following Proposition 10 shows, the simple linear model satisfies Theorem 5 when the parameter θ 0 is suitably restricted. Proposition 10. Consider the simple linear model satisfying the equation
where {W n } are i.i.d. standard Gaussian random variables and |θ 0 | < 2 1 4+2δ -1 for δ > 0. Let F be the space of all scaled Beta distributions on (-1, 1) and suppose the prior π is a uniform distribution on (-1, 1). Then, the conditions of Theorem 5 are satisfied with
n ) , if the initial distribution q (0) satisfies E q (0) [X 4+2δ 0 ] < +∞.
We show next how our results can be extended to the misspecified model setting. Assume that the true data generating distribution is parametrized by θ 0 ∈ Θ. Let θ * n := arg min θ∈Θ K(P
θ ) represent the closest parametrized distribution in the variational family to the data-generating distribution. Further, assume our usual conditions:
Similarly, decomposing the variance it follows that
Using the fact that 2ab ≤ a 2 + b 2 on the covariance term 2Cov[r n (θ,
Integrating both sides with respect to ρ n (dθ) we get
Consequently, we arrive at the following result: Theorem 6. Let F be a subset of all probability distributions parameterized by Θ. Let θ * n = arg min θ∈Θ K(P (n) θ 0 , P (n) θ ) and assume there exist ǫ n > 0 and ρ n ∈ F such that i.
Var(r n (θ, θ * n ))ρ n (dθ) ≤ nǫ n , and iii.
K(ρ n , π) ≤ nǫ n .
Then, for any α re ∈ (0, 1) and (ǫ, η) ∈ (0, 1) × (0, 1),
The proof of this theorem is straightforward and follows from the proof of Theorem 1 by plugging in the upper bounds for KL-divergence from Equation (23), and variance from Equation (26) to Equation (A13). A sketch of the proof is presented in the appendix.
Concentration of the KL-VB model risk, in terms of the expected α re -Rényi divergence, is well established under the i.i.d. data generating model assumption. Here, we extended this to the setting of Markov data generating models, linking the concentration rate to the mixing and ergodic properties of the Markov model. Our results apply to both stationary and non-stationary Markov chains, as well as to the situation with misspecified models. There remain a number of open questions. An immediate one is to extend the current analysis to continuous-time Markov chains and Markov jump processes, possibly using uniformization of the continuous time model. Another direction is to extend this to the setting of non-homogeneous Markov chains, where analogues of notions such as stationarity are less straightforward. Further, as noted in the introduction, [14] establish PAC-Bayes bounds under slightly weaker 'existence of test functions' conditions, while our results are established under the stronger conditions used by [15] for the i.i.d. setting. Weakening the conditions in our analysis is important, but complicated. A possible path is to build on results from [22], who provides conditions form the existence of exponentially powerful test functions exist for distinguishing between two Markov chains. It is also known that there exists a likelihood ratio test separating any two ergodic measures [23]. However, leveraging these to establish the PAC-Bayes bounds for the KL-VB posterior is a challenging effort that we leave to future papers. Finally it is of interest to generalize our PAC-bounds to posterior approximations beyond KL-variational inference, such as α re -Rényi posterior approximations [6], and loss-calibrated posterior approximations [24,25].
Definition A.1.2 ( f -geometric ergodicity). For any function f , Markov chain {X n } parameterized by θ ∈ Θ is said to be f -geometrically ergodic if it is positive Harris and there exists a constant r f > 1, that depends on f , such that for any A ∈ B(X),
It is straightforward to see that this is equivalent to
showing that, under geometric ergodicity, the α-mixing coefficients raised to any positive power υ are finitely summable. We note here that the most standard procedure to establish f -geometric ergodicity for any Markov chain is through the verification of the drift condition. The drift condition is a sufficient condition for a Markov chain to be f -geometrically ergodic, as long as there exists a set (called petite set) towards which the Markov chain drifts to (see Assumption A.1.1 in the appendix). If a Markov chain is f -geometrically ergodic with f ≡ V, for some particular function V, then we call it V-geometrically ergodic. We defined V-geometric ergodicity in the previous sections. In this section, we provide a sufficient condition for a Markov chain to be V-geometrically ergodic. First, we recall the definition of resolvent from [20] (Chapter 5).
Definition A.1.3 (Resolvent). Let n ∈ {0, 1, 2, . . . } and q n be such that q n ≥ 0 ∀ n and ∑ ∞ n=1 q n = 1. Note that q n can be thought of being a probability mass function for a random variable "q" taking values on non-negative integers. Then, the resolvent of a Markov chain with respect to q is given by K q (x, A) where,
Then, the definition of petite sets follows (see, for Reference, [20] (Chapter 5)).
Definition A.1.4 (Petite Sets). Let X 0 , . . . , X n be n samples from a Markov chain taking values on the state space X . Let C be a set. We shall call C to be v q petite if K q (x, B) ≥ υ q (B)
for all x ∈ C and B ∈ B(X ), and a non-trivial measure υ q on B(X ), and a probability mass function q on {1, 2, 3, . . . } Suppose that {X n } is satisfies Assumption A.1.1. Then, the set S V = {x : V(x) < ∞} is absorbing, i.e., P θ (X andfull, i.e., ψ(S c V ) = 0. Furthermore, ∃ constants r > 1, R < ∞ such that, for any A ∈ B(S),
Any aperiodic and ψ-irreducible Markov chain satisfying the drift condition is geometrically ergodic. A consequence of Equation (A2) is that if, {X n } is V-geometrically ergodic, then for any other function U, such that |U| < V, it is also U-geometrically ergodic. In essence, a geometrically ergodic Markov chain is asymptotically uncorrelated in a precise sense. Recall ρ-mixing coefficients defined as follows. Let A be a sigma field and L 2 (A) be the set of square integrable, real valued, A measurable functions. Definition A.1.5 (ρ-mixing coefficient). Let M j i denote the sigma field generated by the measures X k , where i ≤ k ≤ j. Then,
where Corr is the correlation function.
Theorem A.1.2. If X n is geometrically ergodic, then it is α-mixing. That is, there exists a constant c > 0 such that α k = O(e -ck ).
Proof. By [26] (Theorem 2) it follows that a geometrically ergodic Markov chain is asymptotically uncorrelated with ρ-mixing coefficients (see Definition A.1.5) that satisfy ρ k = O(e -ck ). Furthermore, it is well known that [18,26]
The following results will be utilized in the proofs of Propositions 8-10.
Lemma A.2.1. Let θ 0 ∈ (0, 1). Let, ρ n be a sequence of Beta distributions with parameters a n = nθ 0 and b n = n(1θ 0 ). Let π denote an uniform distribution, U(0, 1). Then, K(ρ n , π) < C + 1 2 log(n), for some constant C > 0.
Proof. Without loss of generality, we can assume a n > 1 and b n > 1. The same form of the result can be obtained in all the other cases, by appropriate use of the bounds presented in the proof. We write the KL divergence K(ρ n , π) as log ρ n π ρ n (dθ). Since π is uniform, π(θ) = 1 whenever θ ∈ (0, 1). Hence, the KL-divergence can be written as the negative of the entropy of ρ n 1 0 log(ρ n (θ))ρ n (dθ), which can be written as
where ψ is the digamma function. Using Stirling's approximation on Beta(a n , b n ) yields,
Hence, setting
From [27] we have that log(x) -
Finally, using the fact that log(x) -1 x < ψ(x), we get,
Therefore, after much cancellation, the KL-divergence
Now, plugging in the values of a n and b n , we get Plugging in the values of a n and b n , we get as upper bound for the KL-divergence as,
for some large enough positive constant C. This completes our proof.
Proposition A.2.1. Let θ 0 ∈ (0, 1). Let, ρ n be a sequence of Beta distributions with parameters a n = nθ 0 and b n = n(1θ 0 ). Let π denote an Beta distribution, with parameters (a, b). Then, K(ρ n , π) < C + 1 2 log(n), for some constant C > 0.
Proof. Without loss of generality, we assume a > 1 and b > 1. As mentioned in the proof of Lemma A.2.1, the other cases follows similarly. We write the KL-divergence between ρ n and π as,
where, U is an uniform distribution on (0, 1). We analyze the second term in the above expression. The second term can be written as,
where C 1 is log(Beta(a, b)). Since, ρ n follows a Beta distribution with parameters a n = nθ 0 and b n = n(1θ 0 ), we get that,
Using the lower bound on ψ(nθ 0 ) and the upper bound on ψ(n), we get
Furthermore, similarly, we get that,
for a large positive constant C. Using the above bounds, we finally show that,
which can be upper bounded by C ′ for some large constant C ′ . Finally, we upper bound log
θ 0 ). Then, applying Lemma B.1.1 to the integrand on the left hand side (l.h.s.) above, it follows that
Multiply both sides of this equation by ǫ > 0 to obtain
Now, by Markov's inequality, we have
Thus, it follows via complementation that
Recall the definition of the fractional posterior and the VB approximation,
It follows by definition of the KL divergence that πn,
where π is the prior distribution. Following Proposition 1 it follows that for any ǫ > 0
with probability 1ǫ. We fix an η ∈ (0, 1). Using Chebychev's inequality, we have
Note that α re 1-α re E(r n (θ, θ 0 ))ρ n (dθ) and K(ρ n ,π) 1-α re are constants with respect to the data, implying
Therefore, we have
From Proposition 1, with probability 1ǫ the following holds
We also need to establish the following technical lemma.
Lemma B.1.5. Let {X t } be an α-mixing Markov Chain with mixing coefficients {α t }. Then the process {Y t } where Y t := log
is also α-mixing with mixing coefficients {α t } where αt = α t-1 .
Proof. By Z i denote the paired random measure (X i , X i-1 ). Let M j i denote the sigma field generated by the measures X k , where i ≤ k ≤ j. By G j i denote the sigma field generated by the measures Z k , where i ≤ k ≤ j. Let C ∈ M j i-1 . Then, C can be expressed as
. . and so on. Now, consider a map.
) is obtained by applying the map T j i to each element of M j i-1 . If we assume this latter set to be the range and M j i-1 to be the domain, then, by construction, T j i is a bijection. Furthermore, the two classes are made of disjoint sets, i.e., if
Now consider the α-mixing coefficients for Z i . By definition, it is given by
where,
Then, the expression for the α-mixing coefficient can be reduced into
Note that, by bijection property of T j i , we can find
is just a function of the paired Markov chain Z i , therefore it has α-mixing coefficient α k-1 .
We now proceed to the proof of Proposition 3. Let {X k } be a stationary α-mixing Markov chain under θ 1 with mixing coefficients {α k }. Observe that the log-likelihood can be expressed as
Therefore, the variance of the log-likelihood ratio is simply
It follows from Lemma B.1.5 that {Y k } is a stochastic process with α-mixing coefficients α k-1 . Therefore, using Lemma B.1.4 we have
Similarly, as above we can also say
Combining, the two upper bounds above, we get the first result:
θ 1 ,θ 2 ∀ i, and
Finally, using Equations (A31) and (A32) we have We substitute the true parameter θ 0 for θ 1 and θ for θ 2 . We also set q (0)
1 to be the invariant distribution of the Markov chain under θ 0 , q 0 , and q (0) 2 as the invariant distribution of the Markov chain under θ, q θ . Applying the fact that these Markov chains are stationary to Proposition 2, we have
where the inequality follows from Assumption 1. Therefore, it follows that
k (θ, θ 0 )|ρ n (dθ).
By Assumption 1(i), it follows that
n , where ǫ
(1)
Part 2: Verifying condition (ii) of Corollary 1. Again, using Proposition 3 along with the fact that the Markov chain is stationary we have
It then follows that
(1)
θ 0 ,θ ρ n (dθ)
First, consider the term C
(1) θ 0 ,θ ρ n (θ), and observe that C
(1)
By Assumption 1, we have
Since the function x → x 2+δ is convex, we can apply Jensen's inequality to obtain,
Therefore, it follows that
) . Now, applying Assumption 1, we can bound the previous equation as follows,
k 's are bounded there exists a constant Q so that,
k (θ, θ 0 )|ρ n (dθ).
By Assumption 19 in Assumption 1, it follows that
n , for some ǫ
(1)
First, consider the term C (i) θ 0 ,θ ρ n (dθ). Using Assumption 1, we can upper bound C (i) θ 0 ,θ as,
k 's are upper bounded by Q, it follows from the previous expression that,
Hence, from Assumption 1, we get,
Using the upper bound above, we can say for an L large enough, C
(i)
Next, by the Cauchy-Schwarz inequality, we have that
n . Thus, we have the following upper bound.
< ∞, we have that for some ǫ
(2)
n .
Since K(ρ n , π) ≤ √ nC, following the concluding argument in Theorem 2 completes the proof.
Finally, we take the KL-divergence K(ρ n , π). ρ n follows a scaled Beta distribution on (0, 1/2) with parameters a n = n(θ 0 /2) and b n = n(1θ 0 /2), while π follows a scaled Beta distribution on (0, 1/2) with parameters a and b. Thus,
which, by substituting t = 2θ, we get,
is the KL-divergence between a Beta distribution with parameters a n and b n and a Beta distribution with parameters a and b. An application of Proposition A.2.1 gives us for a constant
Hence we can say, K(ρ n , π) < 2 C 1 + 1 2 log(n) . Thus, we now get that for some constant
we satisfy all of the conditions of Assumption 1 and thus by Theorem 3, we are complete the proof.
Part 1: Verifying condition (i) of Corollary 1 As in the proof of Theorem 2 substitute the true parameter θ 0 for θ 1 and θ for θ 2 . We also set our initial distributions q (0)
1 and q (0)
2 to the known initial distribution q (0) . A method similar to Equation (A35), yields
k s satisfy Assumption 2, it follows by the application of Theorem 2.3, [21] that ∃ λ > 0 such that for any 0 < κ ≤ λ, and for some ζ ∈ (0, 1) possibly depending upon λ, E e κM (1) k
We rewrite E M
k (X i , X i-1 )|X 1 , X 0 as follows:
Since, ζ ∈ (0, 1), ζ i < 1. Hence, we can write that,
for a large constant L. Therefore K(P
θ )ρ n (dθ) can be upper bounded as follows,
k (θ, θ 0 )|ρ n (dθ).
By Assumption 1, | f
Hence, for some ǫ
(1)
n . Part 2: Verifying condition (ii) of Corollary 1: Similar to as in the proof of Theorem 3, we upper bound Var
, where C θ 0 ,θ is upper bounded as
There exists a constant C δ depending upon δ such that, [M
By expressing E M
k (X i , X i-1 ) 2+δ |X 1 , X 0 and following a method similar to the previous part, we get,
The fact that 0 < ζ < 1 implies that 0 < ζ i < ζ. This gives us the following,
Since κ < λ, by the application of Jensen's inequality, we get
We know that | f
k (θ, θ 0 )| 2+δ ρ n (dθ) < C n . Thus, following Assumption 1 we can say that, for a large constant L, C
The rest of the proof follows similarly as in the proof of Theorem 3, and we obtain an ǫ
n .
Since, K(ρ n , π) ≤ √ nC, similar arguments as in the proof of Theorem 2 holds. The theorem is thus proved. Appendix B.3.5. Proof of Theorem 5 Part 1: Verifying condition (i) of Corollary 1 As in the proof of Theorem 2 substitute the true parameter θ 0 for θ 1 and θ for θ 2 . We also set q (0) 1 and q (0) 2 to the known initial distribution q (0) . Similar to the steps leading to Equation (A35), we get
Recall that the marginal density satisfies q
Since the Markov chain {X n } satisfies Assumption A.1.1, we know by the application of Theorem
where q θ 0 is the stationary distribution, implying that
By the application of Jensen's inequality we get E M
k (X 1 , X 0 )] + τ i-1 RV(x 0 )q (0) (x 0 )dx 0 .
Summing from i = 1 to n, we get
k (X 1 , X 0 )] + n ∑ i=1 τ i-1 RV(x 0 )q (0) (x 0 )dx 0
k (X 1 , X 0 )] + 1τ n 1τ RV(x 0 )q (0) (x 0 )dx 0 .
This gives us the following bound on K(P
θ )ρ n (dθ):
k (θ, θ 0 )|ρ n (dθ).
By Assumption 1, | f
k (θ, θ 0 )|ρ n (dθ) < C √ n . Hence, we can rewrite the previous expression as
Since, τ < 1, 0 < 1τ n < 1, and we rewrite the previous equation as,
k (X 1 , X 0 )] +
Hence, there exists an ǫ , where C θ 0 ,θ is upper bounded as
k (θ, θ 0 )| 2+δ .
Since E M
k (X i , X i-1 ) 2+δ |X i-1 < V(X i-1 ), by a similar application of V-geometric er- godicity, we can say that, ∃ 0 < τ < 1, such that
k (X 1 , X 0 )] 2+δ + τ i-1 RV(x 0 )D(x 0 )dx 0 , which, by the fact that τ i-1 < τ, gives us,
k (X 1 , X 0 )] 2+δ + τ RV(x 0 )D(x 0 )dx 0 .
By Assumption 1, we know that, | f For the purpose of the proof, we choose ρ n 's with scaled Beta distribution with parameters a n = n 1+θ 0 2 and b n = n 1-θ 0 2 . Since, ρ n is a scaled Beta distribution with the scaling factors m = 2 and c = -1, the pdf of ρ n is given by
Since this is a scaled distribution, E ρ n [θ] = 2 a n a n +b n -1 = θ 0 and there exists a constant σ > 0, Var ρ n [θ] = σ 2 n . We now analyse the log-ratio of the transition probabilities for the Markov chain, log p θ 0 (X n |X n-1 )log p θ (X n |X n-1 ) = 2X n X n-1 (θθ 0 ) + X 2 n-1 (θ 2 0θ 2 ).
Observe that in this setting, M
1 (X n , X n-1 ) = |X n X n-1 | and M
(1)
2 (X n , X n-1 ) = X 2 n . Next, using the fact that
and by an application of triangle inequality, we obtain
|X n-1 .
Now by using Jensen's inequality we get,
We know if Y ∼ N(µ, σ 2 ), then E|Y -µ| p = σ p 2
. Consequently,
Hence, we have shown that,
Since |θ 0 | < 2 1 4+2δ -1 , 2 3+2δ |θ 0 | 4+2δ < 1, and we can express the above equation as,
Define the set C(m) := {x : |x| 4+2δ + 1 ≤ m}. From Proposition 11.4.2, [20], for a large enough m, C(m) forms a petite set. Thus, we have proved that V(x) as defined in this example satisfies Assumption A.1.1, and {X n } is V-geometrically ergodic. The f
(1) j 's corresponding to Assumption 1 are given by f
2 (θ 0 , θ 0 ) = 0, We just showed that they also have bounded partial derivatives. We also know that |θ| < 1. Hence, by Proposition 4 f The invariant distribution for the simple linear model Markov-chain under parameter θ is given by a gaussian distribution with mean 0 and variance 1 1-θ 2 . In other words,
Analyzing the log likelihood yields, log q 0 (x)log q θ (x) = -
(1θ 2 0 ) +