中文

论信道容量在高斯混合模型学习中的作用

信息论 2022-06-15 v2 机器学习 math.IT 统计理论 机器学习 统计理论

摘要

本文研究了在 Rd\mathbb{R}^d 中具有球协方差矩阵 σ2I\sigma^2\mathbf{I} 的平衡高斯混合模型(GMM)的 kk 个未知中心的学习样本复杂度。具体而言,我们关注以下问题:在何种最大噪声水平 σ2\sigma^2 下,样本复杂度与从带标签测量中估计中心时基本相当?为此,我们将问题限制于贝叶斯公式,其中中心均匀分布在球面 dSd1\sqrt{d}\mathcal{S}^{d-1} 上。我们的主要结果刻画了精确的噪声阈值 σ2\sigma^2:在其以下,GMM 学习问题在大型系统极限 d,kd,k\to\infty 下与从带标签观测中学习一样容易,而在其以上则明显更难。该阈值出现在 logkd=12log(1+1σ2)\frac{\log k}{d} = \frac12\log\left( 1+\frac{1}{\sigma^2} \right),即加性白高斯噪声(AWGN)信道的容量。将 kk 个中心视为一个码,该噪声阈值可解释为码在 AWGN 信道上错误概率较小的最大噪声水平。先前关于 GMM 学习问题的工作已确定中心间最小距离作为决定相应 GMM 统计学习难度的关键参数。尽管我们的结果仅针对中心均匀分布于球面上的 GMM 证明,但它们暗示或许正是中心星座作为信道码的解码错误概率决定了相应 GMM 的统计学习难度,而非仅由最小距离决定。

关键词

引用

@article{arxiv.2202.07707,
  title  = {On the Role of Channel Capacity in Learning Gaussian Mixture Models},
  author = {Elad Romanov and Tamir Bendory and Or Ordentlich},
  journal= {arXiv preprint arXiv:2202.07707},
  year   = {2022}
}

备注

COLT 2022