TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models

We ask whether tiny decoder-only language models benefit from feed-forward layers that multiply learned feature projections directly. TriPLU (Trilinear Product Linear Unit) is a product-only degree-3 FFN branch that multiplies three projected streams coordinatewise. In a character-level TinyStories 1M-byte prefix study, TriPLU reaches mean best validation loss 1.0637, versus 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 compatible with 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, while final BPB still degrades under hot schedules. The claim is 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, or broad LLM performance.

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-06-17
DOI
https://doi.org/10.5281/zenodo.20725234
Primary Topic
Topic Modeling
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models

He Zhang
Zenodo (CERN European Organization for Nuclear Research)
Topic Modeling
article

TriPLU: Bypassing the Gate with Direct Trilinear Product FFNs in Tiny Language Models

He Zhang
article en

Abstract

We ask whether tiny decoder-only language models benefit from feed-forward layers that multiply learned feature projections directly. TriPLU (Trilinear Product Linear Unit) is a product-only degree-3 FFN branch that multiplies three projected streams coordinatewise. In a character-level TinyStories 1M-byte prefix study, TriPLU reaches mean best validation loss 1.0637, versus 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 compatible with 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, while final BPB still degrades under hot schedules. The claim is 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, or broad LLM performance.

Zenodo (CERN European Organization for Nuclear Research)
Openalex Percentile: Top 5%
Topic Modeling
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.