跳转至

LTFP Ch3 §3.5-3.9:岭回归与极小极大下界

本文由手写 PDF 整理而成。符号、命题编号与主要推导按照 Francis Bach 的 Learning Theory from First Principles(LTFP)第 3 章校正。原笔记中跳过的定义、缺失的中间步骤和若干矩阵次序问题已补全。

0. 统一记号

设特征映射为 \(\varphi:X\to\mathbb R^d\),设计矩阵为

\[ \Phi= \begin{bmatrix} \varphi(x_1)^\top\\ \vdots\\ \varphi(x_n)^\top \end{bmatrix} \in\mathbb R^{n\times d}, \]

并记非中心经验协方差矩阵为

\[ \widehat\Sigma=\frac1n\Phi^\top\Phi\in\mathbb R^{d\times d}. \]

对于半正定矩阵 \(M\),记

\[ \|u\|_M^2=u^\top M u. \]

3.5 Fixed Design Setting

3.5.1 模型与风险分解

在固定设计情形中,\(\Phi\) 被视为确定矩阵,随机性仅来自噪声。假设

\[ y=\Phi\theta^*+\varepsilon, \qquad \mathbb E[\varepsilon]=0, \qquad \mathbb E[\varepsilon\varepsilon^\top]=\sigma^2 I_n. \]

固定设计风险为

\[ R(\theta)=\mathbb E_y\left[\frac1n\|y-\Phi\theta\|_2^2\right]. \]

将 \(y=\Phi\theta^*+\varepsilon\) 代入,得到

\[ \begin{aligned} R(\theta) &=\frac1n\mathbb E\|\Phi(\theta^*-\theta)+\varepsilon\|_2^2\\ &=\frac1n\|\Phi(\theta-\theta^*)\|_2^2 +\frac1n\mathbb E\|\varepsilon\|_2^2\\ &=(\theta-\theta^*)^\top\widehat\Sigma(\theta-\theta^*)+\sigma^2. \end{aligned} \]

因此

\[ R^*=\sigma^2, \qquad R(\theta)-R^*=\|\theta-\theta^*\|_{\widehat\Sigma}^2. \]

若估计量 \(\widehat\theta\) 是随机的,则通常的偏差--方差分解给出

\[ \mathbb E[R(\widehat\theta)]-R^* = \|\mathbb E[\widehat\theta]-\theta^*\|_{\widehat\Sigma}^2 + \mathbb E\|\widehat\theta-\mathbb E[\widehat\theta]\|_{\widehat\Sigma}^2. \]

3.5.2 OLS 的统计性质

当 \(\widehat\Sigma\) 可逆时,OLS 估计量为

\[ \widehat\theta =(\Phi^\top\Phi)^{-1}\Phi^\top y =\frac1n\widehat\Sigma^{-1}\Phi^\top y. \]

由 \(y=\Phi\theta^*+\varepsilon\),

\[ \widehat\theta-\theta^* =(\Phi^\top\Phi)^{-1}\Phi^\top\varepsilon. \]

因此

\[ \mathbb E[\widehat\theta]=\theta^*, \qquad \operatorname{Var}(\widehat\theta) =\sigma^2(\Phi^\top\Phi)^{-1} =\frac{\sigma^2}{n}\widehat\Sigma^{-1}. \]

于是

\[ \begin{aligned} \mathbb E[R(\widehat\theta)]-R^* &=\mathbb E\|\widehat\theta-\theta^*\|_{\widehat\Sigma}^2\\ &=\operatorname{tr}\!\left( \widehat\Sigma\operatorname{Var}(\widehat\theta) \right)\\ &=\frac{\sigma^2}{n}\operatorname{tr}(I_d) =\frac{\sigma^2d}{n}. \end{aligned} \]

3.6 Ridge Least-Squares Regression

3.6.1 定义与闭式解

给定 \(\lambda>0\),岭回归估计量定义为

\[ \widehat\theta_\lambda \in\arg\min_{\theta\in\mathbb R^d} \left\{ \frac1n\|y-\Phi\theta\|_2^2+\lambda\|\theta\|_2^2 \right\}. \]

一阶最优性条件为

\[ \frac2n\Phi^\top(\Phi\widehat\theta_\lambda-y) +2\lambda\widehat\theta_\lambda=0, \]

因此

\[ \widehat\theta_\lambda =\frac1n(\widehat\Sigma+\lambda I)^{-1}\Phi^\top y =(\Phi^\top\Phi+n\lambda I)^{-1}\Phi^\top y. \]

利用矩阵求逆引理,还可以写成

\[ \widehat\theta_\lambda =\Phi^\top(\Phi\Phi^\top+n\lambda I)^{-1}y. \]

当 \(d\gg n\) 时,后一形式只需求逆 \(n\times n\) 矩阵,计算上更合适。

3.6.2 Proposition 3.7:岭回归的偏差--方差分解

由

\[ \widehat\theta_\lambda =(\widehat\Sigma+\lambda I)^{-1}\widehat\Sigma\theta^* +\frac1n(\widehat\Sigma+\lambda I)^{-1}\Phi^\top\varepsilon, \]

可得

\[ \mathbb E[\widehat\theta_\lambda] =(\widehat\Sigma+\lambda I)^{-1}\widehat\Sigma\theta^* =\theta^*-\lambda(\widehat\Sigma+\lambda I)^{-1}\theta^*. \]

偏差项

\[ \begin{aligned} B &=\|\mathbb E[\widehat\theta_\lambda]-\theta^*\|_{\widehat\Sigma}^2\\ &=\lambda^2\theta^{*\top} (\widehat\Sigma+\lambda I)^{-1} \widehat\Sigma (\widehat\Sigma+\lambda I)^{-1}\theta^*. \end{aligned} \]

由于 \(\widehat\Sigma\) 与 \((\widehat\Sigma+\lambda I)^{-1}\) 可交换,也可写成

\[ B=\lambda^2\theta^{*\top} (\widehat\Sigma+\lambda I)^{-2} \widehat\Sigma\theta^*. \]

方差项

随机部分为

\[ \widehat\theta_\lambda- \mathbb E[\widehat\theta_\lambda] =\frac1n(\widehat\Sigma+\lambda I)^{-1}\Phi^\top\varepsilon. \]

因此

\[ \begin{aligned} V &=\mathbb E\left\| \widehat\theta_\lambda- \mathbb E[\widehat\theta_\lambda] \right\|_{\widehat\Sigma}^2\\ &=\frac{\sigma^2}{n} \operatorname{tr}\left[ \widehat\Sigma^2(\widehat\Sigma+\lambda I)^{-2} \right]. \end{aligned} \]

综上,

\[ \boxed{ \mathbb E[R(\widehat\theta_\lambda)]-R^* = \lambda^2\theta^{*\top}(\widehat\Sigma+\lambda I)^{-2} \widehat\Sigma\theta^* + \frac{\sigma^2}{n} \operatorname{tr}\left[ \widehat\Sigma^2(\widehat\Sigma+\lambda I)^{-2} \right] }. \]

3.6.3 Proposition 3.8:正则化参数的理论选择

对任意标量特征值 \(u\ge 0\),有

\[ \frac{\lambda u}{(u+\lambda)^2}\le\frac12. \]

因此

\[ B =\lambda\theta^{*\top} \left[\lambda\widehat\Sigma(\widehat\Sigma+\lambda I)^{-2}\right] \theta^* \le \frac\lambda2\|\theta^*\|_2^2. \]

同理,

\[ \begin{aligned} V &=\frac{\sigma^2}{\lambda n} \operatorname{tr}\left[ \widehat\Sigma\, \lambda\widehat\Sigma(\widehat\Sigma+\lambda I)^{-2} \right]\\ &\le \frac{\sigma^2\operatorname{tr}(\widehat\Sigma)}{2\lambda n}. \end{aligned} \]

故

\[ \mathbb E[R(\widehat\theta_\lambda)]-R^* \le \frac\lambda2\|\theta^*\|_2^2 + \frac{\sigma^2\operatorname{tr}(\widehat\Sigma)}{2\lambda n}. \]

右端是 \(a\lambda+b/\lambda\) 型。取

\[ \boxed{ \lambda^* = \frac{\sigma\sqrt{\operatorname{tr}(\widehat\Sigma)}} {\|\theta^*\|_2\sqrt n} } \]

得到

\[ \boxed{ \mathbb E[R(\widehat\theta_{\lambda^*})]-R^* \le \frac{ \sigma\sqrt{\operatorname{tr}(\widehat\Sigma)}\|\theta^*\|_2 }{\sqrt n} }. \]

该 \(\lambda^*\) 优化的是风险上界,不一定是真实期望风险的精确最优值。

3.6.4 Exercise 3.6:一般二次正则项

考虑

\[ \widehat\theta_\Lambda \in\arg\min_\theta \left\{ \frac1n\|y-\Phi\theta\|_2^2+\theta^\top\Lambda\theta \right\}, \]

其中 \(\Lambda\in\mathbb R^{d\times d}\) 对称正定。则

\[ \boxed{ \widehat\theta_\Lambda =(\widehat\Sigma+\Lambda)^{-1}\frac1n\Phi^\top y }. \]

其偏差向量为

\[ \mathbb E[\widehat\theta_\Lambda]-\theta^* =-(\widehat\Sigma+\Lambda)^{-1}\Lambda\theta^*. \]

因此偏差项为

\[ \boxed{ B_\Lambda = \theta^{*\top}\Lambda(\widehat\Sigma+\Lambda)^{-1} \widehat\Sigma (\widehat\Sigma+\Lambda)^{-1}\Lambda\theta^* }. \]

方差项为

\[ \boxed{ V_\Lambda = \frac{\sigma^2}{n} \operatorname{tr}\left[ \widehat\Sigma(\widehat\Sigma+\Lambda)^{-1} \widehat\Sigma(\widehat\Sigma+\Lambda)^{-1} \right] }. \]

修正说明: 当 \(\Lambda\) 与 \(\widehat\Sigma\) 不可交换时,不能把上式随意合并为 \((\widehat\Sigma+\Lambda)^{-2}\widehat\Sigma^2\)。只有两者可交换时才可这样简化。

3.6.5 Exercise 3.7:留一法公式

对每个 \(i\),定义

\[ \widehat\theta_\lambda^{-i} \in\arg\min_\theta \left\{ \frac1n\sum_{j\ne i}(y_j-\theta^\top\varphi(x_j))^2 +\lambda\|\theta\|_2^2 \right\}. \]

记

\[ H=\Phi(\Phi^\top\Phi+n\lambda I)^{-1}\Phi^\top, \qquad h=\operatorname{diag}(H). \]

令

\[ A=\Phi^\top\Phi+n\lambda I, \qquad b=\Phi^\top y, \qquad \phi_i=\varphi(x_i). \]

则

\[ \widehat\theta_\lambda=A^{-1}b, \]

而删去第 \(i\) 个样本后,

\[ \widehat\theta_\lambda^{-i} =(A-\phi_i\phi_i^\top)^{-1}(b-\phi_i y_i). \]

由 Sherman--Morrison 公式,

\[ (A-\phi_i\phi_i^\top)^{-1} =A^{-1}+\frac{A^{-1}\phi_i\phi_i^\top A^{-1}} {1-\phi_i^\top A^{-1}\phi_i}. \]

又因为

\[ h_i=\phi_i^\top A^{-1}\phi_i, \qquad \widehat y_i=\phi_i^\top\widehat\theta_\lambda=(Hy)_i, \]

整理可得

\[ \widehat\theta_\lambda^{-i} =\widehat\theta_\lambda +\frac{A^{-1}\phi_i}{1-h_i}(\widehat y_i-y_i). \]

故

\[ \boxed{ y_i-\phi_i^\top\widehat\theta_\lambda^{-i} = \frac{y_i-\widehat y_i}{1-h_i} = \frac{[(I-H)y]_i}{1-h_i} }. \]

平方并求和得到

\[ \boxed{ \frac1n\sum_{i=1}^n (y_i-\phi_i^\top\widehat\theta_\lambda^{-i})^2 = \frac1n \left\| (I-\operatorname{Diag}(h))^{-1}(I-H)y \right\|_2^2 }. \]

3.7 Lower Bound

3.7.1 极小极大问题

为证明固定设计下 OLS 的 \(\sigma^2d/n\) 风险是不可改进的,进一步假设

\[ \varepsilon\sim\mathcal N(0,\sigma^2I_n). \]

对于真实参数 \(\theta^*\),记

\[ R_{\theta^*}(\theta)-R^* =\|\theta-\theta^*\|_{\widehat\Sigma}^2. \]

任意估计算法可以写成映射

\[ A:\mathbb R^n\to\mathbb R^d, \qquad \widehat\theta=A(y). \]

目标是研究

\[ \inf_A\sup_{\theta^*\in\mathbb R^d} \mathbb E_\varepsilon \left[ R_{\theta^*}(A(\Phi\theta^*+\varepsilon))-R^* \right]. \]

3.7.2 用先验平均下界化最坏情形

选择高斯先验

\[ \theta^*\sim\mathcal N\left(0,\frac{\sigma^2}{\lambda n}I_d\right), \qquad \lambda>0. \]

对任意 \(A\),都有

\[ \sup_{\theta^*} \mathbb E_\varepsilon[\text{excess risk}] \ge \mathbb E_{\theta^*}\mathbb E_\varepsilon[\text{excess risk}]. \]

因此

\[ \inf_A\sup_{\theta^*}\mathbb E_\varepsilon[\text{excess risk}] \ge \inf_A \mathbb E_{\theta^*,\varepsilon} \|A(y)-\theta^*\|_{\widehat\Sigma}^2. \]

给定 \(y\) 后,使条件平方损失最小的估计量是后验均值

\[ A^*(y)=\mathbb E[\theta^*\mid y]. \]

3.7.3 后验均值等于岭回归

似然与先验分别满足

\[ p(y\mid\theta^*) \propto \exp\left(-\frac{\|y-\Phi\theta^*\|_2^2}{2\sigma^2}\right), \]
\[ p(\theta^*) \propto \exp\left(-\frac{\lambda n\|\theta^*\|_2^2}{2\sigma^2}\right). \]

因此后验密度为

\[ p(\theta^*\mid y) \propto \exp\left[ -\frac{ \|y-\Phi\theta^*\|_2^2+n\lambda\|\theta^*\|_2^2 }{2\sigma^2} \right]. \]

后验为高斯分布,所以后验均值等于后验众数。后验众数正是岭回归解:

\[ \boxed{ A^*(y) =(\Phi^\top\Phi+n\lambda I)^{-1}\Phi^\top y }. \]

3.7.4 贝叶斯风险计算

将

\[ A^*(y)-\theta^* =(\Phi^\top\Phi+n\lambda I)^{-1}\Phi^\top\varepsilon -n\lambda(\Phi^\top\Phi+n\lambda I)^{-1}\theta^* \]

代入。由于 \(\varepsilon\) 与 \(\theta^*\) 独立且均值为零,交叉项为零。计算偏差与方差后得到

\[ \inf_A \mathbb E_{\theta^*,\varepsilon} \|A(y)-\theta^*\|_{\widehat\Sigma}^2 = \frac{\sigma^2}{n} \operatorname{tr}\left[ (\widehat\Sigma+\lambda I)^{-1}\widehat\Sigma \right]. \]

若 \(\Phi\) 满列秩,则令 \(\lambda\downarrow0\),有

\[ (\widehat\Sigma+\lambda I)^{-1}\widehat\Sigma\to I_d. \]

因此

\[ \boxed{ \inf_A\sup_{\theta^*\in\mathbb R^d} \mathbb E_\varepsilon \left[ R_{\theta^*}(A(\Phi\theta^*+\varepsilon))-R^* \right] \ge \frac{\sigma^2d}{n} }. \]

OLS 恰好达到同样的上界,因而在该固定设计、高斯噪声、最坏参数意义下是极小极大最优的。

若 \(\Phi\) 不满列秩,上述极限为

\[ \frac{\sigma^2}{n}\operatorname{rank}(\Phi). \]

3.8 Random Design Analysis

3.8.1 随机设计模型

现在 \((x_i,y_i)\) 为 i.i.d. 样本,并假设

\[ y_i=\varphi(x_i)^\top\theta^*+\varepsilon_i, \]

其中 \(\varepsilon_i\) 与 \(x_i\) 独立,且

\[ \mathbb E[\varepsilon_i]=0, \qquad \mathbb E[\varepsilon_i^2]=\sigma^2. \]

定义总体非中心协方差矩阵

\[ \Sigma=\mathbb E[\varphi(x)\varphi(x)^\top]. \]

3.8.2 Proposition 3.9:随机设计风险

对独立测试样本 \((x_0,y_0)\),

\[ \begin{aligned} R(\theta) &=\mathbb E[(y_0-\theta^\top\varphi(x_0))^2]\\ &=\mathbb E[((\theta^*-\theta)^\top\varphi(x_0)+\varepsilon_0)^2]\\ &=(\theta-\theta^*)^\top\Sigma(\theta-\theta^*)+\sigma^2. \end{aligned} \]

故

\[ \boxed{ R^*=\sigma^2, \qquad R(\theta)-R^*=\|\theta-\theta^*\|_\Sigma^2 }. \]

固定设计与随机设计的关键差别是:风险中的矩阵由 \(\widehat\Sigma\) 换成了 \(\Sigma\)。

3.8.3 Proposition 3.10:随机设计下 OLS 风险

条件于训练输入 \(x_1,\ldots,x_n\),有

\[ \widehat\theta =\theta^*+\frac1n\widehat\Sigma^{-1}\Phi^\top\varepsilon. \]

因此

\[ \begin{aligned} \mathbb E[R(\widehat\theta)]-R^* &=\mathbb E\|\widehat\theta-\theta^*\|_\Sigma^2\\ &=\frac{\sigma^2}{n} \mathbb E\operatorname{tr}(\Sigma\widehat\Sigma^{-1}). \end{aligned} \]

即

\[ \boxed{ \mathbb E[R(\widehat\theta)]-R^* = \frac{\sigma^2}{n} \mathbb E\left[\operatorname{tr}(\Sigma\widehat\Sigma^{-1})\right] }. \]

这里的期望同时包含训练输入与噪声的随机性。

3.8.4 Exercise 3.8:随机设计下岭回归

条件于 \(\Phi\),

\[ \widehat\theta_\lambda-\theta^* =-\lambda(\widehat\Sigma+\lambda I)^{-1}\theta^* + \frac1n(\widehat\Sigma+\lambda I)^{-1}\Phi^\top\varepsilon. \]

因此

\[ \boxed{ \begin{aligned} \mathbb E[R(\widehat\theta_\lambda)]-R^* ={}&\lambda^2\mathbb E\left[ \theta^{*\top}(\widehat\Sigma+\lambda I)^{-1} \Sigma (\widehat\Sigma+\lambda I)^{-1}\theta^* \right]\\ &+\frac{\sigma^2}{n}\mathbb E\left[ \operatorname{tr}\left( (\widehat\Sigma+\lambda I)^{-2}\Sigma\widehat\Sigma \right) \right]. \end{aligned} }. \]

第二项的写法利用了迹的循环不变性,以及 \(\widehat\Sigma\) 与 \((\widehat\Sigma+\lambda I)^{-1}\) 可交换;\(\Sigma\) 一般不与它们交换。

3.8.5 Gaussian Designs 与 Wishart 分布

假设

\[ \varphi(x)\sim\mathcal N(0,\Sigma). \]

令

\[ z=\Sigma^{-1/2}\varphi(x)\sim\mathcal N(0,I_d), \]

并令 \(Z\in\mathbb R^{n\times d}\) 的每一行为 \(z_i^\top\)。则

\[ \Phi=Z\Sigma^{1/2}, \]

从而

\[ \widehat\Sigma =\frac1n\Sigma^{1/2}Z^\top Z\Sigma^{1/2}. \]

因此

\[ \operatorname{tr}(\Sigma\widehat\Sigma^{-1}) =n\operatorname{tr}((Z^\top Z)^{-1}). \]

注意

\[ \mathbb E[Z^\top Z]=nI_d. \]

函数 \(M\mapsto\operatorname{tr}(M^{-1})\) 在正定锥上是凸的,所以 Jensen 不等式给出

\[ \mathbb E\operatorname{tr}((Z^\top Z)^{-1}) \ge \operatorname{tr}\left((\mathbb E[Z^\top Z])^{-1}\right) =\frac dn. \]

但这是下界,方向不能用于证明所需的风险上界。

由于

\[ Z^\top Z\sim W_d(n,I_d) \]

服从 Wishart 分布,并且当 \(n>d+1\) 时

\[ \mathbb E[(Z^\top Z)^{-1}] =\frac1{n-d-1}I_d, \]

所以

\[ \mathbb E\operatorname{tr}((Z^\top Z)^{-1}) =\frac d{n-d-1}. \]

最终得到精确风险

\[ \boxed{ \mathbb E[R(\widehat\theta)]-R^* =\frac{\sigma^2d}{n-d-1} =\frac{\sigma^2d/n}{1-(d+1)/n} }, \qquad n>d+1. \]

当 \(n\to\infty\) 且 \(d\) 固定时,它渐近等价于 \(\sigma^2d/n\);当 \(d/n\) 接近 \(1\) 时,风险会显著放大。

3.8.6 Proposition 3.11:矩阵 Bernstein 不等式

设 \(M_1,\ldots,M_n\in\mathbb R^{d\times d}\) 为独立对称随机矩阵,满足

\[ \mathbb E[M_i]=0, \qquad \lambda_{\max}(M_i)\le b \quad\text{a.s.} \]

定义

\[ \tau^2 =\lambda_{\max}\left( \frac1n\sum_{i=1}^n\mathbb E[M_i^2] \right). \]

则对所有 \(t\ge0\),

\[ \boxed{ \mathbb P\left( \lambda_{\max}\left(\frac1n\sum_{i=1}^nM_i\right)\ge t \right) \le d\exp\left( -\frac{nt^2/2}{\tau^2+bt/3} \right) }. \]

推导骨架

矩阵情形不能直接把 \(\exp(s\sum_iM_i)\) 分解成乘积。关键工具是 Tropp 型矩阵 Laplace 变换界:

\[ \mathbb E\operatorname{tr}\exp\left(s\sum_{i=1}^nM_i\right) \le \operatorname{tr}\exp\left( \sum_{i=1}^n\log\mathbb E e^{sM_i} \right). \]

由 \(\lambda_{\max}(M_i)\le b\) 和 \(\mathbb E[M_i]=0\),可得矩阵 mgf 上界

\[ \log\mathbb E e^{sM_i} \preceq \frac{e^{sb}-sb-1}{b^2}\mathbb E[M_i^2]. \]

于是

\[ \mathbb P\left( \lambda_{\max}\left(\sum_iM_i\right)\ge nt \right) \le d\inf_{s>0} \exp\left[ -nst+\frac{n\tau^2}{b^2}(e^{sb}-sb-1) \right]. \]

优化得到 Bennett 形式

\[ d\exp\left[-\frac{n\tau^2}{b^2}h\left(\frac{bt}{\tau^2}\right)\right], \]

其中

\[ h(u)=(1+u)\log(1+u)-u. \]

再使用

\[ h(u)\ge\frac{u^2}{2(1+u/3)}, \qquad u\ge0, \]

即得到上述 Bernstein 形式。

3.8.7 Proposition 3.12:经验协方差的下谱界

设

\[ \Sigma=\mathbb E[\varphi(x)\varphi(x)^\top] \]

可逆,并假设存在 \(\rho>0\),使得

\[ \boxed{ \mathbb E\left[ \varphi(x)^\top\Sigma^{-1}\varphi(x) \,\varphi(x)\varphi(x)^\top \right] \preceq \rho d\,\Sigma }. \]

令

\[ z=\Sigma^{-1/2}\varphi(x). \]

则

\[ \mathbb E[zz^\top]=I, \qquad \mathbb E[\|z\|_2^2zz^\top]\preceq\rho dI. \]

取

\[ M_i=I-z_iz_i^\top. \]

有

\[ \mathbb E[M_i]=0, \qquad \lambda_{\max}(M_i)\le1, \]

且

\[ \mathbb E[M_i^2] = \mathbb E[\|z_i\|_2^2z_iz_i^\top]-I \preceq\rho dI. \]

由矩阵 Bernstein,

\[ \mathbb P\left( \lambda_{\max}\left( I-\frac1n\sum_{i=1}^nz_iz_i^\top \right)\ge t \right) \le d\exp\left[-\frac{nt^2/2}{\rho d+t/3}\right]. \]

取 \(t=3/4\)。若

\[ \boxed{ n\ge5\rho d\log\frac d\delta }, \]

则以至少 \(1-\delta\) 的概率,

\[ \frac1n\sum_{i=1}^nz_iz_i^\top\succeq\frac14I. \]

等价地,

\[ \boxed{ \Sigma^{-1/2}\widehat\Sigma\Sigma^{-1/2} \succeq\frac14I }. \]

在该事件上,

\[ \widehat\Sigma^{-1} \preceq4\Sigma^{-1}, \]

所以条件于训练输入的 OLS 超额风险满足

\[ \frac{\sigma^2}{n} \operatorname{tr}(\Sigma\widehat\Sigma^{-1}) \le \frac{4\sigma^2d}{n}. \]

3.9 Principal Component Analysis

3.9.1 PCA 的优化问题

给定 \(\Phi\in\mathbb R^{n\times d}\),希望找到一个 \(k\) 维线性子空间,使所有特征向量到该子空间的投影误差最小。

用列正交矩阵

\[ V\in\mathbb R^{d\times k}, \qquad V^\top V=I_k \]

表示该子空间。投影矩阵为 \(VV^\top\),于是 PCA 求解

\[ \min_{V^\top V=I_k} \|\Phi-\Phi VV^\top\|_F^2. \]

展开:

\[ \begin{aligned} \|\Phi-\Phi VV^\top\|_F^2 &=\operatorname{tr}\left[(\Phi-\Phi VV^\top)^\top (\Phi-\Phi VV^\top)\right]\\ &=\operatorname{tr}(\Phi^\top\Phi) -\operatorname{tr}(V^\top\Phi^\top\Phi V). \end{aligned} \]

因此等价于

\[ \max_{V^\top V=I_k} \operatorname{tr}(V^\top\Phi^\top\Phi V). \]

若

\[ \widehat\Sigma =Q\operatorname{Diag}(\lambda_1,\ldots,\lambda_d)Q^\top, \qquad \lambda_1\ge\cdots\ge\lambda_d, \]

则最优 \(V\) 可取为前 \(k\) 个特征向量

\[ V_k=[q_1,\ldots,q_k]. \]

最小重构误差为

\[ \boxed{ \min_{V^\top V=I_k} \|\Phi-\Phi VV^\top\|_F^2 =n\sum_{j=k+1}^d\lambda_j }. \]

若使用奇异值分解

\[ \Phi=U\operatorname{Diag}(s_1,\ldots,s_r)V^\top, \]

则 \(\lambda_j=s_j^2/n\),PCA 重构矩阵为

\[ \Phi V_kV_k^\top =U_k\operatorname{Diag}(s_1,\ldots,s_k)V_k^\top. \]

3.9.2 Exercise 3.9:低秩分解与交替最小化

考虑

\[ \min_{A\in\mathbb R^{n\times k},\,D\in\mathbb R^{k\times d}} \|\Phi-AD\|_F^2. \]

因为 \(\operatorname{rank}(AD)\le k\),该问题等价于寻找 \(\Phi\) 的最佳秩 \(k\) 逼近。由 Eckart--Young 定理,最优乘积为

\[ \boxed{ A^*D^*=U_k\operatorname{Diag}(s_1,\ldots,s_k)V_k^\top =\Phi V_kV_k^\top }. \]

分解 \((A^*,D^*)\) 本身不唯一:对任意可逆 \(T\in\mathbb R^{k\times k}\),

\[ A^*D^*=(A^*T)(T^{-1}D^*). \]

交替最小化更新

固定 \(D_t\),最小化 \(\|\Phi-AD_t\|_F^2\):

\[ \boxed{ A_t=\Phi D_t^\top(D_tD_t^\top)^{-1} }, \]

假设 \(D_t\) 满行秩。

固定 \(A_t\),最小化 \(\|\Phi-A_tD\|_F^2\):

\[ \boxed{ D_{t+1}=(A_t^\top A_t)^{-1}A_t^\top\Phi }, \]

假设 \(A_t\) 满列秩。

行空间的演化

令

\[ C=\Phi^\top\Phi. \]

代入 \(A_t\) 后可化为

\[ D_{t+1}=L_tD_tC, \]

其中 \(L_t\in\mathbb R^{k\times k}\) 可逆。因此左乘 \(L_t\) 不改变行空间,故

\[ \operatorname{row}(D_{t+1}) = \operatorname{row}(D_tC). \]

递推得到

\[ \operatorname{row}(D_t) = \operatorname{row}(D_0C^t). \]

这正是对子空间进行的幂迭代。

设

\[ C=V \begin{bmatrix} \Lambda_1&0\\ 0&\Lambda_2 \end{bmatrix} V^\top, \]

其中 \(\Lambda_1=\operatorname{Diag}(s_1^2,\ldots,s_k^2)\)。写成

\[ D_0V=[B_1\;B_2]. \]

只要 \(B_1\) 可逆,就有

\[ \operatorname{row}(D_0C^t) = \operatorname{row}\left( [I\;E_t]V^\top \right), \]

其中

\[ E_t =\Lambda_1^{-t}B_1^{-1}B_2\Lambda_2^t. \]

若存在谱隙 \(s_k>s_{k+1}\),则

\[ \|E_t\|_{\mathrm{op}} \le \|B_1^{-1}B_2\|_{\mathrm{op}} \left(\frac{s_{k+1}^2}{s_k^2}\right)^t \longrightarrow0. \]

于是

\[ \operatorname{row}(D_t)\to\operatorname{span}(v_1,\ldots,v_k), \]

从而

\[ A_tD_t =\Phi P_{\operatorname{row}(D_t)} \longrightarrow \Phi V_kV_k^\top. \]

“几乎所有初始化”来自条件 \(\det(B_1)\ne0\):使 \(B_1\) 奇异的初始化构成测度为零的代数集合。

若 \(s_k=s_{k+1}\),最优主子空间不唯一;算法仍可收敛到某个最优的主不变子空间,但不能要求收敛到唯一指定的 \(\operatorname{span}(v_1,\ldots,v_k)\)。

3.9.3 Exercise 3.10:K-means 的交替更新

考虑

\[ \min_{A,D}\|\Phi-AD\|_F^2, \]

其中

\[ A\in\{0,1\}^{n\times k}, \qquad \sum_{j=1}^kA_{ij}=1, \]

而 \(D\in\mathbb R^{k\times d}\) 的第 \(j\) 行 \(d_j^\top\) 表示第 \(j\) 个聚类中心。

目标函数展开为

\[ \|\Phi-AD\|_F^2 = \sum_{i=1}^n\sum_{j=1}^k A_{ij}\|\varphi(x_i)-d_j\|_2^2. \]

固定 \(D\) 更新 \(A\)

每个样本分配给最近的中心:

\[ \boxed{ A_{ij}=1 \quad\Longleftrightarrow\quad j\in\arg\min_{\ell\in\{1,\ldots,k\}} \|\varphi(x_i)-d_\ell\|_2^2 }. \]

固定 \(A\) 更新 \(D\)

对每个非空簇,中心取簇内均值:

\[ \boxed{ d_j = \frac{\sum_{i=1}^nA_{ij}\varphi(x_i)} {\sum_{i=1}^nA_{ij}} }. \]

若某个簇为空,则分母为零,需要重新初始化该中心或采用其他空簇处理规则。

3.9.4 PCA 后的最小二乘回归

取 PCA 的前 \(k\) 个特征向量组成 \(V\in\mathbb R^{d\times k}\),使用降维特征矩阵 \(\Phi V\)。考虑

\[ \min_{\eta\in\mathbb R^k} \|y-\Phi V\eta\|_2^2. \]

闭式解为

\[ \widehat\eta =(V^\top\Phi^\top\Phi V)^{-1}V^\top\Phi^\top y, \]

预测向量为

\[ \widehat y_{ m PCA} = \Phi V(V^\top\Phi^\top\Phi V)^{-1}V^\top\Phi^\top y. \]

在固定设计线性模型 \(y=\Phi\theta^*+\varepsilon\) 下,

\[ \begin{aligned} \frac1n\mathbb E_\varepsilon \|\widehat y_{\rm PCA}-\Phi\theta^*\|_2^2 ={}&\frac{\sigma^2k}{n}\\ &+\frac1n\left\| \Phi V(V^\top\Phi^\top\Phi V)^{-1} V^\top\Phi^\top\Phi\theta^* -\Phi\theta^* \right\|_2^2. \end{aligned} \]

由于 \(V\) 由 \(\widehat\Sigma\) 的特征向量组成,

\[ \widehat\Sigma V=VD, \qquad V^\top\widehat\Sigma V=D, \]

因此偏差项化简为

\[ \theta^{*\top}(I-VV^\top) \widehat\Sigma (I-VV^\top)\theta^*. \]

于是

\[ \boxed{ \frac1n\mathbb E_\varepsilon \|\widehat y_{\rm PCA}-\Phi\theta^*\|_2^2 = \frac{\sigma^2k}{n} + \theta^{*\top}(I-VV^\top) \widehat\Sigma (I-VV^\top)\theta^* }. \]

若 \(\lambda_{k+1}\) 是 \(\widehat\Sigma\) 的第 \(k+1\) 大特征值,则

\[ \theta^{*\top}(I-VV^\top) \widehat\Sigma (I-VV^\top)\theta^* \le \lambda_{k+1}\|\theta^*\|_2^2. \]

又因为

\[ \lambda_{k+1} \le\frac{\operatorname{tr}(\widehat\Sigma)}{k+1} \le\frac{\operatorname{tr}(\widehat\Sigma)}k, \]

所以

\[ \boxed{ \frac1n\mathbb E_\varepsilon \|\widehat y_{\rm PCA}-\Phi\theta^*\|_2^2 \le \frac{\sigma^2k}{n} + \frac{\operatorname{tr}(\widehat\Sigma)}k\|\theta^*\|_2^2 }. \]

最小化 \(ak+b/k\) 型上界,理论上应选

\[ \boxed{ k^* \approx \frac{\|\theta^*\|_2 \sqrt{n\operatorname{tr}(\widehat\Sigma)}}{\sigma} }. \]

代入后得到与岭回归同阶的界

\[ \boxed{ \frac1n\mathbb E_\varepsilon \|\widehat y_{\rm PCA}-\Phi\theta^*\|_2^2 \lesssim \frac{\sigma\|\theta^*\|_2 \sqrt{\operatorname{tr}(\widehat\Sigma)}}{\sqrt n} }. \]

PCA 是“硬截断”小特征值方向;岭回归则对各特征方向进行连续的“软收缩”。