TRUST-LAPSE: An Explainable and Actionable Mistrust Scoring Framework for Model Monitoring
arXiv:2207.11290 · doi:10.1109/TAI.2023.3272876
Abstract
Continuous monitoring of trained ML models to determine when their predictions should and should not be trusted is essential for their safe deployment. Such a framework ought to be high-performing, explainable, post-hoc and actionable. We propose TRUST-LAPSE, a "mistrust" scoring framework for continuous model monitoring. We assess the trustworthiness of each input sample's model prediction using a sequence of latent-space embeddings. Specifically, (a) our latent-space mistrust score estimates mistrust using distance metrics (Mahalanobis distance) and similarity metrics (cosine similarity) in the latent-space and (b) our sequential mistrust score determines deviations in correlations over the sequence of past input representations in a non-parametric, sliding-window based algorithm for actionable continuous monitoring. We evaluate TRUST-LAPSE via two downstream tasks: (1) distributionally shifted input detection, and (2) data drift detection. We evaluate across diverse domains - audio and vision using public datasets and further benchmark our approach on challenging, real-world electroencephalograms (EEG) datasets for seizure detection. Our latent-space mistrust scores achieve state-of-the-art results with AUROCs of 84.1 (vision), 73.9 (audio), and 77.1 (clinical EEGs), outperforming baselines by over 10 points. We expose critical failures in popular baselines that remain insensitive to input semantic content, rendering them unfit for real-world model monitoring. We show that our sequential mistrust scores achieve high drift detection rates; over 90% of the streams show < 20% error for all domains. Through extensive qualitative and quantitative evaluations, we show that our mistrust scores are more robust and provide explainability for easy adoption into practice.
arXiv commentKeywords: Mistrust Scores, Latent-Space, Model monitoring, Trustworthy AI, Explainable AI, Semantic-guided AI
References in corpus (10)
- Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
- Speech Commands: A Dataset for Limited-Vocabulary Speech Recognition
- Captum: A unified and generic model interpretability library for PyTorch
- Deep k-Nearest Neighbors: Towards Confident, Interpretable and Robust Deep Learning
- Learning Confidence for Out-of-Distribution Detection in Neural Networks
- WAIC, but Why? Generative Ensembles for Robust Anomaly Detection
- Input complexity and out-of-distribution detection with likelihood-based generative models
- Investigating the influence of noise and distractors on the interpretation of neural networks
- A Variational Dirichlet Framework for Out-of-Distribution Detection
- Label-Free Explainability for Unsupervised Models