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

AdaSCALE: Adaptive Scaling for OOD Detection

About

The ability of the deep learning model to recognize when a sample falls outside its learned distribution is critical for safe and reliable deployment. Recent state-of-the-art out-of-distribution (OOD) detection methods leverage activation shaping to improve the separation between in-distribution (ID) and OOD inputs. These approaches resort to sample-specific scaling but apply a static percentile threshold across all samples regardless of their nature, resulting in suboptimal ID-OOD separability. In this work, we propose \textbf{AdaSCALE}, an adaptive scaling procedure that dynamically adjusts the percentile threshold based on a sample's estimated OOD likelihood. This estimation leverages our key observation: OOD samples exhibit significantly more pronounced activation shifts at high-magnitude activations under minor perturbation compared to ID samples. AdaSCALE enables stronger scaling for likely ID samples and weaker scaling for likely OOD samples, yielding highly separable energy scores. Our approach achieves state-of-the-art OOD detection performance, outperforming the latest rival OptFS by 14.94% in near-OOD and 21.67% in far-OOD datasets in average FPR@95 metric on the ImageNet-1k benchmark across eight diverse architectures. The code is available at: https://github.com/sudarshanregmi/AdaSCALE/

Sudarshan Regmi• 2025

Related benchmarks

TaskDatasetResultRank
Out-of-Distribution DetectionImageNet-1K OOD Average
AUROC85.74
31
Out-of-Distribution DetectionOpenOOD ImageNet-1K v1 (test)
SSB-Hard85.89
22
Adversarial and Jailbreaking Attack DetectionAdvBench
AUROC0.5894
20
Adversarial and Jailbreaking Attack DetectionBeavertails
AUROC0.2833
20
Adversarial and Jailbreaking Attack DetectionMaliciousInstruct
AUROC0.2994
20
Adversarial and Jailbreaking Attack DetectionHarmBench
AUROC0.4341
20
Adversarial and Jailbreaking Attack DetectionXSTest
AUROC0.2803
20
Adversarial and Jailbreaking Attack DetectionJailbreakBench
AUROC0.35
20
Safety DetectionPolyguard Code
AUROC0.7029
18
Safety DetectionPolyguard Cyber
AUROC0.6707
18
Showing 10 of 18 rows

Other info

Follow for update