之前的几篇博客的一个共同点就是梯度下降法,梯度下降法是用来求解无约束最优化问题的一个数值方法,简单实用,几乎是大部分算法的基础,下面来利用梯度下降法优化BP神经网络。
[TOC]

梯度公式

下面的BP神经网络结构为最简单的三层网络,各层的神经元数量分别为B1,B2,B3。其中X,H,b2,O,b3均为行向量,W12,W23大小分别为(B1,B2)和(B2,B3)
这里写图片描述
BP神经网络的基本原理,通过输入X,经过非线性映射到输出O(样本大小为m),误差为:

J=i=1m12k=1B3(OkYik)2
<script type="math/tex; mode=display" id="MathJax-Element-1">J = \sum_{i=1}^m{\frac{1}{2}\sum_{k=1}^{B3}{(O_k-Y_{ik})^2}}</script>
显然,我们想要的是J越小越好。
根据上面的网络结构可得H、O的计算公式:
H=f(XW12+b2)
<script type="math/tex; mode=display" id="MathJax-Element-2">H = f(XW12+b2)</script>
f函数为:f(x)=1(1+ex)<script type="math/tex" id="MathJax-Element-3">f(x) = \frac{1}{(1+e^{-x})}</script>,f函数导数为:f1=f(1f)<script type="math/tex" id="MathJax-Element-4">f1= f(1-f)</script>
O=HW23+b3
<script type="math/tex; mode=display" id="MathJax-Element-5">O = HW23+b3</script>

下面采用梯度下降法求解J的最小值时对应的网络的权阈值:

Jb3l=i=1mk=1B3(OkYik)Okb3l.........l=1,2,...B3=i=1mk=1B3(OkYik)((HW23)k+b3k)b3l=i=1mk=1B3(OkYik)b3kb3l=i=1m(OlYil).........l=1,2,...B3
<script type="math/tex; mode=display" id="MathJax-Element-6"> \begin{align} \frac{\partial{J}}{\partial{b3_{l}}} &= \sum_{i=1}^m{\sum_{k=1}^{B3}{(O_k-Y_{ik})\frac{\partial{O_k}}{\partial{b3_l}}}} .........l=1,2,...B3\\ &= \sum_{i=1}^m{\sum_{k=1}^{B3}{(O_k-Y_{ik})\frac{\partial{((HW23)_k+b3_k)}}{\partial{b3_l}}}}\\ &= \sum_{i=1}^m{\sum_{k=1}^{B3}{(O_k-Y_{ik})\frac{\partial{b3_k}}{\partial{b3_l}}}}\\ &= \sum_{i=1}^m{{(O_l-Y_{il})}} .........l=1,2,...B3\\ \end{align} </script>
如果数据集较小时,采用上述公式还可以,但是,当数据集特别大时,也就是m很大,那么梯度的计算将耗费大量时间,所以我们采用单样本误差来调整网络的权阈值。即,每使用一个样本就调整权阈值,那么J函数的形式更改如下:
J损失函数

J=12k=1B3(OkYik)2
<script type="math/tex; mode=display" id="MathJax-Element-7">J =\frac{1}{2}\sum_{k=1}^{B3}{(O_k-Y_{ik})^2} </script>
权阈值梯度公式

下面就新的J函数来推导梯度公式:

Jb3l=OlYl.........l=1,2,...,B3
<script type="math/tex; mode=display" id="MathJax-Element-8">\frac{\partial{J}}{\partial{b3_l}} = O_l-Y_l.........l=1,2,...,B3</script>

J(b3)=Jb3=OY
<script type="math/tex; mode=display" id="MathJax-Element-9">\nabla{J(b3) = \frac{\partial{J}}{\partial{b3}}}=O-Y</script>
JW23pl=k=1B3(OkYk)OkW23pl.........p=1,2,...,B2;l=1,2,...,B3=k=1B3(OkYk)(HW23W23pl)k=k=1B3(OkYk)[0,...Hp,...0]k......Hpl=(OlYl)Hp.........p=1,2,...,B2;l=1,2,...,B3
<script type="math/tex; mode=display" id="MathJax-Element-10"> \begin{align} \frac{\partial{J}}{\partial{W23_{pl}}} &=\sum_{k=1}^{B3}(O_k-Y_k)\frac{\partial{O_k}}{\partial{W23_{pl}}}.........p=1,2,...,B2;l=1,2,...,B3\\ &=\sum_{k=1}^{B3}(O_k-Y_k)(H\frac{\partial{W23}}{\partial{W23pl}})_k\\ &=\sum_{k=1}^{B3}(O_k-Y_k)[0,...H_p,...0]_k......H_p为第l列\\ &=(O_l-Y_l)H_p.........p=1,2,...,B2;l=1,2,...,B3\\ \end{align} </script>
即:
J(W23)=JW23=[HT,...,HT][(OY)T,...,(OY)T]T......H(1,B2);OY(1,B3)(B2,B3)(B2,B3)(B2,B3)
<script type="math/tex; mode=display" id="MathJax-Element-11"> \begin{align} \nabla{J(W23)}&=\frac{\partial{J}}{\partial{W23}}\\ &=[H^T,...,H^T]点乘[(O-Y)^T,...,(O-Y)^T]^T......H为(1,B2);O-Y为(1,B3);左边矩阵为(B2,B3),右边矩阵为(B2,B3),两矩阵点乘结果为(B2,B3)\\ \end{align} </script>
Jb2p=k=1B3(OkYk)Okb2p=k=1B3(OkYk)(HW23)kb2p=k=1B3(OkYk)HW23(:,k)b2p=k=1B3(OkYk)Hb2pW23(:,k)=k=1B3(OkYk){H(1H)b2b2p}W23(:,k)=k=1B3(OkYk){H(1H)[0,...,1,...,0]}W23(:,k)......1p=k=1B3(OkYk)Hp(1Hp)W23pk
<script type="math/tex; mode=display" id="MathJax-Element-12"> \begin{align} \frac{\partial{J}}{\partial{b2_p}}&=\sum_{k=1}^{B3}(O_k-Y_k)\frac{\partial{O_k}}{\partial{b2_p}}\\ &=\sum_{k=1}^{B3}(O_k-Y_k)\frac{\partial{(HW23)_k}}{\partial{b2_p}}\\ &=\sum_{k=1}^{B3}(O_k-Y_k)\frac{\partial{HW23(:,k)}}{\partial{b2_p}}\\ &=\sum_{k=1}^{B3}(O_k-Y_k)\frac{\partial{H}}{\partial{b2_p}}W23(:,k)\\ &=\sum_{k=1}^{B3}(O_k-Y_k)\{H点乘(1-H)点乘\frac{\partial{b2}}{\partial{b2_p}}\}W23(:,k)\\ &=\sum_{k=1}^{B3}(O_k-Y_k)\{H点乘(1-H)点乘[0,...,1,...,0]\}W23(:,k)......中间矩阵的1为第p列\\ &=\sum_{k=1}^{B3}(O_k-Y_k)Hp(1-Hp)W23_{pk}\\ \end{align} </script>
即,
J(b2)=Jb2=H(1H)((OY)W23T)
<script type="math/tex; mode=display" id="MathJax-Element-13"> \begin{align} \nabla{J(b2)}&=\frac{\partial{J}}{\partial{b2}}\\ &=H点乘(1-H)点乘((O-Y)W23^T)\\ \end{align} </script>
JW12op=k=1B3(OkYk)OkW12op.........o=1,2,...,B1;p=1,2,...,B2=k=1B3(OkYk){H(1H)XW12W12op}W23(:,k)=k=1B3(OkYk)[0,...,Hp(1Hp)Xo,...,0]W23(:,k)=k=1B3(OkYK)Hp(1Hp)XoW23pk=XoHp(1Hp)((OY)W23T)p
<script type="math/tex; mode=display" id="MathJax-Element-14"> \begin{align} \frac{\partial{J}}{\partial{W12_{op}}}&=\sum_{k=1}^{B3}(Ok-Yk)\frac{\partial{Ok}}{\partial{W12_{op}}}.........o=1,2,...,B1;p=1,2,...,B2\\ &=\sum_{k=1}^{B3}(Ok-Yk)\{H点乘(1-H)点乘\frac{\partial{XW12}}{\partial{W12_{op}}}\}W23(:,k)\\ &=\sum_{k=1}^{B3}(Ok-Yk)[0,...,H_p(1-H_p)X_o,...,0]W23(:,k)\\ &=\sum_{k=1}^{B3}(O_k-Y_K)H_p(1-H_p)X_oW23_{pk}\\ &=X_oH_p(1-H_p)((O-Y)W23^T)_p\\ \end{align} </script>
即,
J(W12)=JW12=[XT,...,XT][(H(1H)((OY)W23T))T,...,(H(1H)((OY)W23T))T]T......(B1,B2)(B1,B2),(B1,B2)
<script type="math/tex; mode=display" id="MathJax-Element-15"> \begin{align} \nabla{J(W12)}&=\frac{\partial{J}}{\partial{W12}}\\ &=[X^T,...,X^T]点乘[(H点乘(1-H)点乘((O-Y)W23^T))^T,...,(H点乘(1-H)点乘((O-Y)W23^T))^T]^T......左边矩阵为(B1,B2)点乘右边矩阵(B1,B2),结果为(B1,B2)\\ \end{align} </script>

代码实现

下面是matlab的具体实现

准备数据
%% 三层神经网络算法的matlab实现
clear,clc,close all
% 构造样例数据
x = linspace(-10,10,2000)';
y = sin(x);
% 训练测试集分割
a = rand(length(x),1);
[m,n] = sort(a);
x_train = x(n(1:floor(0.7*length(a))));
x_test = x(n(floor(0.7*length(a))+1:end));
y_train = y(n(1:floor(0.7*length(a))));
y_test = y(n(floor(0.7*length(a)+1):end));
% 数据归一化
[x_train_regular,x_train_maxmin] = mapminmax(x_train');
x_train_regular = x_train_regular';
x_test_regular = mapminmax('apply',x_test',x_train_maxmin);
x_test_regular = x_test_regular';
基于梯度下降法的训练函数
function model = BP_train( net_structure,x,y )
[sample_size,n] = size(x);
B1 = n;
B2 = net_structure.hiden_num;
[~,n] = size(y);
B3 = n;
maxgen = net_structure.maxgen;
% 初始化权重和阈值
W12 = rands(B1,B2);
b2 = rands(1,B2);
W23 = rands(B2,B3);
b3 = rands(1,B3);
E = [];
for i = 1:1:maxgen
    e = 0;
    for j = 1:1:sample_size
        alpha = 0.5*rand;
%         alpha = 1/i+0.1;
        H = x(j,:)*W12+b2;
        H = 1./(1+exp(-H));
        O = H*W23+b3;
        delta_W12 = mat_seq(x(j,:)',B2,'h').*mat_seq(H.*(1-H),B1,'v').*mat_seq((O-y(j,:))*W23',B1,'v');
        delta_b2 = H.*(1-H).*((O-y(j,:))*W23');
        delta_W23 = mat_seq(H',B3,'h').*mat_seq(O-y(j,:),B2,'v');
        delta_b3 = O-y(j,:);
        % 更新权阈值
        W12 = W12-alpha*delta_W12;
        b2 = b2-alpha*delta_b2;
        W23 = W23-alpha*delta_W23;
        b3 = b3-alpha*delta_b3;
        e = e+sum((O-y(j,:)).^2);
    end
    E = [E,e];
    disp(['迭代次数:',num2str(i)])
end
model = struct('W12',W12,'b2',b2,'W23',W23,'b3',b3,'E',E);
end

% 矩阵复制成序列
function out_mat = mat_seq(mat,num,axis)
mat0 = mat;
if axis == 'h' % 表示横向复制矩阵
    for i = 1:1:(num-1)
        mat0 = [mat0,mat];
    end
else
    for i = 1:1:(num-1)
        mat0 = [mat0;mat];
    end
end
out_mat = mat0;
end
运行结果

这里写图片描述
这里写图片描述
这里写图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐