arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2110.02459cs.CVcs.LG
收藏

机器学习推理性能估计的事后模型

Post-hoc Models for Performance Estimation of Machine Learning Inference

University of California, Riverside(加州大学河滨分校)

Xuechen Zhang, Samet Oymak, Jiasi Chen

摘要:

估计机器学习模型在推理过程中的表现在各种场景中至关重要(例如,量化不确定性,或从可用模型库中进行选择)。然而,softmax置信度的标准准确度估计不是通用的,并且不能可靠地预测不同的性能度量(例如,F1-分数,召回)或在不同应用场景或输入域中的表现。在这项工作中,我们系统地概括性能估计的一组不同的指标和方案,并讨论广义概念的不确定性校准。我们建议使用事后模型来实现这一目标,并研究设计参数,包括模型类型,功能工程和性能指标,以实现最佳的估计质量。重点是目标检测问题,与以前的工作不同,我们的方法使每图像的指标,如召回和F1分数的估计。通过在三个用例(移动边缘卸载、模型选择和数据集转移)中对计算机视觉模型和数据集进行广泛实验,我们发现提出的事后模型始终优于标准校准的置信基线。据我们所知,这是第一个开发统一框架来解决机器学习推理的不同性能估计问题的工作。

英文摘要:

Estimating how well a machine learning model performs during inference is critical in a variety of scenarios (for example, to quantify uncertainty, or to choose from a library of available models). However, the standard accuracy estimate of softmax confidence is not versatile and cannot reliably predict different performance metrics (e.g., F1-score, recall) or the performance in different application scenarios or input domains. In this work, we systematically generalize performance estimation to a diverse set of metrics and scenarios and discuss generalized notions of uncertainty calibration. We propose the use of post-hoc models to accomplish this goal and investigate design parameters, including the model type, feature engineering, and performance metric, to achieve the best estimation quality. Emphasis is given to object detection problems and, unlike prior work, our approach enables the estimation of per-image metrics such as recall and F1-score. Through extensive experiments with computer vision models and datasets in three use cases -- mobile edge offloading, model selection, and dataset shift -- we find that proposed post-hoc models consistently outperform the standard calibrated confidence baselines. To the best of our knowledge, this is the first work to develop a unified framework to address different performance estimation problems for machine learning inference.

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

↑