考虑以下(凸)优化问题:
minimize 0.5 * y.T * y
s.t. A*x - b == y
其中优化(向量)变量为x
和y
以及A
,b
分别是具有适当维度的矩阵和向量。
以下代码使用Scipy中的SLSQP
方法轻松找到解决方案:
import numpy as np
from scipy.optimize import minimize
# problem dimensions:
n = 10 # arbitrary integer set by user
m = 2 * n
# generate parameters A, b:
np.random.seed(123) # for reproducibility of results
A = np.random.randn(m,n)
b = np.random.randn(m)
# objective function:
def obj(z):
vy = z[n:]
return 0.5 * vy.dot(vy)
# constraint function:
def cons(z):
vx = z[:n]
vy = z[n:]
return A.dot(vx) - b - vy
# constraints input for SLSQP:
cons = ({'type': 'eq','fun': cons})
# generate a random initial estimate:
z0 = np.random.randn(n+m)
sol = minimize(obj, x0 = z0, constraints = cons, method = 'SLSQP', options={'disp': True})
Optimization terminated successfully. (Exit mode 0) Current function value: 2.12236220865 Iterations: 6 Function evaluations: 192 Gradient evaluations: 6
请注意,约束函数是一个方便的数组输出'功能。
现在,代替约束的数组输出函数,原则上可以使用一组等效的标量输出'约束函数(实际上,scipy.optimize文档仅讨论此类约束函数作为minimize
的输入)。
以下是等效约束集,后跟minimize
的输出(相同A
,b
,以及上面列出的初始值):
# this is the i-th element of cons(z):
def cons_i(z, i):
vx = z[:n]
vy = z[n:]
return A[i].dot(vx) - b[i] - vy[i]
# listable of scalar-output constraints input for SLSQP:
cons_per_i = [{'type':'eq', 'fun': lambda z: cons_i(z, i)} for i in np.arange(m)]
sol2 = minimize(obj, x0 = z0, constraints = cons_per_i, method = 'SLSQP', options={'disp': True})
Singular matrix C in LSQ subproblem (Exit mode 6) Current function value: 6.87999270692 Iterations: 1 Function evaluations: 32 Gradient evaluations: 1
显然,算法失败(返回的目标值实际上是给定初始化的目标值),我觉得有点奇怪。请注意,运行[cons_per_i[i]['fun'](sol.x) for i in np.arange(m)]
表示使用数组输出约束公式获得的sol.x
满足cons_per_i
的所有标量输出约束(在数值公差范围内)。
如果有人对此问题有任何解释,我将不胜感激。
答案 0 :(得分:7)
您已遇到"late binding closures" gotcha。所有对cons_i
的调用都是在第二个参数等于19的情况下进行的。
修复是使用字典中的args
字典元素来定义约束而不是lambda函数闭包:
cons_per_i = [{'type':'eq', 'fun': cons_i, 'args': (i,)} for i in np.arange(m)]
有了这个,最小化就起作用了:
In [417]: sol2 = minimize(obj, x0 = z0, constraints = cons_per_i, method = 'SLSQP', options={'disp': True})
Optimization terminated successfully. (Exit mode 0)
Current function value: 2.1223622086
Iterations: 6
Function evaluations: 192
Gradient evaluations: 6
您还可以使用链接文章中提出的建议,即使用具有所需默认值的第二个参数的lambda表达式:
cons_per_i = [{'type':'eq', 'fun': lambda z, i=i: cons_i(z, i)} for i in np.arange(m)]