用于求解非线性方程的levenberg-Marquardt方法

时间:2016-09-29 21:49:15

标签: algorithm math julia numerical-methods levenberg-marquardt

我尝试使用levenberg-marquardt方法基于Numerical Optimization using the Levenberg-Marquardt Algorithm演示来解决Julia上的非线性方程。这是我的代码:

function get_J(ArrOfFunc,X,delta)
  N = length(ArrOfFunc)
  J = zeros(Float64,N,N)
  for i = 1:N
    for j=1:N
      Temp = copy(X);
      Temp[j]=Temp[j]+delta;
      J[i,j] = (ArrOfFunc[i](Temp)-ArrOfFunc[i](X))/delta;
    end
  end
  return J
end

function get_resudial(ArrOfFunc,Arg)
  return  map((x)->x(Arg),ArrOfFunc)
end

function lm_solve(Funcs,Init)
  X = copy(Init)
  delta = 0.01;
  Lambda = 0.01;
  Factor = 2;
  J = get_J(Funcs,X,delta)
  R = get_resudial(Funcs,X)
  N = 5
  for t = 1:N

    G = J'*J+Lambda.*eye(length(X))
    dC = J'*R
    C = sum(R.*R)/2;
    Xnew = X-(inv(G)\dC);
    Rnew = get_resudial(Funcs,Xnew)
    Cnew =  sum(Rnew.*Rnew)/2;
    if ( Cnew < C)
      X = Xnew;
      R = Rnew;
      Lambda = Lambda/Factor;
      J = get_J(Funcs,X,delta)
    else
      Lambda = Lambda*Factor;
    end
    if(maximum(abs(Rnew)) < 0.001)
      return X
    end
  end
  return X
end

function test()
  ArrOfFunc = [
  (X)->X[1]+X[2]-2;
  (X)->X[1]-X[2]
  ];

  X = lm_solve(ArrOfFunc,Float64[3;3])
  println(X)
  return X
end

但从任何起点都不接受这一步骤。我做错了什么? 任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:2)

我目前没办法测试这个,但是一行在数学上没有意义:

Xnew的计算中,它应该是inv(G)*dCG\dC,而不是两者的混合。优选地,第二个,因为线性系统的解不需要计算逆矩阵。

在迭代的中心有一个错误的计算,计算的轨迹几乎肯定会误入歧途。