Train Smarter, Not Harder: Switching Signal-Guided Training in Active Learning
Organizations: Faculty of Data and Decision Sciences, Technion
Abstract
Training strategy, namely whether to retrain from scratch or fine-tune from the previous checkpoint, is an overlooked decision variable in active learning. We show that this choice has exploitable structure: retraining is most useful in early rounds, when each batch can substantially reshape the labeled distribution, while fine-tuning becomes safer once the model trajectory stabilizes. We propose HybridAL, an adaptive training schedule that monitors an online stabilization signal and switches from retraining to fine-tuning after sustained stabilization. Two complementary signals, spectral exponent change (weight-based) and accuracy change Acc (validation-based), span different points on the time-calibration trade-off. Across three encoder backbones and six text-classification tasks (five seeds each), HybridAL keeps endpoint macro-F1 non-inferior to retraining and fine-tuning at a 0.010 margin, saves up to 49% of retraining time, and recovers a substantial fraction of retraining's calibration advantage as measured by negative log-likelihood (NLL). Compared with schedules that switch at a pre-committed round, HybridAL obtains lower NLL at moderate additional cost, showing that trajectory-dependent switching provides a stronger time-calibration trade-off than fixed early switching.
Figures & tables
| Dataset | Domain | C | Size |
| Binary | |||
| IMDb ( Maas et al., 2011 ) | sentiment analysis | 2 | 50,000 |
| SST-2 ( Socher et al., 2013 ) | sentiment analysis | 2 | 68,221 |
| Jigsaw ( Wulczyn et al., 2017 ) | toxicity detection | 2 | 159,571 |
| Multi-class | |||
| TweetEval ( Barbieri et al., 2020 ) | sentiment analysis | 3 | 59,899 |
| Signal | Type | Extra val. pass? | Fire rate | Mean |
| Acc ⋆ | performance | yes | 93% | 9.1 |
| F1 | performance | yes | 90% | 10.0 |
| Loss | performance | yes | 43% | 13.4 |
| Grad. norm | performance | yes | 40% | 8.5 |
| model-based | no | 97% | 7.9 | |
| distance | model-based | no | 0% | ✗ |
Appendix figures & tables27 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Class | Train ( ) | Train (%) | Val. ( ) | Val. (%) | Test ( ) | Test (%) |
| IMDb | negative | 19,750 | 50.00 | 250 | 50.00 | 5,000 | 50.00 |
| positive | 19,750 | 50.00 | 250 | 50.00 | 5,000 | 50.00 | |
| Total | 39,500 | — | 500 | — | 10,000 | — | |
| Jigsaw | non-toxic | 113,978 | 90.42 | 1,443 | 90.41 | 28,856 | 90.42 |
| toxic | 12,082 | 9.58 | 153 | 9.59 | 3,059 | 9.58 | |
| Total | 126,060 | — | 1,596 | — | 31,915 | — |
| DistilBERT | ||||||
| Method | IMDb | AGNews | Jigsaw | SST-2 | TwtEv | Yahoo |
| Retrain | 0.819 0.021 | 0.900 0.004 | 0.881 0.003 | 0.835 0.021 | 0.634 0.023 | 0.667 0.004 ‡§¶⋆∙∘ |
| FineTune | 0.823 0.008 ¶ | 0.904 0.005 | 0.883 0.004 | 0.858 0.018 | 0.626 0.021 | 0.655 0.006 § |
| NewOnly | 0.823 0.006 | 0.899 0.001 | 0.882 0.007 | 0.854 0.009 | 0.552 0.117 | 0.637 0.009 |
| HybridAL ( ) | 0.814 0.014 | 0.902 0.007 | 0.881 0.004 ∘ | 0.844 0.022 | 0.627 0.025 | 0.656 0.004 § |
| HybridAL ( Acc) | 0.827 0.006 | 0.905 0.005 | 0.880 0.005 | 0.857 0.012 | 0.629 0.018 | 0.654 0.005 § |
| Variant | Reference | Lower bound | ||
| Retrain | ✓ | ✓ | ||
| FineTune | ✓ | ✓ | ||
| Acc | Retrain | ✓ | ✓ | |
| Acc | FineTune | ✗ | ✓ | |
| per-cell best | ✗ | ✓ | ||
| Acc | per-cell best | ✗ | ✓ |
| Method | IMDb | AGNews | Jigsaw | SST-2 | TwtEv | Yahoo | Mean |
| DistilBERT | |||||||
| Retrain | 0.426 0.039 ‡¶∙∘⋄△ | 0.320 0.008 ‡§¶⋆∙∘⋄△ | 0.133 0.008 ‡⋆∙⋄ | 0.391 0.029 | 0.814 0.066 ‡¶⋆∙∘⋄△ | 1.108 0.018 ‡§¶⋆∙∘⋄△ | 0.532 0.336 ‡§¶⋆∙∘⋄△ |
| FineTune | 0.614 0.075 | 0.448 0.042 | 0.153 0.016 | 0.460 0.053 | 1.379 0.075 | 1.609 0.030 | 0.777 0.540 |
| NewOnly | 0.501 0.036 ‡∙∘⋄△ | 0.362 0.013 ‡⋆∙∘⋄△ | 0.133 0.007 ‡⋆∙⋄ | 0.435 0.041 | 0.880 0.105 ‡∙∘⋄△ | 1.265 0.046 ‡⋆∙∘⋄△ | 0.596 0.382 ‡¶⋆∙∘⋄△ |
| HybridAL ( ) | 0.641 0.140 | 0.408 0.059 | 0.150 0.023 ∘ | 0.412 0.038 △ | 1.288 0.321 | 1.441 0.192 | 0.724 0.508 ∘ |
| HybridAL ( Acc) | 0.457 0.079 ‡∙∘⋄△ | 0.433 0.023 | 0.156 0.008 | 0.393 0.032 △ | 1.163 0.246 | 1.490 0.175 | 0.682 0.498 ‡∙∘⋄△ |
| Method | IMDb | AGNews | Jigsaw | SST-2 | TwtEv | Yahoo | Mean |
| DistilBERT | |||||||
| Retrain | 526 19 | 690 33 | 719 41 | 834 39 | 886 22 | 1375 44 | 838 272 |
| FineTune | 367 11 †¶⋆⋄△ | 407 17 †¶⋆∘⋄△ | 459 12 †⋆⋄△ | 621 14 †¶⋆∘⋄△ | 564 13 †⋆∘⋄△ | 675 19 †¶⋆∘⋄△ | 515 115 †¶⋆∘⋄△ |
| NewOnly | 153 6 †‡¶⋆∙∘⋄△ | 324 15 †‡¶⋆∙∘⋄△ | 374 13 †‡¶⋆∙∘⋄△ | 122 3 †‡¶⋆∙∘⋄△ | 94 7 †‡¶⋆∙∘⋄△ | 162 12 †‡¶⋆∙∘⋄△ | 205 107 †‡¶⋆∙∘⋄△ |
| HybridAL ( ) | 478 84 | 564 102 | 572 115 † | 780 97 | 687 126 † | 1061 258 † | 690 234 † |
| HybridAL ( Acc) | 510 70 | 512 42 † | 523 21 † | 727 84 † | 765 102 | 1000 189 † | 673 204 † |
| ( ) | Acc ( ) | |||||
| Dataset | max post | % | max post | % | ||
| IMDb | 8.4 | 4.2e-4 | 43 | 17.8 | 3.8e-2 | 83 |
| AG News | 10.7 | 4.4e-4 | 46 | 8.0 | 2.5e-2 | 54 |
| Jigsaw | 6.9 | 2.9e-4 | 31 | 9.0 | 1.6e-2 | 36 |
| SST-2 | 8.4 | 4.6e-4 | 33 | 10.8 | 3.9e-2 | 67 |
| TweetEval | 6.3 | 4.5e-4 | 41 | 12.5 | 6.7e-2 | 80 |
| HybridAL ( ) | HybridAL ( Acc) | |||||||
| Dataset | DistilBERT | BERT | RoBERTa | Mean | DistilBERT | BERT | RoBERTa | Mean |
| IMDb | 12.3 | 8.5 | 4.5 | 8.4 | 15.5 | 23.0 | 14.8 | 17.8 |
| AG News | 11.3 | 8.8 | 12.0 | 10.7 | 8.6 | 8.0 | 7.4 | 8.0 |
| Jigsaw | 11.2 | 5.4 | 4.0 | 6.9 | 7.4 | 9.6 | 10.0 | 9.0 |
| SST-2 | 7.5 | 9.6 | 8.0 | 8.4 | 9.8 | 9.0 | 13.5 | 10.8 |
| TweetEval | 7.5 | 6.6 | 4.7 | 6.3 | 14.0 | ✗ | 11.0 | 12.5 |
| Method | NLL | NLL ( ) | ECE | ECE ( ) | |
| Retrain | 0.97 | 0.532 | 0.517 | 5.4% | 2.6% |
| FineTune | 2.29 | 0.777 | 0.533 | 13.7% | 2.5% |
| NewOnly | 1.20 | 0.596 | 0.560 | 8.0% | 3.3% |
| HybridAL ( ) | 1.97 | 0.722 | 0.532 | 12.1% | 2.8% |
| HybridAL ( Acc) | 1.80 | 0.682 | 0.525 | 11.8% | 3.0% |
| Dataset | Fire rate | Test F1 | ||
| AG News | 25 | 100% | ||
| 50 | 100% | |||
| 100 | 67% | |||
| 200 | 33% | |||
| full ( ) | 100% | |||
| IMDb | 25 | 33% |
| Comparison vs. Retrain | Pre-switch | Post-switch |
| HybridAL ( ) | ||
| HybridAL ( Acc) |
| Signal | Dataset | Pre-switch | Post-switch |
| IMDb | |||
| AG News | |||
| Acc | IMDb | ||
| Acc | AG News |
| Method | Pre-switch | Post-switch | Change |
| HybridAL ( ) | |||
| HybridAL ( Acc) | |||
| Retrain | — | — |
| Round | Retrain | Acc | |
| 5 | |||
| 10 | |||
| 15 | |||
| 20 | |||
| 24 | |||
| Last 5 |
| Test | |||||||
| Signal | IMDb | AGNews | Jigsaw | SST-2 | TwtEv | Yahoo | Mean |
| Retrain | 0.819 0.021 | 0.900 0.004 | 0.881 0.003 | 0.835 0.021 | 0.634 0.023 | 0.667 0.004 | 0.789 0.105 |
| Acc | 0.825 0.009 | 0.904 0.003 | 0.881 0.007 | 0.847 0.023 | 0.633 0.009 | 0.657 0.006 | 0.791 0.109 |
| 0.826 0.006 | 0.901 0.005 | 0.880 0.007 | 0.861 0.010 | 0.633 0.012 | 0.657 0.004 | 0.793 0.109 | |
| F1 | 0.830 0.011 | 0.904 0.003 | 0.880 0.005 | 0.847 0.023 | 0.628 0.011 | 0.656 0.006 | 0.791 0.111 |
| Loss | 0.804 0.022 | 0.901 0.005 | 0.883 0.005 | 0.853 0.020 | 0.636 0.006 | 0.662 0.005 | 0.790 0.107 |
| mean val F1 | mean val NLL | mean time (s) | ||||||||||
| 5e-5 | 0.858 | 0.857 | 0.855 | 0.855 | 0.447 | 0.385 | 0.370 | 0.370 | 290.8 | 318.0 | 331.9 | 322.5 |
| 1e-4 | 0.854 | 0.865 | 0.856 | 0.855 | 0.492 | 0.455 | 0.388 | 0.370 | 246.3 | 284.8 | 313.7 | 320.8 |
| 1.5e-4 | 0.855 | 0.855 | 0.858 | 0.855 | 0.498 | 0.511 | 0.435 | 0.370 | 214.3 | 229.4 | 289.3 | 320.9 |
| 2e-4 | 0.855 | 0.855 | 0.859 | 0.861 | 0.498 | 0.511 | 0.436 | 0.416 | 214.3 | 229.3 | 278.9 | 302.3 |
| Configuration | Method | AG News | IMDb |
| fixed ep | FineTune | ||
| fixed ep | Retrain | ||
| max + ES | FineTune | ||
| max + ES | Retrain |
| Method | AG News | IMDb | Yahoo | |
| Retrain | 50 | |||
| HybridAL | 50 | |||
| Retrain | 100 | |||
| HybridAL | 100 | |||
| Retrain | 200 | |||
| HybridAL | 200 | ∗ |
| Method | AG News | IMDb | Yahoo | |
| Retrain | 16 | |||
| HybridAL | 16 | |||
| Retrain | 32 | |||
| HybridAL | 32 | ∗ | ||
| Retrain | 64 | |||
| HybridAL | 64 |
| Sampler | Method | AG News | IMDb | Yahoo |
| Entropy | Retrain | |||
| Entropy | HybridAL | ∗ | ||
| Random | Retrain | |||
| Random | HybridAL | |||
| BADGE | Retrain | |||
| BADGE | HybridAL | ∗ |
Active Learning
Active learning selects the examples, labels, features, or experiments most useful for improving a model under limited budgets.
See all 296 papers