julia - 用线性/非线性回归拟合两条曲线

Question

我需要用 JuMP 将两条曲线(都应该属于三次函数)拟合成一组点。

我已经完成了一条曲线的拟合，但我正在努力将 2 条曲线拟合到同一个数据集中。

我想，如果我可以将点分布到曲线上——所以如果每个点只能使用一次——我可以像下面那样做，但它没有用。 (我知道我可以使用更复杂的东西，我想保持简单。)

这是我当前代码的一部分:

# cubicFunc is a two dimensional array which accepts cubicFunc[x,degree]

@variable(m, mult1[1:4]) // 0:3 because it's cubic
@variable(m, mult2[1:4]) // 0:3 because it's cubic

@variable(m, 0 <= includeIn1[1:numOfPoints] <= 1, Int)
@variable(m, 0 <= includeIn2[1:numOfPoints] <= 1, Int)

# some kind of hack to force one of them to 0 and other one to 1
@constraint(m, loop[i in 1:numOfPoints], includeIn1[i] + includeIn2[i] == 1)

@objective(m, Min, sum( (yPoints - cubicFunc*mult1).*includeIn1 .^2 ) + sum( (yPoints - cubicFunc*mult2).*includeIn2 .^2 ))

但它会根据我的尝试给出各种错误； *includeIn1和，.*includeIn1不起作用，我尝试通过 @NLobjective 来实现但它给了我大约 50 行错误等。

我的想法现实吗？我可以把它写进代码吗？

任何帮助将不胜感激。非常感谢。

Answer 1

您可以写下问题，例如像这样:

using JuMP, Ipopt

m = Model(with_optimizer(Ipopt.Optimizer))

@variable(m, mult1[1:4])
@variable(m, mult2[1:4])
@variable(m, 0 <= includeIn1[1:numOfPoints] <= 1)
@variable(m, 0 <= includeIn2[1:numOfPoints] <= 1)

@NLconstraint(m, loop[i in 1:numOfPoints], includeIn1[i] + includeIn2[i] == 1)

@NLobjective(m, Min, sum(includeIn1[i] * (yPoints[i] - sum(cubicFunc[i,j]*mult1[j] for j in 1:4)) ^2 for i in 1:numOfPoints) +
                     sum(includeIn2[i] * (yPoints[i] - sum(cubicFunc[i,j]*mult2[j] for j in 1:4)) ^2 for i in 1:numOfPoints))

optimize!(m)

鉴于约束 includeIn1和 includeIn2将是 1或 0处于最佳状态(如果它们不是，这意味着您将点分配给哪个组无关紧要)，因此我们不必将它们限制为二进制。此外，我使用非线性求解器，因为该问题似乎无法重新表述为线性或二次优化任务。

不过，我上面的代码只是作为一个例子，你如何写下来。您制定的任务没有唯一的局部最小值(当时是全局最小值)，而是几个局部最小值。因此，使用 JuMP 支持的标准非线性凸求解器只能找到一个局部最优解(不一定是全局最优解)。为了寻找全局最优解，您需要切换到全局求解器，例如https://github.com/robertfeldt/BlackBoxOptim.jl .