An In-Depth Examination of Risk Assessment in Multi-Class Classification Algorithms

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ghandwani, Disha, Sarna, Neeraj, Li, Yuanyuan, Lin, Yang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910728638693376
author Ghandwani, Disha
Sarna, Neeraj
Li, Yuanyuan
Lin, Yang
author_facet Ghandwani, Disha
Sarna, Neeraj
Li, Yuanyuan
Lin, Yang
contents Advanced classification algorithms are being increasingly used in safety-critical applications like health-care, engineering, etc. In such applications, miss-classifications made by ML algorithms can result in substantial financial or health-related losses. To better anticipate and prepare for such losses, the algorithm user seeks an estimate for the probability that the algorithm miss-classifies a sample. We refer to this task as the risk-assessment. For a variety of models and datasets, we numerically analyze the performance of different methods in solving the risk-assessment problem. We consider two solution strategies: a) calibration techniques that calibrate the output probabilities of classification models to provide accurate probability outputs; and b) a novel approach based upon the prediction interval generation technique of conformal prediction. Our conformal prediction based approach is model and data-distribution agnostic, simple to implement, and provides reasonable results for a variety of use-cases. We compare the different methods on a broad variety of models and datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2412_04166
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An In-Depth Examination of Risk Assessment in Multi-Class Classification Algorithms
Ghandwani, Disha
Sarna, Neeraj
Li, Yuanyuan
Lin, Yang
Machine Learning
Numerical Analysis
Advanced classification algorithms are being increasingly used in safety-critical applications like health-care, engineering, etc. In such applications, miss-classifications made by ML algorithms can result in substantial financial or health-related losses. To better anticipate and prepare for such losses, the algorithm user seeks an estimate for the probability that the algorithm miss-classifies a sample. We refer to this task as the risk-assessment. For a variety of models and datasets, we numerically analyze the performance of different methods in solving the risk-assessment problem. We consider two solution strategies: a) calibration techniques that calibrate the output probabilities of classification models to provide accurate probability outputs; and b) a novel approach based upon the prediction interval generation technique of conformal prediction. Our conformal prediction based approach is model and data-distribution agnostic, simple to implement, and provides reasonable results for a variety of use-cases. We compare the different methods on a broad variety of models and datasets.
title An In-Depth Examination of Risk Assessment in Multi-Class Classification Algorithms
topic Machine Learning
Numerical Analysis
url https://arxiv.org/abs/2412.04166