arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.01433stat.MLcs.LG

Consistent Batch Normalization for Weighted Loss in Imbalanced-Data Environment

Graduate School of Science and Engineering, Yamagata University(山形大学理工学研究生院);FUJITSU BROAD SOLUTION & CONSULTING Inc.(富士通广义解决方案咨询有限公司)

Muneki Yasuda, Yeo Xian En, Seishirou Ueno

英文摘要:

In this study, classification problems based on feedforward neural networks in a data-imbalanced environment are considered. Learning from an imbalanced dataset is one of the most important practical problems in the field of machine learning. A weighted loss function (WLF) based on a cost-sensitive approach is a well-known and effective method for imbalanced datasets. A combination of WLF and batch normalization (BN) is considered in this study. BN is considered as a powerful standard technique in the recent developments in deep learning. A simple combination of both methods leads to a size-inconsistency problem due to a mismatch between the interpretations of the effective size of the dataset in both methods. A simple modification to BN, called weighted BN (WBN), is proposed to correct the size mismatch. The idea of WBN is simple and natural. The proposed method in a data-imbalanced environment is validated using numerical experiments.

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

↑