Drift-to-Action Controllers: Budgeted Interventions with Online Risk Certificates
Organizations: Multidisciplinary Faculty of Nador, Mohammed First University, Oujda 60000, Morocco · Laboratory LaSTI, ENSAK, Sultan Moulay Slimane University, Khouribga 54000, Morocco
Abstract
Deployed machine learning systems face distribution drift, yet most monitoring pipelines stop at alarms and leave the response underspecified under labeling, compute, and latency constraints. We introduce Drift2Act, a drift-to-action controller that treats monitoring as constrained decision-making with explicit safety. Drift2Act combines a sensing layer that maps unlabeled monitoring signals to a belief over drift types with an active risk certificate that queries a small set of delayed labels from a recent window to produce an anytime-valid upper bound on current risk. The certificate gates operation: if , the controller selects low-cost actions (e.g., recalibration or test-time adaptation); if , it activates abstain/handoff and escalates to rollback or retraining under cooldowns. In a realistic streaming protocol with label delay and explicit intervention costs, Drift2Act achieves near-zero safety violations and fast recovery at moderate cost on WILDS Camelyon17, DomainNet, and a controlled synthetic drift stream, outperforming alarm-only monitoring, adapt-always adaptation, schedule-based retraining, selective prediction alone, and an ablation without certification. Overall, online risk certification enables reliable drift response and reframes monitoring as decision-making with safety.
Figures & tables
| Method | |||||
|---|---|---|---|---|---|
| Camelyon17 stream | |||||
| Alarm-only | 3.2 | 46 | 22 | 210 | 0.71 |
| Adapt-always (TTA) | 58.0 | 3 | 18 | 74 | 0.79 |
| Retrain-on-schedule | 41.5 | 9 | 35 | 96 | 0.77 |
| Selective prediction only | 9.8 | 14 | 24 | 160 | 0.74 |
| Controller (no certificate) | 18.7 | 7 | 19 | 98 | 0.78 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Action | Name | Operational effect | Cost (units) |
|---|---|---|---|
| No-op | Continue inference with current parameters and current output policy. | ||
| Recalibrate | Update a lightweight calibrator using ; logits unchanged. | ||
| Test-time adapt | Apply a small number of adaptation steps on ; update . | ||
| Query labels | Request labels from recent inputs; add to when they arrive. | ||
| Retrain | Trigger retraining using plus logged data; deploy updated . | ||
| Rollback | Revert to a previously validated checkpoint . |
| Act. | Name | Prereq./Effect | Cooldown | Cost |
|---|---|---|---|---|
| No-op | Keep . | – | ||
| Recalib. | Update from . | – | ||
| TTA | Adapt on . | – | ||
| Query | Request ; delay . | – | ||
| Retrain | Retrain + deploy. | |||
| Rollback | Restore . |
| Baseline | Configuration |
|---|---|
| Alarm-only | Alarm when with ; no actions beyond logging. |
| Adapt-always | Apply every step; 1 gradient step per batch; learning rate . |
| Retrain-on-schedule | Trigger every steps, respecting cooldown . |
| Selective prediction only | Predict if with ; otherwise abstain ( ). |
| Controller (no cert.) | Same as ours but remove gating: replace branch by normal utility maximization. |
| Drift type | (recalib.) | (TTA) | (query) | (retrain) | (rollback) | (abstain) | |
|---|---|---|---|---|---|---|---|
| none | 0.00 | 0.10 | 0.05 | 0.08 | 0.12 | 0.10 | 0.15 |
| covariate | 0.00 | 0.35 | 0.70 | 0.25 | 0.85 | 0.40 | 0.55 |
| concept | 0.00 | 0.20 | 0.30 | 0.75 | 1.05 | 0.60 | 0.65 |
| subgroup | 0.00 | 0.25 | 0.35 | 0.85 | 0.95 | 0.55 | 0.80 |
| Method variant | FIR | |||||
|---|---|---|---|---|---|---|
| Camelyon17 stream | ||||||
| Alarm-only | ||||||
| Adapt-always (TTA) | ||||||
| Retrain-on-schedule | ||||||
| Selective prediction only | ||||||
| Controller (no certificate) | ||||||
| Variant | FIR | |||||
|---|---|---|---|---|---|---|
| Camelyon17 stream | ||||||
| Drift2Act (full) | ||||||
| No belief (single score) | ||||||
| No active labeling (fixed rate) | ||||||
| No certificate gating | ||||||
| MMD-only monitors | ||||||
| Camelyon17 | DomainNet | SyntheticDrift-CIFAR | ||||
|---|---|---|---|---|---|---|
| Method | FIR | Heavy-FIR | FIR | Heavy-FIR | FIR | Heavy-FIR |
| Alarm-only | ||||||
| Adapt-always (TTA) | ||||||
| Retrain-on-schedule | ||||||
| Selective prediction only | ||||||
| Controller (no certificate) | ||||||
| Delay | ||||
|---|---|---|---|---|
| 0 | ||||
| 25 | ||||
| 50 | ||||
| 100 |
| Budget | ||||
|---|---|---|---|---|
| 1000 | ||||
| 2000 | ||||
| 3000 | ||||
| 5000 |
| Drift pattern | FIR | |||||
|---|---|---|---|---|---|---|
| Sudden drift (baseline setting) | ||||||
| Gradual drift (slow ramp) | ||||||
| Gradual drift (fast ramp) | ||||||
| Recurring drift (short period) | ||||||
| Recurring drift (long period) |
| Per-label cost | FIR | |||||
|---|---|---|---|---|---|---|
| 0.01 | ||||||
| 0.05 | ||||||
| 0.10 |
| Noise level | FIR | |||||
|---|---|---|---|---|---|---|
| 0.00 | ||||||
| 0.25 | ||||||
| 0.50 | ||||||
| 0.75 |
| Method | FIR | |||||
|---|---|---|---|---|---|---|
| Alarm-only | ||||||
| Adapt-always (TTA) | ||||||
| Retrain-on-schedule | ||||||
| Controller (no certificate) | ||||||
| Controller + certificate (ours) |
Drifting Models
Drifting models learn transport fields that move generated samples toward a target distribution, often enabling one-step generation.
See all 53 papers