arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1906.04026cs.LGstat.ML
收藏

CRCEN:一种用于不平衡分类的广义代价敏感神经网络方法

CRCEN: A Generalized Cost-sensitive Neural Network Approach for Imbalanced Classification

Department of Computer Science, Wayne State University(韦恩州立大学计算机科学系)

Xiangrui Li, Dongxiao Zhu

摘要:

不平衡数据集的分类在实际应用中是一项具有挑战性的任务。在这种情况下,直接通过最小化分类误差来训练传统的分类算法可能会损害少数类的模型性能,同时优化多数类的性能。解决不平衡问题的传统方法包括重新采样和成本敏感方法。本文提出了一种新的损失函数的神经网络模型CRCEN,用于不平衡分类。基于交叉熵损失的加权形式,给出了模型预测概率、不平衡率和加权机制的理论关系。为了证明我们所提出的模型的有效性,CRCEN进行了测试,在几个基准数据集,并与基线模型进行了比较。

英文摘要:

Classification on imbalanced datasets is a challenging task in real-world applications. Training conventional classification algorithms directly by minimizing classification error in this scenario can compromise model performance for minority class while optimizing performance for majority class. Traditional approaches to the imbalance problem include re-sampling and cost-sensitive methods. In this paper, we propose a neural network model with novel loss function, CRCEN, for imbalanced classification. Based on the weighted version of cross entropy loss, we provide a theoretical relation for model predicted probability, imbalance ratio and the weighting mechanism. To demonstrate the effectiveness of our proposed model, CRCEN is tested on several benchmark datasets and compared with baseline models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑