On the Normalization of Confusion Matrices: Methods and Geometric Interpretations

View PDF HTML (experimental)

Abstract:The confusion matrix is a standard tool for evaluating classifiers by providing insights into class-level errors. In heterogeneous settings, its values are shaped by two main factors: class similarity -- how easily the model confuses two classes -- and distribution bias, arising from skewed distributions in the training and test sets. However, confusion matrix values reflect a mix of both factors, making it difficult to disentangle their individual contributions. To address this, we introduce bistochastic normalization using Iterative Proportional Fitting, a generalization of row and column normalization. Unlike standard normalizations, this method recovers the underlying structure of class similarity. By disentangling error sources, it enables more accurate diagnosis of model behavior and supports more targeted improvements. We also show a correspondence between confusion matrix normalizations and the model's internal class representations. Both standard and bistochastic normalizations can be interpreted geometrically in this space, offering a deeper understanding of what normalization reveals about a classifier.

Subjects:

Machine Learning (cs.LG)

Cite as: arXiv:2509.04959 [cs.LG]

(or arXiv:2509.04959v2 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2509.04959

arXiv-issued DOI via DataCite

Submission history

From: Johan Erbani Leydet [view email] [v1] Fri, 5 Sep 2025 09:36:51 UTC (4,198 KB) [v2] Mon, 30 Mar 2026 14:43:06 UTC (3,589 KB)