中文

MAPIE:一个用于无分布不确定性量化的开源库

机器学习 2022-07-26 v1 机器学习

摘要

估计机器学习(ML)模型预测相关的不确定性,对于评估其鲁棒性和预测能力至关重要。在本投稿中,我们介绍 MAPIE(Model Agnostic Prediction Interval Estimator,模型无关预测区间估计器),一个用于量化单输出回归和多类分类任务中 ML 模型不确定性的开源 Python 库。MAPIE 实现了保形预测方法,使用户能够轻松计算不确定性,并在边际覆盖率上具有强有力的理论保证,且对模型或底层数据分布的假设较弱。MAPIE 托管于 scikit-learn-contrib,并完全“兼容 scikit-learn”。因此,它接受任何带有 scikit-learn API 的回归器或分类器。该库可在 https://github.com/scikit-learn-contrib/MAPIE/ 获取。

关键词

引用

@article{arxiv.2207.12274,
  title  = {MAPIE: an open-source library for distribution-free uncertainty quantification},
  author = {Vianney Taquet and Vincent Blot and Thomas Morzadec and Louis Lacombe and Nicolas Brunel},
  journal= {arXiv preprint arXiv:2207.12274},
  year   = {2022}
}

备注

Submitted to the 2022 ICML workshop "Distribution-free uncertainty quantification"