|国家预印本平台
首页|TorchCP: A Python Library for Conformal Prediction

TorchCP: A Python Library for Conformal Prediction

TorchCP: A Python Library for Conformal Prediction

来源:Arxiv_logoArxiv
英文摘要

Conformal prediction (CP) is a robust statistical framework that generates prediction intervals or sets with guaranteed coverage probability, addressing the challenge of quantifying predictive uncertainty in deep learning. Despite advancements in deep learning architectures and datasets, reliable uncertainty estimation remains elusive, making CP increasingly vital. This paper introduces TorchCP, a PyTorch-native library designed to integrate state-of-the-art CP algorithms into deep learning tasks, including classification, regression, graph neural networks, and large language models. TorchCP offers a comprehensive suite of advanced methodologies, a modular design for easy customization, and full GPU-accelerated scalability. Released under the LGPL-3.0 license, TorchCP has gained widespread adoption with over 12,582 PyPi downloads. It is supported by approximately 16,132 lines of code, 564 unit tests achieving 100\% coverage, and comprehensive documentation. By bridging statistics and computer science, TorchCP empowers researchers and practitioners to advance conformal prediction in diverse deep learning applications.

Jianguo Huang、Jianqing Song、Xuanning Zhou、Bingyi Jing、Hongxin Wei

计算技术、计算机技术

Jianguo Huang,Jianqing Song,Xuanning Zhou,Bingyi Jing,Hongxin Wei.TorchCP: A Python Library for Conformal Prediction[EB/OL].(2025-07-15)[2025-07-25].https://arxiv.org/abs/2402.12683.点此复制

评论