Beyond Augmented-Action Surrogates for Multi-Expert Learning-to-Defer

TL;DR

This paper introduces a novel decoupled surrogate approach for multi-expert learning-to-defer systems, addressing optimization issues in existing methods and providing theoretical guarantees of stability and improved performance.

Contribution

It proposes a decoupled surrogate model with a theoretical excess-risk bound, outperforming existing methods in stability and effectiveness as the expert pool expands.

Findings

The new method remains stable as the expert pool grows.

It preserves rare specialists effectively.

It outperforms standalone classifiers on multiple benchmarks.

Abstract

A learning-to-defer (L2D) system decides, for each input, whether to predict on its own or to hand it to one of several available experts. The very well established recipe trains classifier and router jointly by treating the $K$ classes and $J$ experts as competing actions in one shared $(K{+}J)$-action geometry. Subsequent work has proposed a series of incremental fixes within this geometry; we show that each still suffers, to varying severity, from an optimization-level pathology (target distortion, gradient amplification, winner-take-all starvation, set-mass collapse, or class--expert coupling) even under statistical consistency. We step outside the augmented-action family entirely and propose a decoupled surrogate: a softmax classifier head and an independent sigmoid head per expert, mirroring the two natural objects of the problem. We show that per-sample updates are then coordinatewise and the class--expert Hessian block is identically zero, and prove an excess-risk bound with calibration constant $\max\{2\sqrt{2},\sqrt{2J/\lambda}\}$ -- to our knowledge the first multi-expert L2D guarantee whose constant does not grow with the expert pool when the per-expert weight is held fixed. On controlled synthetic studies and on CIFAR-10, CIFAR-10H, and Covertype, it is the only method in our comparison that remains stable as the expert pool grows, preserves rare specialists, and improves over a standalone classifier on every real-data benchmark.