【机器学习】如何处理数据不均衡问题

2018-07-25 11:36:42 浏览数 (1)

阅读大概需要2分钟 跟随小博主,每天进步一丢丢

最近在打比赛期间,遇到了数据不均衡问题。于是网上查了查资料,便做了以下总结。嘿嘿

什么是数据不均衡问题

举个小栗子:

如果100个人中喜欢吃梨的有90人,喜欢吃苹果的有10人。

那么如果你随便在大街上找一个人,预测他是喜欢吃梨还是苹果?肯定的,傻子都知道猜他喜欢吃梨呀。因为这样基本不用学习什么知识,就能将准确率控制在90%左右。

用图表示就为:

真实值

预测值

像这种问题,傻子都会,还要机器干什么呢?所以,我们要找到真的学习并预测梨和苹果,且不受数据不均衡影响的分类器。

解决办法

1.想办法获取更多的数据

获取更多的数据,从数据来源查找更多的数据,来弥补数据的短缺。

2.换个评测方式

以前都用准确率accuracy,也就是预测对的个数 / 总个数。但是这个评测方式在高的准确率和低的误差并没有那么有说服力。那么怎么办呢?这个时候就该想起来准确率Precision和召回率Recall,从而求得的F1值。P,R,F1值是对所有类别的测评值。如果想了解更多,我之前写了该方法的理论与实践。

详谈P(查准率),R(查全率),F1值

3.重组数据

这个方法就简单粗暴了。比如原始数据为:

方式一

将少的蓝色数据复制多份,使得数量和多的数据相当即可。

蓝色数据复制多份

注:这样做的缺点是可能会过拟合。

方式二

砍掉一些橙色部分,使得和蓝色基本持平。

注:这样做的缺点一般使得数据预测的不那么准确,因为数据丢失。

个人认为,建议优先选择方式一。

4.换机器学习算法

神经网络,对于数据不均衡问题真的是束手无策。可以换其他的机器学习方法,比如决策树之类,不受该问题影响。

5.修改算法

这个就腻害了,直接将算法改了,来改变数据不均衡带来的影响。

比如你使用的是sigmoid算法:

x=0为分界线,当x小于0时为梨,大于0的时候为苹果。但是梨比较多,那么可以将阈值x=0向右平移:

使得绝大多数预测的结果为梨,只有一些特别情况,极其确定的时候预测为苹果。

补充

对于上述的所有方法,我做一个补充。

1.选择复制小数据方法的时候,没必要非得要和大数据1:1才好。可以比例慢慢靠近,说不定期间就有一个比值就很好呢。毕竟实践才是检验真理的唯一标准。

2.砍掉大数据的方法尽量不要用,很可能会影响分类器的效果。

参考资料 morvanzhou https://www.bilibili.com/video/av16009140?from=search&seid=1236544046869302415

0 人点赞