English

Adversarial Robustness in Two-Stage Learning-to-Defer: Algorithms and Guarantees

Machine Learning 2025-08-26 v4 Machine Learning

Abstract

Two-stage Learning-to-Defer (L2D) enables optimal task delegation by assigning each input to either a fixed main model or one of several offline experts, supporting reliable decision-making in complex, multi-agent environments. However, existing L2D frameworks assume clean inputs and are vulnerable to adversarial perturbations that can manipulate query allocation--causing costly misrouting or expert overload. We present the first comprehensive study of adversarial robustness in two-stage L2D systems. We introduce two novel attack strategie--untargeted and targeted--which respectively disrupt optimal allocations or force queries to specific agents. To defend against such threats, we propose SARD, a convex learning algorithm built on a family of surrogate losses that are provably Bayes-consistent and (R,G)(\mathcal{R}, \mathcal{G})-consistent. These guarantees hold across classification, regression, and multi-task settings. Empirical results demonstrate that SARD significantly improves robustness under adversarial attacks while maintaining strong clean performance, marking a critical step toward secure and trustworthy L2D deployment.

Keywords

Cite

@article{arxiv.2502.01027,
  title  = {Adversarial Robustness in Two-Stage Learning-to-Defer: Algorithms and Guarantees},
  author = {Yannis Montreuil and Axel Carlier and Lai Xing Ng and Wei Tsang Ooi},
  journal= {arXiv preprint arXiv:2502.01027},
  year   = {2025}
}

Comments

Accepted at the 42nd International Conference on Machine Learning (ICML 2025)

R2 v1 2026-06-28T21:29:55.061Z