A Short Note on Concentration Inequalities for Random Vectors with SubGaussian Norm

Chi Jin, Praneeth Netrapalli, Rong Ge, Sham M. Kakade, Michael I. Jordan

Introduction

Concentration (large deviation) inequalities are one of the most important subjects of study in probability theory. A class of distributions for which sharp concentration inequalities have been developed is the class of subGaussian distributions.

A random variable X∈RX\in R is subGaussian, if there exists σ∈R\sigma\in R so that:

A random vector X∈Rd\mathbf{X}\in R^{d} is subGaussian, if there exists σ∈R\sigma\in R so that:

The concentration bounds of subGaussian random vectors/variables depends on the parameter σ\sigma – smaller the σ\sigma better the concentration bounds. While subGaussian distributions arise naturally in several applications, there are settings where the random vectors have nice concentration properties but the subGaussian parameter σ\sigma is very large (so that applying concentration bounds for general subGaussian random vectors gives loose bounds). In this short note, we consider a related but different class of distributions, called norm-subGaussian random vectors and establish tighter concentration bounds for them.

Organization: In Section 2, we introduce norm subGaussian random vectors and some of their properties and we prove our main results in Section 3. We conclude in Section 4.

Norm SubGaussian Random Vector

The norm subGaussian random vector is defined as follows.

Norm subGaussian includes both subGaussian (with a smaller σ\sigma parameter) and bounded norm random vectors as special cases.

There exists absolute constant cc so that following random vectors are all nSG(c⋅σ)\text{nSG}(c\cdot\sigma).

Then let v(X)=X/∥X∥\mathbf{v}(\mathbf{X})=\mathbf{X}/\|{\mathbf{X}}\|, since {vi}\{\mathbf{v}_{i}\} is a 1/21/2-cover, there always exists a j(X)j(\mathbf{X}) so that vj(X)\mathbf{v}_{j(\mathbf{X})} in cover and ∥v(X)−vj(X)∥≤1/2\|{\mathbf{v}(\mathbf{X})-\mathbf{v}_{j(\mathbf{X})}}\|\leq 1/2. Therefore, we have:

Now we are ready to check the second claim of Lemma 1, when t2≤8σ2ln⁡4t^{2}\leq 8\sigma^{2}\ln 4, we have,

when t2>8σ2ln⁡4t^{2}>8\sigma^{2}\ln 4, we let t2=8σ2ln⁡4+st^{2}=8\sigma^{2}\ln 4+s where s>0s>0, then:

In sum, this proves that X\mathbf{X} is nSG(22⋅σ)\text{nSG}(2\sqrt{2}\cdot\sigma). ∎

The following lemma gives equivalent characterizations of norm subGaussian in terms of moments and moment generating function (MGF).

Note ∥X∥\|{\mathbf{X}}\| is a 1-dimensional random variable. This lemma directly follows from the equivalent properties of 11-dimensional subGaussian, for instance, Lemma 5.5 in (Vershynin, 2010). ∎

The following lemma says that if a random vector is nSG(σ)\text{nSG}(\sigma), then its norm squared is subexponential and its projection on any direction a is subGaussian random variable.

The undesirable thing about the MGF characterization in Lemma 2 is that even if X\mathbf{X} is a zero mean random vector, ∥X∥\|{\mathbf{X}}\| is not zero mean, so it is difficult to directly work with MGF of ∥X∥\|{\mathbf{X}}\|. Instead, we first convert the random vector X\mathbf{X} to a matrix Y\mathbf{Y} and characterize the MGF of Y\mathbf{Y}.

There is an absolute constant cc, if random vector X∈Rd\mathbf{X}\in R^{d} is zero-mean nSG(σ)\text{nSG}(\sigma), then let

where in the last inequality we used the fact that pp(2p)!≤1p!\frac{p^{p}}{(2p)!}\leq\frac{1}{p!}, this finishes the proof. ∎

Vector Martingales with SubGaussian Norm

In this section, we will prove our main result (Lemma 6, Corollaries 7 and 8) giving concentration bounds for norm subGaussian random vectors. The main tool we use is Lieb’s concavity theorem.

Let A\mathbf{A} be a fixed symmetric matrix, and let Y\mathbf{Y} be a random symmetric matrix. Then,

We will prove our concentration result for norm subGaussian random vectors in a general setting where the subGaussian parameter σi\sigma_{i} for the ithi^{\textrm{th}} vector can itself be a random variable.

where step (1) is due to Theorem 5, and step (2) used the fact that if matrix A⪯B\mathbf{A}\preceq\mathbf{B}, then eC+A⪯eC+Be^{\mathbf{C}+\mathbf{A}}\preceq e^{\mathbf{C}+\mathbf{B}}. On the other hand, since identity matrix commutes with any matrix, we know:

Therefore, for any t≥0t\geq 0, θ≥0\theta\geq 0, by Markov’s inequality, we have:

where step (1) is because ∑i=1nYi\sum_{i=1}^{n}\mathbf{Y}_{i} is a rank-2 matrix whose eigenvalues are ∥∑i=1nXi∥,−∥∑i=1nXi∥\|{\sum_{i=1}^{n}\mathbf{X}_{i}}\|,-\|{\sum_{i=1}^{n}\mathbf{X}_{i}}\|; step (2) is due to all preconditions are symmetric with respect to 0. Finally, setting RHS equal to δ\delta, we finish the proof. ∎

Since now {σi}\{\sigma_{i}\} are fixed which are not random, we can pick θ\theta in Lemma 6 as a function of {σi}\{\sigma_{i}\}. Indeed, pick θ=1∑i=1nσi2log⁡2dδ\theta=\sqrt{\frac{1}{\sum_{i=1}^{n}\sigma_{i}^{2}}\log\frac{2d}{\delta}} finishes the proof. ∎

For simplicity, denote log factor ι:=log⁡2dδ+log⁡log⁡Bb\iota\mathrel{\mathop{:}}=\log\frac{2d}{\delta}+\log\log\frac{B}{b} By Lemma 6, we know for any fixed θ\theta, with probability 1−δ⋅log⁡−1(B/b)1-\delta\cdot\log^{-1}(B/b), we have:

Construct two sets of Ψ={ψ1,…,ψs}\Psi=\{\psi_{1},\ldots,\psi_{s}\} and Θ={θ1,…,θs}\Theta=\{\theta_{1},\ldots,\theta_{s}\}, where ψj=2j−1⋅b\psi_{j}=2^{j-1}\cdot b and θj=ιψj\theta_{j}=\sqrt{\frac{\iota}{\psi_{j}}} with last element ψs≤B\psi_{s}\leq B, 2ψs>B2\psi_{s}>B. It is easy to see ∣Ψ∣=∣Θ∣≤log⁡(B/b)|\Psi|=|\Theta|\leq\log(B/b). By union bound, we have with probability 1−δ1-\delta:

Consider following two cases: (1) ∑i=1nσi2∈[b,B]\sum_{i=1}^{n}\sigma_{i}^{2}\in[b,B]. Then, there exists j∈[s]j\in[s] such that ψj≤∑i=1nσi2<2ψj\psi_{j}\leq\sum_{i=1}^{n}\sigma_{i}^{2}<2\psi_{j}:

(2) ∑i=1nσi2∈[0,b)\sum_{i=1}^{n}\sigma_{i}^{2}\in[0,b). In this case we know ψ1=b\psi_{1}=b and:

Conclusion

In this short note, we introduced the notion of norm subGaussian random vectors, which include subGaussian random vectors and bounded random vectors as special cases. While it is true that subGaussian(σd)⊆nSG(σ)⊆subGaussian(σ)\textrm{subGaussian}\left(\frac{\sigma}{\sqrt{d}}\right)\subseteq\text{nSG}(\sigma)\subseteq\textrm{subGaussian}(\sigma), applying concentration bounds for subGaussian(σ)\textrm{subGaussian}(\sigma) would yield bounds which have at least linear dependence on dd. In contrast, the bounds we develop (in Lemma 6 and Corollaries 7 and 8) have only logarithmic dependence on dd. It is not clear if this logarithmic dependence is tight – totally eliminating this dependence is an interesting open problem.

Acknowledgements

We thank Gabor Lugosi and Nilesh Tripuraneni for helpful discussions.

References