GPU上经典机器学习问题的优化
机器学习
2022-03-31 v1 最优化与控制
机器学习
摘要
约束优化问题在经典机器学习中频繁出现。存在解决约束优化的框架，例如CVXPY和GENO。然而，与深度学习框架相比，GPU支持有限。在此，我们扩展了GENO框架以在GPU上求解约束优化问题。该框架允许用户以易读的建模语言指定约束优化问题。然后从该规范自动生成求解器。在GPU上运行时，该求解器比最先进的方法（如CVXPY结合cuOSQP或SCS等GPU加速求解器）快几个数量级。
引用
@article{arxiv.2203.16340,
title = {Optimization for Classical Machine Learning Problems on the GPU},
author = {Sören Laue and Mark Blacher and Joachim Giesen},
journal= {arXiv preprint arXiv:2203.16340},
year = {2022}
}
备注
Appeared in AAAI 2022