机器学习--神经网络算法系列--交叉熵(Cross_entropy)

本文对比了二次代价函数和交叉熵代价函数在人工神经网络训练中的表现,指出交叉熵能更好地加速训练过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

原文:http://blog.youkuaiyun.com/u014313009/article/details/51043064

        交叉熵代价函数(Cross-entropy cost function)是用来衡量人工神经网络(ANN)的预测值与实际值的一种方式。与二次代价函数相比,它能更有效地促进ANN的训练。在介绍交叉熵代价函数之前,本文先简要介绍二次代价函数,以及其存在的不足。


1. 二次代价函数的不足

        ANN的设计目的之一是为了使机器可以像人一样学习知识。人在学习分析新事物时,当发现自己犯的错误越大时,改正的力度就越大。比如投篮:当运动员发现自己的投篮方向离正确方向越远,那么他调整的投篮角度就应该越大,篮球就更容易投进篮筐。同理,我们希望:ANN在训练时,如果预测值与实际值的误差越大,那么在反向传播训练的过程中,各种参数调整的幅度就要更大,从而使训练更快收敛。然而,如果使用二次代价函数训练ANN,看到的实际效果是,如果误差越大,参数调整的幅度可能更小,训练更缓慢。

        以一个神经元的二类分类训练为例,进行两次实验(ANN常用的激活函数为sigmoid函数,该实验也采用该函数):输入一个相同的样本数据x=1.0(该样本对应的实际分类y=0);两次实验各自随机初始化参数,从而在各自的第一次前向传播后得到不同的输出值,形成不同的代价(误差):

实验1:第一次输出值为0.82

     

实验2:第一次输出值为0.98

        

        在实验1中,随机初始化参数,使得第一次输出值为0.82(该样本对应的实际值为0);经过300次迭代训练后,输出值由0.82降到0.09,逼近实际值。而在实验2中,第一次输出值为0.98,同样经过300迭代训练,输出值只降到了0.20。

        从两次实验的代价曲线中可以看出:实验1的代价随着训练次数增加而快速降低,但实验2的代价在一开始下降得非常缓慢;直观上看,初始的误差越大,收敛得越缓慢

        其实,误差大导致训练缓慢的原因在于使用了二次代价函数。二次代价函数的公式如下:



        其中,C表示代价,x表示样本,y表示实际值,a表示输出值,n表示样本的总数。为简单起见,同样一个样本为例进行说明,此时二次代价函数为:



        目前训练ANN最有效的算法是反向传播算法。简而言之,训练ANN就是通过反向传播代价,以减少代价为导向,调整参数。参数主要有:神经元之间的连接权重w,以及每个神经元本身的偏置b。调参的方式是采用梯度下降算法(Gradient descent),沿着梯度方向调整参数大小。w和b的梯度推导如下:



        其中,z表示神经元的输入,表示激活函数。从以上公式可以看出,w和b的梯度跟激活函数的梯度成正比,激活函数的梯度越大,w和b的大小调整得越快,训练收敛得就越快。而神经网络常用的激活函数为sigmoid函数,该函数的曲线如下所示:



        如图所示,实验2的初始输出值(0.98)对应的梯度明显小于实验1的输出值(0.82),因此实验2的参数梯度下降得比实验1慢。这就是初始的代价(误差)越大,导致训练越慢的原因。与我们的期望不符,即:不能像人一样,错误越大,改正的幅度越大,从而学习得越快。

        可能有人会说,那就选择一个梯度不变化或变化不明显的激活函数不就解决问题了吗?图样图森破,那样虽然简单粗暴地解决了这个问题,但可能会引起其他更多更麻烦的问题。而且,类似sigmoid这样的函数(比如tanh函数)有很多优点,非常适合用来做激活函数,具体请自行google之。



2. 交叉熵代价函数

        换个思路,我们不换激活函数,而是换掉二次代价函数,改用交叉熵代价函数:



        其中,x表示样本,n表示样本的总数。那么,重新计算参数w的梯度:



        其中(具体证明见附录):


        因此,w的梯度公式中原来的被消掉了;另外,该梯度公式中的表示输出值与实际值之间的误差。所以,当误差越大,梯度就越大,参数w调整得越快,训练速度也就越快。同理可得,b的梯度为:



        实际情况证明,交叉熵代价函数带来的训练效果往往比二次代价函数要好。



3. 交叉熵代价函数是如何产生的?

        以偏置b的梯度计算为例,推导出交叉熵代价函数:



        在第1小节中,由二次代价函数推导出来的b的梯度公式为:



        为了消掉该公式中的,我们想找到一个代价函数使得:



        即:



        对两侧求积分,可得:



        而这就是前面介绍的交叉熵代价函数。




附录:

        sigmoid函数为:


        可证:







### 关于交叉熵损失函数 #### 交叉熵损失函数定义 交叉熵损失函数是一种衡量实际输出与期望输出之间差异的方法,在机器学习领域特别是分类问题中广泛应用。该方法通过比较预测概率分布和真实标签的概率分布之间的距离来进行评估。 #### 公式表达 对于二分类问题中的单个样本而言,其交叉熵损失可以由下述公式给出: \[ L(y, \hat{y}) = -[ y \log(\hat{y}) + (1-y)\log(1-\hat{y}) ] \] 其中 \(y\) 表示真实的类别标签(0 或者 1),而 \(\hat{y}\) 则代表模型对该类别的预测概率[^2]。 当处理多分类情况时,则会采用更一般的软最大似然估计(Softmax)配合交叉熵的形式: \[ C=-\sum_{i=1}^{N}{t_i log(p_i)} \] 这里\(p_i\) 是经过 Softmax 转换后的第 i 类的预测概率;\(t_i\) 是对应的真实标签向量中的元素值,通常是一个 one-hot 编码形式的数据集的一部分[^3]。 #### 应用场景 - **逻辑回归** 和其他线性判别分析算法经常使用二元交叉熵作为目标函数。 - 在深度学习框架内,无论是卷积神经网络还是循环神经网络都可能涉及到 softmax 层加上交叉熵损失组合来完成最终决策任务。 - 对于自然语言处理(NLP),图像识别等领域来说,这种类型的损失函数有助于提高模型性能并加速收敛过程。 ```python import torch.nn as nn # 创建一个简单的二分类交叉熵损失实例 criterion = nn.BCELoss() output = torch.tensor([0.7], requires_grad=True) target = torch.tensor([1.]) loss = criterion(output, target) print(f'Binary Cross Entropy Loss: {loss.item()}') ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值