Azuma-Hoeffding Inequality

Let X0,…,XnX_0, \ldots, X_n be a martingale such that

Ak≤Xk−Xk−1≤Bk A_k \le X_k - X_{k - 1} \le B_k

for some random variables AkA_k and BkB_k that may be functions of X0,X1,…Xk−1X_0, X_1, \ldots X_{k - 1}. If Bk−Ak≤ckB_k - A_k \le c_k for some constants ckc_k, then, for all t≥0t \ge 0 and any η>0\eta > 0,

Pr⁡[∣Xt−X0∣≥η]≤2exp⁡(−2η2∑k∈[t]ck2) \Pr[ | X_t - X_0 | \ge \eta ] \le 2 \exp \left( - \frac{ 2 \eta^2 }{ \sum_{ k \in [t] } c_k^2 }\right)

Hoeffding's Lemma

Let XX be any real-valued random variable with mean 00, such that a≤X≤b{\displaystyle a\leq X\leq b} almost surely, i.e. with probability one. Then, for all λ∈R{\displaystyle \lambda \in \mathbb {R} },

E[eλX]≤exp⁡(λ2(b−a)28). {\displaystyle \mathbb {E} \left[ e^{\lambda X} \right] \le \exp {\Big (} {\frac{\lambda^{2}(b-a)^{2}} {8}} {\Big ) }. }

Proof. We prove a version with a weaker factor of 12\frac{1}{2} but simpler analysis. The proof for the stronger version is included in the appendix.

First, consider a special case of Rademacher random variable σ\sigma that equals −1-1 and 11 with equal probability. For all λ∈R\lambda \in \mathbb{R},

E[eλσ]=e−λ+eλ2=∑i≥0λ2i(2i)!≤∑i≥0(λ2/2)ii!=eλ2/2. \begin{aligned} \mathbb{E} [ e^{ \lambda \sigma} ] &= \frac{ e^{ - \lambda } + e^{ \lambda } }{ 2 } &= \sum_{ i \ge 0 } \frac{ \lambda^{ 2i } }{ (2i)! } &\le \sum_{ i \ge 0 } \frac{ ( \lambda^2 / 2)^i }{ i ! } &= e^{ \lambda^2 / 2}. \end{aligned}

Next, let YY be identical copy of XX. By Jensen's inequality,

E[eλX]=E[eλ(X−E[X])]=EX[eλ(X−EY[Y])]≤EX,Y[eλ(X−Y)] \begin{aligned} \mathbb{E} [ e^{ \lambda X }] &= \mathbb{E} [ e^{ \lambda ( X - \mathbb{E}[ X ] ) }] &= \mathbb{E}_{X} [ e^{ \lambda ( X - \mathbb{E}_{Y} [ Y ] ) }] &\le \mathbb{E}_{ X, Y } [ e^{ \lambda ( X - Y ) }] \end{aligned}

Since XX and YY has identical distribution, the random variable σ(X−Y)\sigma( X - Y) has the same distribution as X−YX - Y. Hence,

EX,Y[eλ(X−Y)]=EX,Y[Eσ[eλ(X−Y)σ|X,Y]]. \mathbb{E}_{ X, Y } [ e^{ \lambda ( X - Y ) }] = \mathbb{E}_{ X, Y } \left[ \mathbb{E}_\sigma [ e^{ \lambda ( X - Y ) \sigma } \middle\vert X, Y] \right].

Finally, by previous argument,

Eσ[eλ(X−Y)σ∣X,Y]≤eλ2(X−Y)2/2≤eλ2(b−a)2/2. \begin{aligned} \mathbb{E}_{\sigma } [ e^{ \lambda ( X - Y ) \sigma } \mid X, Y ] \le e^{ \lambda^2 ( X - Y )^2 / 2 } \le e^{ \lambda^2 (b - a)^2 / 2 }. \end{aligned}

It concludes that EX,Y[eλ(X−Y)]≤eλ2(b−a)2/2\mathbb{E}_{ X, Y } [ e^{ \lambda ( X - Y ) }] \le e^{ \lambda^2 (b - a)^2 / 2 }.

□\square

Proof of the Theorem

Define Zk=Xk−Xk−1Z_k = X_k - X_{k - 1} for each k∈[n]k \in [n]. Since X1,…,XnX_1, \ldots, X_n is a martingale,

E[Zk∣X0,…,Xk−1]=0. \mathbb{E}[ Z_k \mid X_0, \ldots, X_{k - 1} ] = 0.

Via Hoeffding's Lemma, for each k∈[n]k \in [n], and for all λ∈R\lambda \in \mathbb{R}, it holds that

E[eλZt]≤eλ2ck2/8. \mathbb{E} \left[ e^{\lambda Z_t } \right] \le e^{ \lambda^2 c_k^2 / 8 }.

It follows that

E[eλ∑k∈[t]Zk]=E[E[eλ∑k∈[t]Zk|Z1,…,Zt−1]]=E[E[eλZt]⋅E[eλ∑k∈[t−1]Zk|Z1,…,Zt−1]]≤eλ2ct2/8⋅E[E[eλ∑k∈[t−1]Zk|Z1,…,Zt−1]] \begin{aligned} \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t]} Z_k } \right] &=\mathbb{E} \left[ \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t] } Z_k } \middle\vert Z_1, \ldots, Z_{t - 1} \right] \right] \\ &=\mathbb{E} \left[ \mathbb{E} \left[ e^{\lambda Z_t } \right] \cdot \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t - 1] } Z_k } \middle\vert Z_1, \ldots, Z_{t - 1} \right] \right] \\ &\le e^{ \lambda^2 c_t^2 / 8 } \cdot \mathbb{E} \left[ \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t - 1] } Z_k } \middle\vert Z_1, \ldots, Z_{t - 1} \right] \right] \end{aligned}

By induction, it holds that

E[eλ∑k∈[t]Zk]≤eλ2∑k∈[t]ck2/8. \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t]} Z_k } \right] \le e^{ \lambda^2 \sum_{k \in [t] } c_k^2 / 8 }.

For all λ>0\lambda > 0, via Markov's inequality,

Pr⁡[∑k∈[t]Zk≥η]=Pr⁡[eλ∑k∈[t]Zk≥eλη]≤E[eλ∑k∈[t]Zk]E[eλη]≤eλ2∑k∈[t]ck2/8−λη. \begin{aligned} \Pr \left[ \sum_{k \in [t] } Z_k \ge \eta \right] &= \Pr \left[ e^{ \lambda \sum_{k \in [t] } Z_k } \ge e^{ \lambda \eta } \right] \\ &\le \frac{ \mathbb{E} \left[ e^{ \lambda \sum_{k \in [t]} Z_k } \right] }{ \mathbb{E}[ e^{ \lambda \eta } ] } \\ &\le e^{ \lambda^2 \sum_{k \in [t] } c_k^2 / 8 - \lambda \eta }. \end{aligned}

When λ=4η∑k∈[t]ck2\lambda = \frac{4 \eta}{ \sum_{k \in [t] } c_k^2 }, the probability is minimized to

exp⁡{2η2∑k∈[t]ck2−4η2∑k∈[t]ck2}=exp⁡{−2η2∑k∈[t]ck2}. \exp \left\{ \frac{2 \eta^2 }{ \sum_{k \in [t] } c_k^2 } - \frac{4 \eta^2 }{ \sum_{k \in [t] } c_k^2 } \right\} = \exp \left\{ - \frac{2 \eta^2 }{ \sum_{k \in [t] } c_k^2 } \right\}.

□\square

Hoeffding's Lemma

Proof 1

Suppose that b≠ab \neq a (otherwise, XX is a constant). Define

  1. θ=−ab−a=1−bb−a∈(0,1)\theta = - \frac{a}{b - a} = 1 - \frac{b}{b - a} \in (0, 1);
  2. c=(b−a)>0c = (b - a) > 0.

By Jensen's inequality,

E[eλX]=E[exp⁡(b−Xb−a⋅λa+X−ab−a⋅λb)]≤E[b−Xb−a⋅exp⁡(λa)+X−ab−a⋅exp⁡(λb)]=(1−θ)⋅exp⁡(λa)+θ⋅exp⁡(λb)=(1−θ+θeλ(b−a))exp⁡(λa)=(1−θ+θe(λc))e−θ(λc) \begin{aligned} \mathbb{E} [ e^{\lambda X } ] &= \mathbb{E} \left[ \exp \left( \frac{b - X}{ b - a} \cdot \lambda a + \frac{X - a}{b - a} \cdot \lambda b \right) \right] \\ &\le \mathbb{E} \left[ \frac{b - X}{ b - a} \cdot \exp \left( \lambda a \right) + \frac{X - a}{b - a} \cdot \exp \left( \lambda b \right) \right] \\ &= (1 - \theta) \cdot \exp \left( \lambda a \right) + \theta \cdot \exp \left( \lambda b \right) \\ &= \left( 1 - \theta + \theta e^{\lambda (b - a)} \right) \exp \left( \lambda a \right) \\ &= \left( 1 - \theta + \theta e^{ (\lambda c) } \right) e^ { - \theta (\lambda c) } \end{aligned}

Let f(z)≐−θz+ln⁡(1−θ+θez)f(z) \doteq - \theta z + \ln \left( 1 - \theta + \theta e^{ z} \right). The function is well-defined, as 1−θ+θez>01 - \theta + \theta e^{ z} > 0 for z∈Rz \in \mathbb{R}. Observe that

exp⁡{f(λc)}=(1−θ+θeλc)exp⁡(−λθc). \exp \Big \{ f(\lambda c) \Big \} = \left( 1 - \theta + \theta e^{\lambda c } \right) \exp \left( - \lambda \theta c \right).

Further,

f′(z)=−θ+θez1−θ+θez, f'(z) = - \theta + \frac{ \theta e^{ z} }{ 1 - \theta + \theta e^{ z} },

and

f′′(z)=θez(1−θ+θez)−(θez)2(1−θ+θez)2=(1−θ)θez(1−θ+θez)2≤14. f''(z) = \frac{ \theta e^{ z} ( 1 - \theta + \theta e^{ z} ) - ( \theta e^z )^2 }{ ( 1 - \theta + \theta e^{ z} )^2 } = \frac{ ( 1 - \theta ) \theta e^{ z} }{ ( 1 - \theta + \theta e^{ z} )^2 } \le \frac{1}{4}.

Finally, by Taylor's expansion, there exists some z∗∈[0,λc]z^* \in [0, \lambda c], such that

f(λc)=f(0)+f′(0)⋅(λc)+f′′(z∗)2(λc)2=f′′(z∗)2(λc)2≤18(λc)2=λ2(b−a)28. \begin{aligned} f(\lambda c) &= f(0) + f' (0) \cdot (\lambda c) + \frac{ f''(z^*) }{ 2 } (\lambda c)^2 \\ &= \frac{ f''(z^*) }{ 2 } (\lambda c)^2 \\ &\le \frac{1}{8 } (\lambda c)^2 \\ &= \frac{ \lambda^2 (b - a)^2 }{8 }. \end{aligned}

The picture above shows that the bound might not be tight.

□\square

Proof 2

Preliminaries

The proof relies on the following lemma.

Lemma. Let (Ω,F,P)(\Omega, \mathcal{F}, \mathbb{P}) be a probability space. Let YY be a nonnegative F\mathcal{F}-measurable function, and define the measure μ\mu by

μ(A)≐∫AY d P,A∈F. \mu(A) \doteq \int_A Y \, d\,\mathbb{P}, \qquad A \in \mathcal{F}.

Then for nonnegative F\mathcal{F}-measurable function ZZ, it holds that

∫Z d μ=∫Z⋅Y d P. \int Z \, d\, \mu = \int Z \cdot Y \, d\, \mathbb{P}.

Proof of the Lemma. If ZZ is an indicator function 1A1_A for some A∈FA \in \mathcal{F}, then the lemma clearly holds. It follows that the lemma holds for nonnegative simple functions. For general nonnegative ZZ, there is a sequence of nonnegative simple functions {Zn}\{ Z_n \}, such that Zn↑ZZ_n \uparrow Z. Via monotone convergence theorem, we obtain

lim⁡n∫Zn d μ=∫Z d μ,lim⁡n∫Zn⋅Y d P=∫Z⋅Y d P. \lim_n \int Z_n \, d\, \mu = \int Z \, d\, \mu, \\ \lim_n \int Z_n \cdot Y \, d\, \mathbb{P} = \int Z \cdot Y \, d\, \mathbb{P}.

■\blacksquare

Formal Proof

Let XX be a random variable such that a≤X≤ba \le X \le b. Define the function f:R≥0→Rf: \mathbb{R}_{\ge 0} \rightarrow \mathbb{R} as

f(λ)=E[exp⁡(λX)],  f(\lambda) = \mathbb{E}[\exp(\lambda X)],\,

and let

h(λ)=ln⁡f(λ). h(\lambda) = \ln f(\lambda).

We will justify the interchange of differentiation and expectation, and show that

f′(λ)=E[Xexp⁡(λX)], f′′(λ)=E[X2exp⁡(λX)]. f'(\lambda) = \mathbb{E}\left[ X \exp ( \lambda X) \right],\, \\ f''(\lambda) = \mathbb{E}\left[ X^2 \exp ( \lambda X) \right].

Therefore,

h′(λ)=f′(λ)f(λ)=E[Xexp⁡(λX)]E[exp⁡(λX)] ,h′′(λ)=f′′(λ)⋅f(λ)−(f′(λ))2(f(λ))2=E[X2exp⁡(λX)]⋅E[exp⁡(λX)]−(E[Xexp⁡(λX)])2(E[exp⁡(λX)])2=E[X2exp⁡(λX)]E[exp⁡(λX)]−(E[Xexp⁡(λX)])2(E[exp⁡(λX)])2. \begin{aligned} h'(\lambda) &= \frac{ f'(\lambda) }{ f(\lambda) } = \frac{ \mathbb{E}\left[ X \exp ( \lambda X) \right] }{ \mathbb{E}[\exp(\lambda X)] }\,, \\ h''(\lambda) &= \frac{ f''(\lambda) \cdot f(\lambda) - \left( f'(\lambda) \right)^2 }{ \left( f(\lambda) \right)^2 } \\ &= \frac{ \mathbb{E}[X^2 \exp(\lambda X)] \cdot \mathbb{E}[\exp(\lambda X)] - \left( \mathbb{E}[ X \exp(\lambda X)] \right)^2 }{ \left( \mathbb{E}[\exp(\lambda X)] \right)^2 } \\ &= \frac{ \mathbb{E}[X^2 \exp(\lambda X)] }{ \mathbb{E}[\exp(\lambda X)] } - \frac{ \left( \mathbb{E}[ X \exp(\lambda X)] \right)^2 }{ \left( \mathbb{E}[\exp(\lambda X)] \right)^2 }. \end{aligned}

Suppose (Ω,F,P)(\Omega, \mathcal{F}, \mathbb{P}) is the probability space on which XX is defined, i.e., XX is a function from F\mathcal{F} to R\mathbb{R}. Now we define a new measure Pλ:F→R\mathbb{P}_\lambda: \mathcal{F} \rightarrow \mathbb{R} as

Pλ(A)≐∫Aexp⁡(λX)E[exp⁡(λX)] dP,∀A∈F. \mathbb{P}_{\lambda}(A) \doteq \int_A \frac{\exp(\lambda X)}{ \mathbb{E}[\exp(\lambda X)] } \, d \mathbb{P}, \quad \forall A \in \mathcal{F}.

It is easy to see that Pλ\mathbb{P}_{\lambda} is a new probability measure. The function δ(λ)≐exp⁡(λX)E[exp⁡(λX)]\delta(\lambda) \doteq \frac{\exp(\lambda X)}{ \mathbb{E}[\exp(\lambda X)] } is called the density function of Pλ\mathbb{P}_{\lambda}. We see that

Eλ[X]≐∫XdPλ=∫Xexp⁡(λX)E[exp⁡(λX)] dP=E[Xexp⁡(λX)]E[exp⁡(λX)] ,Eλ[X2]≐∫X2dPλ=∫X2exp⁡(λX)E[exp⁡(λX)] dP=E[X2exp⁡(λX)]E[exp⁡(λX)] . \begin{aligned} \mathbb{E}_{\lambda} [X] \doteq \int X d { \mathbb{P}_{\lambda} } &= \int X \frac{\exp(\lambda X)}{ \mathbb{E}[\exp(\lambda X)] } \, d \mathbb{P} = \frac{ \mathbb{E}\left[ X \exp ( \lambda X) \right] }{ \mathbb{E}[\exp(\lambda X)] }\,, \\ \mathbb{E}_{\lambda} [X^2] \doteq \int X^2 d { \mathbb{P}_{\lambda} } &= \int X^2 \frac{\exp(\lambda X)}{ \mathbb{E}[\exp(\lambda X)] } \, d \mathbb{P} = \frac{ \mathbb{E}\left[ X^2 \exp ( \lambda X) \right] }{ \mathbb{E}[\exp(\lambda X)] }\,. \end{aligned}

It follows that

h′′(λ)=Varλ[X]≐Eλ[X2]−(Eλ[X])2. \begin{aligned} h''(\lambda) =\mathbb{Var}_{\lambda} [ X ] \doteq \mathbb{E}_{\lambda} [X^2] - \big( \mathbb{E}_{\lambda} [X] \big)^2. \end{aligned}

Therefore,

Varλ[X]=Varλ[X−a+b2]=Eλ[(X−a+b2)2]≤(b−a)24. \mathbb{Var}_{\lambda} [ X ] = \mathbb{Var}_{\lambda} \left[ X - \frac{a + b}{2} \right] = \mathbb{E}_{\lambda} \left[ \left( X - \frac{a + b}{2} \right)^2 \right] \le \frac{(b - a)^2}{4}.

Finally, via Taylor's expansion,

h(λ)=h(0)+λ⋅h′(0)+λ22h′′(λ∗). h(\lambda) = h(0) + \lambda \cdot h'(0) + \frac{\lambda^2}{2} h''(\lambda^*).

Since h(0)=0h(0) = 0, h′(0)=E[X]=0h'(0) = \mathbb{E}[X] = 0, and h′′(λ∗)=Varλ∗[X]≤(b−a)2/4h''(\lambda^*) = \mathbb{Var}_{\lambda^*}[X] \le (b - a)^2 / 4, we conclude that

h(λ)≤(b−a)28. h(\lambda) \le \frac{ (b - a)^2 }{ 8 }.

First Derivative of ff.

Denote M≐max⁡{∣a∣,∣b∣}M \doteq \max\{ |a|, |b| \}. For each n∈N+n \in \mathbb{N}_+,

g(n)≐f(λ+1/n)−f(λ)1/n=E[exp⁡((λ+1/n)⋅X)−exp⁡(λX)1/n] g(n) \doteq \frac{ f(\lambda + 1 / n) - f(\lambda) }{ 1 / n} = \mathbb{E}\left[ \frac{ \exp( (\lambda + 1 / n) \cdot X) - \exp(\lambda X) }{ 1 / n} \right]

Via mean value theorem, there exists λ′∈[λ,λ+1/n]\lambda' \in [\lambda, \lambda + 1 / n], s.t.,

exp⁡((λ+1/n)⋅X)−exp⁡(λX)1/n=X⋅exp⁡(λ′⋅X)≤M⋅exp⁡((λ+1)⋅M). \frac{ \exp( (\lambda + 1 / n) \cdot X) - \exp(\lambda X) }{ 1 / n} = X \cdot \exp ( \lambda' \cdot X) \le M \cdot \exp( (\lambda + 1) \cdot M).

Since the RHS is integrable, via dominated convergence theorem,

f′(λ)=lim⁡ng(n)=E[lim⁡nexp⁡((λ+1/n)⋅X)−exp⁡(λX)1/n]=E[Xexp⁡(λX)]. f'(\lambda) = \lim_n g(n) = \mathbb{E}\left[ \lim_n \frac{ \exp( (\lambda + 1 / n) \cdot X) - \exp(\lambda X) }{ 1 / n} \right] = \mathbb{E}\left[ X \exp ( \lambda X) \right] .

Second Derivative of ff.

Similarly, for each n∈Nn \in \mathbb{N}, there exists λ′∈[λ,λ+1/n]\lambda' \in [\lambda, \lambda + 1 / n], s.t.,

X⋅exp⁡((λ+1/n)⋅X)−X⋅exp⁡(λX)1/n=X2⋅exp⁡(λ′⋅X)≤M2⋅exp⁡((λ+1)⋅M). \frac{ X \cdot \exp( (\lambda + 1 / n) \cdot X) - X \cdot \exp(\lambda X) }{ 1 / n} = X^2 \cdot \exp ( \lambda' \cdot X) \le M^2 \cdot \exp( (\lambda + 1) \cdot M).

It follows via dominated convergence theorem that

f′′(λ)=lim⁡nE[X⋅exp⁡((λ+1/n)⋅X)−X⋅exp⁡(λX)1/n]=E[lim⁡nX⋅exp⁡((λ+1/n)⋅X)−X⋅exp⁡(λX)1/n]=E[X2exp⁡(λX)]. \begin{aligned} f''(\lambda) &= \lim_n \mathbb{E}\left[ \frac{ X \cdot \exp( (\lambda + 1 / n) \cdot X) - X \cdot \exp(\lambda X) }{ 1 / n} \right] \\ &= \mathbb{E}\left[ \lim_n \frac{ X \cdot \exp( (\lambda + 1 / n) \cdot X) - X \cdot \exp(\lambda X) }{ 1 / n} \right] \\ &= \mathbb{E}\left[ X^2 \exp ( \lambda X) \right] . \end{aligned}

□\square