一杯清酒邀明月
天下本无事,庸人扰之而烦耳。

上一篇博客中介绍的高斯牛顿算法可能会有J'*J为奇异矩阵的情况,这时高斯牛顿法稳定性较差,可能导致算法不收敛。比如当系数都为7或更大的时候,算法无法给出正确的结果。

Levenberg-Marquardt法一定程度上修正了这个问题。

计算迭代系数deltaX公式如下:

当lambda很小的时候,H占主要地位,公式变为高斯牛顿法,当lambda很大的时候,H可以忽略,公式变为最速下降法。该方法提供了更稳定的deltaX。

算法步骤如下:

1.给定初始系数,以及初始优化半径u。

2.计算使用当前系数的模型得到的结果与测量结果差值e。

3.使用迭代公式更新带解算系数。

4.计算更新后系数的模型得到的结果与测量结果差值ecur。

5.如果ecur>e,则u=2*u;否则u=u/2,并且更新模型系数x(k+1)=x(k)+deltaX。

6.判断算法是否收敛,不收敛返回2,否则结束。

代码如下:

 1 clear all;
 2 close all;
 3 clc;
 4 warning off all;
 5 
 6 a=7;b=7;c=7;              %待求解的系数
 7 
 8 x=(0:0.01:1)';
 9 w=rand(length(x),1)*2-1;   %生成噪声
10 y=exp(a*x.^2+b*x+c)+w;     %带噪声的模型 
11 plot(x,y,'.')
12 
13 pre=rand(3,1);             
14 update=1;
15 u=0.1;
16 for i=1:100    
17     if update==1
18         f = exp(pre(1)*x.^2+pre(2)*x+pre(3));
19         g = y-f;                                        %计算误差 
20 
21         p1 = exp(pre(1)*x.^2+pre(2)*x+pre(3)).*x.^2;    %对a求偏导
22         p2 = exp(pre(1)*x.^2+pre(2)*x+pre(3)).*x;       %对b求偏导
23         p3 = exp(pre(1)*x.^2+pre(2)*x+pre(3));          %对c求偏导
24         J = [p1 p2 p3];                                 %计算雅克比矩阵
25         H=J'*J;
26         if i==1
27             e=dot(g,g);
28         end          
29     end
30     
31     delta = inv(H+u*eye(length(H)))*J'* g;      
32     pcur = pre+delta;                           %迭代
33     fcur = exp(pcur(1)*x.^2+pcur(2)*x+pcur(3)); 
34     ecur = dot(y-fcur,y-fcur);
35     
36     if ecur<e                                   %比较两次差值,新模型好则使用
37         if norm(pre-pcur)<1e-10
38            break; 
39         end
40         u=u/2;  
41         pre=pcur;
42         e=ecur;
43         update=1;    
44     else
45         u=u*2;        
46         update=0;
47     end 
48 end
49 
50 hold on;
51 plot(x,exp(a*x.^2+b*x+c),'r');
52 plot(x,exp(pre(1)*x.^2+pre(2)*x+pre(3)),'g');
53 
54 %比较一下
55 [a b c]
56 pre'

迭代结果,其中散点为带噪声数据,红线为原始模型,绿线为解算模型

posted on 2020-09-10 14:41  一杯清酒邀明月  阅读(897)  评论(0编辑  收藏  举报