Confidence interval for micro-averaged F1 and macro-averaged F1 scores

作者:Kanae Takahashi, Kouji Yamamoto, Aya Kuchiba, Tatsuki Koyama

摘要

A binary classification problem is common in medical field, and we often use sensitivity, specificity, accuracy, negative and positive predictive values as measures of performance of a binary predictor. In computer science, a classifier is usually evaluated with precision (positive predictive value) and recall (sensitivity). As a single summary measure of a classifier’s performance, F1 score, defined as the harmonic mean of precision and recall, is widely used in the context of information retrieval and information extraction evaluation since it possesses favorable characteristics, especially when the prevalence is low. Some statistical methods for inference have been developed for the F1 score in binary classification problems; however, they have not been extended to the problem of multi-class classification. There are three types of F1 scores, and statistical properties of these F1 scores have hardly ever been discussed. We propose methods based on the large sample multivariate central limit theorem for estimating F1 scores with confidence intervals.

论文关键词:Precision, Recall, Machine learning, F 1 measures, Multi-class classification, Delta-method

论文评审过程:

论文官网地址:https://doi.org/10.1007/s10489-021-02635-5