HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Watermarking for Out-of-distribution Detection

Qizhou Wang Feng Liu Yonggang Zhang Jing Zhang Chen Gong Tongliang Liu Bo Han

Watermarking for Out-of-distribution Detection

Abstract

Out-of-distribution (OOD) detection aims to identify OOD data based on representations extracted from well-trained deep models. However, existing methods largely ignore the reprogramming property of deep models and thus may not fully unleash their intrinsic strength: without modifying parameters of a well-trained deep model, we can reprogram this model for a new purpose via data-level manipulation (e.g., adding a specific feature perturbation to the data). This property motivates us to reprogram a classification model to excel at OOD detection (a new task), and thus we propose a general methodology named watermarking in this paper. Specifically, we learn a unified pattern that is superimposed onto features of original data, and the model's detection capability is largely boosted after watermarking. Extensive experiments verify the effectiveness of watermarking, demonstrating the significance of the reprogramming property of deep models in OOD detection.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
out-of-distribution-detection-on-imagenet-1k-10Watermarking (WRN-40-2 w/ MSP)
AUROC: 84.00
FPR95: 61.2
out-of-distribution-detection-on-imagenet-1k-10Watermarking (WRN-40-2 w/ Energy)
AUROC: 80.80
FPR95: 67.75
out-of-distribution-detection-on-imagenet-1k-9Watermarking (WRN-40-2 w/ Energy)
AUROC: 79.85
FPR95: 71.85
out-of-distribution-detection-on-imagenet-1k-9Watermarking (WRN-40-2 w/ MSP)
AUROC: 82.03
FPR95: 70.59

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp