Home

Awesome

Multi-class probabilistic classification using Venn-ABERS (Conformal) prediction

DOI

Please cite the code - BIBTEX citation

@misc{https://doi.org/10.5281/zenodo.6685149, doi = {10.5281/ZENODO.6685149}, url = {https://zenodo.org/record/6685149}, author = {{Valeman}}, title = {valeman/Multi-class-probabilistic-classification: v0.1.0}, publisher = {Zenodo}, year = {2022}, copyright = {Open Access} }

Log loss wavefront dataset

Brier loss wavefront dataset

Implementation of multi-class probabilistic classification using inductive and cross Venn–Abers predictors described in Multi-class probabilistic classification using inductive and cross Venn–Abers predictors. This code has also been released on Papers with Code.

The VennABERS.py file is a pure Python implementation of the fast Venn-ABERS Predictor for binary classification implemented in https://github.com/ptocca/VennABERS/ as published in NeurIPS paper Large-scale probabilistic predictors with and without guarantees of validity

Unlike other methods such as Platt's scaler and Isotonic Regression Venn-ABERS predictors (being a form of conformal prediction) contain inbuilt mathematical guarantees of validity (lack of bias). In addition Venn-ABERS predictors are multi-output predictors that output two probability predictions of class 1. Such two probability prediction of assigning label 1 for each test objectc can be considered prediction interval. The interval width contains valuable information about degree of certainty of prediction, such information is not available when using other calibration methods such as Platt's and isotonic regression.

In addition Venn-ABERS is a more advanced and regularised form of isotonic regression that, unlike isotonic regression, does not suffer from overfitting in non-large size datasets.

The Venn-ABERS predictor can be viewed as a distribution-free calibration function that maps scores output by a scoring classifier to well-calibrated probabilities. A gentle introduction can be found in the tutorial by Paolo Toccaceli (2017) listed on Awesome Conformal Prediction https://github.com/valeman/awesome-conformal-prediction