TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models
Not on the map. Why?
Abstract (arXiv)
We study whether tiny decoder-only language models benefit from feed-forward layers that directly multiply learned feature projections. TriPLU, a Trilinear Product Linear Unit, replaces the usual gated FFN branch with a product-only degree-3 branch that multiplies three projected streams coordinatewise. In a character-level TinyStories 1M-byte prefix study, TriPLU reaches a mean best validation loss of 1.0637, compared with 1.1017 for closely matched SwiGLU, 1.0780 for a degree-4 product control, and 1.1026 for a degree-2 control. In train-only Byte-BPE experiments, TriPLU also lowers validation and heldout bits per byte on TinyStories and WikiText-2 raw under low-learning-rate settings, with PMI-slice evidence suggesting gains on seen middle- and high-PMI adjacent-token pairs. Constant-learning-rate diagnostics show that product-branch normalization can reduce the high-learning-rate best-checkpoint gap, although final BPB still degrades under hot schedules. The resulting claim is deliberately narrow: direct product FFNs can improve fixed-budget small-model loss in specific low-compute regimes, but the branch is optimization-sensitive and does not establish FLOP-normalized efficiency, scaling behavior, or broad LLM performance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 22 references that are in the map. First posted on arXiv 17 Jun 2026.
Builds on
14 of 22 references.
References · 14
- 2017Attention Is All You Needcited by 60,361
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2020GLU Variants Improve Transformercited by 1,804
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
- 2018Neural Arithmetic Logic Unitscited by 70
- 2020Neural Arithmetic Unitscited by 28
- 2020Neural Power Unitscited by 7
- 2021A Primer for Neural Arithmetic Logic Modulescited by 7
- 2024Dissecting Multiplication in Transformers: Insights into LLMscited by 7
- 2026PowLU: An Activation Function for Stable Pre-Training of LLMscited by 4
- 2026More Expressive Feedforward Layers: Part I. Token-Adaptive Mixing of Activationscited by 3
- 2025Sample-Efficient Language Modeling with Linear Attention and Lightweight Enhancementscited by 1
Built on by
Not cited yet.