Learning from Imbalanced Data Sets with Weighted Cross-Entropy Function
Learning from Imbalanced Data Sets with Weighted Cross-Entropy Function
复制标题
DOI:
10.1007/s11063-018-09977-1
复制
发表时间:
2019-10-01
影响因子:
3.1
通讯作者:
Braga, Antonio Padua
中科院分区:
文献类型:
--
作者:
Aurelio, Yuri Sousa;de Almeida, Gustavo Matheus;Braga, Antonio Padua
This paper presents a novel approach to deal with the imbalanced data set problem in neural networks by incorporating prior probabilities into a cost-sensitive cross-entropy error function. Several classical benchmarks were tested for performance evaluation using different metrics, namely G-Mean, area under the ROC curve (AUC), adjusted G-Mean, Accuracy, True Positive Rate, True Negative Rate and F1-score. The obtained results were compared to well-known algorithms and showed the effectiveness and robustness of the proposed approach, which results in well-balanced classifiers given different imbalance scenarios.