-
Notifications
You must be signed in to change notification settings - Fork 12
Expand file tree
/
Copy pathlearning.tex
More file actions
320 lines (289 loc) · 22.3 KB
/
Copy pathlearning.tex
File metadata and controls
320 lines (289 loc) · 22.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
\documentclass[10pt]{article}
\def\withcolors{1}
\def\withnotes{1}
\def\withindex{0}
\input{packages}
\input{preamble}
\newcommand{\dst}{\varepsilon}
\newcommand{\ab}{k}
\newcommand{\ns}{n}
\usepackage{filecontents}
\title{A short note on learning discrete distributions}
\date{February, 2020}
\begin{document}
\begin{flushleft}\sf\footnotesize
\makeatletter
\@date~- \today \hfill \@title
\makeatother
\end{flushleft}
\vspace{5mm}
The goal of this short note is to provide simple proofs for the ``folklore facts'' on the sample complexity of learning a discrete probability distribution over a known domain of size $\ab$ to various distances $\dst$, with error probability $\delta$. Thanks to \href{http://www.gautamkamath.com/}{Gautam Kamath} and \href{http://www.mit.edu/~jswright/}{John Wright} for suggesting ``someone should write this up as a note,'' and to \href{https://people.eecs.berkeley.edu/~jiantao/}{Jiantao Jiao} for discussions about the Hellinger case.\medskip
For a given distance measure $\operatorname{d}$, we write $\Phi(\operatorname{d},\ab,\dst,\delta)$ for the sample complexity of learning discrete distributions over a known domain of size $\ab$, to accuracy $\dst>0$, with error probability $\delta\in(0,1]$. As usual, asymptotics will be taken with regard to $\ab$ going to infinity, $\dst$ going to $0$, and $\delta$ going to $0$, in that order.
\noindent Without loss of generality, we hereafter assume the domain is the set $[\ab]\eqdef \{1,\dots,\ab\}$.
\section{Total variation distance}
Recall that $\totalvardist{p}{q} = \sup_{S\subseteq [\ab]} (p(S)-q(S)) = \frac{1}{2}\normone{p-q}\in[0,1]$ for any $p,q\in\distribs{[\ab]}$.
\begin{theorem}\label{theo:learning:tv}
$\Phi(\dtv,\ab,\dst,\delta) = \bigTheta{\frac{\ab+\log(1/\delta)}{\dst^2}}$.
\end{theorem}
\noindent We focus here on the upper bound. The lower bound can be proven, e.g., via Assouad's lemma (for the $\ab/\dst^2$ term), and from the hardness of estimating the bias of a coin ($\ab=2$) with high probability (for the $\log(1/\delta)/\dst^2$ term).
\begin{proof}[First proof]
Consider the empirical distribution $\tilde{p}$ obtained by drawing $\ns$ independent samples $s_1,\dots,s_\ns$ from the underlying distribution $p\in\distribs{[\ab]}$:
\begin{equation}\label{def:empirical}
\tilde{p}(i) = \frac{1}{\ns} \sum_{j=1}^\ns \indic{s_j=i}, \qquad i\in [\ab]
\end{equation}
\begin{itemize}
\item First, we bound the \emph{expected} total variation distance between $\tilde{p}$ and $p$, by using $\lp[2]$ distance as a proxy:
\[
\expect{ \totalvardist{p}{\tilde{p}} }
=\frac{1}{2}\expect{ \normone{p-\tilde{p}}}
=\frac{1}{2}\sum_{i=1}^\ab\expect{ \abs{p(i)-\tilde{p}(i)}}
\leq\frac{1}{2}\sum_{i=1}^\ab\sqrt{\expect{ (p(i)-\tilde{p}(i))^2} }
\]
the last inequality by Jensen. But since, for every $i\in[\ab]$, $\ns\tilde{p}(i)$ follows a $\binomial{\ns}{p(i)}$ distribution, we have
$\expect{ (p(i)-\tilde{p}(i))^2} = \frac{1}{\ns^2}\var[\ns\tilde{p}(i)] = \frac{1}{\ns}p(i)(1-p(i))$, from which
\[
\expect{ \totalvardist{p}{\tilde{p}} } \leq\frac{1}{2\sqrt{\ns}}\sum_{i=1}^\ab\sqrt{p(i)} \leq \frac{1}{2}\sqrt{\frac{\ab}{\ns}}
\]
the last inequality this time by Cauchy--Schwarz. Therefore, for $\ns\geq \frac{\ab}{\dst^2}$ we have $\expect{ \totalvardist{p}{\tilde{p}} }\leq \frac{\dst}{2}$.
\item Next, to convert this expected result to a \emph{high probability} guarantee, we apply McDiarmid's inequality to the random variable $f(s_1,\dots,s_\ns) \eqdef \totalvardist{p}{\tilde{p}}$, noting that changing any single sample cannot change its value by more than $c\eqdef 1/\ns$:
\[
\probaOf{ \abs{f(s_1,\dots,s_\ns) - \expect{f(s_1,\dots,s_\ns)}} \geq \frac{\dst}{2} } \leq 2e^{-\frac{2\left(\frac{\dst}{2}\right)^2}{\ns c^2}} = 2e^{-\frac{1}{2}\ns\dst^2}
\]
and therefore as long as $\ns\geq \frac{2}{\dst^2}\ln\frac{2}{\delta}$, we have $\abs{f(s_1,\dots,s_\ns) - \expect{f(s_1,\dots,s_\ns)}} \leq \frac{\dst}{2}$ with probability at least $1-\delta$.
\end{itemize}
Putting it all together, we obtain that $\totalvardist{p}{\tilde{p}} \leq \dst$ with probability at least $1-\delta$, as long as $\ns\geq \max\left( \frac{\ab}{\dst^2},\frac{2}{\dst^2}\ln\frac{2}{\delta} \right)$.
\end{proof}
\begin{proof}[Second proof -- the ``fun'' one]
Again, we will analyze the behavior of the empirical distribution $\tilde{p}$ over $\ns$ i.i.d. samples from the unknown $p$ (cf.~\eqref{def:empirical}) -- because it is simple, efficiently computable, and \emph{it works}. Recalling the definition of total variation distance, note that $\totalvardist{p}{\tilde{p}} > \dst$ literally means there exists a subset $S\subseteq [\ab]$ such that $\tilde{p}(S) > p(S) + \dst$. There are $2^\ab$ such subsets, so\dots{} let us do a union bound.
Fix any $S\subseteq[\ab]$. We have
\[
\tilde{p}(S) = \sum_{i\in S} \tilde{p}(i) \operatorname*{=}^{\eqref{def:empirical}} \frac{1}{\ns} \sum_{i\in S} \sum_{j=1}^\ns \indic{s_j=i}
\]
and so, letting $X_j \eqdef \sum_{i\in S}\indic{s_j=i}$ for $j\in [\ns]$, we have
$
\tilde{p}(S) = \frac{1}{\ns}\sum_{j=1}^\ns X_j
$ where the $X_j$'s are i.i.d. Bernoulli random variable with parameter $p(S)$. Here comes the Chernoff bound (actually, Hoeffding, the \emph{other} Chernoff):
\[
\probaOf{ \tilde{p}(S) > p(S) + \dst } = \probaOf{ \frac{1}{\ns}\sum_{j=1}^\ns X_j > \expect{\frac{1}{\ns}\sum_{j=1}^\ns X_j} + \dst } \leq e^{-2\dst^2 \ns}
\]
and therefore $\probaOf{ \tilde{p}(S) > p(S) + \dst } \leq \frac{\delta}{2^\ab}$ for any $\ns\geq \frac{\ab\ln 2+\log(1/\delta)}{2\dst^2}$. A union bound over these $2^\ab$ possible sets $S$ concludes the proof:
\[
\probaOf{ \exists S\subseteq [\ab] \text{ s.t. }\tilde{p}(S) > p(S) + \dst } \leq 2^\ab\cdot \frac{\delta}{2^\ab} = \delta
\]
and we are done. \emph{Badda bing badda boom}, as someone\footnote{John Wright.} would say.
\end{proof}
\section{Hellinger distance}
Recall that $\hellinger{p}{q} = \frac{1}{\sqrt{2}}\sqrt{\sum_{i=1}^\ab (\sqrt{p(i)}-\sqrt{q(i)})^2} = \frac{1}{\sqrt{2}}\normtwo{\sqrt{p}-\sqrt{q}}\in[0,1]$ for any $p,q\in\distribs{[\ab]}$. The Hellinger distance has many nice properties: it is well-suited to manipulating product distributions, its square is subadditive, and is always within a quadratic factor of the total variation distance; see, e.g.,~\cite[Appendix~C.2]{Canonne:15}.
\begin{theorem}\label{theo:learning:hellinger}
$\Phi(\dhell,\ab,\dst,\delta) = \bigTheta{\frac{\ab+\log(1/\delta)}{\dst^2}}$.
\end{theorem}
This theorem is ``highly non-trivial'' to establish, however; for the sake of exposition, we will show increasingly stronger bounds, starting with the easiest to establish.
\begin{proposition}[Easy bound]\label{theo:learning:hellinger:easy}
$\Phi(\dhell,\ab,\dst,\delta) = \bigO{\frac{\ab+\log(1/\delta)}{\dst^4}}$, and $\Phi(\dhell,\ab,\dst,\delta) = \bigOmega{\frac{\ab+\log(1/\delta)}{\dst^2}}$.
\end{proposition}
\begin{proof}
This is immediate from~\autoref{theo:learning:tv}, recalling that $\frac{1}{2}\dtv^2\leq \dhell^2\leq \dtv$.
\end{proof}
\begin{proposition}[More involved bound]\label{theo:learning:hellinger:intermediate}
$\Phi(\dhell,\ab,\dst,\delta) = \bigO{\frac{\ab}{\dst^2}+\frac{\log(1/\delta)}{\dst^4}}$.
\end{proposition}
\begin{proof}
As for total variation distance, we consider the empirical distribution $\widehat{p}$ (cf.~\eqref{def:empirical}) obtained by drawing $\ns$ independent samples $s_1,\dots,s_\ns$ from $p\in\distribs{[\ab]}$.
\begin{itemize}
\item First, we bound the \emph{expected} squared Hellinger distance between $\widehat{p}$ and $p$: using the simple fact that
$\hellinger{p}{q}^2 = 1-\sum_{i=1}^\ab \sqrt{p(i)q(i)}$ for any $p,q\in\distribs{[\ab]}$,
\[
\expect{ \hellinger{p}{\widehat{p}}^2 } = 1-\sum_{i=1}^\ab \sqrt{p(i)}\cdot \expect{\sqrt{\widehat{p}(i)}}\,.
\]
Now we would like to handle the square root inside the expectation, and \emph{of course} Jensen's inequality is in the wrong direction. However, for every nonnegative r.v. $X$ with positive expectation, letting $Y\eqdef X/\expect{X}$, we have that
\begin{align*}
\expect{\sqrt{X}}
&= \sqrt{\expect{X}}\cdot\expect{\sqrt{Y}}
= \sqrt{\expect{X}}\cdot\expect{\sqrt{1+(Y-\expect{Y})})} \\
&\geq \sqrt{\expect{X}} \mleft( 1+ \frac{1}{2}\expect{Y-\expect{Y}} - \frac{1}{2} \expect{(Y-\expect{Y})^2} \mright)
= \sqrt{\expect{X}}\mleft(1-\frac{\var X}{2\expect{X}^2}\mright)
\end{align*}
where we used the inequality $\sqrt{1+x} \geq 1+\frac{x}{2}-\frac{x^2}{2}$, which holds for $x\geq -1$.\footnote{And is inspired by the Tayor expansion $\sqrt{1+x} = 1+\frac{x}{2} - \frac{x^2}{8} +o(x^2)$: there is \emph{some} intuition for it.}{} Since, for every $i\in[\ab]$, $\ns\widehat{p}(i)$ follows a $\binomial{\ns}{p(i)}$ distribution, we get
\[
\expect{ \hellinger{p}{\widehat{p}}^2 } \leq 1-\frac{1}{\sqrt{\ns}}\sum_{i=1}^\ab \sqrt{p(i)}\cdot\sqrt{\ns p(i)} \mleft(1-\frac{\ns p(i)(1-p(i))}{2\ns^2 p(i)^2}\mright)
\leq 1 - \sum_{i=1}^\ab p(i) \mleft(1-\frac{1}{2\ns p(i)}\mright) = \frac{\ab}{2\ns}\,.
\]
Therefore, for $\ns\geq \frac{\ab}{\dst^2}$, we have $\expect{ \hellinger{p}{\widehat{p}}^2 }\leq \frac{\dst^2}{2}$.
\item Next, to convert this expected result to a high probability guarantee, we \emph{would like} to apply McDiarmid's inequality to the random variable $f(s_1,\dots,s_\ns) \eqdef \hellinger{p}{\widehat{p}}^2$ as in the (first) proof of~\autoref{theo:learning:tv}; unfortunately, changing a sample can change the value by up to $c \approx 1/\sqrt{\ns}$, and McDiarmid will yield
only a vacuous bound.\footnote{Try it: it's a real bummer.}{} Instead, we will use a stronger, more involved concentration inequality:
\begin{theorem}[{\cite[Theorem~8.6]{BLM:13}}]\label{theo:stronger:mcdiarmid}
Let $f\colon \mathcal{X}^\ns \to \R$ be a measurable function, and let $X_1,\dots,X_\ns$ be independent random variables taking values in $\mathcal{X}$. Define $Z\eqdef f(X_1,\dots,X_\ns)$. Assume that there exist measurable functions $c_i\colon \mathcal{X}^\ns \to [0,\infty)$ such that, for all $x,y\in\mathcal{X}^\ns$,
\[
f(y) - f(x) \leq \sum_{i=1}^\ns c_i(x) \indic{x_i \neq y_i}\,.
\]
Then, setting $v \eqdef \shortexpect \sum_{i=1}^\ns c_i(x)^2$ and $v_\infty \eqdef \sup_{x\in\mathcal{X}^\ns} \sum_{i=1}^\ns c_i(x)^2$, we have, for all $t>0$,
\[
\probaOf{ Z \geq \expect{Z} + t } \leq e^{-\frac{t^2}{2v}}\,\qquad \probaOf{ Z \leq \expect{Z} - t } \leq e^{-\frac{t^2}{2v_\infty}}\,.
\]
\end{theorem}
For our $f$ above, we have, for two any different $x, y \in [\ab]^\ns$ , that
\begin{align*}
f(y)-f(x)
&= \frac{1}{\sqrt{\ns}} \sum_{i=1}^\ab \sqrt{p(i)} \mleft( \sqrt{\sum_{j=1}^\ns \indic{x_j=i}} - \sqrt{\sum_{j=1}^\ns \indic{y_j=i}} \mright) \\
&= \frac{1}{\sqrt{\ns}} \sum_{i=1}^\ab \sqrt{p(i)} \frac{\sum_{j=1}^\ns (\indic{x_j=i}-\indic{y_j=i})}{ \sqrt{\sum_{j=1}^\ns \indic{x_j=i}} + \sqrt{\sum_{j=1}^\ns \indic{y_j=i}} } \\
&\leq \frac{1}{\sqrt{\ns}} \sum_{i=1}^\ab \sqrt{p(i)} \frac{\sum_{j=1}^\ns \indic{x_j=i}\indic{y_j\neq x_j}}{ \sqrt{\sum_{j=1}^\ns \indic{x_j=i}} }
= \sum_{j=1}^\ns \underbrace{\sqrt{ \frac{ p_{x_j} }{ \ns\sum_{\ell=1}^\ns \indic{x_\ell=x_j} } }}_{c_j(x)} \cdot \indic{x_j\neq y_j}\,.
\end{align*}
In view of~\autoref{theo:stronger:mcdiarmid}, we then must evaluate
\[
v \eqdef \sum_{j=1}^\ns \expect{c_j(X)^2} = \frac{1}{\ns} \sum_{j=1}^\ns \sum_{i=1}^\ab p(i)^2 \cdot \expect{ \frac{1}{1+ \sum_{\ell\neq j} \indic{X_\ell=i} } }
\]
where that last expectation is over $(x_\ell)_{\ell\neq j}$ drawn from $p^{\otimes(\ns-1)}$. Since $\sum_{\ell\neq j} \indic{X_\ell=i}$ is Binomially distributed with parameters $\ns-1$ and $p(i)$, we can use the simple fact that, for $N\sim\binomial{r}{\rho}$,
\[
\expect{\frac{1}{N+1}} = \frac{1-(1-\rho)^{r+1}}{\rho(r+1)} \leq \frac{1}{\rho(r+1)}
\]
to conclude that $v \leq \frac{1}{\ns^2} \sum_{j=1}^\ns \sum_{i=1}^\ab p(i) = \frac{1}{\ns}$. By~\autoref{theo:stronger:mcdiarmid}, we obtain
\[
\probaOf{ \abs{f(s_1,\dots,s_\ns)-\expect{f(s_1,\dots,s_\ns)}} \geq \frac{\dst^2}{2} } \leq e^{-\frac{1}{8}\ns\dst^4}
\]
and therefore, as long as $\ns \geq \frac{8}{\dst^4}\ln\frac{1}{\delta}$, we have $\abs{f(s_1,\dots,s_\ns)-\expect{f(s_1,\dots,s_\ns)}} \leq \frac{\dst^2}{2}$ with probability at least $1-\delta$.
\end{itemize}
Putting it all together, we obtain that $\hellinger{p}{\widehat{p}}^2 \leq \dst^2$ with probability at least $1-\delta$, as long as $\ns \geq \max\mleft( \frac{\ab}{\dst^2}, \frac{8}{\dst^4}\ln\frac{1}{\delta}\mright)$.
\end{proof}
\noindent We finally get to the final, optimal bound:
\begin{proofof}{\autoref{theo:learning:hellinger}}
We will rely on a recent~--~and quite involved~--~result due to Agrawal [Agr19], analyzing the concentration of the empirical distribution $\widehat{p}$ in terms of its Kullback--Leibler (KL) divergence with regard to the true $p$,
\[
\kldiv{\widehat{p}}{p} = \sum_{i=1}^\ab \widehat{p}(i) \ln \frac{\widehat{p}(i)}{p(i)} \in[0,\infty]\,.
\]
Observing that $\hellinger{p}{q}^2 \leq \frac{1}{2} \kldiv{p}{q}$ for any distributions $p, q$, the aforementioned result is actually stronger than what we need:
\begin{theorem}[{\cite[Theorem~1.2]{Agrawal:19}}]
Suppose $\ns \geq \frac{\ab-1}{\alpha}$. Then
\[
\probaOf{ \kldiv{\widehat{p}}{p} \geq \alpha } \leq e^{-\ns\alpha}\mleft(\frac{e\alpha\ns}{\ab-1}\mright)^{\ab-1}\,.
\]
\end{theorem}
\noindent In view of the above relation between Hellinger and KL, we will apply this convergence result with $\alpha \eqdef 2\dst^2$, obtaining
\[
\probaOf{ \hellinger{\widehat{p}}{p} \geq \dst } \leq e^{-2\ns\dst^2 + (\ab-1)\ln\frac{2e\ns\dst^2}{\ab-1}}\,.
\]
\begin{fact}
For $\ns \geq \frac{15}{2e}\frac{\ab}{\dst^2}$, we have $(\ab-1)\ln\frac{2e\ns\dst^2}{\ab-1}\leq \ns\dst^2$.
\end{fact}
\begin{proof}
The conclusion is equivalent to $2e\cdot\ln\frac{2e\ns\dst^2}{\ab-1}\leq \frac{2e\ns\dst^2}{\ab-1}$, and thus follows from the fact that $x\geq 2e \ln x$ for $x\geq 15$.
\end{proof}
\noindent This fact implies that, for $\ns \geq \frac{15\ab}{2\dst^2}$, $\probaOf{ \hellinger{\widehat{p}}{p} \geq \dst } \leq e^{-\ns\dst^2}$. Overall, we obtain that $\hellinger{p}{\widehat{p}} \leq \dst$ with probability at least $1-\delta$ as long as $\ns \geq \max\mleft( \frac{15\ab}{2e\dst^2}, \frac{1}{\dst^2}\ln\frac{1}{\delta}\mright)$, as desired.
\end{proofof}
\section{$\chi^2$ and Kullback–-Leibler divergences}
In view of the previous section, some remarks on Kullback--Leibler (KL) and chi-squared ($\chi^2$) divergences. Recall their definition, for $p,q\in\distribs{[\ab]}$,
\[
\kldiv{p}{q} = \sum_{i=1}^\ab p(i) \ln \frac{p(i)}{q(i)}\,,\qquad \chisquare{p}{q} = \sum_{i=1}^\ab \frac{(p(i)-q(i))^2}{q(i)}
\]
both taking values in $[0,\infty]$; as well as the chain of (easily checked) inequalities
\[
2\totalvardist{p}{q}^2 \leq \kldiv{p}{q} %\leq \ln(1+\chisquare{p}{q})
\leq \chisquare{p}{q}\,,
\]
where the first one is Pinsker's. Importantly, KL and $\chi^2$ divergences are unbounded and asymmetric, so
the order of p and q matters \emph{a lot}: for instance, it is easy to show that, without strong assumptions on the
unknown distribution $p\in\distribs{[\ab]}$, the empirical estimator $\widehat{p}$ cannot achieve $\kldiv{p}{\widehat{p}} < \infty$ (resp., $\chisquare{p}{\widehat{p}} < \infty$) with any finite number of samples.\footnote{You can verify this: intuitively, the issue boils down to having to non-trivially learn even the elements of the support of $p$ that have arbitrarily small probability.}{} So, that's uplifting. (On the other hand, \emph{other} estimators than the empirical one, e.g., add-constant estimators, do provide good learning guarantees for those distance measures: see for instance~\cite{Kamath:15}).\smallskip
We are going to focus here on getting $\kldiv{\widehat{p}}{p}$ and $\chisquare{\widehat{p}}{p}$ down to $\dst$. Of course, in view of the inequalities above, the latter is at least as hard as the former, and a lower bound on both follows from that on $\dtv$: $\bigOmega{(\ab+\log(1/\delta))/\dst^2}$. And, behold! The result of Agrawal~\cite{Agrawal:19} used in the proof of~\autoref{theo:learning:hellinger} does provide the optimal upper bound on learning in KL divergence~--~and it is achieved by the usual suspect, the empirical estimator:
\begin{theorem}\label{theo:learning:kl}
$\Phi(\kl,\ab,\dst,\delta) = \bigTheta{\frac{\ab+\log(1/\delta)}{\dst}}$, where by $\kl$ we refer to minimizing $\kldiv{\widehat{p}}{p}$.
\end{theorem}
\noindent The optimal sample complexity of learning in $\chi^2$ as a function of $\ab,\dst,\delta$, however, remains open.
\section{Briefly: Kolmogorov, $\lp[\infty]$, and $\lp[2]$ distances}
To conclude, let us briefly discuss three other distance measures: Kolmogorov (a.k.a., ``$\lp[\infty]$ between cumulative distribution functions''), $\lp[\infty]$, and $\lp[2]$:
\[
\kolmogorov{p}{q} = \max_{i\in[\ab]} \abs{\sum_{j=1}^i p(j) - \sum_{j=1}^i q(j)}
\] and
\[
\lp[2](p,q) = \normtwo{p-q} = \sqrt{\sum_{i=1}^\ab (p(i)-q(i))^2},\qquad
\lp[\infty](p,q) = \norminf{p-q} = \max_{i\in[\ab]} \abs{p(i)-q(i)}\,.
\]
A few remarks first. The Kolmogorov distance is actually defined for any distribution on $\R$, not necessarily discrete; one can equivalently define it as
$
\kolmogorov{p}{q} =\sup_{i} ( \shortexpect_p[\indicSet{(-\infty,i]}] - \shortexpect_q[\indicSet{(-\infty,i]}] )$. This has a nice interpretation: recalling the definition of TV distance, both are of the form $\sup_{f \in\mathcal{C}} (\shortexpect_p[f] - \shortexpect_q[f])
$ where $\mathcal{C}$ is a class of measurable functions.\footnote{Such metrics on the space of probability distributions are called \emph{integral probability metrics}.}{} For TV distance, $\mathcal{C}$ is the class of indicators of all measurable subsets; for Kolmogorov, this is the (smaller) class of indicators of intervals of the form $(-\infty, a]$. (For Wasserstein/EMD distance, this will be the class of continuous, $1$-Lipschitz functions.)
Second, because of the above, and also monotonicity of $\lp[p]$ norms, Cauchy--Schwarz, the fact that $\lp[1](p,q) = 2\totalvardist{p}{q}$, and elementary manipulations, we have
\[
\lp[\infty](p,q)\leq \lp[2](p,q)\leq 2\totalvardist{p}{q} \leq \sqrt{\ab}\lp[2](p,q), \quad
\lp[2](p,q)\leq \sqrt{\lp[\infty](p,q)}, \quad \frac{1}{2}\lp[\infty](p,q)\leq \kolmogorov{p}{q} \leq \totalvardist{p}{q}\,.
\]
That can be useful sometimes. Now, I will only briefly sketch the proof of the next theorem: the lower bounds follow from the simple case $\ab=2$ (estimating the bias of a biased coin), the upper bounds are achieved by the empirical estimator (again). Importantly, the result for Kolmogorov distance \emph{still applies to continuous, arbitrary distributions}.
\begin{theorem}\label{theo:learning:kolmogorov}
$\Phi(\operatorname{d_{\rm{}K}},\ab,\dst,\delta),\Phi(\lp[\infty],\ab,\dst,\delta),\Phi(\lp[2],\ab,\dst,\delta) = \bigTheta{\frac{\log(1/\delta)}{\dst^2}}$, independent of $\ab$.
\end{theorem}
\begin{proof}[Sketch]
The proof for Kolmogorov distance is the most involved, and follows from a \emph{very} useful and non-elementary theorem due to Dvoretzky, Kiefer, and Wolfowitz from 1956~\cite{DKW:56} (with the optimal constant due to Massart, in 1990~\cite{Massart:90}):
\begin{theorem}[DKW Inequality]
Let $\hat{p}$ denote the empirical distribution on $\ns$ i.i.d. samples from $p$ (an \emph{arbitrary} distribution on $\R$). Then, for every $\dst > 0$,
\[
\probaOf{ \kolmogorov{\hat{p}}{p} > \dst } \leq 2e^{-2\ns \dst^2 }\,.
\]
Note, again, that this holds even if $p$ is a \emph{continuous} (or arbitrary) distribution on an unbounded support.
\end{theorem}
The proof for $\lp[\infty]$ just follows the Kolmogorov upper bound and the aforementioned inequality $\lp[\infty](p,q) \leq 2\kolmogorov{p}{q}$ (which hinges on the fact that $p(i) = \sum_{j=1}^i p(i) - \sum_{j=1}^{i-1} p(i)$ and the triangle inequality). Finally, the proof for $\lp[2]$ is a nice exercise involving analyzing the expectation of the $\lp[2]^2$ distance achieved by the empirical estimator, and McDiarmid's inequality.
\end{proof}
%%%%%%%%%% Bibliography
\begin{filecontents}{references-learning.bib}
@article{Agrawal:19,
author = {Rohit Agrawal},
title = {Multinomial Concentration in Relative Entropy at the Ratio of Alphabet and Sample Sizes},
journal = {CoRR},
volume = {abs/1904.02291},
year = {2019}
}
@book{BLM:13,
title={Concentration inequalities: A nonasymptotic theory of independence},
author={Boucheron, St{\'e}phane and Lugosi, G{\'a}bor and Massart, Pascal},
year={2013},
publisher={Oxford University Press}
}
@article{Canonne:15,
author = {Cl{\'{e}}ment L. Canonne},
title = {{A Survey on Distribution Testing: Your Data is Big. But is it Blue?}},
journal = {Electronic Colloquium on Computational Complexity {(ECCC)}},
volume = {22},
pages = {63},
year = {2015}
}
@inproceedings{Kamath:15,
title = {On Learning Distributions from their Samples},
author = {Sudeep Kamath and Alon Orlitsky and Dheeraj Pichapati and Ananda Theertha Suresh},
booktitle = {Proceedings of The 28th Conference on Learning Theory},
pages = {1066--1100},
year = {2015},
volume = {40},
series = {Proceedings of Machine Learning Research},
publisher = {PMLR}
}
@article{DKW:56,
AUTHOR = {Dvoretzky, Aryeh and Kiefer, Jack and Wolfowitz, Jacob},
TITLE = {Asymptotic minimax character of the sample distribution
function and of the classical multinomial estimator},
JOURNAL = {Ann. Math. Statist.},
FJOURNAL = {Annals of Mathematical Statistics},
VOLUME = {27},
YEAR = {1956},
PAGES = {642--669},
ISSN = {0003-4851},
MRCLASS = {62.0X},
MRNUMBER = {83864},
DOI = {10.1214/aoms/1177728174}
}
@article{Massart:90,
AUTHOR = {Massart, Pascal},
TITLE = {The tight constant in the {D}voretzky-{K}iefer-{W}olfowitz
inequality},
JOURNAL = {Ann. Probab.},
FJOURNAL = {The Annals of Probability},
VOLUME = {18},
YEAR = {1990},
NUMBER = {3},
PAGES = {1269--1283},
ISSN = {0091-1798},
MRCLASS = {60E15 (60G50 62G30)},
MRNUMBER = {1062069}
}
\end{filecontents}
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
\bibliographystyle{alpha}
\bibliography{references-learning}
\end{document}