约束非线性优化
问题陈述 :
找到函数 f(x,y)
的最小值(超过 x
,y
),取决于 g(x,y)=0
,其中 f(x,y) = 2 * x**2 + 3 * y**2
和 g(x,y) = x**2 + y**2 - 4
。
解决方案 :我们将通过执行以下步骤来解决此问题:
- 为问题指定拉格朗日函数
- 确定 Karush-Kuhn-
Tucker(KKT)
条件 - 找到满足 KKT 条件的
(x,y)
元组 - 确定哪些
(x,y)
元组对应于f(x,y)
的最小值
首先,定义优化变量以及目标和约束函数:
import sympy as sp
x, y = sp.var('x,y',real=True);
f = 2 * x**2 + 3 * y**2
g = x**2 + y**2 - 4
接下来,定义拉格朗日函数,其包括对应于约束的拉格朗日乘数 lam
lam = sp.symbols('lambda', real = True)
L = f - lam* g
现在,我们可以计算对应于 KKT 条件的方程组。
gradL = [sp.diff(L,c) for c in [x,y]] # gradient of Lagrangian w.r.t. (x,y)
KKT_eqs = gradL + [g]
KKT_eqs
[-2*lambda*x + 4*x, -2*lambda*y + 6*y, x**2 + y**2 - 4]
f
(给定 g=0
)的潜在最小化是通过解决 KKT_eqs
方程得到的 x
,y
,lam
:
stationary_points = sp.solve(KKT_eqs, [x, y, lam], dict=True) # solve the KKT equations
stationary_points
[{x: -2, y: 0, lambda: 2}, {x: 2, y: 0, lambda: 2}, {x: 0, y: -2, lambda: 3}, {x: 0, y: 2, lambda: 3}]
最后,检查上述每个点的目标函数以确定最小值
[f.subs(p) for p in stat_points]
[8, 8, 12, 12]
由此得出,f
的约束最小值等于 8
并且在 (x,y)=(-2,0)
和 (x,y)=(2,0)
处实现。