{"id":1286,"date":"2022-10-04T15:54:58","date_gmt":"2022-10-04T15:54:58","guid":{"rendered":"https:\/\/www.ethanepperly.com\/?p=1286"},"modified":"2024-10-05T20:34:07","modified_gmt":"2024-10-05T20:34:07","slug":"note-to-self-hanson-wright-inequality","status":"publish","type":"post","link":"https:\/\/www.ethanepperly.com\/index.php\/2022\/10\/04\/note-to-self-hanson-wright-inequality\/","title":{"rendered":"Note to Self: Hanson\u2013Wright Inequality"},"content":{"rendered":"\n

<\/p>\n\n\n\n

This post is part of a new series for this blog, Note to Self<\/strong>, where I collect together some notes about an idea related to my research. This content may be much more technical than most of the content of this blog and of much less wide interest. My hope in sharing this is that someone will find this interesting and useful for their own work.<\/em><\/p>\n\n\n\n

\n\n\n\n

This post is about a fundamental tool of high-dimensional probability<\/a>, the Hanson\u2013Wright inequality<\/a>. The Hanson\u2013Wright inequality is a concentration inequality<\/a> for quadratic forms<\/a><\/em> of random vectors\u2014that is, expressions of the form \"x^\top A x\" where \"x\" is a random vector. Many statements of this inequality in the literature have an unspecified constant \"c > 0\"; our goal in this post will be to derive a fairly general version of the inequality with only explicit constants.<\/p>\n\n\n\n

The core object of the Hanson\u2013Wright inequality is a subgaussian<\/a><\/em> random variable<\/a>. A random variable \"Y\" is subgaussian if the probability it exceeds a threshold \"t\" in magnitude decays as<\/p>\n\n\n\n

(1) <\/span>   <\/span>\"\[\mathbb{P}\{|Y|\ge t\} \le \mathrm{e}^{-t^2/a} \quad \text{for some $a>0$ and for all sufficiently large $t$.} \]\"<\/p><\/p>\n\n\n\n

The name subgaussian<\/em> is appropriate as the tail probabilities of Gaussian random variables<\/a> exhibit the same square-exponential decrease \"\mathrm{e}^{-t^2/a}\".<\/p>\n\n\n\n

A (non-obvious) fact is that if \"Y\" is subgaussian in the sense (1) and centered (\"\mathbb{E} Y = 0\"), then \"Y\"‘s cumulant generating function<\/a><\/em> (cgf)<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\xi_Y(t) := \log \mathbb{E} \exp(tY).\]\"<\/p><\/p>\n\n\n\n

is subquadratic: There is a constant \"c > 0\" (independent of \"Y\" and \"a\"), for which<\/p>\n\n\n\n

(2) <\/span>   <\/span>\"\[\xi_Y(t) \le ca t^2 \quad \text{for all $t\in\mathbb{R}$}. \]\"<\/p><\/p>\n\n\n\n

Moreover,1<\/a><\/sup>See Proposition 2.5.2 of Vershynin’s<\/a> High-Dimensional Probability<\/em><\/a>.<\/span> a subquadratic cgf (2) also implies the subgaussian tail property (1), with a different parameter \"a > 0\".<\/p>\n\n\n\n

Since properties (1) and (2) are equivalent (up to a change in the parameter \"a\"), we are free to fix a version of property (2) as our definition for a (centered) subgaussian random variable.<\/p>\n\n\n\n

\n

Definition (subgaussian random variable):<\/strong> A centered random variable \"X\" is said to be \"v\"-subgaussian<\/em> or subgaussian with variance proxy \"v\"<\/em> if its cgf is subquadratic:

(3) <\/span>   <\/span>\"\[\xi_{x}(t) \le\frac{1}{2} vt^2 \quad \text{for all $t\in\mathbb{R}$.} \]\"<\/p><\/p>\n\n\n\n

<\/p>\n<\/blockquote>\n\n\n\n

For instance, a mean-zero Gaussian random variable<\/a> \"X\" with variance \"\sigma^2\" has cgf<\/p>\n\n\n

(4) <\/span>   <\/span>\"\[<\/p><\/p>\n\n\n

and is thus subgaussian with variance proxy \"v = \sigma^2\" equal to its variance.<\/p>\n\n\n\n

Here is a statement of the Hanson\u2013Wright inequality as it typically appears with unspecified constants (see Theorem 6.2.1 of Vershynin’s<\/a> High-Dimensional Probability<\/em><\/a>):<\/p>\n\n\n\n

\n

Theorem (Hanson\u2013Wright):<\/strong> Let \"x\" be a random vector with independent centered \"v\"-subgaussian entries and let \"A\" be a square matrix. Then

  <\/span>   <\/span>\"\[\mathbb{P}\left\{\left|x^\top Ax - \mathbb{E} \left[x^\top A x\right]\right|\ge t \right\} \le 2\exp\left(- \frac{c\cdot t^2}{v^2\left\|A\right\|_{\rm F}^2 + v\left\|A\right\|t} \right),\]\"<\/p> where \"c>0\" is a constant (not depending on \"v\", \"x\", \"t\", or \"A\").2<\/a><\/sup>Here, \"\|\cdot\|_{\rm F}\" and \"\|\cdot\|\" denote the Frobenius<\/a> and spectral<\/a> norms.<\/span>\n<\/blockquote>\n\n\n\n

This type of concentration is exactly the same type as provided by Bernstein’s inequality<\/a> (which I discussed in my post on concentration inequalities<\/a>). In particular, for small deviations \"t\", the tail probabilities decay are subgaussian<\/em> with variance proxy \"\approx v^2\left\|A\right\|_{\rm F}^2\":<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\mathbb{P}\left\{\left|x^\top Ax - \mathbb{E}\left[x^\top Ax\right]\right|\ge t \right\} \stackrel{\text{small $t$}}{\lessapprox} 2\exp\left(- \frac{c\cdot t^2}{v^2\left\|A\right\|_{\rm F}^2} \right)\]\"<\/p><\/p>\n\n\n\n

For large deviations \"t\", this switches to subexponential<\/a><\/em> tail probabilities with decay rate \"\approx v\|A\|\":<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\mathbb{P}\left\{\left|x^\top Ax - \mathbb{E}\left[x^\top Ax\right]\right|\ge t \right\} \stackrel{\text{large $t$}}{\lessapprox} 2\exp\left(- \frac{c\cdot t}{v\|A\|} \right).\]\"<\/p><\/p>\n\n\n\n

Mediating these two parameter regimes are the size of the matrix \"A\", as measured by its Frobenius<\/a> and <\/span>spectral<\/a> norms<\/span>, and the degree of subgaussianity of \"x\", measured by the variance proxy \"v\".<\/p>\n\n\n\n

Diagonal-Free Hanson\u2013Wright<\/h2>\n\n\n\n

Now we come to a first version of the Hanson\u2013Wright inequality with explicit constants, first for a matrix which is diagonal-free\u2014that is, having all zeros on the diagonal. I obtained this version of the inequality myself, though I am very sure that this version of the inequality or an improvement thereof appears somewhere in the literature.<\/p>\n\n\n\n

\n

Theorem (Hanson\u2013Wright, explicit constants, diagonal-free):<\/strong> Let \"x\" random vector with independent centered \"v\"-subguassian entries and let \"A\" be a diagonal-free<\/em> square matrix. Then we have the cgf bound

  <\/span>   <\/span>\"\[\xi_{x^\top Ax}(t) \le \frac{16v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-4v\left\|A\right\|t)}.\]\"<\/p>As a consequence, we have the concentration bound

  <\/span>   <\/span>\"\[\mathbb{P} \{ x^\top A x \ge t \} \le \exp\left( -\frac{t^2/2}{16v^2 \left\|A\right\|_{\rm F}^2+4v\left\|A\right\|t} \right).\]\"<\/p>Similarly, we have the lower tail

  <\/span>   <\/span>\"\[\mathbb{P} \{ x^\top A x \le -t \} \le \exp\left( -\frac{t^2/2}{16v^2 \left\|A\right\|_{\rm F}^2+4v\left\|A\right\|t} \right)\]\"<\/p>and the two-sided bound

  <\/span>   <\/span>\"\[\mathbb{P} \{ |x^\top A x| \ge t \} \le 2\exp\left( -\frac{t^2/2}{16v^2 \left\|A\right\|_{\rm F}^2+4v\left\|A\right\|t} \right).\]\"<\/p><\/p>\n<\/blockquote>\n\n\n\n

Let us begin proving this result. Our proof will follow the same steps as Vershynin’s<\/a> proof in High-Dimensional Probability<\/a><\/em> (which in turn is adapted from an article by Rudelson and Vershynin<\/a>), but taking care to get explicit constants. Unfortunately, proving all of the relevant tools from first principles would easily triple the length of this post, so I make frequent use of results from the literature.<\/p>\n\n\n\n

We begin by the decoupling bound<\/em> (Theorem 6.1.1 in Vershynin’s<\/a> High-Dimensional Probability<\/em><\/a>), which allows us to replace one \"x\" with an independent copy \"\tilde{x}\" at the cost of a factor of four:<\/p>\n\n\n\n

(5) <\/span>   <\/span>\"\[\xi_{x^\top Ax}(t) \le \xi_{\tilde{x}^\top Ax}(4t). \]\"<\/p><\/p>\n\n\n\n

We seek to compare the bilinear form \"\tilde{x}^\top Ax\" to the Gaussian bilinear form \"\tilde{g}^\top Ag\" where \"\tilde{g}\" and \"g\" are independent standard Gaussian vectors. We begin with the following cgf bound for the Gaussian quadratic form \"g^\top Ag\":<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\xi_{g^\top Ag}(t) \le \frac{\left\|A\right\|_{\rm F}^2 \, t^2}{1-2\|A\|\, t}.\]\"<\/p><\/p>\n\n\n\n

This equation is the result of Example 2.12 in Boucheron<\/a>, Lugosi<\/a>, and Massart’s<\/a> Concentration Inequalities<\/a><\/em>. By applying this result to the Hermitian dilation<\/a> of \"A\" in \"A\"‘s place, one obtains a similar result for the decoupled bilinear form \"\tilde{g}^\top Ag\":<\/p>\n\n\n\n

(6) <\/span>   <\/span>\"\[\xi_{\tilde{g}^\top Ag}(t) \le \frac{\left\|A\right\|_{\rm F}^2 \, t^2}{2(1-\|A\|\, t)}. \]\"<\/p><\/p>\n\n\n\n

We now seek to compare \"\xi_{\tilde{x}^\top Ax}(t)\" to \"\xi_{\tilde{g}^\top Ag}(t)\". To do this, we first evaluate the cgf of \"\tilde{x}^\top Ax\" only over the randomness in \"\tilde{x}\". Since we’re only taking an expectation over the random variable \"\tilde{x}\", we can apply the subquadratic tail condition (3) to obtain<\/p>\n\n\n\n

(7) <\/span>   <\/span>\"\[\log \mathbb{E}_{\tilde{x}} \exp(t \, \tilde{x}^\top Ax) = \sum_{i=1}^n \log \mathbb{E}_{\tilde{x}} \exp(t \,\tilde{x}_i (Ax)_i) \le \frac{1}{2} v \left(\sum_{i=1}^n (Ax)_i^2\right)t^2 \le \frac{1}{2} v\left\|Ax\right\|^2 \, t^2. \]\"<\/p><\/p>\n\n\n\n

Now we perform a similar computation for the quantity \"\tilde{g}^\top Ax\" in which \"\tilde{x}\" has been replaced by the Gaussian vector \"\tilde{g}\":<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\log \mathbb{E}_{\tilde{g}} \exp((\sqrt{v} t) \, \tilde{g}^\top Ax) = \frac{1}{2} v \left\|Ax\right\|^2 \, t^2.\]\"<\/p><\/p>\n\n\n\n

We stress that this is an equality<\/em> since the cgf of a Gaussian random variable is given by (4). Thus we can substitute the left-hand side of the above display into the right-hand side of (7), yielding<\/p>\n\n\n\n

(8) <\/span>   <\/span>\"\[\log \mathbb{E}_{\tilde{x}} \exp(t \, \tilde{x}^\top Ax) \le \log \mathbb{E}_{\tilde{g}} \exp((\sqrt{v} t) \, \tilde{g}^\top Ax). \]\"<\/p><\/p>\n\n\n\n

We now perform this same trick again using the randomness in \"x\":<\/p>\n\n\n\n

(9) <\/span>   <\/span>\"\[\log \mathbb{E}_{\tilde{g},x} \exp((\sqrt{v} t) \, \tilde{g}^\top Ax) \le \log \mathbb{E}_{\tilde{g}} \exp \left(\frac{1}{2} v^2 \left\|A^\top \tilde{g}\right\|^2t^2\right) = \log \mathbb{E}_{\tilde{g},g} \exp(v t \, \tilde{g}^\top Ag). \]\"<\/p><\/p>\n\n\n\n

Packaging up (8) and (9) gives<\/p>\n\n\n\n

(10) <\/span>   <\/span>\"\[\xi_{\tilde{x}^\top Ax}(t)\le \xi_{\tilde{g}^\top Ag}(vt). \]\"<\/p><\/p>\n\n\n\n

Combining all these results (5), (6), and (10), we obtain<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\xi_{x^\top Ax}(t) \le \xi_{\tilde{x}^\top Ax}(4t) \le \xi_{\tilde{g}^\top Ag}(4vt) \le \frac{16v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-4v\left\|A\right\|t)}.\]\"<\/p><\/p>\n\n\n\n

This cgf implies the desired probability bound on the upper tail as a consequence of the following fact (see Boucheron<\/a>, Lugosi<\/a>, and Massart’s<\/a> Concentration Inequalities<\/a><\/em> page 29 and Exercise 2.8):<\/p>\n\n\n\n

\n

Fact (Bernstein concentration from Bernstein cgf bound):<\/strong> Suppose that a random variable \"X\" satisfies the cgf bound \"\xi_X(t) \le \tfrac{vt^2}{2(1-ct)}\" for \"0 < t < 1/c\". Then

  <\/span>   <\/span>\"\[\mathbb{P} \left\{ X\ge t \right\} \le \exp\left( -\frac{t^2/2}{v+ct} \right).\]\"<\/p><\/p>\n<\/blockquote>\n\n\n\n

To get the bound on the lower tail, apply the result for the upper tail to the matrix \"-A\" to obtain<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\mathbb{P} \{ x^\top A x \le -t \} = \mathbb{P} \{ x^\top (-A) x \ge t \} \le \exp\left( -\frac{t^2/2}{16v^2 \left\|A\right\|_{\rm F}^2+4v\left\|A\right\|t} \right).\]\"<\/p><\/p>\n\n\n\n

Finally, to obtain the two-sided bound, use a union bound over the upper and lower tails:<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\mathbb{P} \{ |x^\top A x| \ge t \} \le \mathbb{P} \{ x^\top A x \ge t \} + \mathbb{P} \{ x^\top A x \le -t \} \le 2\exp\left( -\frac{t^2/2}{16v^2 \left\|A\right\|_{\rm F}^2+4v\left\|A\right\|t} \right).\]\"<\/p><\/p>\n\n\n\n

General Hanson\u2013Wright<\/h2>\n\n\n\n

Now, here’s a more general result (with worse constants) which permits the matrix \"A\" to possess a diagonal.<\/p>\n\n\n\n

\n

Theorem (Hanson\u2013Wright, explicit constants):<\/strong> Let \"x\" random vector with independent centered \"v\"-subguassian entries and let \"A\" be an arbitrary<\/em> square matrix. Then we have the cgf bound

  <\/span>   <\/span>\"\[\xi_{x^\top Ax-\mathbb{E} [x^\top A x]}(t) \le \frac{40v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-8v\left\|A\right\|t)}.\]\"<\/p>As a consequence, we have the concentration bound

  <\/span>   <\/span>\"\[\mathbb{P} \{ x^\top A x-\mathbb{E} [x^\top A x] \ge t \} \le \exp\left( -\frac{t^2/2}{40v^2 \left\|A\right\|_{\rm F}^2+8v\left\|A\right\|t} \right).\]\"<\/p>Left tail and two-sided bounds versions of this bound also hold:

  <\/span>   <\/span>\"\[\mathbb{P} \{ x^\top A x-\mathbb{E} [x^\top A x] \le -t \} \le \exp\left( -\frac{t^2/2}{40v^2 \left\|A\right\|_{\rm F}^2+8v\left\|A\right\|t} \right)\]\"<\/p> and

  <\/span>   <\/span>\"\[\mathbb{P} \{ |x^\top A x-\mathbb{E} [x^\top A x]| \ge t \} \le 2\exp\left( -\frac{t^2/2}{40v^2 \left\|A\right\|_{\rm F}^2+8v\left\|A\right\|t} \right).\]\"<\/p><\/p>\n<\/blockquote>\n\n\n\n

Decompose the matrix \"A = D+F\" into its diagonal and off-diagonal portions. For any two random variables \"X\" and \"Y\" (possibly highly dependent), we can bound the cgf of their sum using the following “union bound”:<\/p>\n\n\n

(11) <\/span>   <\/span>\"\begin{align*}<\/p><\/p>\n\n\n

The two equality statements are the definition of the cumulant generating function and the inequality is Cauchy\u2013Schwarz<\/a>. <\/p>\n\n\n\n

Using the “union bound”, it is sufficient to obtain bounds for the cgfs of the diagonal and off-diagonal parts \"x^\top D x - \mathbb{E}[x^\top Ax]\" and \"x^\top F x\". We begin with the diagonal part. We compute<\/p>\n\n\n\n

(12) <\/span>   <\/span>\"\begin{align*}\xi_{x^\top D x - \mathbb{E}[x^\top Ax]}(t) &= \log \mathbb{E} \exp\left(t \sum_{i=1}^n A_{ii}(x_i^2 - \mathbb{E}[x_i^2]) \right) \\ &= \sum_{i=1}^n \log \mathbb{E} \exp\left((t A_{ii})\cdot(x_i^2 - \mathbb{E}[x_i^2]) \right). \end{align*}\"<\/p><\/p>\n\n\n\n

For the cgf of \"x_i^2 - \mathbb{E}[x_i^2]\", we use the following bound, taken from Appendix B<\/a> of the following paper<\/a>:<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\log \mathbb{E} \exp\left(t(x_i^2 - \mathbb{E}[x_i^2]) \right) \le \frac{8v^2t^2}{1-2v|t|}.\]\"<\/p><\/p>\n\n\n\n

Substituting this result into (12) gives<\/p>\n\n\n\n

(13) <\/span>   <\/span>\"\[\xi_{x^\top D x - \mathbb{E}[x^\top Ax]}(t) \le \sum_{i=1}^n \frac{8v^2|A_{ii}|^2t^2}{1-2v|A_{ii}|t} \le \frac{8v^2\|A\|_{\rm F}^2t^2}{1-2v\|A\|t}\quad \text{for $t>0$}. \]\"<\/p><\/p>\n\n\n\n

For the second inequality, we used the facts that \"\max_i |A_{ii}| \le \|A\|\" and \"\sum_i |A_{ii}|^2 \le \|A\|_{\rm F}^2\".<\/p>\n\n\n\n

We now look at the off-diagonal part \"x^\top F x\". We use a version of the decoupling bound (5) where we compare \"x^\top F x\" to \"\tilde{x}^\top A x\", where we’ve both replaced one copy of \"x\" with an independent copy and reinstated the diagonal of \"A\" (see Remark 6.1.3 in Vershynin’s<\/a> High-Dimensional Probability<\/a><\/em>):<\/p>\n\n\n\n

  <\/span>   <\/span>\"\[\xi_{x^\top F x}(t) \le \xi_{\tilde{x}^\top Ax}(4t).\]\"<\/p><\/p>\n\n\n\n

We can now just repeat the rest of the argument for the diagonal-free Hanson\u2013Wright inequality, yielding the same conclusion<\/p>\n\n\n

(14) <\/span>   <\/span>\"\[<\/p><\/p>\n\n\n

Combining (11), (13), and (14), we obtain<\/p>\n\n\n\n

  <\/span>   <\/span>\"\begin{align*}\xi_{x^\top Ax-\mathbb{E} [x^\top A x]} &\le \frac{1}{2} \xi_{x^\top D x - \mathbb{E}[x^\top Ax]}(2t) + \frac{1}{2} \xi_{x^\top Fx}(2t) \\&\le \frac{8v^2\|A\|_{\rm F}^2t^2}{2(1-4v\|A\|t)} + \frac{32v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-8v\left|A\right|t)} \\&\le \frac{8v^2\|A\|_{\rm F}^2t^2}{2(1-4v\|A\|t)} + \frac{32v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-8v\left\|A\right\|t)} \\&\le \frac{40v^2\left\|A\right\|_{\rm F}^2\, t^2}{2(1-8v\left\|A\right\|t)}.\end{align*}\"<\/p><\/p>\n\n\n\n

As with above, this cgf bound implies the desired probability bound.<\/p>\n","protected":false},"excerpt":{"rendered":"

This post is part of a new series for this blog, Note to Self, where I collect together some notes about an idea related to my research. This content may be much more technical than most of the content of this blog and of much less wide interest. My hope in sharing this is thatRead more<\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[9],"tags":[],"class_list":["post-1286","post","type-post","status-publish","format-standard","hentry","category-note-to-self"],"_links":{"self":[{"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/posts\/1286","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/comments?post=1286"}],"version-history":[{"count":59,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/posts\/1286\/revisions"}],"predecessor-version":[{"id":1878,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/posts\/1286\/revisions\/1878"}],"wp:attachment":[{"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/media?parent=1286"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/categories?post=1286"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.ethanepperly.com\/index.php\/wp-json\/wp\/v2\/tags?post=1286"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}