中文

图上梯度稀疏参数估计的树投影梯度下降

机器学习 2020-06-03 v1 机器学习 统计理论 统计方法学 统计理论

摘要

我们研究梯度稀疏参数向量 θRp\boldsymbol{\theta}^* \in \mathbb{R}^p 的估计,该向量在底层图 GG 上具有强梯度稀疏性 s:=Gθ0s^*:=\|\nabla_G \boldsymbol{\theta}^*\|_0。给定观测 Z1,,ZnZ_1,\ldots,Z_n 以及一个光滑凸损失函数 L\mathcal{L},其中 θ\boldsymbol{\theta}^* 最小化总体风险 E[L(θ;Z1,,Zn)]\mathbb{E}[\mathcal{L}(\boldsymbol{\theta};Z_1,\ldots,Z_n)],我们提出通过投影梯度下降算法估计 θ\boldsymbol{\theta}^*,该算法迭代地并将梯度步长近似投影到在 GG 的低度生成树上具有小梯度稀疏性的向量空间上。我们证明,在损失的适当限制强凸性和光滑性假设下,所得估计量达到平方误差风险 snlog(1+ps)\frac{s^*}{n} \log (1+\frac{p}{s^*})(直至一个独立于 GG 的乘法常数)。相比之下,以往的多项式时间算法仅被证明在更专门的设定中,或在对 GG 和/或 Gθ\nabla_G \boldsymbol{\theta}^* 的稀疏模式附加假设下达到该保证。作为我们通用框架的应用,我们将结果应用于具有随机设计的线性模型和广义线性模型示例。

关键词

引用

@article{arxiv.2006.01662,
  title  = {Tree-Projected Gradient Descent for Estimating Gradient-Sparse Parameters on Graphs},
  author = {Sheng Xu and Zhou Fan and Sahand Negahban},
  journal= {arXiv preprint arXiv:2006.01662},
  year   = {2020}
}