Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Residual Pattern Learning for Pixel-wise Out-of-Distribution Detection in Semantic Segmentation

About

Semantic segmentation models classify pixels into a set of known (``in-distribution'') visual classes. When deployed in an open world, the reliability of these models depends on their ability not only to classify in-distribution pixels but also to detect out-of-distribution (OoD) pixels. Historically, the poor OoD detection performance of these models has motivated the design of methods based on model re-training using synthetic training images that include OoD visual objects. Although successful, these re-trained methods have two issues: 1) their in-distribution segmentation accuracy may drop during re-training, and 2) their OoD detection accuracy does not generalise well to new contexts (e.g., country surroundings) outside the training set (e.g., city surroundings). In this paper, we mitigate these issues with: (i) a new residual pattern learning (RPL) module that assists the segmentation model to detect OoD pixels without affecting the inlier segmentation performance; and (ii) a novel context-robust contrastive learning (CoroCL) that enforces RPL to robustly detect OoD pixels among various contexts. Our approach improves by around 10\% FPR and 7\% AuPRC the previous state-of-the-art in Fishyscapes, Segment-Me-If-You-Can, and RoadAnomaly datasets. Our code is available at: https://github.com/yyliu01/RPL.

Yuyuan Liu, Choubo Ding, Yu Tian, Guansong Pang, Vasileios Belagiannis, Ian Reid, Gustavo Carneiro• 2022

Related benchmarks

TaskDatasetResultRank
Anomaly SegmentationFishyscapes Lost & Found (val)--
53
Out-of-Distribution DetectionSMIYC Obstacle (test)
AP85.93
22
Anomaly SegmentationSMIYC-RA21 v1 (test)
AP83.49
14
Anomaly SegmentationSMIYC-RO21 v1 (test)
AP85.93
14
Anomaly SegmentationRoadAnomaly v1 (test)
AP0.7161
12
Domain-Generalized SegmentationMUAD
mIoU29.86
10
Anomaly SegmentationMUAD
AP27.7
10
Anomaly SegmentationACDC-POC
AP77.84
10
Domain-Generalized SegmentationACDC-POC
mIoU46.35
10
Out-of-Distribution DetectionSMIYC Obstacle (val)
AuPRC96.91
9
Showing 10 of 26 rows

Other info

Code

Follow for update