计算机科学2021,Vol.48Issue(9) :77-85.DOI:10.11896/jsjkx.200900013

基于代价敏感卷积神经网络的非平衡问题混合方法

Cost-sensitive Convolutional Neural Network Based Hybrid Method for Imbalanced Data Classification

黄颖琦 陈红梅
计算机科学2021,Vol.48Issue(9) :77-85.DOI:10.11896/jsjkx.200900013

基于代价敏感卷积神经网络的非平衡问题混合方法

Cost-sensitive Convolutional Neural Network Based Hybrid Method for Imbalanced Data Classification

黄颖琦 1陈红梅2
扫码查看

作者信息

  • 1. 西南交通大学信息科学与技术学院 成都 611756
  • 2. 西南交通大学云计算与智能技术高校重点实验室 成都 611756
  • 折叠

摘要

非平衡问题是数据挖掘领域中普遍存在的一个问题,数据的偏态分布会使得分类器的分类效果不理想.卷积神经网络作为一种高效的数据挖掘工具,被广泛应用于分类任务,但其训练过程若受到数据非平衡的不利影响,则将导致少数类的分类准确率下降.针对二分类非平衡数据分类问题,文中提出了一种基于代价敏感卷积神经网络的非平衡问题混合方法.首先将密度峰值聚类算法与SMOTE相结合,通过过采样对数据进行预处理,降低原始数据集的不平衡程度;然后利用代价敏感思想对非平衡数据中的不同类别给予不同权重,并考虑预测值与标签值之间的欧氏距离,对非平衡数据中多数类和少数类赋予不同的代价损失,构建代价敏感卷积神经网络模型,以提高卷积神经网络对少数类的识别率.选取6个不同的数据集,用于验证所提方法的有效性.实验结果表明,所提方法可以提高卷积神经网络模型对非平衡数据的分类性能.

关键词

非平衡问题/卷积神经网络/过采样/数据预处理/代价敏感损失函数

引用本文复制引用

基金项目

出版年

2021
计算机科学
重庆西南信息有限公司(原科技部西南信息中心)

计算机科学

CSTPCDCSCD北大核心
影响因子:0.944
ISSN:1002-137X
被引量3
参考文献量39
段落导航相关论文