损失函数的选择

回归问题常用:均值平方差损失函数(MSE);交叉熵问题常用交叉熵损失函数(CE)。

那为啥?

1.从原理上来讲,CE就是分类问题的极大似然估计,并不适用于连续变量的回归问题。

2.对于分类问题,若采用逻辑回归算法,MSE一般是非凸函数,不能取得全局最优解,且MSE容易发生梯度消失;而CE是凸函数,能取得全局最优解。所以分类问题一般用CE。

MSE
CE
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容