1、背景
样本类别不均衡是机器学习实践中经常会遇到的问题,很多时候我们只有少量的正样本和非常多的负样本,当然这种情况在多分类问题中也会经常遇到。
这种现象会对机器学习模型的性能带来很大影响。因为负样本很多,负样本在总的损失函数中的占比很大,使得模型会专注与学习负样本相关的信息来减小损失,对于正样本的关注度不足,导致最终模型对正样本以及与正样本相似的负样本的分类效果较差。
2、数据重采样
为了解决上述数据不均衡的问题,一个被广泛采用的技术叫做数据重采样。它包含从多数类别的样本中删除个别样本(下采样)以及往少数类别的样本中添加更多样本(过采样)。
(1)上采样
通过采样对样本进行扩充,简单的重复样本不是和好方法,最好使用数据增强的方式生成相对自然又有新信息的样本。
缺点: 实现过采样一种简单的方式是重复复制数量较少的类别的样本,这有可能会导致模型过拟合。
(2)下采样
通过随机采样、分层采样、加权采样等方式对样本进行筛选,保留比较有代表性的样本,去掉大量重复的相似的样本。
缺点: 实现下采样最简单的方式是删除数量较多的某类样本,这可能会导致训练过程信息丢失。
3、权重法
样本数量多类别,样本权重小一些;样本数量少的类别,样本权重大一些。
4、不均衡数据集采样器:PyTorch采样器ImbalancedDatasetSampler
特点:
- 能够从不平衡的数据集采样后重新平衡类间分布
- 能够自动估计采样时的权值
- 避免创建新的平衡数据集
- 当它与数据增强技术一起使用时,可以减轻过拟合
5、SMOTE方法
SMOTE算法的基本思想就是对少数类别样本进行分析和模拟,并将人工模拟的新样本添加到数据集中,进而使原始数据中的类别不再严重失衡。是基于随机过采样算法的一种改进方案。
该算法的模拟过程采用了KNN技术,模拟生成新样本的步骤如下:
- 采样最邻近算法,计算出每个少数类样本的K个近邻;
- 从K个近邻中随机挑选N个样本进行随机线性插值;
- 构造新的少数类样本;
- 将新样本与原数据合成,产生新的训练集;
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split
from imblearn.over_sampling import SMOTE
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import confusion_matrix
from sklearn.model_selection import train_test_split
# 读取数据集并处理
data = pd.read_csv("E:/file/creditcard.csv")
data<

文章讨论了机器学习中常见的样本不均衡问题,及其对模型性能的影响。提出了数据重采样策略,包括上采样和下采样,以及SMOTE算法来生成新样本。此外,还介绍了PyTorch的ImbalancedDatasetSampler和权重法来调整不同类别样本的权重。文章通过示例展示了如何使用SMOTE平衡信用卡欺诈检测数据集,并进行了模型训练。
最低0.47元/天 解锁文章
3万+

被折叠的 条评论
为什么被折叠?



