本文介绍了一个名为 rlaopt 的开源软件包,旨在利用随机数值线性代数(RandNLA)解决大规模优化问题。该工具基于 PyTorch 构建,填补了现有 RandNLA 算法在 GPU 加速和用户友好界面方面的空白。rlaopt 提供了针对正定线性系统和带约束及正则化的凸经验风险最小化问题的求解器,通过随机化技术加速共轭梯度、算子分裂和随机梯度等方法。其特色包括自然数学语法建模、自动求解器兼容性检查,以及支持迭代过程微分以实现超参数调优。实验表明,在岭回归、有界多项逻辑回归等任务中,随机预条件能提升性能,且 GPU 执行带来显著加速。该软件以 Apache 许可开源,代码和版本已公开,为大规模科学计算和机器学习优化提供了高效且易用的新工具。
| rlaopt | 一个基于 PyTorch 的软件包,用于大规模优化和科学计算,利用随机数值线性代数并支持 GPU 加速。 |
| RandNLA | 随机数值线性代数,一种利用随机化技术加速线性代数运算(如矩阵分解和求解)的方法。 |
| NystromPCG | 一种结合 Nyström 近似和预条件共轭梯度法的算法,用于加速正定线性系统的求解。 |
| NysADMM | 一种将 Nyström 近似与交替方向乘子法(ADMM)相结合的算子分裂方法,用于求解带约束的优化问题。 |
| SAPPHIRE | 一种随机梯度方法,可能利用随机化技术来加速大规模优化中的梯度计算。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅