Pixel-wise Energy-biased Abstention Learning for Anomaly Segmentation on Complex Urban Driving Scenes

Abstract

State-of-the-art (SOTA) anomaly segmentation approaches on complex urban driving scenes explore pixel-wise classification uncertainty learned from outlier exposure, or external reconstruction models. However, previous uncertainty approaches that directly associate high uncertainty to anomaly may sometimes lead to incorrect anomaly predictions, and external reconstruction models tend to be too inefficient for real-time self-driving embedded systems. In this paper, we propose a new anomaly segmentation method, named pixel-wise energy-biased abstention learning (PEBAL), that explores pixel-wise abstention learning (AL) with a model that learns an adaptive pixel-level anomaly class, and an energy-based model (EBM) that learns inlier pixel distribution. More specifically, PEBAL is based on a non-trivial joint training of EBM and AL, where EBM is trained to output high-energy for anomaly pixels (from outlier exposure) and AL is trained such that these high-energy pixels receive adaptive low penalty for being included to the anomaly class. We extensively evaluate PEBAL against the SOTA and show that it achieves the best performance across four benchmarks. Code is available at https://github.com/tianyu0207/PEBAL.

arXiv commentECCV 2022 Oral

References in corpus (16)

• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• On Calibration of Modern Neural Networks
• A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attacks
• Multi-Task Learning for Dense Prediction Tasks: A Survey
• Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles
• Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networks
• Deep Autoencoding Models for Unsupervised Anomaly Segmentation in Brain MR Images
• Deep Anomaly Detection with Outlier Exposure
• Predictive Uncertainty Estimation via Prior Networks
• Scaling Out-of-Distribution Detection for Real-World Settings
• The Fishyscapes Benchmark: Measuring Blind Spots in Semantic Segmentation
• Evaluating Bayesian Deep Learning Methods for Semantic Segmentation
• Training Confidence-calibrated Classifiers for Detecting Out-of-Distribution Samples
• Your Classifier is Secretly an Energy Based Model and You Should Treat it Like One
• Discriminative out-of-distribution detection for semantic segmentation
• This is not what I imagined: Error Detection for Semantic Segmentation through Visual Dissimilarity