arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2202.04837stat.MLcs.LG
收藏

异构校准:用于改进泛化的事后模型不可知框架

Heterogeneous Calibration: A post-hoc model-agnostic framework for improved generalization

LinkedIn Corporation(领英公司)

David Durfee, Aman Gupta, Kinjal Basu

摘要:

我们引入了异构校准的概念,该概念将事后模型不可知的转换应用于模型输出,以提高AUC在二进制分类任务上的性能。我们考虑了过度自信的模型,其在训练数据上的表现明显优于测试数据,并直观地解释了为什么它们可能没有充分利用数据中的适度有效的简单模式。我们将这些简单的模式称为特征空间的异质分区,并从理论上证明,完美地分别校准每个分区可以优化AUC。这给出了作为事后程序的异构校准的一般范例,通过该事后程序,通过基于树的算法识别特征空间的异构分区,并将事后校准技术应用于每个分区以提高AUC。虽然这个框架的理论最优性适用于任何模型,但我们专注于深度神经网络(DNN),并在各种开源数据集上测试这种范式的最简单实例。实验证明了该框架的有效性,以及未来应用更高性能的分区方案和更有效的校准技术的潜力。

英文摘要:

We introduce the notion of heterogeneous calibration that applies a post-hoc model-agnostic transformation to model outputs for improving AUC performance on binary classification tasks. We consider overconfident models, whose performance is significantly better on training vs test data and give intuition onto why they might under-utilize moderately effective simple patterns in the data. We refer to these simple patterns as heterogeneous partitions of the feature space and show theoretically that perfectly calibrating each partition separately optimizes AUC. This gives a general paradigm of heterogeneous calibration as a post-hoc procedure by which heterogeneous partitions of the feature space are identified through tree-based algorithms and post-hoc calibration techniques are applied to each partition to improve AUC. While the theoretical optimality of this framework holds for any model, we focus on deep neural networks (DNNs) and test the simplest instantiation of this paradigm on a variety of open-source datasets. Experiments demonstrate the effectiveness of this framework and the future potential for applying higher-performing partitioning schemes along with more effective calibration techniques.

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

↑