Coursera吴恩达课程笔记 2.3《优化深度神经网络》-- 超参数调试、Batch正则化和编程框架

原文链接 https://blog.csdn.net/red_stone1/article/details/78403416

上节课我们主要介绍了深度神经网络的优化算法。包括对原始数据集进行分割,使用mini-batch gradient descent。然后介绍了指数加权平均(Exponentially weighted averages)的概念以及偏移校正(bias correction)方法。接着,我们着重介绍了三种常用的加速神经网络学习速度的三种算法:动量梯度下降、RMSprop和Adam算法。其中,Adam结合了动量梯度下降和RMSprop各自的优点,实际应用中表现更好。然后,我们介绍了另外一种提高学习速度的方法:learning rate decay,通过不断减小学习因子,减小步进长度,来减小梯度振荡。最后,我们对深度学习中local optima的概念作了更深入的解释。本节课,我们将重点介绍三个方面的内容:超参数调试、Batch正则化和深度学习编程框架。

1. Tuning Process

深度神经网络需要调试的超参数(Hyperparameters)较多,包括:

  • α \alpha α:学习因子

  • β \beta β:动量梯度下降因子

  • β 1 , β 2 , ε \beta_1,\beta_2,\varepsilon β1,β2,ε:Adam算法参数

  • #layers:神经网络层数

  • #hidden units:各隐藏层神经元个数

  • learning rate decay:学习因子下降参数

  • mini-batch size:批量训练样本包含的样本个数

超参数之间也有重要性差异。通常来说,学习因子 α \alpha α是最重要的超参数,也是需要重点调试的超参数。动量梯度下降因子 β \beta β、各隐藏层神经元个数#hidden units和mini-batch size的重要性仅次于 α \alpha α。然后就是神经网络层数#layers和学习因子下降参数learning rate decay。最后,Adam算法的三个参数 β 1 , β 2 , ε \beta_1,\beta_2,\varepsilon β1,β2,ε一般常设置为0.9,0.999和 1 0 − 8 10^{-8} 108,不需要反复调试。当然,这里超参数重要性的排名并不是绝对的,具体情况,具体分析。

如何选择和调试超参数?传统的机器学习中,我们对每个参数等距离选取任意个数的点,然后,分别使用不同点对应的参数组合进行训练,最后根据验证集上的表现好坏,来选定最佳的参数。例如有两个待调试的参数,分别在每个参数上选取5个点,这样构成了5x5=25中参数组合,如下图所示:

这里写图片描述

这种做法在参数比较少的时候效果较好。但是在深度神经网络模型中,我们一般不采用这种均匀间隔取点的方法,比较好的做法是使用随机选择。也就是说,对于上面这个例子,我们随机选择25个点,作为待调试的超参数,如下图所示:

这里写图片描述

随机化选择参数的目的是为了尽可能地得到更多种参数组合。还是上面的例子,如果使用均匀采样的话,每个参数只有5种情况;而使用随机采样的话,每个参数有25种可能的情况,因此更有可能得到最佳的参数组合。

这种做法带来的另外一个好处就是对重要性不同的参数之间的选择效果更好。假设hyperparameter1为 α \alpha α,hyperparameter2为 ε \varepsilon ε,显然二者的重要性是不一样的。如果使用第一种均匀采样的方法, ε \varepsilon ε的影响很小,相当于只选择了5个 α \alpha α值。而如果使用第二种随机采样的方法, ε \varepsilon ε α \alpha α都有可能选择25种不同值。这大大增加了 α \alpha α调试的个数,更有可能选择到最优值。其实,在实际应用中完全不知道哪个参数更加重要的情况下,随机采样的方式能有效解决这一问题,但是均匀采样做不到这点。

在经过随机采样之后,我们可能得到某些区域模型的表现较好。然而,为了得到更精确的最佳参数,我们应该继续对选定的区域进行由粗到细的采样(coarse to fine sampling scheme)。也就是放大表现较好的区域,再对此区域做更密集的随机采样。例如,对下图中右下角的方形区域再做25点的随机采样,以获得最佳参数。

这里写图片描述

2. Using an appropriate scale to pick hyperparameters

上一部分讲的调试参数使用随机采样,对于某些超参数是可以进行尺度均匀采样的,但是某些超参数需要选择不同的合适尺度进行随机采样。

什么意思呢?例如对于超参数#layers和#hidden units,都是正整数,是可以进行均匀随机采样的,即超参数每次变化的尺度都是一致的(如每次变化为1,犹如一个刻度尺一样,刻度是均匀的)。

但是,对于某些超参数,可能需要非均匀随机采样(即非均匀刻度尺)。例如超参数 α \alpha α,待调范围是[0.0001, 1]。如果使用均匀随机采样,那么有90%的采样点分布在[0.1, 1]之间,只有10%分布在[0.0001, 0.1]之间。这在实际应用中是不太好的,因为最佳的 α \alpha α值可能主要分布在[0.0001, 0.1]之间,而[0.1, 1]范围内 α \alpha α值效果并不好。因此我们更关注的是区间[0.0001, 0.1],应该在这个区间内细分更多刻度。

通常的做法是将linear scale转换为log scale,将均匀尺度转化为非均匀尺度,然后再在log scale下进行均匀采样。这样,[0.0001, 0.001],[0.001, 0.01],[0.01, 0.1],[0.1, 1]各个区间内随机采样的超参数个数基本一致,也就扩大了之前[0.0001, 0.1]区间内采样值个数。

这里写图片描述

一般解法是,如果线性区间为[a, b],令m=log(a),n=log(b),则对应的log区间为[m,n]。对log区间的[m,n]进行随机均匀采样,然后得到的采样值r,最后反推到线性区间,即 1 0 r 10^r 10r 1 0 r 10^r 10r就是最终采样的超参数。相应的Python语句为:

m = np.log10(a)
n = np.log10(b)
r = np.random.rand()
r = m + (n-m)*r
r = np.power(10, r)

除了 α \alpha α之外,动量梯度因子 β \beta β也是一样,在超参数调试的时候也需要进行非均匀采样。一般 β \beta β的取值范围在[0.9, 0.999]之间,那么 1 − β 1-\beta 1β的取值范围就在[0.001, 0.1]之间。那么直接对 1 − β 1-\beta 1β在[0.001, 0.1]区间内进行log变换即可。

这里解释下为什么 β \beta β也需要向 α \alpha α那样做非均匀采样。假设 β \beta β从0.9000变化为0.9005,那么 1 1 − β \frac{1}{1-\beta} 1β1基本没有变化。但假设 β \beta β从0.9990变化为0.9995,那么 1 1 − β \frac{1}{1-\beta} 1β1前后差别1000。 β \beta β越接近1,指数加权平均的个数越多,变化越大。所以对 β \beta β接近1的区间,应该采集得更密集一些。

3. Hyperparameters tuning in practice: Pandas vs. Caviar

经过调试选择完最佳的超参数并不是一成不变的,一段时间之后(例如一个月),需要根据新的数据和实际情况,再次调试超参数,以获得实时的最佳模型。

在训练深度神经网络时,一种情况是受计算能力所限,我们只能对一个模型进行训练,调试不同的超参数,使得这个模型有最佳的表现。我们称之为Babysitting one model。另外一种情况是可以对多个模型同时进行训练,每个模型上调试不同的超参数,根据表现情况,选择最佳的模型。我们称之为Training many models in parallel。

这里写图片描述

因为第一种情况只使用一个模型,所以类比做Panda approach;第二种情况同时训练多个模型,类比做Caviar approach。使用哪种模型是由计算资源、计算能力所决定的。一般来说,对于非常复杂或者数据量很大的模型,使用Panda approach更多一些。

4. Normalizing activations in a network

Sergey Ioffe和Christian Szegedy两位学者提出了Batch Normalization方法。Batch Normalization不仅可以让调试超参数更加简单,而且可以让神经网络模型更加“健壮”。也就是说较好模型可接受的超参数范围更大一些,包容性更强,使得更容易去训练一个深度神经网络。接下来,我们就来介绍什么是Batch Normalization,以及它是如何工作的。

之前,我们在Coursera吴恩达课堂笔记 2.1《优化深度神经网络》– 深度学习的实用层面中提到过在训练神经网络时,标准化输入可以提高训练的速度。方法是对训练数据集进行归一化的操作,即将原始数据减去其均值 μ \mu μ后,再除以其方差 σ 2 \sigma^2 σ2。但是标准化输入只是对输入进行了处理,那么对于神经网络,又该如何对各隐藏层的输入进行标准化处理呢?

其实在神经网络中,第 l l l层隐藏层的输入就是第 l − 1 l-1 l1层隐藏层的输出 A [ l − 1 ] A^{[l-1]} A[l1]。对 A [ l − 1 ] A^{[l-1]} A[l1]进行标准化处理,从原理上来说可以提高 W [ l ] W^{[l]} W[l] b [ l ] b^{[l]} b[l]的训练速度和准确度。这种对各隐藏层的标准化处理就是Batch Normalization。值得注意的是,实际应用中,一般是对 Z [ l − 1 ] Z^{[l-1]} Z[l1]进行标准化处理而不是 A [ l − 1 ] A^{[l-1]} A[l1],其实差别不是很大。

Batch Normalization对第 l l l层隐藏层的输入 Z [ l − 1 ] Z^{[l-1]} Z[l1]做如下标准化处理,忽略上标 [ l − 1 ] [l-1] [l1]

μ = 1 m ∑ i z ( i ) \mu=\frac1m\sum_iz^{(i)} μ=m1iz(i)

σ 2 = 1 m ∑ i ( z i − μ ) 2 \sigma^2=\frac1m\sum_i(z_i-\mu)^2 σ2=m1i(ziμ)2

z n o r m ( i ) = z ( i ) − μ σ 2 + ε z^{(i)}_{norm}=\frac{z^{(i)}-\mu}{\sqrt{\sigma^2+\varepsilon}} znorm(i)=σ2+ε z(i)μ

其中,m是单个mini-batch包含样本个数, ε \varepsilon ε是为了防止分母为零,可取值 1 0 − 8 10^{-8} 108。这样,使得该隐藏层的所有输入 z ( i ) z^{(i)} z(i)均值为0,方差为1。

但是,大部分情况下并不希望所有的 z ( i ) z^{(i)} z(i)均值都为0,方差都为1,也不太合理。通常需要对 z ( i ) z^{(i)} z(i)进行进一步处理:

z ~ ( i ) = γ ⋅ z n o r m ( i ) + β \tilde z^{(i)}=\gamma\cdot z^{(i)}_{norm}+\beta z~(i)=γznorm(i)+β

上式中, γ \gamma γ β \beta β是learnable parameters,类似于W和b一样,可以通过梯度下降等算法求得。这里, γ \gamma γ β \beta β的作用是让 z ~ ( i ) \tilde z^{(i)} z~(i)的均值和方差为任意值,只需调整其值就可以了。例如,令:

γ = σ 2 + ε ,    β = u \gamma=\sqrt{\sigma^2+\varepsilon},\ \ \beta=u γ=σ2+ε ,  β=u

z ~ ( i ) = z ( i ) \tilde z^{(i)}=z^{(i)} z~(i)=z(i),即identity function。可见,设置 γ \gamma γ β \beta β为不同的值,可以得到任意的均值和方差。

这样,通过Batch Normalization,对隐藏层的各个 z [ l ] ( i ) z^{[l](i)} z[l](i)进行标准化处理,得到 z ~ [ l ] ( i ) \tilde z^{[l](i)} z~[l](i),替代 z [ l ] ( i ) z^{[l](i)} z[l](i)

值得注意的是,输入的标准化处理Normalizing inputs和隐藏层的标准化处理Batch Normalization是有区别的。Normalizing inputs使所有输入的均值为0,方差为1。而Batch Normalization可使各隐藏层输入的均值和方差为任意值。实际上,从激活函数的角度来说,如果各隐藏层的输入均值在靠近0的区域即处于激活函数的线性区域,这样不利于训练好的非线性神经网络,得到的模型效果也不会太好。这也解释了为什么需要用 γ \gamma γ β \beta β来对 z [ l ] ( i ) z^{[l](i)} z[l](i)作进一步处理。

5. Fitting Batch Norm into a neural network

我们已经知道了如何对某单一隐藏层的所有神经元进行Batch Norm,接下来将研究如何把Bath Norm应用到整个神经网络中。

对于L层神经网络,经过Batch Norm的作用,整体流程如下:

这里写图片描述

实际上,Batch Norm经常使用在mini-batch上,这也是其名称的由来。

值得注意的是,因为Batch Norm对各隐藏层 Z [ l ] = W [ l ] A [ l − 1 ] + b [ l ] Z^{[l]}=W^{[l]}A^{[l-1]}+b^{[l]} Z[l]=W[l]A[l1]+b[l]有去均值的操作,所以这里的常数项 b [ l ] b^{[l]} b[l]可以消去,其数值效果完全可以由 Z ~ [ l ] \tilde Z^{[l]} Z~[l]中的 β \beta β来实现。因此,我们在使用Batch Norm的时候,可以忽略各隐藏层的常数项 b [ l ] b^{[l]} b[l]。在使用梯度下降算法时,分别对 W [ l ] W^{[l]} W[l] β [ l ] \beta^{[l]} β[l] γ [ l ] \gamma^{[l]} γ[l]进行迭代更新。除了传统的梯度下降算法之外,还可以使用我们之前介绍过的动量梯度下降、RMSprop或者Adam等优化算法。$

6. Why does Batch Norm work?

我们可以把输入特征做均值为0,方差为1的规范化处理,来加快学习速度。而Batch Norm也是对隐藏层各神经元的输入做类似的规范化处理。总的来说,Batch Norm不仅能够提高神经网络训练速度,而且能让神经网络的权重W的更新更加“稳健”,尤其在深层神经网络中更加明显。比如神经网络很后面的W对前面的W包容性更强,即前面的W的变化对后面W造成的影响很小,整体网络更加健壮。

举个例子来说明,假如用一个浅层神经网络(类似逻辑回归)来训练识别猫的模型。如下图所示,提供的所有猫的训练样本都是黑猫。然后,用这个训练得到的模型来对各种颜色的猫样本进行测试,测试的结果可能并不好。其原因是训练样本不具有一般性(即不是所有的猫都是黑猫),这种训练样本(黑猫)和测试样本(猫)分布的变化称之为covariate shift。

这里写图片描述

对于这种情况,如果实际应用的样本与训练样本分布不同,即发生了covariate shift,则一般是要对模型重新进行训练的。在神经网络,尤其是深度神经网络中,covariate shift会导致模型预测效果变差,重新训练的模型各隐藏层的 W [ l ] W^{[l]} W[l] B [ l ] B^{[l]} B[l]均产生偏移、变化。而Batch Norm的作用恰恰是减小covariate shift的影响,让模型变得更加健壮,鲁棒性更强。Batch Norm减少了各层 W [ l ] W^{[l]} W[l] B [ l ] B^{[l]} B[l]之间的耦合性,让各层更加独立,实现自我训练学习的效果。也就是说,如果输入发生covariate shift,那么因为Batch Norm的作用,对个隐藏层输出 Z [ l ] Z^{[l]} Z[l]进行均值和方差的归一化处理, W [ l ] W^{[l]} W[l] B [ l ] B^{[l]} B[l]更加稳定,使得原来的模型也有不错的表现。针对上面这个黑猫的例子,如果我们使用深层神经网络,使用Batch Norm,那么该模型对花猫的识别能力应该也是不错的。

从另一个方面来说,Batch Norm也起到轻微的正则化(regularization)效果。具体表现在:

  • 每个mini-batch都进行均值为0,方差为1的归一化操作

  • 每个mini-batch中,对各个隐藏层的 Z [ l ] Z^{[l]} Z[l]添加了随机噪声,效果类似于Dropout

  • mini-batch越小,正则化效果越明显

但是,Batch Norm的正则化效果比较微弱,正则化也不是Batch Norm的主要功能。

7. Batch Norm at test time

训练过程中,Batch Norm是对单个mini-batch进行操作的,但在测试过程中,如果是单个样本,该如何使用Batch Norm进行处理呢?

首先,回顾一下训练过程中Batch Norm的主要过程:

μ = 1 m ∑ i z ( i ) \mu=\frac1m\sum_iz^{(i)} μ=m1iz(i)

σ 2 = 1 m ∑ i ( z ( i ) − μ ) 2 \sigma^2=\frac1m\sum_i(z^{(i)}-\mu)^2 σ2=m1i(z(i)μ)2

z n o r m ( i ) = z ( i ) − μ σ 2 + ε z_{norm}^{(i)}=\frac{z^{(i)}-\mu}{\sqrt{\sigma^2+\varepsilon}} znorm(i)=σ2+ε z(i)μ

z ~ ( i ) = γ ⋅ z n o r m ( i ) + β \tilde z^{(i)}=\gamma\cdot z^{(i)}_{norm}+\beta z~(i)=γznorm(i)+β

其中, μ \mu μ σ 2 \sigma^2 σ2是对单个mini-batch中所有m个样本求得的。在测试过程中,如果只有一个样本,求其均值和方差是没有意义的,就需要对 μ \mu μ σ 2 \sigma^2 σ2进行估计。估计的方法有很多,理论上我们可以将所有训练集放入最终的神经网络模型中,然后将每个隐藏层计算得到的 μ [ l ] \mu^{[l]} μ[l] σ 2 [ l ] \sigma^{2[l]} σ2[l]直接作为测试过程的 μ \mu μ σ 2 \sigma^2 σ2来使用。但是,实际应用中一般不使用这种方法,而是使用我们之前介绍过的指数加权平均(exponentially weighted average)的方法来预测测试过程单个样本的 μ \mu μ σ 2 \sigma^2 σ2

指数加权平均的做法很简单,对于第 l l l层隐藏层,考虑所有mini-batch在该隐藏层下的 μ [ l ] \mu^{[l]} μ[l] σ 2 [ l ] \sigma^{2[l]} σ2[l],然后用指数加权平均的方式来预测得到当前单个样本的 μ [ l ] \mu^{[l]} μ[l] σ 2 [ l ] \sigma^{2[l]} σ2[l]。这样就实现了对测试过程单个样本的均值和方差估计。最后,再利用训练过程得到的 γ \gamma γ β \beta β值计算出各层的 z ~ ( i ) \tilde z^{(i)} z~(i)值。

8. Softmax Regression

目前我们介绍的都是二分类问题,神经网络输出层只有一个神经元,表示预测输出 y ^ \hat y y^是正类的概率 P ( y = 1 ∣ x ) P(y=1|x) P(y=1x) y ^ > 0.5 \hat y>0.5 y^>0.5则判断为正类, y ^ < 0.5 \hat y<0.5 y^<0.5则判断为负类。

对于多分类问题,用C表示种类个数,神经网络中输出层就有C个神经元,即 n [ L ] = C n^{[L]}=C n[L]=C。其中,每个神经元的输出依次对应属于该类的概率,即 P ( y = c ∣ x ) P(y=c|x) P(y=cx)。为了处理多分类问题,我们一般使用Softmax回归模型。Softmax回归模型输出层的激活函数如下所示:

z [ L ] = W [ L ] a [ L − 1 ] + b [ L ] z^{[L]}=W^{[L]}a^{[L-1]}+b^{[L]} z[L]=W[L]a[L1]+b[L]

a i [ L ] = e z i [ L ] ∑ i = 1 C e z i [ L ] a^{[L]}_i=\frac{e^{z^{[L]}_i}}{\sum_{i=1}^Ce^{z^{[L]}_i}} ai[L]=i=1Cezi[L]ezi[L]

输出层每个神经元的输出 a i [ L ] a^{[L]}_i ai[L]对应属于该类的概率,满足:

∑ i = 1 C a i [ L ] = 1 \sum_{i=1}^Ca^{[L]}_i=1 i=1Cai[L]=1

所有的 a i [ L ] a^{[L]}_i ai[L],即 y ^ \hat y y^,维度为(C, 1)。

下面给出几个简单的线性多分类的例子:

这里写图片描述

如果使用神经网络,特别是深层神经网络,可以得到更复杂、更精确的非线性模型。

9. Training a softmax classifier

Softmax classifier的训练过程与我们之前介绍的二元分类问题有所不同。先来看一下softmax classifier的loss function。举例来说,假如C=4,某个样本的预测输出 y ^ \hat y y^和真实输出 y y y为:

y ^ = [ 0.3 0.2 0.1 0.4 ] \hat y=\left[ \begin{matrix} 0.3 \\ 0.2 \\ 0.1 \\ 0.4 \end{matrix} \right] y^=0.30.20.10.4

y = [ 0 1 0 0 ] y=\left[ \begin{matrix} 0 \\ 1 \\ 0 \\ 0 \end{matrix} \right] y=0100

y ^ \hat y y^值来看, P ( y = 4 ∣ x ) = 0.4 P(y=4|x)=0.4 P(y=4x)=0.4,概率最大,而真实样本属于第2类,因此该预测效果不佳。我们定义softmax classifier的loss function为:

L ( y ^ , y ) = − ∑ j = 1 4 y j ⋅ l o g   y ^ j L(\hat y,y)=-\sum_{j=1}^4y_j\cdot log\ \hat y_j L(y^,y)=j=14yjlog y^j

然而,由于只有当 j = 2 j=2 j=2时, y 2 = 1 y_2=1 y2=1,其它情况下, y j = 0 y_j=0 yj=0。所以,上式中的 L ( y ^ , y ) L(\hat y,y) L(y^,y)可以简化为:

L ( y ^ , y ) = − y 2 ⋅ l o g   y ^ 2 = − l o g   y ^ 2 L(\hat y,y)=-y_2\cdot log\ \hat y_2=-log\ \hat y_2 L(y^,y)=y2log y^2=log y^2

要让 L ( y ^ , y ) L(\hat y,y) L(y^,y)更小,就应该让 y ^ 2 \hat y_2 y^2越大越好。 y ^ 2 \hat y_2 y^2反映的是概率,完全符合我们之前的定义。

所有m个样本的cost function为:

J = 1 m ∑ i = 1 m L ( y ^ , y ) J=\frac1m\sum_{i=1}^mL(\hat y,y) J=m1i=1mL(y^,y)

其预测输出向量 A [ L ] A^{[L]} A[L] Y ^ \hat Y Y^的维度为(4, m)。

softmax classifier的反向传播过程仍然使用梯度下降算法,其推导过程与二元分类有一点点不一样。因为只有输出层的激活函数不一样,我们先推导 d Z [ L ] dZ^{[L]} dZ[L]

d a [ L ] = − 1 a [ L ] da^{[L]}=-\frac{1}{a^{[L]}} da[L]=a[L]1

∂ a [ L ] ∂ z [ L ] = ∂ ∂ z [ L ] ⋅ ( e z i [ L ] ∑ i = 1 C e z i [ L ] ) = a [ L ] ⋅ ( 1 − a [ L ] ) \frac{\partial a^{[L]}}{\partial z^{[L]}}=\frac{\partial}{\partial z^{[L]}}\cdot (\frac{e^{z^{[L]}_i}}{\sum_{i=1}^Ce^{z^{[L]}_i}})=a^{[L]}\cdot (1-a^{[L]}) z[L]a[L]=z[L](i=1Cezi[L]ezi[L])=a[L](1a[L])

d z [ L ] = d a [ L ] ⋅ ∂ a [ L ] ∂ z [ L ] = a [ L ] − 1 = a [ L ] − y dz^{[L]}=da^{[L]}\cdot \frac{\partial a^{[L]}}{\partial z^{[L]}}=a^{[L]}-1=a^{[L]}-y dz[L]=da[L]z[L]a[L]=a[L]1=a[L]y

对于所有m个训练样本:

d Z [ L ] = A [ L ] − Y dZ^{[L]}=A^{[L]}-Y dZ[L]=A[L]Y

可见 d Z [ L ] dZ^{[L]} dZ[L]的表达式与二元分类结果是一致的,虽然推导过程不太一样。然后就可以继续进行反向传播过程的梯度下降算法了,推导过程与二元分类神经网络完全一致。

10. Deep learning frameworks

  • Caffe/Caffe2

  • CNTK

  • DL4J

  • Keras

  • Lasagne

  • mxnet

  • PaddlePaddle

  • TensorFlow

  • Theano

  • Torch

一般选择深度学习框架的基本准则是:

  • Ease of programming(development and deployment)

  • Running speed

  • Truly open(open source with good governance)

实际应用中,我们应该根据自己的需求选择最合适的深度学习框架。

11. TensorFlow

这里简单介绍一下最近几年比较火的一个深度学习框架:TensorFlow。

举个例子来说明,例如cost function是参数w的函数:

J = w 2 − 10 w + 25 J=w^2-10w+25 J=w210w+25

如果使用TensorFlow对cost function进行优化,求出最小值对应的w,程序如下:

import numpy as np
import tensorflow as tf

w = tf.Variable(0,dtype=tf.float32)
#cost = tf.add(tf.add(w**2,tf.multiply(-10,w)),25)
cost = w**2 - 10*w +25
train = tf.train.GradientDescentOptimizer(0.01).minimize(cost)

init = tf.global_variables_initializer()
session = tf.Session()
session.run(init)
print(session.run(w))

>>0.0

session.run(train)
print(session.run(w))

>>0.1

for i in range(1000):
    session.run(train)
print(session.run(w))

>>4.99999

TensorFlow框架内可以直接调用梯度下降优化算法,不需要我们自己再写程序了,大大提高了效率。在运行1000次梯度下降算法后,w的解为4.99999,已经非常接近w的最优值5了。

针对上面这个例子,如果对w前的系数用变量x来代替,程序如下:

import numpy as np
import tensorflow as tf

cofficients = np.array([[1.],[-10.],[25.]])

w = tf.Variable(0,dtype=tf.float32)
x = tf.placeholder(tf.float32,[3,1])
#cost = tf.add(tf.add(w**2,tf.multiply(-10,w)),25)
#cost = w**2 - 10*w +25
cost = x[0][0]*w**2 + x[1][0]*w + x[2][0]
train = tf.train.GradientDescentOptimizer(0.01).minimize(cost)

init = tf.global_variables_initializer()
session = tf.Session()
session.run(init)
print(session.run(w))

>>0.0

session.run(train, feed_dict=(x:coefficients))
print(session.run(w))

>>0.1

for i in range(1000):
    session.run(train, feed_dict=(x:coefficients))
print(session.run(w))

>>4.99999

结果跟之前是一样的。除此之外,我们还可以更改x即cofficients的值,而得到不同的优化结果w。

另外,上段程序中的:

session = tf.Session()
session.run(init)
print(session.run(w))

有另外一种写法:

with tf.Session() as session:
    session.run(init)
    print(session.run(w))

TensorFlow的最大优点就是采用数据流图(data flow graphs)来进行数值运算。图中的节点(Nodes)表示数学操作,图中的线(edges)则表示在节点间相互联系的多维数据数组,即张量(tensor)。而且它灵活的架构让你可以在多种平台上展开计算,例如台式计算机中的一个或多个CPU(或GPU),服务器,移动设备等等。

关于TensorFlow更多的原理和编程技巧这里就不在赘述了,感兴趣的朋友可以关注更详细的TensorFlow相关文档。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值