使用 categorical_crossentropy 进行二元分类

数据挖掘 Python 深度学习
2022-03-06 19:56:18

可以使用 categorical_crossentropy 进行二元分类还是使用 binary_crossentropy 更好

1个回答

二元交叉熵是分类交叉熵的一个特例,只有 2 个类。所以理论上它并没有什么区别。

如果yk是真正的标签并且y^k是类的预测标签k(都是one-hot编码,即kyk=1kyk^=1yk,yk^{0,1}) 的多类别交叉熵K类别是

k=1Kyklog(y^k)

等于二元交叉熵K=2

k=1Kyklog(y^k)=ylog(y^)(1y)log(1y^)

以防万一有任何实现差异,例如速度方面,我仍然会使用二进制交叉熵来解决二进制分类问题,而不是使用多分类问题K=2.