每天给你送来NLP技术干货!
作者 | godweiyang
出品 | 公众号:算法码上来(ID:GodNLP)
- BEGIN -
交叉熵(CrossEntropy)是常见的损失函数,本文详细推导一下它的梯度,面试大厂或者工程实践中都可能会用到。
前向传播
假设分类任务类别数是,隐层输出是维向量,标准的one-hot向量是,正确的类别是。那么交叉熵损失可以定义为:
其中,,是平滑参数。Softmax函数大家都很熟悉了,具体形式为:。
反向传播
对的梯度要分两种情况:
402 Payment Required
推导过程
根据求导法则有:
402 Payment Required
其中就是Softmax函数的梯度(这个推导比较简单,放在了文末):
下面分两种情况讨论:
当时:
402 Payment Required
当时:
402 Payment Required
Softmax梯度
回顾Softmax函数的形式:
这里也分两种情况讨论:
当时:
当时:
- END -
最近文章
EMNLP 2022 和 COLING 2022,投哪个会议比较好?