Consistent Learning-to-Defer with Expert-Conditional Advice
Organizations: School of Computing National University of Singapore Singapore, 117418 · Département de Mathématiques Sorbonne University Paris, 75005, France · Fédération ENAC ISAE-SUPAERO ONERA Université de Toulouse Toulouse, 31555, France · Agency for Science, Technology and Research Institute for Infocomm Research Singapore, 138634
Abstract
We introduce learning-to-defer with advice, a new setting in which a system learns both which expert should handle an input and which information that expert should receive. The policy acts before observing advice and minimizes prediction loss plus expert and acquisition fees. Training may reveal all advice sources or only a subset on each record. We show that composing routing and advice losses can be inconsistent and that augmented training can incur avoidable estimation variance. We propose reference conditional classification, which jointly learns a fixed expert's error probability and each action's error conditional on that expert's success or failure. The reference outcome is observed on every labeled record, so records with missing advice still inform the estimates. We prove a consistency bound for bounded task losses, show that limited training preserves the full population objective, and establish variance reductions for the analyzed fits. On FEVER and Sensitive, full training has the lowest mean dev loss at every evaluated fee, while limited training improves on full training at most matched acquisition budgets.
Figures & tables
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
|---|---|
| Input, label, and advice sources. | |
| Numbers of experts and paid sources; includes no advice; . | |
| Expert , advice vector revealing only source , and task loss. | |
| Bounds on task loss, no-advice total cost, and all pair costs, respectively. | |
| Expert and source fees. | |
| Task loss, total cost, and conditional expected total cost of a pair; subscript abbreviates . |
| Task | Method | Learning rate | Weight decay | Reference | |
|---|---|---|---|---|---|
| Sensitive | RC full | ||||
| Sensitive | RC limited | ||||
| FEVER | RC full | ||||
| FEVER | RC limited |
| Fee | Reference conditional full | Augmented surrogate | Learning-to-Defer |
|---|---|---|---|
| 0 | 0.555 0.004 | 0.563 0.002 | 0.627 0.002 |
| 0.5 | 0.560 0.003 | 0.565 0.006 | 0.627 0.002 |
| 1 | 0.566 0.006 | 0.572 0.002 | 0.627 0.002 |
| 2 | 0.576 0.005 | 0.584 0.004 | 0.627 0.002 |
| 4 | 0.592 0.003 | 0.600 0.003 | 0.627 0.002 |
| 6 | 0.603 0.004 | 0.614 0.006 | 0.627 0.002 |
| Fee | Reference conditional full | Augmented surrogate | Learning-to-Defer |
|---|---|---|---|
| 0 | 0.244 0.002 | 0.248 0.001 | 0.273 0.001 |
| 0.04 | 0.247 0.002 | 0.252 0.001 | 0.273 0.001 |
| 0.12 | 0.253 0.001 | 0.258 0.002 | 0.273 0.001 |
| 0.2 | 0.256 0.000 | 0.262 0.001 | 0.273 0.001 |
| 1 | 0.266 0.001 | 0.277 0.001 | 0.273 0.001 |
| 2 | 0.266 0.001 | 0.273 0.002 | 0.273 0.001 |
| Fee | Decision loss | Task loss | Expert cost | Advice cost | Advice rate |
|---|---|---|---|---|---|
| 0.2 | 0.2561 | 0.2127 | 0.0329 | 0.0105 | 36.17% |
| 1 | 0.2660 | 0.2273 | 0.0299 | 0.0088 | 7.81% |
| 2 | 0.2659 | 0.2366 | 0.0292 | 0.0001 | 0.03% |
| 5 | 0.2659 | 0.2364 | 0.0295 | 0.0000 | 0.00% |
| Resource | Reference conditional full | Reference conditional limited |
|---|---|---|
| Training labels used | ||
| Paid source requests | ||
| Expert outcomes under paid sources | ||
| No-advice expert outcomes | ||
| Dev source requests | ||
| Optimizer updates |
| Budget | RC full | RC limited | Augmented | Limited full | Limited lower |
|---|---|---|---|---|---|
| 1404 | 0.685 0.013 | 0.674 0.024 | 0.651 0.007 | -0.0116 | 6/8 |
| 2812 | 0.668 0.021 | 0.656 0.011 | 0.644 0.005 | -0.0125 | 6/8 |
| 5624 | 0.643 0.004 | 0.623 0.008 | 0.639 0.006 | -0.0201 | 8/8 |
| 11248 | 0.627 0.006 | 0.601 0.007 | 0.624 0.006 | -0.0267 | 8/8 |
| 16872 | 0.607 0.007 | 0.589 0.006 | 0.614 0.006 | -0.0181 | 8/8 |
| 22500 | 0.597 0.005 | 0.579 0.005 | 0.606 0.004 | -0.0180 | 8/8 |
| Budget | RC full | RC limited | Augmented | Limited full | Limited lower |
|---|---|---|---|---|---|
| 1000 | 0.277 0.007 | 0.273 0.004 | 0.285 0.005 | -0.0037 | 6/8 |
| 2000 | 0.272 0.005 | 0.269 0.003 | 0.279 0.003 | -0.0029 | 6/8 |
| 4000 | 0.266 0.002 | 0.264 0.002 | 0.274 0.002 | -0.0022 | 6/8 |
| 8000 | 0.263 0.004 | 0.256 0.002 | 0.270 0.003 | -0.0065 | 8/8 |
| 12000 | 0.259 0.003 | 0.254 0.002 | 0.266 0.002 | -0.0043 | 8/8 |
| 16000 | 0.257 0.002 | 0.254 0.001 | 0.264 0.002 | -0.0031 | 7/8 |
| Augmented HT | Acquired-only mean | RC | ||
| Informative reference: | ||||
| 512 | 1 | |||
| 512 | 2 | |||
| 512 | 4 | |||
| 512 | 8 | |||
| 512 | 16 | |||
Learning to Defer
Learning to defer trains AI systems to decide when to predict autonomously and when to route cases to human or machine experts.
See all 26 papers