<h1 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">共形预测</h1>
<h3 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">适用于任何机器学习任务的严格不确定性量化</h3>
<p align="center">
<a style="text-decoration:none !important;" href="https://arxiv.org/abs/2107.07511" alt="arXiv"><img src="https://img.shields.io/badge/paper-arXiv-red" /></a>
<a style="text-decoration:none !important;" href="https://people.eecs.berkeley.edu/%7Eangelopoulos/blog/posts/gentle-intro" alt="website"><img src="https://img.shields.io/badge/website-Berkeley-yellow" /></a>
<a style="text-decoration:none !important;" href="https://docs.conda.io/en/latest/miniconda.html" alt="package management"> <img src="https://img.shields.io/badge/conda-env-green" /></a>
<a style="text-decoration:none !important;" href="https://opensource.org/licenses/MIT" alt="License"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/1ee37478-7984-44e4-b3d0-959c97037d55.svg" /></a>
<a style="text-decoration:none !important;" href="https://www.youtube.com/watch?v=nql000Lu_iE" alt="arXiv"><img src="https://img.shields.io/youtube/views/nql000Lu_iE?style=social" /></a>
<a style="text-decoration:none !important;" href="http://hits.dwyl.com/aangelopoulos/conformal-prediction" alt="arXiv"><img src="https://yellow-cdn.veclightyear.com/0a4dffa0/3a09388a-10ad-4f8a-9efb-38e82859bf20.svg?style=flat-square" /></a>
</p>
<p>
本代码库是在真实数据上开始使用共形预测(又称共形推理)的最简单方法。
每个<code>notebooks</code>都将共形预测应用于一个使用最先进机器学习模型的实际预测问题。
</p>
<p align="center"><b>无需下载模型或数据即可运行一致性预测</b></p>
<p>
笔记本会自动下载几个大规模真实世界数据集的原始模型输出和每个数据集的少量样本数据。您可以完全在这个沙盒中开发和测试一致性预测方法,无需运行原始模型或下载原始数据。打开笔记本即可查看预期输出。您可以使用这些笔记本来试验现有方法或作为开发自己方法的模板。
</p>
<h3 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">示例笔记本</h3>
<ul>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/imagenet-smallest-sets.ipynb"><code>notebooks/imagenet-smallest-sets.ipynb</code></a>:使用ResNet152分类器进行Imagenet分类。预测集以90%的概率保证包含真实类别。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/meps-cqr.ipynb"><code>notebooks/meps-cqr.ipynb</code></a>:使用梯度提升回归器和一致性分位数回归进行医疗支出回归。预测区间以90%的概率保证包含真实美元价值。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/multilabel-classification-mscoco.ipynb"><code>notebooks/multilabel-classification-mscoco.ipynb</code></a>:在Microsoft Common Objects in Context (MS-COCO)数据集上进行多标签图像分类。集值预测保证包含90%的真实类别。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/toxic-text-outlier-detection.ipynb"><code>notebooks/toxic-text-outlier-detection.ipynb</code></a>:通过一致性异常检测识别有毒或仇恨的在线评论。不超过10%的分布内数据会被标记为有毒。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/tumor-segmentation.ipynb"><code>notebooks/tumor-segmentation.ipynb</code></a>:从内窥镜图像中分割肠息肉。分割掩码包含90%的真实肿瘤像素。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/weather-time-series-distribution-shift.ipynb"><code>notebooks/weather-time-series-distribution-shift</code></a>:使用时间序列数据和加权一致性预测来预测全球未来温度。预测区间包含90%的真实温度。</li>
<li><a href="https://github.com/aangelopoulos/conformal-prediction/blob/main/notebooks/imagenet-selective-classification.ipynb"><code>notebooks/imagenet-selective-classification.ipynb</code></a>:当Imagenet分类器不确定时,它会放弃预测。否则,它将有90%的准确率,尽管基础模型的准确率只有77%。</li>
<li>...以及更多!</li>
</ul>
<h3 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">笔记本可以使用提供的Google Colab链接立即运行</h3>
<h5 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">Colab链接在每个笔记本的顶部单元格中</h5>
<p>
要在本地运行这些笔记本,您只需安装正确的依赖项并按下<code>运行所有单元格</code>即可!笔记本将自动下载所有必需的数据和模型输出。您的计算机需要1.5GB的空间来存储自动下载的数据。如果您想了解我们如何生成预先计算的模型输出和数据子样本,请查看<code>generation-scripts</code>中的文件。每个数据集都有一个对应的脚本。要创建具有正确依赖项的<code>conda</code>环境,请运行<code>conda env create -f environment.yml</code>。如果您仍然遇到依赖项错误,请确保在Jupyter笔记本中激活<code>conformal</code>环境。
</p>
<h3 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">引用</h3>
<p>
这个仓库旨在配合我们的论文<a href="https://arxiv.org/abs/2107.07511">一致性预测和无分布不确定性量化的温和介绍</a>。
该论文中有每个示例的详细解释和归属说明。
如果您发现这个仓库有用,除了相关方法和数据集外,请引用:
</p>
<pre><code>@article{angelopoulos2021gentle,
title={A gentle introduction to conformal prediction and distribution-free uncertainty quantification},
author={Angelopoulos, Anastasios N and Bates, Stephen},
journal={arXiv preprint arXiv:2107.07511},
year={2021}
}</code></pre>
<h3 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">视频</h3>
如果您有兴趣通过视频形式了解一致性预测,请观看我们下面的视频!
<h4 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">共形预测教程</h4>
<p align="center"><a href="https://www.youtube.com/watch?v=nql000Lu_iE"> <img width="350" src="https://yellow-cdn.veclightyear.com/0a4dffa0/95cf0dc5-1e07-4e64-98ea-3961e233fb9e.jpg" /> </a></p>
<h4 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">共形预测教程 第二部分:条件覆盖</h4>
<p align="center"><a href="https://www.youtube.com/watch?v=TRx4a2u-j7M"> <img width="350" src="https://yellow-cdn.veclightyear.com/0a4dffa0/9d65c587-2b79-44f1-8b4b-eb52a2d66f43.jpg" /> </a></p>
<h4 align="center" style="margin-bottom:0px; border-bottom:0px; padding-bottom:0px">共形预测教程 第三部分:超越共形预测</h4>
<p align="center"><a href="https://www.youtube.com/watch?v=37HKrmA5gJE"> <img width="350" src="https://yellow-cdn.veclightyear.com/0a4dffa0/7683fba4-883e-4791-a8fd-73649293e30e.jpg" /> </a></p>