CAFE: Confining Feature Distillation to Language-Routed Adapters for Compact Multilingual Automatic Speech Recognition
Organizations: Tether AI Research
Abstract
Feature distillation is usually expected to add to logit distillation. We find that this expectation does not hold for two common feature losses in compact multilingual automatic speech recognition (ASR), and analyze why through their gradients. The gradient of a projection-based loss is orthogonal to the task gradient, while that of Centered Kernel Alignment (CKA), a similarity-based loss, is weakly aligned with it. CKA lowers the word error rate (WER) on non-English languages, but at the expense of data-rich English. Three loss-level remedies (one trained, two probed) all fall short, and the CKA gradient overlaps with the subspace English uses in the shared encoder by 97.5% to 99.5% (about 81% by chance). Guided by this diagnosis, we propose CAFE (Confined-Adapter Feature distillation), which confines the CKA gradient to adapters that English bypasses, addressing the trade-off in the parameters rather than in the loss. On the Open ASR Leaderboard, CAFE lowers the average WER from 8.55% to 8.02%, beats an adapter-only capacity control at 8.51%, and makes English more stable across checkpoints. At 254M parameters, it trails the 0.6B teacher by 2 WER points and is competitive with much larger models across 48 test sets in 24 languages. We use 564k hours of training data and state-of-the-art models to keep the findings relevant to practice.
Figures & tables
| feature term | ||
| VkD variant | ||
| VkD w/o norm | ||
| CKA |
| System | EN-7 | ML-9 | Avg |
| Baseline | 9.85 | 7.24 | 8.55 |
| VkD variant | 10.18 | 7.22 | 8.70 |
| CKA | 11.25 | 6.83 | 9.04 |
| Lang-gated CKA | 11.21 | 7.18 | 9.19 |
| English (7 sets) | Multilingual (9 sets) | Avg (EN, ML) | |||||||
| system | Lang. | Adapter | feature-KD | mean std | CA | mean std | CA | mean std | CA |
| Baseline | 25 | – | – | 10.33 0.82 | 9.85 | 7.59 0.20 | 7.24 | 8.96 0.39 | 8.55 |
| CKA | 25 | – | CKA | 11.42 0.67 | 11.25 | 7.22 0.12 | 6.83 | 9.32 0.33 | 9.04 |
| Capacity | 25 | ✓ | – | 9.98 0.53 | 9.67 | 7.77 0.11 | 7.34 | 8.88 0.27 | 8.51 |
| CAFE w/o sg | 25 | ✓ | CKA (no sg) | 10.75 0.69 | 10.44 | 7.37 0.12 | 6.95 | 9.06 0.32 | 8.70 |
| CAFE (ours) | 25 | ✓ | CKA (confined) | 9.32 0.32 | 9.03 | 7.43 0.13 | 7.00 | 8.38 0.19 | 8.02 |
| CAFE | Baseline | |
| Params | 254.45M | 253.95M |
| RTFx bs1 | 212.9 6.9 | 221.3 5.2 |
| RTFx bs32 | 2602.3 68.9 | 2656.6 79.8 |
| P50 lat. (s) | 0.089 0.001 | 0.088 0.000 |
| Peak mem (MB) | 2593.4 | 2592.5 |
| Parakeet (teacher) | Whisper medium | omniASR-LLM 300M v2 | OWSM- CTC v4 | Qwen3-ASR 0.6B | omniASR- CTC-1B v2 | Whisper small | omniASR-CTC 300M v2 | OWSM v4 small | CAFE (Ours) | CAFE- 8lang | |
| Params (M) | 627 | 764 | 1,632 | 1,011 | 782 | 976 | 242 | 326 | 370 | 254 | 254 |
| Number of languages | 25 | 100 | 1600 | 75 | 30 | 1600 | 100 | 1600 | 75 | 25 | 8 |
| RTFx bs1 | 198 | 21 | 10 | 590 | 18 | 513 | 35 | 848 | 39 | 213 | 213 |
| RTFx bs32 | 2,520 | 386 | 117 | 1,282 | 310 | 2,142 | 546 | 3,405 | 37 | 2,602 | 2,602 |
| P50 lat. bs1 (s) | 0.095 | 0.704 | 1.990 | 0.033 | 1.070 | 0.037 | 0.417 | 0.024 | 0.501 | 0.089 | 0.089 |
| Peak mem (MB) | 3,310 | 6,932 | 17,310 | 25,803 | 6,266 | 16,033 | 2,617 | 14,802 | 2,505 | 2,593 | 2,593 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Parakeet (teacher) | Whisper medium | omniASR-LLM 300M v2 | OWSM- CTC v4 | Qwen3-ASR 0.6B | omniASR- CTC-1B v2 | Whisper small | omniASR-CTC 300M v2 | OWSM v4 small | CAFE (ours) | CAFE- 8lang | |
| Number of languages | 25 | 100 | 1600 | 75 | 30 | 1600 | 100 | 1600 | 75 | 25 | 8 |
| Params (M) | 627 | 764 | 1,632 | 1,011 | 782 | 976 | 242 | 326 | 370 | 254 | 254 |
| RTFx bs1 | 198 | 21 | 10 | 590 | 18 | 513 | 35 | 848 | 39 | 213 | 213 |
| RTFx bs32 | 2,520 | 386 | 117 | 1,282 | 310 | 2,142 | 546 | 3,405 | 37 | 2,602 | 2,602 |
| P50 lat. bs1 (s) | 0.095 | 0.704 | 1.990 | 0.033 | 1.070 | 0.037 | 0.417 | 0.024 | 0.501 | 0.089 | 0.089 |
| Peak mem (MB) | 3,310 | 6,932 | 17,310 | 25,803 | 6,266 | 16,033 | 2,617 | 14,802 | 2,505 | 2,593 | 2,593 |
| Lang. | Utterances | Hours | % |
| en | 62,165,596 | 200,020.7 | 35.46% |
| de | 4,702,308 | 26,909.2 | 4.77% |
| fr | 4,511,551 | 25,603.1 | 4.54% |
| es | 4,354,218 | 23,614.3 | 4.19% |
| it | 3,183,254 | 22,482.3 | 3.99% |
| pl | 2,774,073 | 20,770.6 | 3.68% |
| Lang. Samples Hours % VoxPopuli (23 langs, 373,686 h, 66.24%) en 3,053,532 24,103.2 6.45% de 2,979,597 23,198.5 6.21% fr 2,955,798 22,777.4 6.10% it 2,814,827 21,788.2 5.83% es 2,788,598 21,348.5 5.71% pl 2,652,542 20,556.1 5.50% nl 2,474,233 18,882.2 5.05% cs 2,333,712 18,125.7 4.85% ro 2,296,650 17,717.2 4.74% el 2,303,443 17,563.8 4.70% hu 2,278,164 17,484.2 4.68% pt 2,282,071 17,334.2 4.64% bg 2,234,897 17,095.7 4.57% sv 2,054,815 15,741.4 4.21% fi 1,799,765 13,894.5 3.72% da 1,763,822 13,356.2 3.57% lt 1,754,809 13,331.6 3.57% lv 1,595,156 12,292.1 3.29% sk 1,476,106 11,398.3 3.05% sl 1,433,486 10,979.5 2.94% et 1,281,814 9,788.0 2.62% hr 1,004,410 7,817.1 2.09% mt 919,343 7,112.8 1.90% YODAS (23 langs, 104,306 h, 18.49%) en 39,801,853 100,102.2 95.97% ru 1,873,248 1,732.4 1.66% es 947,615 779.4 0.75% fr 557,519 715.5 0.69% de 415,050 496.2 0.48% pt 727,788 262.1 0.25% it 86,580 120.2 0.12% nl 71,477 37.8 0.04% pl 28,678 21.1 0.02% uk 9,479 10.2 0.01% hu 6,530 8.5 0.01% ro 2,303 5.8 0.01% fi 4,637 3.8 0.00% cs 4,185 3.8 0.00% sv 3,189 1.9 0.00% bg 1,533 1.8 0.00% el 514 1.0 0.00% lt 390 0.6 0.00% da 656 0.6 0.00% et 513 0.5 0.00% sk 287 0.5 0.00% hr 261 0.2 0.00% lv 75 0.1 0.00% | Lang. Samples Hours % MLS (8 langs, 50,686 h, 8.99%) en 10,808,037 44,659.7 88.11% de 469,940 1,966.5 3.88% nl 374,287 1,554.2 3.07% fr 258,213 1,076.6 2.12% es 220,701 917.7 1.81% it 59,623 247.4 0.49% pt 37,533 161.0 0.32% pl 25,043 103.6 0.20% Common Voice (24 langs, 6,492 h, 1.15%) en 1,500,870 2,225.7 34.28% de 837,721 1,248.0 19.22% fr 740,021 1,033.6 15.92% es 397,304 568.7 8.76% it 222,224 326.5 5.03% ru 135,348 193.1 2.97% lv 135,129 176.4 2.72% pt 121,417 136.0 2.09% hu 72,866 108.4 1.67% pl 67,810 89.8 1.38% nl 66,017 81.8 1.26% uk 53,832 67.5 1.04% cs 40,203 50.6 0.78% et 18,482 33.8 0.52% sv 29,351 31.0 0.48% sk 25,398 30.3 0.47% lt 14,258 20.0 0.31% ro 12,363 13.6 0.21% el 11,317 12.9 0.20% bg 7,950 11.8 0.18% sl 10,168 10.8 0.17% fi 7,316 9.1 0.14% da 6,219 7.5 0.12% mt 4,368 5.5 0.08% Monolingual English corpora Libri-Light (en, 23,735 h, 4.21%) NSC (en, 3,098 h, 0.55%) Europarl-ASR (en, 1,016 h, 0.18%) LibriSpeech (en, 961 h, 0.17%) AMI (en, 77 h, 0.01%) VCTK (en, 41 h, 0.01%) |
| Test set / Language | Samples | Hours | Test set | Samples | Hours |
| From Open ASR Leaderboard | From the official dataset’s test set | ||||
| English short-form (7) | Multilingual LibriSpeech (8) | ||||
| ami | 11,653 | 8.54 | en | 3,769 | 15.55 |
| earnings22 | 2,737 | 5.43 | de | 3,394 | 14.29 |
| gigaspeech | 19,898 | 35.35 | nl | 628 | 2.51 |
| LibriSpeech.clean | 2,620 | 5.40 | pl | 284 | 1.17 |
| system | EN-7 | ML-9 | Avg | |
| CAFE | 5 | 9.10 | 7.02 | 8.06 |
| 7 | 9.03 | 7.00 | 8.02 | |
| 10 | 9.05 | 7.03 | 8.04 | |
| Baseline | 5 | 9.85 | 7.25 | 8.55 |
| 7 | 9.85 | 7.24 | 8.55 | |
| 10 | 9.99 | 7.17 | 8.58 |
| Overall | English | Non-English |
| 98.9% | 97.6% | 99.9% |
| Baseline | CAFE (w/ label) | CAFE + gate (w/o label) | Adapters always on | Adapters never on | |
| EN-7 | 9.85 | 9.03 | 9.11 | 15.95 | 9.03 |
| FLEURS (5) | 7.22 | 7.00 | 7.04 | 7.00 | 8.46 |
| MLS (4) | 7.27 | 7.01 | 7.01 | 7.01 | 7.62 |
| ML-9 | 7.24 | 7.00 | 7.03 | 7.00 | 8.04 |
| Avg (EN, ML) | 8.55 | 8.02 | 8.07 | 11.48 | 8.54 |
| EN-7 | WER | 95% CI | Perm. | Ckpt | Wilcoxon |
| CKA vs Baseline | 0/7 | 0.016 | |||
| CAFE vs Baseline | 7/7 | 0.016 | |||
| CAFE vs Capacity | 6/7 | 0.047 | |||
| ML-9 | |||||
| CKA vs Baseline | 7/7 | 0.016 | |||
| CAFE vs Baseline | 0.0028 | 5/7 | 0.156 |
Feature-Based Knowledge Distillation
Feature-based knowledge distillation transfers internal representations from large teacher models to smaller students for efficient learning.
See all 49 papers