On Model Evaluation Under Non-constant Class Imbalance

被引:22
作者
Brabec, Jan [1 ,2 ]
Komarek, Tomas [1 ,2 ]
Franc, Vojtech [2 ]
Machlica, Lukas [3 ]
机构
[1] Cisco Syst Inc, Karlovo Namesti 10 St, Prague, Czech Republic
[2] Czech Tech Univ, Fac Elect Engn, Prague, Czech Republic
[3] Resistant AI, Prague, Czech Republic
来源
COMPUTATIONAL SCIENCE - ICCS 2020, PT IV | 2020年 / 12140卷
关键词
Evaluation metrics; Imbalanced data; Precision; ROC;
D O I
10.1007/978-3-030-50423-6_6
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Many real-world classification problems are significantly class-imbalanced to detriment of the class of interest. The standard set of proper evaluation metrics is well-known but the usual assumption is that the test dataset imbalance equals the real-world imbalance. In practice, this assumption is often broken for various reasons. The reported results are then often too optimistic and may lead to wrong conclusions about industrial impact and suitability of proposed techniques. We introduce methods (Supplementary code related to techniques described in this paper is available at: https://github.com/CiscoCTA/nci eval) focusing on evaluation under non-constant class imbalance. We show that not only the absolute values of commonly used metrics, but even the order of classifiers in relation to the evaluation metric used is affected by the change of the imbalance rate. Finally, we demonstrate that using sub-sampling in order to get a test dataset with class imbalance equal to the one observed in the wild is not necessary, and eventually can lead to significant errors in classifier's performance estimate.
引用
收藏
页码:74 / 87
页数:14
相关论文
共 20 条
  • [1] [Anonymous], SIGKDD Explor. Newsl, DOI DOI 10.1145/1007730.1007738
  • [2] Axelsson S., 2006, UNDERSTANDING INTRUS, P31
  • [3] Decision-forest voting scheme for classification of rare classes in network intrusion detection
    Brabec, Jan
    Machlica, Lukas
    [J]. 2018 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN, AND CYBERNETICS (SMC), 2018, : 3325 - 3330
  • [4] Chawla NV, 2010, DATA MINING AND KNOWLEDGE DISCOVERY HANDBOOK, SECOND EDITION, P875, DOI 10.1007/978-0-387-09823-4_45
  • [5] A comparison of static, dynamic, and hybrid analysis for malware detection
    Damodaran A.
    Troia F.D.
    Visaggio C.A.
    Austin T.H.
    Stamp M.
    [J]. Journal of Computer Virology and Hacking Techniques, 2017, 13 (1) : 1 - 12
  • [6] Davis J., 2006, P 23 INT C MACH LEAR, V23, P233, DOI [10.1145/1143844.1143874, DOI 10.1145/1143844.1143874]
  • [7] An introduction to ROC analysis
    Fawcett, Tom
    [J]. PATTERN RECOGNITION LETTERS, 2006, 27 (08) : 861 - 874
  • [8] Learning from class-imbalanced data: Review of methods and applications
    Guo Haixiang
    Li Yijing
    Shang, Jennifer
    Gu Mingyun
    Huang Yuanyue
    Bing, Gong
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2017, 73 : 220 - 239
  • [9] Learning from Imbalanced Data
    He, Haibo
    Garcia, Edwardo A.
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2009, 21 (09) : 1263 - 1284
  • [10] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778