Distribution-free Conformal Prediction for Ordinal Classification

Subhrasish Chakraborty, Chhavi Tyagi, Haiyan Qiao, Wenge Guo

Research output: Contribution to journalConference articlepeer-review

Abstract

Conformal prediction is a general distribution-free approach for constructing prediction sets combined with any machine learning algorithm that achieve valid marginal or conditional coverage in finite samples. Ordinal classification is common in real applications where the target variable has natural ordering among the class labels. In this paper, we discuss constructing distribution-free prediction sets for such ordinal classification problems by leveraging the ideas of conformal prediction and multiple testing with FWER control. Newer conformal prediction methods are developed for constructing contiguous and non-contiguous prediction sets based on marginal and conditional (class-specific) conformal p-values, respectively. Theoretically, we prove that the proposed methods respectively achieve satisfactory levels of marginal and class-specific conditional coverages. Through simulation study and real data analysis, these proposed methods show promising performance compared to the existing conformal method.

Original languageEnglish (US)
Pages (from-to)120-139
Number of pages20
JournalProceedings of Machine Learning Research
Volume230
StatePublished - 2024
Event13th Symposium on Conformal and Probabilistic Prediction with Applications, COPA 2024 - Milano, Italy
Duration: Sep 9 2024Sep 11 2024

All Science Journal Classification (ASJC) codes

  • Artificial Intelligence
  • Software
  • Control and Systems Engineering
  • Statistics and Probability

Keywords

  • class-specific conditional coverage
  • Conformal prediction
  • FWER control
  • marginal coverage
  • multiple testing
  • ordinal classification

Fingerprint

Dive into the research topics of 'Distribution-free Conformal Prediction for Ordinal Classification'. Together they form a unique fingerprint.

Cite this