MLLGJun 16, 2021

Silhouettes and quasi residual plots for neural nets and tree-based classifiers

arXiv:2106.08814v29 citations
Originality Synthesis-oriented
AI Analysis

This work provides a visualization tool for practitioners to assess classifier performance and data quality, but it is incremental as it adapts existing silhouette plots from cluster analysis to classification tasks.

The paper tackles the problem of visualizing classification results and potential label errors by introducing silhouette plots and quasi residual plots based on the probability of the alternative class (PAC). The method is demonstrated on benchmark datasets containing images, mixed features, and tweets, showing how high PAC values can indicate mislabeled cases.

Classification by neural nets and by tree-based methods are powerful tools of machine learning. There exist interesting visualizations of the inner workings of these and other classifiers. Here we pursue a different goal, which is to visualize the cases being classified, either in training data or in test data. An important aspect is whether a case has been classified to its given class (label) or whether the classifier wants to assign it to different class. This is reflected in the (conditional and posterior) probability of the alternative class (PAC). A high PAC indicates label bias, i.e. the possibility that the case was mislabeled. The PAC is used to construct a silhouette plot which is similar in spirit to the silhouette plot for cluster analysis (Rousseeuw, 1987). The average silhouette width can be used to compare different classifications of the same dataset. We will also draw quasi residual plots of the PAC versus a data feature, which may lead to more insight in the data. One of these data features is how far each case lies from its given class. The graphical displays are illustrated and interpreted on benchmark data sets containing images, mixed features, and tweets.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes