中文

$\ell_{1,\infty}$球上的近线性时间投影及其在稀疏自编码器中的应用

机器学习 2025-02-28 v2 最优化与控制

摘要

当今寻求稀疏性对于加速大规模神经网络训练至关重要。到1,2\ell_{1,2}1,\ell_{1,\infty}的投影是稀疏化并降低神经网络总体成本的最高效技术之一。本文中,我们引入了一种用于1,\ell_{1,\infty}范数球的新投影算法。对于Rn×m\mathbb{R}^{n\times m}中的矩阵,该算法的最坏情况时间复杂度为O(nm+Jlog(nm))\mathcal{O}\big(nm+J\log(nm)\big)JJ是一项在稀疏性高时趋于0、稀疏性低时趋于nmnm的项。其实现简单,且保证在有限时间内收敛到精确解。此外,我们提出在训练自编码器时融入1,\ell_{1,\infty}球投影,以强制特征选择与权重稀疏性。由于我们在生物学中的应用仅有一小部分(<2%<2\%)数据相关,稀疏化出现在编码器中以主要进行特征选择。我们表明在生物学案例和稀疏性的一般案例中,我们的方法都是最快的。

关键词

引用

@article{arxiv.2307.09836,
  title  = {Near-Linear Time Projection onto the $\ell_{1,\infty}$ Ball; Application to Sparse Autoencoders},
  author = {Guillaume Perez and Laurent Condat and Michel Barlaud},
  journal= {arXiv preprint arXiv:2307.09836},
  year   = {2025}
}

备注

22 pages, 8 figures