This repository is the official implementation of Conformal Prediction for Deep Classifier via Label Ranking at ICML'2024
This code is built on the awesome toolbox TorchCP that you need to install first. We use Python 3.9, and TorchCP can be installed by:
pip install torchcp
Simply follow the instructions described here to install TorchCP as well as PyTorch. After that, other packages can be installed by:
pip install -r requirements.txt
Then, you are ready to go.
Producing the prediction sets:
python main.py --dataset imagenet --trials 10
with the following arguments:
- dataset: the name of the dataset.
- trials: the number of trials.
If you find this useful in your research, please consider citing:
@inproceedings{huang2023conformal,
title={Conformal Prediction for Deep Classifier via Label Ranking},
author={Jianguo Huang and Huajun Xi and Linjun Zhang and Huaxiu Yao and Yue Qiu and Hongxin Wei},
booktitle = {International Conference on Machine Learning (ICML)},
year = {2024}
}