Entropy Maximization and Meta Classification for Out-Of-Distribution Detection in Semantic Segmentation

TL;DR

This paper proposes a two-step method combining entropy maximization and meta classification to improve out-of-distribution detection in semantic segmentation, enhancing safety and reliability of DNNs in open-world applications.

Contribution

The authors introduce a novel training objective for entropy maximization on OoD samples and a post-processing meta classification step, significantly improving OoD detection performance.

Findings

Up to 52% reduction in detection errors.

Consistent OoD detection improvement across multiple datasets.

Marginal impact on original segmentation accuracy.

Abstract

Deep neural networks (DNNs) for the semantic segmentation of images are usually trained to operate on a predefined closed set of object classes. This is in contrast to the "open world" setting where DNNs are envisioned to be deployed to. From a functional safety point of view, the ability to detect so-called "out-of-distribution" (OoD) samples, i.e., objects outside of a DNN's semantic space, is crucial for many applications such as automated driving. A natural baseline approach to OoD detection is to threshold on the pixel-wise softmax entropy. We present a two-step procedure that significantly improves that approach. Firstly, we utilize samples from the COCO dataset as OoD proxy and introduce a second training objective to maximize the softmax entropy on these samples. Starting from pretrained semantic segmentation networks we re-train a number of DNNs on different in-distribution datasets and consistently observe improved OoD detection performance when evaluating on completely disjoint OoD datasets. Secondly, we perform a transparent post-processing step to discard false positive OoD samples by so-called "meta classification". To this end, we apply linear models to a set of hand-crafted metrics derived from the DNN's softmax probabilities. In our experiments we consistently observe a clear additional gain in OoD detection performance, cutting down the number of detection errors by up to 52% when comparing the best baseline with our results. We achieve this improvement sacrificing only marginally in original segmentation performance. Therefore, our method contributes to safer DNNs with more reliable overall system performance.