Natural Language Understanding on Downstream Benchmarks (HellaSwag, SIQA, PIQA, OBQA, WinoGrande, RACE)
42.96HellaSwagFlashMHF
Evaluation Results
| Method | Links | |||||||
|---|---|---|---|---|---|---|---|---|
| FlashMHFScale=1.3B, Head Dimension (dh)=128, Evaluation Protocol=Zero-shot2025.12 | 42.96 | 44.17 | 68.44 | 27.8 | 54.46 | 22.26 | 43.35 | |
| FlashMHFScale=1.3B, Head Dimension (dh)=256, Evaluation Protocol=Zero-shot2025.12 | 41.8 | 42.32 | 68.88 | 27.6 | 53.35 | 22.01 | 42.66 | |
| FlashMHFScale=1.3B, Head Dimension (dh)=64, Evaluation Protocol=Zero-shot2025.12 | 39.8 | 42.17 | 67.08 | 27.2 | 51.3 | 22.62 | 41.7 | |
| BaselineScale=1.3B, Evaluation Protocol=Zero-shot2025.12 | 39.47 | 41.76 | 67.3 | 27.6 | 52.64 | 21.73 | 41.75 | |
| FlashMHFScale=370M, Head Dimension (dh)=128, Evaluation Protocol=Zero-shot2025.12 | 33.97 | 40.63 | 66.27 | 27.8 | 52.41 | 21.8 | 40.48 | |
| FlashMHFScale=370M, Head Dimension (dh)=256, Evaluation Protocol=Zero-shot2025.12 | 33.6 | 39.71 | 64.85 | 27.3 | 52.17 | 21.94 | 39.92 | |
| MH-FFNScale=370M, Evaluation Protocol=Zero-shot2025.12 | 33.45 | 41.3 | 65.18 | 27.6 | 51.46 | 21.87 | 40.14 | |
| BaselineScale=370M, Evaluation Protocol=Zero-shot2025.12 | 33.2 | 40.94 | 64.53 | 27.5 | 51.7 | 21.66 | 39.92 | |
| FlashMHFScale=370M, Head Dimension (dh)=64, Evaluation Protocol=Zero-shot2025.12 | 32.57 | 41.5 | 65.34 | 27.6 | 53.35 | 21.59 | 40.32 | |
| Parametric KVScale=370M, Evaluation Protocol=Zero-shot2025.12 | 27.85 | 39.61 | 60.5 | 27.2 | 51.78 | 21.67 | 38.1 |