MHLA: Restoring Expressivity of Linear Attention via Token-Level Multi-Head

Benchmark Model Rank Results
text-to-video-generation-on-vbenchWan-MHLA-H (Wan2.1-1.3B)#20VBench Total Score: 83.82
image-classification-on-imagenetMHLA-VLT-S (27M)#329Top 1 Accuracy: 84.6%