Reduced Matrix Multiplication: Input-Adaptive Matrix-Product Reduction for LLM Inference

Zixuan Lan, Yanhong Li, Jiawei Zhou


Abstract
Transformer-based language models achieve strong performance but incur substantial inference cost due to repeated high-dimensional matrix multiplications. We propose Reduced Matrix Multiplication (RMM), a training free, input-adaptive inference method that reduces Transformer matrix products by selecting informative slices along their contraction dimensions, without modifying model weights. Under a simple retention-ratio control, RMM provides a smooth and predictable accuracy–efficiency trade-off.Across language models ranging from 1B to 70B parameters, we find that reduction tolerance depends on the model family, task, component, and retention ratio, although it often improves with model scale. Under moderate reduction, RMM remains robust across the evaluated discriminative, autoregressive-generation, and long-context settings. We further show that the same principle extends to multimodal vision–language inference. Mechanistic ablations reveal a structural asymmetry within Transformers: attention-side computations are substantially more reducible than MLP components. Finally, wall-clock benchmarks with custom kernels on an NVIDIA A100 show that these computational savings can translate into practical runtime gains, especially at longer sequence lengths. Together, these results position RMM as a scalable direction for input-adaptive inference-time optimization.
Anthology ID:
2026.findings-emnlp.1322
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
28024–28048
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1322/
DOI:
10.18653/2026.findings-emnlp.1322
Bibkey:
Cite (ACL):
Zixuan Lan, Yanhong Li, and Jiawei Zhou. 2026. Reduced Matrix Multiplication: Input-Adaptive Matrix-Product Reduction for LLM Inference. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 28024–28048, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Reduced Matrix Multiplication: Input-Adaptive Matrix-Product Reduction for LLM Inference (Lan et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1322.pdf