中文

矩阵乘积态机器学习架构中的泛化与过拟合

机器学习 2022-08-10 v1 量子物理

摘要

尽管过拟合以及更一般的双下降在机器学习中无处不在,但在以往研究中,增加最广泛使用的张量网络——矩阵乘积态(MPS)的参数数量通常导致测试性能单调提升。为了更好地理解由 MPS 参数化的架构的泛化性质,我们构造了可被 MPS 精确建模的人工数据,并用不同参数数量训练模型。我们观察到一维数据上的模型过拟合,但也发现对于更复杂的数据过拟合不那么显著,而在 MNIST 图像数据上我们未发现任何过拟合迹象。我们推测 MPS 的泛化性质依赖于数据的性质:对于一维数据(MPS 拟设最适用)MPS 易于过拟合,而对于不能被 MPS 精确拟合的更复杂数据,过拟合可能远不显著。

关键词

引用

@article{arxiv.2208.04372,
  title  = {Generalization and Overfitting in Matrix Product State Machine Learning Architectures},
  author = {Artem Strashko and E. Miles Stoudenmire},
  journal= {arXiv preprint arXiv:2208.04372},
  year   = {2022}
}

备注

Main text: 8 pages and 6 figures. Supplementary material: 3 pages, 3 figures