Positive-Unlabeled Learning for Agent Safety False Alarm Auditing
Organizations: Jinan University · Northwestern University · Analogy AI, Inc. · Shenzhen Technology University · Tsinghua University · The Chinese University of Hong Kong, Shenzhen
Abstract
Safety monitors help safeguard language-model agents interacting with external tools and environments, but conservative monitoring can generate many false alarms, consuming extensive review resources and weakening trust in alerts. Because false and genuine alarms often remain interleaved in native monitor scores, obtaining a reliable cutoff still requires substantial manual verification. In practice, a small set of verified-safe non-alarmed trajectories may be available while alarms remain unlabeled, naturally casting false-alarm auditing as a positive-unlabeled (PU) ranking problem. The key challenge is monitor-induced selection, since observed safe references are accepted by the monitor, while the hidden safe alarms of interest are precisely those it incorrectly flags, making the observed positives poorly representative of the positives to be recovered. To address this challenge, we propose a two-stage framework in which Trust-aware PU Supervision adapts safe references toward the alarm domain and protects plausible false alarms from excessive negative pressure, while Reliability-gated Rank Distillation consolidates consistent ordering preferences from multiple PU reference models into a single student. Consensus-guided Structural Refinement then improves the student ranking using hierarchical safe-reference support, alarm relations, and predicted reference consensus. The framework requires no alarm safety labels for fitting and leaves the underlying monitor unchanged. Across mainstream safety monitors, our method achieves a macro AUPRC of , outperforming eight evaluated PU baselines by 5.27--16.98 absolute percentage points; compared with PULDA, the strongest evaluated PU baseline, it recovers 33.3% more false alarms at a 5% review budget.
Figures & tables
| ATBench AgentDoG1.5-Qwen3.5-2B | ATBench Granite-Guardian-3.1-2B | |||||
| Method (year) | AUPRC | AUROC | AUGRC | AUPRC | AUROC | AUGRC |
| nnPU (2017) | 61.75 | 65.94 | 19.66 | 60.73 | 50.97 | 21.09 |
| Dist-PU (2022) | 70.04 | 70.00 | 18.65 | 68.16 | 57.81 | 19.43 |
| Robust-PU (2023) | 62.02 | 66.75 | 19.46 | 60.24 | 50.78 | 21.14 |
| PULDA (2023) | 72.89 | 73.59 | 17.76 | 69.82 | 58.43 | 19.28 |
| HolisticPU (2023) | 61.85 | 65.77 | 19.71 | 57.97 | 48.06 | 21.80 |
| Method | AUPRC | AUROC | AUGRC | FP@5% | FP@10% | FP@20% |
|---|---|---|---|---|---|---|
| Dist-PU | 24.16 | 57.20 | 40.97 | 66 | 110 | 165 |
| PULDA | 24.39 | 58.57 | 40.74 | 66 | 106 | 186 |
| PULDA ensemble | 25.42 | 59.45 | 40.21 | 71 | 114 | 195 |
| Our method ( ) | 26.15 | 60.32 | 39.68 | 76 | 125 | 208 |
| Our method ( ) | 28.72 | 63.45 | 37.82 | 94 | 152 | 241 |
| Macro ranking ( ) | FPs recovered | |||||
|---|---|---|---|---|---|---|
| Method | AUPRC | AUROC | AUGRC | Top 5% | Top 10% | Top 20% |
| Native monitor score | 53.39 | 55.21 | 24.71 | 12.00 | 24.00 | 50.00 |
| PULDA | 59.17 | 59.38 | 23.41 | 17.00 | 31.00 | 53.33 |
| Qwen2.5-7B judge | 57.78 | 57.17 | 23.97 | 18.00 | 28.00 | 50.00 |
| Our method | 64.44 | 61.70 | 22.85 | 22.67 | 37.00 | 65.00 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Component | Setting |
|---|---|
| PU scorer | two hidden layers; width each; Adam, learning rate , weight decay ; gradient clip |
| Training | epochs; TN/alarm batches ; updates per epoch; -epoch warm-up |
| Ranking and mixing | margin ; post-warm-up ; mixing; protection-memory EMA |
| Distribution alignment | smoothing temperature ; alarm-mean EMA coefficient |
| Role and transfer | five group-disjoint role folds; logistic ; L-BFGS tolerance ; role scale ; displacement scale |
| Reference bank | ; midpoint ranks; coordinatewise median; protection mass |
| Dataset | Monitor | |
|---|---|---|
| ATBench | AgentDoG1.5-Qwen3.5-2B | 0.02381 |
| ATBench | Granite-Guardian-3.1-2B | 0.02368 |
| TraceSafe | AgentDoG-Qwen3-4B | 0.01000 |
| TraceSafe | StepGuard-4B | 0.01266 |
| Method | |||||
|---|---|---|---|---|---|
| nnPU | 53.36 | 53.41 | 53.18 | 52.86 | 52.87 |
| DC-PU | 50.48 | 47.46 | 48.69 | 49.22 | 51.49 |
| Dist-PU | 58.86 | 57.78 | 58.09 | 57.88 | 57.71 |
| Robust-PU | 53.04 | 53.14 | 53.21 | 52.91 | 53.45 |
| PUL-CPBF | 52.96 | 52.96 | 52.96 | 52.96 | 52.96 |
| PULDA | 58.23 | 59.17 | 58.48 | 58.38 | 56.13 |
| Dataset | Monitor | Training alarms | Target alarms | Target FPs |
|---|---|---|---|---|
| ATBench | AgentDoG1.5-Qwen3.5-2B | 249 | 56 | 30 |
| ATBench | Granite-Guardian-3.1-2B | 300 | 76 | 44 |
| TraceSafe | AgentDoG-Qwen3-4B | 949 | 269 | 122 |
| TraceSafe | StepGuard-4B | 504 | 109 | 43 |
| Total target pool | — | 510 | 239 | |
| Resource | |
|---|---|
| Total elapsed time (s) | 162.86 |
| Peak GPU allocation (MiB) | 138.93 |
| Peak GPU reservation (MiB) | 164.00 |
| Peak process memory (MiB) | 2223.91 |
| Method | AUPRC | AUROC | AUGRC | Top 5% | Top 10% | Top 20% |
|---|---|---|---|---|---|---|
| Native monitor score | — | — | — | — | — | 50.00 |
| TN centroid distance | 55.44 | 57.98 | 23.75 | 13.00 | 28.00 | 53.00 |
| -NN to TNs | 54.69 | 57.74 | 23.82 | 14.00 | 27.00 | 50.00 |
| One-class SVM ( Schölkopf et al., 2001 ) | 57.74 | 59.30 | 23.43 | 14.00 | 29.00 | 52.00 |
| TN PCA support | 54.97 | 57.55 | 23.86 | 14.00 | 29.00 | 51.00 |
| Role-propensity score | 53.89 | 55.93 | 24.26 | 13.00 | 26.00 | 53.00 |
| Method | Architecture / compute | Macro AUPRC | Macro AUROC | Top- FP yield | Top- FP yield |
|---|---|---|---|---|---|
| (%) | (%) | ||||
| Dist-PU (Single) | one model; three-run mean | 57.78 | 58.08 | 16.00 | 29.33 |
| Dist-PU (Ensemble) | three-logit average | 58.65 | 58.74 | 17.00 | 30.67 |
| PULDA (Single) | one model; three-run mean | 59.17 | 59.38 | 17.00 | 31.00 |
| PULDA (Ensemble) | three-logit average | 60.24 | 60.18 | 18.33 | 32.67 |
| Ours ( ) | single model | 62.35 | 60.82 | 20.67 | 34.67 |
| Method | AUPRC | AUROC | AUGRC | FP@1% | FP@2% | FP@5% | FP@10% | FP@20% |
|---|---|---|---|---|---|---|---|---|
| Native monitor score | 16.98 | 49.36 | 42.42 | 5 | 10 | 34 | 65 | 121 |
| nnPU | 19.40 | 54.98 | 41.34 | 8 | 18 | 45 | 78 | 143 |
| DC-PU | 15.32 | 45.35 | 42.84 | 5 | 9 | 22 | 36 | 76 |
| Dist-PU | 24.16 | 57.20 | 40.97 | 24 | 42 | 66 | 110 | 165 |
| Robust-PU | 20.26 | 55.01 | 41.28 | 13 | 22 | 53 | 85 | 149 |
| PULDA | 24.39 | 58.57 | 40.74 | 23 | 38 | 66 | 106 | 186 |
| Monitor | Alarms | FPs | Native | PULDA | Our method |
|---|---|---|---|---|---|
| AgentDoG-Qwen3-4B | 2,427 | 213 | 14.92 | 19.70 | 22.85 |
| Granite-Guardian-3.1-2B | 1,627 | 425 | 19.05 | 29.09 | 34.58 |
| Risk category | Alarms | FPs | Native | PULDA | Our method |
|---|---|---|---|---|---|
| Unsafe information / misinformation | 606 | 329 | 63.31 | 59.09 | 69.85 |
| Property loss | 503 | 71 | 19.58 | 23.34 | 27.12 |
| Sensitive-data leak | 496 | 65 | 17.34 | 27.73 | 30.45 |
| Availability compromise | 455 | 46 | 11.24 | 13.19 | 15.60 |
| Physical harm | 439 | 39 | 16.73 | 22.93 | 24.88 |
| Harmful / vulnerable code | 716 | 62 | 14.74 | 27.69 | 29.35 |