Estimating Renyi Entropy of Discrete Distributions

Jayadev Acharya, Alon Orlitsky, Ananda Theertha Suresh, Himanshu Tyagi

I Introduction

A popular generalization of Shannon entropy is the Rényi entropy of order α≥0\alpha\geq 0, defined for α≠1\alpha\neq 1 by

Motivated by these and other applications, unbiased and heuristic estimators of Rényi entropy have been studied in the physics literature following , and asymptotically consistent and normal estimates were proposed in . However, no systematic study of the complexity of estimating Rényi entropy is available. For example, it was hitherto unknown if the number of samples needed to estimate the Rényi entropy of a given order α\alpha differs from that required for Shannon entropy, or whether it varies with the order α\alpha, or how it depends on the alphabet size kk.

I-B Definitions and results

We show that Sα(k)S_{\alpha}(k) behaves differently in three ranges of α\alpha. For 0≤α<10\leq\alpha<1,

namely the sample complexity grows super-linearly in kk and estimating the Rényi entropy of these orders is even more difficult than estimating the Shannon entropy. In fact, the upper bound follows from a corresponding result on estimation of power sums considered in (see Section III-C for further discussion). For completeness, we show in Theorem 10 that the empirical estimator requires O(k1/α)O(k^{1/\alpha}) samples and in Theorem 14 prove the improvement by a factor of log⁡k\log k. The lower bound is proved in Theorem 22.

namely as with Shannon entropy, the sample complexity grows roughly linearly in the alphabet size. The lower bound is proved in Theorem 21. In a conference version of this paper , a weaker O(k)O(k) upper bound was established using the empirical-frequency estimator. For the sake of completeness, we include this result as Theorem 9. The tighter upper bound reported here uses the best polynomial approximation based estimator of and is proved in Theorem 13. In fact, in the Appendix we show that the empirical estimator can’t attain this log⁡k\log k improvement and requires Ω(k/δ)\Omega(k/\delta) and Ω((k/δ)1/α)\Omega((k/\delta)^{1/\alpha}) samples for α>1\alpha>1 and α<1\alpha<1, respectively.

and in particular, the sample complexity is strictly sublinear in the alphabet size. The upper and lower bounds are shown in Theorems 12 and 16, respectively. Figure 1 illustrates our results for different ranges of α\alpha.

Of the three ranges, the most frequently used, and coincidentally the one for which the results are most surprising, is the last with α=2,3,…\alpha=2,3,\ldots. Some elaboration is in order.

It should also be noted that the estimators achieving the upper bounds are simple and run in time linear in the number of samples. Furthermore, the estimators are universal in that they do not require the knowledge of kk. On the other hand, the lower bounds on Sα(k)S_{\alpha}(k) hold even if the estimator knows kk.

I-C The estimators

and is related to the Rényi entropy for α≠1\alpha\neq 1 via

We construct estimators for the power-sums of distributions with a multiplicative-accuracy of (1±δ)(1\pm\delta) and hence obtain an additive-accuracy of Θ(δ)\Theta(\delta) for Rényi entropy estimation. We consider the following three different estimators for different ranges of α\alpha and with different performance guarantees.

Bias-corrected estimator

Polynomial approximation estimator

where dd and τ\tau are both O(log⁡n)O(\log n) and chosen appropriately.

Theorem 13 and Theorem 14 show that for α>1\alpha>1 and α<1\alpha<1, respectively, the sample complexity of P^αd,τ\widehat{P}_{\alpha}^{d,\tau} is O(k/log⁡k)O(k/\log k) and O(k1α/log⁡k)O(k^{\frac{1}{\alpha}}/\log k), resulting in a reduction in sample complexity of O(log⁡k)O(\log k) over the empirical estimator.

Table I summarizes the performance of these estimators in terms of their sample complexity. The last column denote the lower bounds from Section V.

I-D Organization

The rest of the paper is organized as follows. Section II presents basic properties of power sums of distributions and moments of Poisson random variables, which may be of independent interest. The estimation algorithms are analyzed in Section III, in Section III-A we show results for the empirical or plug-in estimate, in Section III-B we provide optimal results for integral α\alpha and finally we provide an improved estimator for non-integral α>1\alpha>1. Examples and simulation of the proposed estimators are given in Section IV. Section V contains our lower bounds for the sample complexity of estimating Rényi entropy. Furthermore, in the Appendix we analyze the performance of the empirical estimator for power-sum estimation with an additive-accuracy and also derive lower bounds for its sample complexity.

II Technical preliminaries

Further, for α>1\alpha>1 and 0≤β≤α0\leq\beta\leq\alpha,

The first inequality follows upon choosing β=α\beta=\alpha. For 1<α1<\alpha and 0≤β≤α0\leq\beta\leq\alpha, we get the second by (4). Note that by the monotonicity of Hölder means, we have

The final inequality follows upon rearranging the terms and using (4). ∎

II-B Bounds on moments of a Poisson random variable

We start with the expected value and the variance of falling powers of a Poisson random variable.

The next result establishes a bound on the moments of a Poisson random variable.

Let X∼Poi(λ)X\sim{\rm Poi}(\lambda) and let β\beta be a positive real number. Then,

Let Z=max⁡{λ1/β,λ}Z=\max\{\lambda^{1/\beta},\lambda\}.

The first inequality follows from the fact that either X/Z>1X/Z>1 or ≤1\leq 1. The equality follows from the fact that the integer moments of Poisson distribution are Touchard polynomials in λ\lambda. The second inequality uses the property that λ/Z≤1\lambda/Z\leq 1. Multiplying both sides by ZβZ^{\beta} results in the lemma. ∎

For α≤1\alpha\leq 1, (1+y)α≥1+αy−y2(1+y)^{\alpha}\geq 1+\alpha y-y^{2} for all y∈[−1,∞]y\in[-1,\infty], hence,

Since xαx^{\alpha} is a concave function and XX is nonnegative, the previous bound yields

where the last-but-one inequality is by Lemma 3. ∎

There is a constant cα′c^{\prime}_{\alpha} such that for any d>0d>0,

To obtain an estimator which does not require a knowledge of the support size kk, we seek a polynomial approximation qα(x)q_{\alpha}(x) of xαx^{\alpha} with qα(0)=0q_{\alpha}(0)=0. Such a polynomial qα(x)q_{\alpha}(x) can be obtained by a minor modification of the polynomial qα′(x)=∑j=0dqjxjq^{\prime}_{\alpha}(x)=\sum^{d}_{j=0}q_{j}x^{j} satisfying the error bound in Lemma 5. Specifically, we use the polynomial qα(x)=qα′(x)−q0q_{\alpha}(x)=q^{\prime}_{\alpha}(x)-q_{0} for which the approximation error is bounded as

To bound the variance of the proposed polynomial approximation estimator, we require a bound on the absolute values of the coefficients of qα(x)q_{\alpha}(x). The following inequality due to Markov serves this purpose.

Let p(x)=∑j=0dcjxjp(x)=\sum_{j=0}^{d}c_{j}x^{j} be a degree-dd polynomial so that ∣p(x)∣≤1|p(x)|\leq 1 for all x∈x\in. Then for all j=0,…,mj=0,\ldots,m

Since ∣xα∣≤1|x^{\alpha}|\leq 1 for x∈x\in, the approximation bound (5) implies ∣qα(x)∣<1+cαd2α|q_{\alpha}(x)|<1+\frac{c_{\alpha}}{d^{2\alpha}} for all x∈x\in. It follows from Lemma 6 that

III Upper bounds on sample complexity

In this section, we analyze the performances of the estimators we proposed in Section I-C. Our proofs are based on bounding the bias and the variance of the estimators under Poisson sampling. We first describe our general recipe and then analyze the performance of each estimator separately.

The following reduction to Poisson sampling is well-known.

(Poisson approximation 1) For n≥8log⁡(2/ϵ)n\geq 8\log(2/\epsilon) and N∼Poi(n/2)N\sim{\rm Poi}(n/2),

It remains to bound the probability on the right-side above, which can be done provided the bias and the variance of the estimator are bounded.

For N∼Poi(n)N\sim{\rm Poi}(n), let the power sum estimator P^α=P^α(n,XN)\widehat{\rm P}_{\alpha}=\widehat{\rm P}_{\alpha}(n,X^{N}) have bias and variance satisfying

Then, there exists an estimator P^α′\widehat{\rm P}^{\prime}_{\alpha} that uses 18nlog⁡(1/ϵ)18n\log(1/\epsilon) samples and ensures

In the remainder of the section, we bound the bias and the variance for our estimators when the number of samples nn are of the appropriate order. Denote by fαef_{\alpha}^{\text{e}}, fαuf_{\alpha}^{\text{u}}, and fαd,τf_{\alpha}^{d,\tau}, respectively, the empirical estimator 11−αlog⁡P^αe\frac{1}{1-\alpha}\log\widehat{P}_{\alpha}^{\text{e}}, the bias-corrected estimator 11−αlog⁡P^αu\frac{1}{1-\alpha}\log\widehat{P}_{\alpha}^{\text{u}}, and the polynomial approximation estimator 11−αlog⁡P^αd,τ\frac{1}{1-\alpha}\log\widehat{P}_{\alpha}^{d,\tau}. We begin by analyzing the performances of fαef_{\alpha}^{\text{e}} and fαuf_{\alpha}^{\text{u}} and build-up on these steps to analyze fαd,τf_{\alpha}^{d,\tau}.

The empirical estimator was presented in (1). Using the Poisson sampling recipe given above, we derive upper bound for the sample complexity of the empirical estimator by bounding its bias and variance. The resulting bound for α>1\alpha>1 is given in Theorem 9 and for α<1\alpha<1 in Theorem 10.

For α>1\alpha>1, 0<δ<1/20<\delta<1/2, and 0<ϵ<10<\epsilon<1, the estimator fαef_{\alpha}^{\text{e}} satisfies

Similarly, to bound the variance, using independence of multiplicities:

where the equality holds for kk sufficiently large. The theorem follows by using Lemma 8. ∎

For α<1\alpha<1, δ>0\delta>0, and 0<ϵ<10<\epsilon<1, the estimator fαef_{\alpha}^{\text{e}} satisfies

For α<1\alpha<1, once again we take a recourse to Lemma 4 to bound the bias as follows:

for every subset A⊂[k]A\subset[k]. Upon choosing A={x:λx≥1}A=\{x:\lambda_{x}\geq 1\}, we get

where the last inequality uses (4). For bounding the variance, note that

Consider the first term on the right-side. For α≤1/2\alpha\leq 1/2, it is bounded above by since z2αz^{2\alpha} is concave in zz, and for α>1/2\alpha>1/2 the bound in (8) and Lemma 1 applies to give

where (a)(a) is from (9) and (b)(b) from the concavity of zαz^{\alpha} in zz. The proof is completed by combining the two bounds above and using Lemma 8. ∎

In fact, we show in the appendix that the dependence on kk implied by the previous two results are optimal.

Given a sufficiently small δ\delta, the sample complexity Sαfαe(k,δ,ϵ)S_{\alpha}^{f_{\alpha}^{\text{e}}}(k,\delta,\epsilon) of the empirical estimator fαef_{\alpha}^{\text{e}} is bounded below as

While the performance of the empirical estimator is limited by these bounds, below we exhibit estimators that beat these bounds and thus outperform the empirical estimator.

III-B Performance of bias-corrected estimator for integral α𝛼\alpha

To reduce the sample complexity for integer orders α>1\alpha>1 to below kk, we follow the development of Shannon entropy estimators. Shannon entropy was first estimated via an empirical estimator, analyzed in, for instance, . However, with o(k)o(k) samples, the bias of the empirical estimator remains high . This bias is reduced by the Miller-Madow correction , but even then, O(k)O(k) samples are needed for a reliable Shannon-entropy estimation .

The next result provides a bound for the number of samples needed for the bias-corrected estimator.

For an integer α>1\alpha>1, any δ>0\delta>0, and 0<ϵ<10<\epsilon<1, the estimator fαuf_{\alpha}^{\text{u}} satisfies

Since the bias is 0, we only need to bound the variance to use Lemma 8. To that end, we have

where the inequality uses Lemma 2. It follows from Lemma 1 that

which is less than δ2/12\delta^{2}/12 if α2k1−1/α/n\alpha^{2}k^{1-1/\alpha}/n, for all δ\delta sufficiently small. Applying Lemma 8 completes the proof. ∎

III-C The polynomial approximation estimator

We first give a brief description of the polynomial estimator of and then in Theorem 13 prove that for α>1\alpha>1 the sample complexity of P^αd,τ\widehat{P}_{\alpha}^{d,\tau} is O(k/log⁡k)O(k/\log k). For completeness, we also include a proof for the case α<1\alpha<1, which is slightly different from the one in .

Therefore, for a given τ\tau and dd the combined estimator P^αd,τ\widehat{P}_{\alpha}^{d,\tau} is

We derive upper bounds for the sample complexity of the polynomial approximation estimator.

For α>1\alpha>1, δ>0\delta>0, 0<ϵ<10<\epsilon<1, there exist constants c1c_{1} and c2c_{2} such that the estimator P^αd,τ\widehat{P}_{\alpha}^{d,\tau} with τ=c1log⁡n\tau=c_{1}\log n and d=c2log⁡nd=c_{2}\log n satisfies

Let q(x)=∑m=0damxmq(x)=\sum_{m=0}^{d}a_{m}x^{m} satisfy the polynomial approximation error bound guaranteed by Lemma 5, i.e.i.e.,

For Nx′>τN_{x}^{\prime}>\tau, the bias of empirical part of the power sum is bounded as

For variance, independence of multiplicities under Poisson sampling gives

The two bounds above along with Lemma 1 and (4) yield

For d=τ/8=12log⁡nd=\tau/8=\frac{1}{2}\log n, the last terms in (15) are o(1)o(1) which gives

Recall from (6) that a<(1+cα/d2α)(2+1)da<(1+c_{\alpha}/d^{2\alpha})(\sqrt{2}+1)^{d}, and therefore, a2=O((2+1)log⁡n)=nc0a^{2}=O((\sqrt{2}+1)^{\log n})=n^{c_{0}} for some c0<1c_{0}<1. Using (18) we get

Therefore, the result follows from Lemma 8 for kk sufficiently large. ∎

We now prove an analogous result for α<1\alpha<1.

For α<1\alpha<1, δ>0\delta>0, 0<ϵ<10<\epsilon<1, there exist constants c1c_{1} and c2c_{2} such that the estimator P^αd,τ\widehat{P}_{\alpha}^{d,\tau} with τ=c1log⁡n\tau=c_{1}\log n and d=c2log⁡nd=c_{2}\log n satisfies

We proceed as in the previous proof and set τ\tau to be 4log⁡n4\log n. The contribution to the bias of the estimator for a symbol xx with Nx′<τN_{x}^{\prime}<\tau remains bounded as in (14). For a symbol xx with Nx′>τN_{x}^{\prime}>\tau, the bias contribution of the empirical estimator is bounded as

The first term can be bounded in the manner of (11) as

and, using (17), the following bound for the variance:

Here a2a^{2} is the largest squared coefficient of the approximating polynomial and, by (6), is O(22c0d)=O(nc0α)O(2^{2c_{0}d})=O(n^{c_{0}\alpha}) for some c0<1c_{0}<1. Thus, a2=o(nα)a^{2}=o(n^{\alpha}) and the proof follows by Lemma 8. ∎

IV Examples and experiments

The uniform distribution UkU_{k} over [k]={1,…,k}[k]=\left\{1,\ldots,k\right\} is given by

Its Rényi entropy for every order 1≠α≥01\neq\alpha\geq 0, and hence for all α≥0\alpha\geq 0, is

The Zipf distribution Zβ,kZ_{\beta,k} for β>0\beta>0 and k∈[k]k\in[k] is given by

Its Rényi entropy of order α≠1\alpha\neq 1 is

Table II summarizes the leading term g(k)g(k) in the approximationWe say f(n)∼g(n)f(n)\sim g(n) to denote lim⁡n→∞f(n)/g(n)=1\lim_{n\rightarrow\infty}f(n)/g(n)=1. Hα(Zβ,k)∼g(k)H_{\alpha}(Z_{\beta,k})\sim g(k).

We now illustrate the performance of the proposed estimators for various distributions for α=2\alpha=2 in Figures 2 and α=1.5\alpha=1.5 in Figures 3. For α=2\alpha=2, we compare the performance of bias-corrected and empirical estimators. For α=1.5\alpha=1.5, we compare the performance of the polynomial-approximation and the empirical estimator. For the polynomial-approximation estimator, the threshold τ\tau is chosen as τ=ln⁡(n)\tau=\ln(n) and the approximating polynomial degree is chosen as d=⌈1.5τ⌉d=\lceil 1.5\tau\rceil.

We test the performance of these estimators over six different distributions: the uniform distribution, a step distribution with half of the symbols having probability 1/(2k)1/(2k) and the other half have probability 3/(2k)3/(2k), Zipf distribution with parameter 3/43/4 (pi∝i−3/4p_{i}\propto i^{-3/4}), Zipf distribution with parameter 1/21/2 (pi∝i−1/2p_{i}\propto i^{-1/2}), a randomly generated distribution using the uniform prior on the probability simplex, and another one generated using the Dirichlet-1/21/2 prior.

In both the figures the true value is shown in black and the estimated values are color-coded, with the solid line representing their mean estimate and the shaded area corresponding to one standard deviation. As expected, bias-corrected estimators outperform empirical estimators for α=2\alpha=2 and polynomial-approximation estimators perform better than empirical estimators for α=1.5\alpha=1.5.

V Lower bounds on sample complexity

We first prove the lower bound for integers α>1\alpha>1, which matches the upper bound in Theorem 12 up to a constant factor.

where the constant implied by Ω\Omega may depend on ϵ\epsilon.

Since for small values of δ\delta we have (1+δ)1/2<1+δ(1+\delta)^{1/2}<1+\delta,

Next, we lower bound Sα(k)S_{\alpha}(k) for noninteger α>1\alpha>1 and show that it must be almost linear in kk. While we still rely on Lemma 15 for our lower bound, we take recourse to Poisson sampling to simplify our calculations.

(Poisson approximation 2) Suppose there exist δ,ϵ>0\delta,\epsilon>0 such that, with N∼Poi(2n)N\sim{\rm Poi}(2n), for all estimators f^\hat{f} we have

(Sufficiency of profiles). Consider an estimator f^\hat{f} such that

for Poisson sampling with N∼Poi(n)N\sim{\rm Poi}(n), it holds that

Let x=(1,...,d))\mathbf{x}=(1,...,{d})). Consider the polynomial

and q(z)=p(z)−Δq(z)=p(z)-\Delta, where Δ\Delta is chosen small enough so that q(z)q(z) has d{d} positive roots. Let y1,...,ydy_{1},...,y_{d} be the roots of the polynomial q(z)q(z). By Newton-Girard identities, while the sum of d{d}th power of roots of a polynomial does depend on the constant term, the sum of first d−1{d}-1 powers of roots of a polynomial do not depend on it. Since p(z)p(z) and q(z)q(z) differ only by a constant, it holds that

Furthermore, using a first order Taylor approximation, we have

and so, the left side above is nonzero for all Δ\Delta sufficiently small provided

Denoting the right side above by h(α)h(\alpha), note that h(i)=0h(i)=0 for i=1,...,d−1i=1,...,{d}-1. Since h(α)h(\alpha) is a linear combination of d{d} exponentials, it cannot have more than d−1{d}-1 zeros (see, for instance, ). Therefore, h(α)≠0h(\alpha)\neq 0 for all α∉{1,...,d−1}\alpha\notin\{1,...,{d}-1\}; in particular, ∥x∥α≠∥y∥α\|\mathbf{x}\|_{\alpha}\neq\|\mathbf{y}\|_{\alpha} for all Δ\Delta sufficiently small. ∎

We are now in a position to prove our converse results.

Given a nonintegral α>1\alpha>1, for any fixed 0<ϵ<1/20<\epsilon<1/2, we have

Finally, we show that Sα(k)S_{\alpha}(k) must be super-linear in kk for α<1\alpha<1.

Given α<1\alpha<1, for every 0<ϵ<1/20<\epsilon<1/2, we have

where the vectors x\mathbf{x} and y\mathbf{y} are given by Lemma 20 and β\beta satisfies α(1+β)<1\alpha(1+\beta)<1, and

Therefore, for sufficiently large kk, (20) holds by Lemma 20 since α(1+β)<1\alpha(1+\beta)<1, and for n<C2k(1+β−1/d)n<C_{2}k^{(1+\beta-1/d)} we get (21) by Theorem 19 as

The theorem follows since dd and β<1/α−1\beta<1/\alpha-1 are arbitrary. ∎

Acknowledgements

The authors thank Chinmay Hegde and Piotr Indyk for helpful discussions and suggestions.

References

Appendix A: Estimating power sums

namely for an additive accuracy in this range, Rényi entropy requires more samples than power sums.

It follows that the power sum estimation results in and the Rényi-entropy estimation results in this paper complement each other in several ways. For example, for α<1\alpha<1,

where the first inequality follows from Theorem 22 and the last follows from the upper-bound (22) derived in using a polynomial approximation estimator. Hence, for α<1\alpha<1, estimating power sums to additive and multiplicative accuracy require a comparable number of samples.

On the other hand, for α>1\alpha>1, Theorems 9 and 21 imply that for non integer α\alpha, Ω∼∼(k)≤SαP×(k)≤O(k),\stackrel{{\scriptstyle\sim}}{{\smash{\stackrel{{\scriptstyle\sim}}{{\smash{\Omega}\rule{0.0pt}{4.73611pt}}}}\rule{0.0pt}{6.45831pt}}}\left(k\right)\leq S^{P\times}_{\alpha}(k)\leq O\left(k\right), while in the Appendix we show that for 1<α1<\alpha, SαP+(k)S^{P+}_{\alpha}(k) is a constant. Hence in this range, power sum estimation to a multiplicative accuracy requires considerably more samples than estimation to an additive accuracy.

The next result shows that the bias and the variance of the empirical estimator are o(1)o(1).

For an appropriately chosen constant c>0c>0, the bias and the variance of the empirical estimator are bounded above as

where the last inequality holds by Lemma 4 and Lemma 2since xαx^{\alpha} is convex in xx. Noting ∑iλx=n\sum_{i}\lambda_{x}=n, we get

Similarly, proceeding as in the proof of Theorem 9, the variance of the empirical estimator is bounded as

Further, since xα−1/2x^{\alpha-1/2} is convex for α≥3/2\alpha\geq 3/2, the summation above is maximized when one of the λx\lambda_{x}’s is nn and the remaining equal which yields

Appendix B: Lower bound for sample complexity of empirical estimator

Given α<1\alpha<1 and δ<cα\delta<c_{\alpha} for a constant cαc_{\alpha} depending only on α\alpha, the sample complexity Sαfαe(k,δ,ϵ)S_{\alpha}^{f_{\alpha}^{\text{e}}}(k,\delta,\epsilon) of the empirical estimator fαef_{\alpha}^{\text{e}} is bounded below as

We prove the lower bound for the uniform distributon over kk symbols in two steps. We first show that for any constant c1>1c_{1}>1 if n<k/c1n<k/c_{1} then the additive approximation error is at least δ\delta with probability one, for every δ<log⁡c1\delta<\log c_{1}. Then, assuming that n≥k/c1n\geq k/c_{1}, we show that the additve approximation error is at least δ\delta with probability greater than 0.90.9 if n<k/δn<k/\delta.

Hence, for any c1>1c_{1}>1 and n<k/c1n<k/c_{1} and any 0≤δ≤log⁡c10\leq\delta\leq\log c_{1}, the additive approximation error is more than δ\delta with probability one.

Moving to the second claim, suppose now n>k/c1n>k/c_{1}. We first show that with high probability, the multiplicities of a linear fraction of kk symbols should be at least a factor of standard deviaton higher than the mean. Specifically, let

where QQ denotes the QQ-function, i.e.i.e., the tail of the standard normal random variable, and the final inequality uses Slud’s inequality [38, Theorem 2.1].

Note that AA is a function of nn i.i.d. random variables X1,X2,…,XnX_{1},X_{2},\ldots,X_{n}, and changing any one XiX_{i} changes AA by at most 22. Hence, by McDiarmid’s inequality,

where the second inequality is by Bernoulli’s inequality and the third inequality holds for every c4≤α(α−1)(c2c1)α−2/2c_{4}\leq\alpha(\alpha-1)(c_{2}\sqrt{c_{1}})^{\alpha-2}/2. Therefore, with probability ≥0.9\geq 0.9,

Given α<1\alpha<1 and δ<cα\delta<c_{\alpha} for a constant cαc_{\alpha} depending only on α\alpha, the sample complexity Sαfαe(k,δ,ϵ)S_{\alpha}^{f_{\alpha}^{\text{e}}}(k,\delta,\epsilon) of the empirical estimator fαef_{\alpha}^{\text{e}} is bounded as

We proceed as in the proof of the previous lemma. However, instead of using the uniform distribution, we use a distribution which has one “heavy element” and is uniform conditioned on the occurance of the remainder. The key observation is that there will be roughly nαn^{\alpha} occurances of the “light elements”. Thus, when we account for the error in the estimation of the contribution of light elements to the power sum, we can replace nn with n1/αn^{1/\alpha} in our analysis of the previous lemma, which yields the required bound for sample complexity.

Specifically, consider a distribution with one heavy element such that

We begin by analyzing the estimate of the second term in power sum, namely

Let R=∑i∈[k]NiR=\sum_{i\in[k]}N_{i} be the total number of occurances of light elements. Since RR is a binomial (n,δnα−1)(n,\delta n^{\alpha-1}) random variable, for every constant c>0c>0

In the remainder of the proof, we shall assume that this large probability event holds.

where the last inequality uses R≤(1+c)nα≤2nαR\leq(1+c)n^{\alpha}\leq 2n^{\alpha}. Thus, the empirical estimate is at most 33 with probability close to 11 when kk (and therefore nn) large. It follows from (23) that

holds. Note that conditioned on each value of RR, the random variables (Ni,i∈[k])(N_{i},i\in[k]) have a multinomial distribution with uniform probabilities, i.e.i.e., these random variables behave as if we drew RR i.i.d. samples from a uniform distribution on [k][k] elements. Thus, we can follow the proof of the previous lemma mutatis mutandis. We now define AA as

To lower bound p(Nx≤nk−c2nk(1−1k))p\left(N_{x}\leq\frac{n}{k}-c_{2}\sqrt{\frac{n}{k}\left(1-\frac{1}{k}\right)}\right) Slud’s inequality is no longer available (since it may not hold for Bin(n,p){\tt Bin}(n,p) with p>1/2p>1/2 and that is the regime of interest for the lower tail probability bounds needed here). Instead we take recourse to a combination of Bohman’s inequality and Anderson-Samuel inequality, as suggested in [38, Eqns. (i) and (ii)]. It can be verified that the condition for [38, Eqns. (ii)] holds, and therefore,

Continuing as in the proof of the previous lemma, we get that the following holds with conditional probability greater than 0.90.9 given each value of RR satisfying (24):

where c3c_{3} is a sufficiently small constant such that (1+x)α≤1+αx−c3x2(1+x)^{\alpha}\leq 1+\alpha x-c_{3}x^{2} for all x≥0x\geq 0 and c4=c3/(1+c)c_{4}=c_{3}/(1+c). Thus,

Denoting y=(k/nα)y=(k/n^{\alpha}) and choosing c1c_{1} and cc small enough such that P^αe≤2\widehat{P}_{\alpha}^{\text{e}}\leq 2, for all sufficiently large nn we get from (23) that