English

Multiclass threshold-based classification and model evaluation

Machine Learning 2025-12-01 v1

Abstract

In this paper, we introduce a threshold-based framework for multiclass classification that generalizes the standard argmax rule. This is done by replacing the probabilistic interpretation of softmax outputs with a geometric one on the multidimensional simplex, where the classification depends on a multidimensional threshold. This change of perspective enables for any trained classification network an \textit{a posteriori} optimization of the classification score by means of threshold tuning, as usually carried out in the binary setting, thus allowing for a further refinement of the prediction capability of any network. Our experiments show indeed that multidimensional threshold tuning yields performance improvements across various networks and datasets. Moreover, we derive a multiclass ROC analysis based on \emph{ROC clouds} -- the attainable (FPR,TPR) operating points induced by a single multiclass threshold -- and summarize them via a \emph{Distance From Point} (DFP) score to (0,1)(0,1). This yields a coherent alternative to standard One-vs-Rest (OvR) curves and aligns with the observed tuning gains.

Keywords

Cite

@article{arxiv.2511.21794,
  title  = {Multiclass threshold-based classification and model evaluation},
  author = {Edoardo Legnaro and Sabrina Guastavino and Francesco Marchetti},
  journal= {arXiv preprint arXiv:2511.21794},
  year   = {2025}
}

Comments

arXiv admin note: substantial text overlap with arXiv:2505.11276

R2 v1 2026-07-01T07:56:57.097Z