如何开发针对不平衡分类的成本敏感神经网络
深度学习神经网络是一类灵活的机器学习算法,可以在各种问题上表现良好。
神经网络使用误差反向传播算法进行训练,该算法涉及计算模型在训练数据集上产生的误差,并根据这些误差的比例更新模型权重。这种训练方法的局限性在于,每个类别的示例都被视为相同,对于不平衡的数据集,这意味着模型对一个类别的适应性要强得多,而对另一个类别的适应性则弱得多。
反向传播算法可以更新,以根据类别的重要性对错误分类误差进行加权,称为加权神经网络或成本敏感神经网络。这可以使模型在类别分布严重倾斜的数据集中更加关注少数类别的样本,而不是多数类别的样本。
在本教程中,您将发现用于不平衡分类的加权神经网络。
完成本教程后,您将了解:
- 标准神经网络算法为何不支持不平衡分类。
- 如何修改神经网络训练算法,以根据类别重要性对错误分类误差进行加权。
- 如何为神经网络配置类权重并评估对模型性能的影响。
教程概述
本教程分为四个部分:
- 不平衡分类数据集
- Keras 中的神经网络模型
- 深度学习解决不平衡分类问题
- 使用 Keras 的加权神经网络
不平衡分类数据集
在深入研究如何修改神经网络以解决不平衡分类问题之前,让我们首先定义一个不平衡分类数据集。
我们可以使用make_classification() 函数来定义一个合成的不平衡二分类数据集。我们将生成 10,000 个示例,少数类与多数类的比例约为 1:100。
...
# define dataset
X, y = make_classification(n_samples=10000, n_features=2, n_redundant=0,n_clusters_per_class=2, weights=[0.99], flip_y=0, random_state=4)
一旦生成,我们可以总结类分布以确认数据集是按照我们预期创建的。
...
# summarize class distribution
counter = Counter(y)
print(counter)
最后,我们可以创建示例的散点图并按类标签对其进行着色,以帮助理解从该数据集对示例进行分类的挑战。
...
# scatter plot of examples by class label
for label, _ in counter.items():row_ix = where(y == label)[0]pyplot.scatter(X[row_ix, 0], X[row_ix, 1], label=str(label))
pyplot.legend()
pyplot.show()
综合以上几点,生成合成数据集和绘制示例的完整示例如下所示。
# Generate and plot a synthetic imbalanced classification dataset
from collections import Counter
from sklearn.datasets import make_classification
from matplotlib import pyplot
from numpy import where
# define dataset
X, y = make_classification(n_samples=10000, n_features=2, n_redundant=0,n_clusters_per_class=2, weights=[0.99], flip_y=0, random_state=4)
# summarize class distribution
counter = Counter(y)
print(counter)
# scatter plot of examples by class label
for label, _ in counter.items():row_ix = where(y == label)[0]pyplot.scatter(X[row_ix, 0], X[row_ix, 1], label=str(label))
pyplot.legend()
pyplot.show()
首先运行示例会创建数据集并汇总类分布。
我们可以看到,数据集的类别分布近似于 1:100,其中多数类别有不到 10,000 个示例,少数类别有 100 个示例。
Counter({0: 9900, 1: 100})
接下来,创建数据集的散点图,显示多数类别的大量示例(蓝色)和少数类别的少量示例(橙色),其中有一些适度的类别重叠。
具有 1 到 100 类不平衡的二分类数据集的散点图
Keras 中的神经网络模型
接下来,我们可以在数据集上拟合标准的神经网络模型。
首先,我们可以定义一个函数来创建合成数据集,并将其分成单独的训练数据集和测试数据集,每个数据集有 5,000 个示例。
# prepare train and test dataset
def prepare_data():# generate 2d classification datasetX, y = make_classification(n_samples=10000, n_features=2, n_redundant=0,n_clusters_per_class=2, weights=[0.99], flip_y=0, random_state=4)# split into train and testn_train = 5000trainX, testX = X[:n_train, :], X[n_train:, :]trainy, testy = y[:n_train], y[n_train:]return trainX, trainy, testX, testy
可以使用Keras 深度学习库定义多层感知器神经网络。我们将定义一个神经网络,该网络需要两个输入变量,一个包含 10 个节点的隐藏层,然后是一个预测类标签的输出层。
我们将在隐藏层中使用流行的ReLU 激活函数,在输出层中使用 sigmoid 激活函数,以确保预测的概率在 [0,1] 范围内。该模型将使用具有默认学习率的随机梯度下降进行拟合,并根据交叉熵损失进行优化。
网络架构和超参数并未针对该问题进行优化;相反,当训练算法稍后被修改来处理倾斜的类分布时,网络提供了比较的基础。
*下面的define_model()*函数定义并返回模型,将网络输入变量的数量作为参数。
# define the neural network model
def define_model(n_input):# define modelmodel = Sequential()# define first hidden layer and visible layermodel.add(Dense(10, input_dim=n_input, activation='relu', kernel_initializer='he_uniform'))# define output layermodel.add(Dense(1, activation='sigmoid'))# define loss and optimizermodel.compile(loss='binary_crossentropy', optimizer='sgd')return model
一旦定义了模型,它就可以适合训练数据集。
我们将使用默认的批量大小对模型进行 100 个训练周期的拟合。
...
# fit model
model.fit(trainX, trainy, epochs=100, verbose=0)
一旦拟合,我们就可以使用该模型对测试数据集进行预测,然后使用ROC AUC分数评估预测。
...
# make predictions on the test dataset
yhat = model.predict(testX)
# evaluate the ROC AUC of the predictions
score = roc_auc_score(testy, yhat)
print('ROC AUC: %.3f' % score)
综合以上内容,下面列出了在不平衡分类数据集上拟合标准神经网络模型的完整示例。
# standard neural network on an imbalanced classification dataset
from sklearn.datasets import make_classification
from sklearn.metrics import roc_auc_score
from keras.layers import Dense
from keras.models import Sequential# prepare train and test dataset
def prepare_data():# generate 2d classification datasetX, y = make_classification(n_samples=10000, n_features=2, n_redundant=0,n_clusters_per_class=2, weights=[0.99], flip_y=0, random_state=4)# split into train and testn_train = 5000trainX, testX = X[:n_train, :], X[n_train:, :]trainy, testy = y[:n_train], y[n_train:]return trainX, trainy, testX, testy# define the neural network model
def define_model(n_input):# define modelmodel = Sequential()# define first hidden layer and visible layermodel.add(Dense(10, input_dim=n_input, activation='relu', kernel_initializer='he_uniform'))# define output layermodel.add(Dense(1, activation='sigmoid'))# define loss and optimizermodel.compile(loss='binary_crossentropy', optimizer='sgd')return model# prepare dataset
trainX, trainy, testX, testy = prepare_data()
# define the model
n_input = trainX.shape[1]
model = define_model(n_input)
# fit model
model.fit(trainX, trainy, epochs=100, verbose=0)
# make predictions on the test dataset
yhat = model.predict(testX)
# evaluate the ROC AUC of the predictions
score = roc_auc_score(testy, yhat)
print('ROC AUC: %.3f' % score)
运行该示例可在不平衡数据集上评估神经网络模型并报告 ROC AUC。
注意:由于算法或评估程序的随机性,或数值精度的差异,您的结果可能会有所不同。考虑运行示例几次并比较平均结果。
在这种情况下,该模型的 ROC AUC 达到约 0.949。这表明,与 ROC AUC 为 0.5 的朴素分类器相比,该模型具有一定的技巧。
ROC AUC: 0.949
这为对标准神经网络训练算法进行的任何修改提供了比较的baseline。
深度学习解决不平衡分类问题
神经网络模型通常使用误差反向传播算法进行训练。
这涉及使用模型的当前状态对训练集示例进行预测,计算预测的误差,然后使用误差更新模型权重,并将误差信用分配给从输出层到输入层的不同节点和层。
鉴于对错误分类错误的平衡关注,大多数标准神经网络算法并不适用于类别分布严重偏斜的数据集。
现有的深度学习算法大多没有考虑数据不平衡问题,导致这些算法在平衡的数据集上能够表现良好,而在不平衡的数据集上性能却无法保证。
—在不平衡数据集上训练深度神经网络,2016 年。
这个训练过程可以进行修改,使得一些例子比其他例子有更多或更少的错误。
还可以通过改变最小化的误差函数来考虑误分类成本。反向传播学习过程应该最小化误分类成本,而不是最小化平方误差。
—使用神经网络进行成本敏感学习,1998 年。
实现这一点的最简单方法是根据样本的类别对误差分数使用固定的权重,其中对于更重要的类别中样本的预测误差会增加,而对于不太重要的类别中样本的预测误差则会减少或保持不变。
…成本敏感学习方法基于对错误分类样本相关成本的考虑来解决数据不平衡问题。具体来说,它为样本的错误分类分配不同的成本值。
—在不平衡数据集上训练深度神经网络,2016 年。
可以对少数类中的例子应用较大的误差权重,因为在不平衡分类问题中,它们通常比多数类中的例子更重要。
- 大权重:分配给少数类的例子。
- 小权重:分配给来自多数类别的例子。
这种对神经网络训练算法的修改被称为加权神经网络或成本敏感神经网络。
通常,在定义用于成本敏感型学习的成本或“权重”时需要小心谨慎。但是,对于仅关注错误分类的不平衡分类,权重可以使用在训练数据集中观察到的类分布的逆。
使用 Keras 的加权神经网络
Keras Python 深度学习库提供了对类权重的支持。
用于训练 Keras 神经网络模型的fit() 函数采用一个名为class_weight的参数。此参数允许您定义一个字典,将类整数值映射到要应用于每个类的重要性。
此函数用于训练每种不同类型的神经网络,包括多层感知器、卷积神经网络和循环神经网络,因此类加权功能适用于所有这些网络类型。
例如,每个类 0 和 1 的 1 比 1 权重可以定义如下:
...
# fit model
weights = {0:1, 1:1}
history = model.fit(trainX, trainy, class_weight=weights, ...)
类别权重可以通过多种方式定义;例如:
- 领域专业知识,通过与主题专家交谈确定。
- 调整,由超参数搜索(例如网格搜索)确定。
- 启发式,使用一般的最佳实践来指定。
使用类权重的最佳实践是使用训练数据集中存在的类分布的倒数。
例如,测试数据集的类分布为少数类与多数类的比例为 1:100。该比例的倒数可以为 1 表示多数类,100 表示少数类,例如:
...
# fit model
weights = {0:1, 1:100}
history = model.fit(trainX, trainy, class_weight=weights, ...)
表示相同比例的分数不会产生相同的效果。例如,分别对多数类和少数类使用 0.01 和 0.99 可能会导致比使用 1 和 100 更差的性能(在本例中确实如此)。
...
# fit model
weights = {0:0.01, 1:0.99}
history = model.fit(trainX, trainy, class_weight=weights, ...)
原因是从多数类和少数类中抽取的示例的误差都减少了。此外,多数类的误差减少幅度显著缩小到非常小的数字,这些数字对模型权重的影响可能有限或非常小。
因此,建议使用整数来表示类权重,例如 1 表示无变化,100 表示类 1 的错误分类错误,其影响或惩罚比类 0 的错误分类错误大 100 倍。
我们可以使用上一节中定义的相同评估程序来评估具有类权重的神经网络算法。
我们期望神经网络的类加权版本比没有任何类加权的训练算法版本表现更好。
完整的示例如下。
# class weighted neural network on an imbalanced classification dataset
from sklearn.datasets import make_classification
from sklearn.metrics import roc_auc_score
from keras.layers import Dense
from keras.models import Sequential# prepare train and test dataset
def prepare_data():# generate 2d classification datasetX, y = make_classification(n_samples=10000, n_features=2, n_redundant=0,n_clusters_per_class=2, weights=[0.99], flip_y=0, random_state=4)# split into train and testn_train = 5000trainX, testX = X[:n_train, :], X[n_train:, :]trainy, testy = y[:n_train], y[n_train:]return trainX, trainy, testX, testy# define the neural network model
def define_model(n_input):# define modelmodel = Sequential()# define first hidden layer and visible layermodel.add(Dense(10, input_dim=n_input, activation='relu', kernel_initializer='he_uniform'))# define output layermodel.add(Dense(1, activation='sigmoid'))# define loss and optimizermodel.compile(loss='binary_crossentropy', optimizer='sgd')return model# prepare dataset
trainX, trainy, testX, testy = prepare_data()
# get the model
n_input = trainX.shape[1]
model = define_model(n_input)
# fit model
weights = {0:1, 1:100}
history = model.fit(trainX, trainy, class_weight=weights, epochs=100, verbose=0)
# evaluate model
yhat = model.predict(testX)
score = roc_auc_score(testy, yhat)
print('ROC AUC: %.3f' % score)
运行该示例准备合成的不平衡分类数据集,然后评估神经网络训练算法的类加权版本。
注意:由于算法或评估程序的随机性,或数值精度的差异,您的结果可能会有所不同。考虑运行示例几次并比较平均结果。
报告了 ROC AUC 分数,在这种情况下显示出比训练算法的未加权版本更好的分数,或者约为 0.973,而约为 0.949。
ROC AUC: 0.973