用机器学习自动学出快速求解凸优化问题的参数,仅需10个样本即可
Learning Algorithm Hyperparameters for Fast Parametric Convex Optimization
- 设计分阶段超参数学习框架,迭代初期可变,后期固定以保证收敛
- 在梯度下降等算法上实现闭式最优步长,训练时通过反向传播优化误差
- 仅用10个问题实例就能高效训练,适用于控制、信号处理等多领域
我们提出一种机器学习框架,用于学习一阶优化方法(如梯度下降)的超参数序列,以快速求解参数化凸优化问题。该计算架构采用固定点迭代,包含两个阶段:第一阶段超参数随迭代变化,第二阶段保持恒定。所学优化器可适应任意迭代次数,且保证收敛至最优解。训练目标为最小化与真实解的均方误差;在梯度下降中,单步最优步长可通过最小二乘求解;对于无约束二次优化,两步和三步最优解可闭式计算。其他情形则通过算法步骤反向传播优化目标函数。我们在梯度下降、近端梯度下降及两种基于ADMM的求解器(OSQP和SCS)上验证了该方法。利用样本收敛界,我们为未见数据提供了性能的上下界一般化保证。实验展示其在控制、信号处理和机器学习中的有效性,尤其表现出极高数据效率——所有案例仅用10个问题实例进行训练。
原文摘要 · Abstract (English)
We introduce a machine-learning framework to learn the hyperparameter sequence of first-order methods (e.g., the step sizes in gradient descent) to quickly solve parametric convex optimization problems. Our computational architecture amounts to running fixed-point iterations where the hyperparameters are the same across all parametric instances and consists of two phases. In the first step-varying phase the hyperparameters vary across iterations, while in the second steady-state phase the hyperparameters are constant across iterations. Our learned optimizer is flexible in that it can be evaluated on any number of iterations and is guaranteed to converge to an optimal solution. To train, we minimize the mean square error to a ground truth solution. In the case of gradient descent, the one-step optimal step size is the solution to a least squares problem, and in the case of unconstrained quadratic minimization, we can compute the two and three-step optimal solutions in closed-form. In other cases, we backpropagate through the algorithm steps to minimize the training objective after a given number of steps. We show how to learn hyperparameters for several popular algorithms: gradient descent, proximal gradient descent, and two ADMM-based solvers: OSQP and SCS. We use a sample convergence bound to obtain generalization guarantees for the performance of our learned algorithm for unseen data, providing both lower and upper bounds. We showcase the effectiveness of our method with many examples, including ones from control, signal processing, and machine learning. Remarkably, our approach is highly data-efficient in that we only use $10$ problem instances to train the hyperparameters in all of our examples.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。