Pairwise AUC Optimization Needs Corrective Power: A Unified View
Organizations: School of Computer Science and Technology, University of Chinese Academy of Sciences, Beijing, China · Institute of Computing Technology, Chinese Academy of Sciences, Beijing, China
Abstract
Direct pairwise AUC-surrogate training is attractive because it trains toward the ranking metric used at test time, especially under class imbalance. This creates a direct alignment between the training objective and the evaluation metric, but such metric alignment alone does not guarantee that stochastic training remains corrective. We study this gap through Corrective Power: whether pairwise mini-batch updates produce stronger and more reliable improvements for more severe positive-negative ranking errors. Our main observation is that direct pairwise training can enter regions where severe errors remain common but no longer produce useful corrective motion. In such regions, even the mini-batch noise that would normally help SGD move along the steepest severe-error correction direction can become much weaker. In this paper, we seek a theoretical explanation for this by exploring the degenerate property of the U-statistics formed from the pairwise gradients therein. Our results provide a mechanistic explanation for several empirical patterns: (a) squared-hinge loss is often an effective AUC surrogate; (b) cross-entropy warm-up can help models that enter the AUC phase from a poor starting region; and (c) parameter-efficient fine-tuning from a pretrained model can reduce that need by starting closer to a favorable ranking geometry. Experiments on 8 CV tasks and 8 NLP tasks, using ResNet50, DenseNet121, DistilBERT-base, and Qwen2.5-1.5B-Instruct, support these predictions.
Figures & tables
| Task | Mode | CE | Square | Hinge | Exp. | Logistic | Sq.-hinge |
| DenseNet121 | |||||||
| Apple Disease | Direct | 60.64 ±2.63 | 61.10 ±1.85 | 58.81 ±2.46 | 60.18 ±1.21 | 60.46 ±3.09 | 62.69 ±4.86 |
| Warm-up | – | 61.10 ±1.85 | 59.91 ±2.60 | 61.03 ±1.92 | 60.54 ±2.02 | 62.69 ±4.86 | |
| Rice Disease | Direct | 72.45 ±6.57 | 71.43 ±7.23 | 70.92 ±7.66 | 70.61 ±6.81 | 71.63 ±7.30 | 71.53 ±7.30 |
| Warm-up | – | 74.90 ±6.31 | 73.98 ±6.26 | 74.90 ±6.41 | 71.63 ±7.30 | 75.71 ±8.61 | |
| APTOS Sev./Mod. | Direct | 82.75 ±0.95 | 83.83 ±1.41 | 85.10 ±0.76 | 83.68 ±1.66 | 84.18 ±1.37 | 84.20 ±1.12 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Source | Binary construction |
|---|---|---|
| ImageNet-LT Dogs | ImageNet-LT split files with ImageNet images | Tail dog classes versus sibling dog classes. |
| Dog Breeds | Stanford Dogs | Spitz, husky, and malamute versus Eskimo dog and Samoyed. |
| CRJ700/CRJ900 | FGVC-Aircraft | Bombardier CRJ700 versus Bombardier CRJ900. |
| APTOS Sev./Mod. | APTOS diabetic-retinopathy fundus images | Severe-or-worse retinopathy versus moderate retinopathy. |
| DeepWeeds P/S | DeepWeeds | Parthenium weed versus Siam weed, with low-shot training construction. |
| Apple Disease | D-KAP apple leaf disease archive | Apple scab versus apple rot, with low-shot training construction. |
| Task | Source | Binary construction |
|---|---|---|
| GoEmotions Annoyance | GoEmotions, simplified | Annoyance label versus all other examples. |
| GoEmotions Fear | GoEmotions, simplified | Fear label versus all other examples. |
| ANLI R1 Ent./Neu. | ANLI Round 1 prepared split | Entailment versus neutral. |
| ANLI R3 Con./Neu. | ANLI Round 3 prepared split | Contradiction versus neutral. |
| ANLI R3 Ent./Neu. | ANLI Round 3 prepared split | Entailment versus neutral. |
| SNLI High-Overlap | SNLI prepared high-overlap split | Entailment versus neutral on a high-overlap slice. |
| CV task | Train ( ; pos./neg.) | Val. ( ; pos./neg.) | Test ( ; pos./neg.) |
|---|---|---|---|
| ImageNet-LT Dogs | ( ) | ( ) | ( ) |
| Dog Breeds | ( ) | ( ) | ( ) |
| CRJ700/CRJ900 | ( ) | ( ) | ( ) |
| APTOS Sev./Mod. | ( ) | ( ) | ( ) |
| DeepWeeds P/S | ( ) | ( ) | ( ) |
| Apple Disease | ( ) | ( ) | ( ) |
| NLP task | Train ( ; pos./neg.) | Val. ( ; pos./neg.) | Test ( ; pos./neg.) |
|---|---|---|---|
| GoEmotions Annoy. | ( ) | ( ) | ( ) |
| GoEmotions Fear | ( ) | ( ) | ( ) |
| ANLI R1 E/N | ( ) | ( ) | ( ) |
| ANLI R3 C/N | ( ) | ( ) | ( ) |
| ANLI R3 E/N | ( ) | ( ) | ( ) |
| SNLI High-Overlap | ( ) | ( ) | ( ) |
| Regime | Models/tasks | Optimization settings |
|---|---|---|
| CV, ImageNet-LT | ResNet50 on ImageNet-LT Dogs | epochs, batch/eval batch , SGD with momentum , learning rate , weight decay , cosine annealing, positive batch fraction , CE-to-AUC switch LR scale . |
| CV, other ResNet50 | Dog Breeds, CRJ700/CRJ900, APTOS Sev./Mod., DeepWeeds P/S | epochs, batch/eval batch , AdamW [ 85 ] , learning rate , weight decay , cosine annealing, positive batch fraction , CE-to-AUC switch LR scale . |
| CV, DenseNet121 | Apple Disease, Rice Disease, APTOS Sev./Mod., HAM10000 M/N | epochs, batch/eval batch , AdamW, learning rate , weight decay , cosine annealing, positive batch fraction , CE-to-AUC switch LR scale . |
| DistilBERT NLP | Eight NLP tasks in Appendix B.4 | epochs, batch/eval batch , gradient accumulation , AdamW, learning rate , weight decay , cosine schedule with warm-up steps equal to of total optimizer updates, maximum length , positive batch fraction , max gradient norm , CE-to-AUC switch LR scale . |
| Qwen NLP | Same eight NLP tasks | epochs, batch/eval batch , gradient accumulation , AdamW, learning rate , weight decay , cosine schedule with warm-up steps equal to of total optimizer updates, maximum length , positive batch fraction , max gradient norm , CE-to-AUC switch LR scale . |
| Regime | Grid |
|---|---|
| CV | Warm-up ratios . Margins . |
| DistilBERT | Warm-up ratios . Margins . |
| Qwen | Warm-up ratios . Margins . |
| Task / surrogate | Region | Severe projected drift | Log-variance–log- slope |
|---|---|---|---|
| APTOS / direct square | Good | to | to |
| APTOS / direct square | Bad | to | to |
| APTOS / direct exponential | Good | to | to |
| APTOS / direct exponential | Bad | to | to |
| APTOS / direct logistic | Good | to | to |
| APTOS / direct logistic | Bad | to | to |
| Task | Directional R/H/N | Isotropic R/H/N | Mean AUC D/I | Mean D/I | |
|---|---|---|---|---|---|
| ImageNet-LT dog-tail | 16 | 2/6/8 | 0/0/16 | ||
| APTOS | 24 | 6/13/5 | 1/1/22 | ||
| Total | 40 | 8/19/13 | 1/1/38 | — | — |
| Multiclass task | CE | Direct hinge | Warm-up hinge | Direct squared-hinge | Warm-up squared-hinge |
|---|---|---|---|---|---|
| Aircraft Airbus-6 LT | 60.92 | 61.43 | 61.61 | 61.68 | 60.91 |
| CIFAR-100 Medium-Mammals-5 LT | 81.08 | 81.13 | 82.91 | 83.62 | 84.44 |
| CIFAR-100 Vehicles-10 LT | 86.47 | 88.30 | 88.64 | 87.97 | 89.08 |
| Multilabel task | BCE | Direct hinge | Warm-up hinge | Direct squared-hinge | Warm-up squared-hinge |
|---|---|---|---|---|---|
| Competition-5 | 80.86 | 82.75 | 83.40 | 81.51 | 83.09 |
| Cardiac/Fluid-4 | 80.76 | 81.14 | 82.13 | 79.96 | 81.44 |
| Consolidation/Cardiac-2 | 51.52 | 57.28 | 56.09 | 57.08 | 59.69 |
| Dataset | Metric | CE | Hinge direct | Hinge warm-up | Sq.-hinge direct | Sq.-hinge warm-up |
|---|---|---|---|---|---|---|
| iNaturalist | Macro-OVO-AUC | 96.64 | 92.45 | 99.70 | 89.18 | 99.74 |
| ImageNet-LT | Macro-OVO-AUC | 98.50 | 97.87 | 98.45 | 97.69 | 98.91 |
| iNaturalist | Macro-OVR-AUC | 97.70 | 88.72 | 97.31 | 79.94 | 98.13 |
| ImageNet-LT | Macro-OVR-AUC | 94.80 | 93.21 | 94.81 | 92.95 | 96.08 |
| Task–model setting | Deep-AUC | Ours |
|---|---|---|
| Apple Disease–DenseNet121 | 55.56 | 62.69 |
| Rice Disease–DenseNet121 | 77.76 | 75.71 |
| APTOS Sev./Mod.–DenseNet121 | 82.12 | 84.20 |
| HAM10000 M/N–DenseNet121 | 97.16 | 97.70 |
| ImageNet-LT Dogs–ResNet50 | 94.56 | 95.83 |
| Dog Breeds–ResNet50 | 85.74 | 87.00 |