30

Randomized multi-class classification under system constraints: a unified approach via post-processing

Evgenii Chzhen
Mohamed Hebiri
Gayane Taturyan
Main:18 Pages
2 Figures
Bibliography:4 Pages
Appendix:2 Pages
Abstract

We study the problem of multi-class classification under system-level constraints expressible as linear functionals over randomized classifiers. We propose a post-processing approach that adjusts a given base classifier to satisfy general constraints without retraining. Our method formulates the problem as a linearly constrained stochastic program over randomized classifiers, and leverages entropic regularization and dual optimization techniques to construct a feasible solution. We provide finite-sample guarantees for the risk and constraint satisfaction for the final output of our algorithm under minimal assumptions. The framework accommodates a broad class of constraints, including fairness, abstention, and churn requirements.

View on arXiv
Comments on this paper