PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
June 23, 20260 citationsOpen Access

HFFN-Transformer: Integrating Exact Sparse Feed-Forward Kernels into Transformer Inference

View Full Paper
MPMaxime Parker

Key Points

  • This research aims to determine the effectiveness of Hidden-FFN in full Transformer inference scenarios.
  • Introduced a runtime evaluation that replaces Transformer FFN sublayers with HFFN.
  • Maintained attention, embeddings, normalization, residuals, and language-model head in the setup.
  • Evaluated on 17 toy Transformer models with varying configurations and sequence lengths.
  • HFFN showed significant performance improvements in specific Transformer models.
  • Evaluation conducted across 3 dense models and 14 sparse models, showcasing versatility.
  • Results were assessed on an Apple M1 CPU for efficiency.

Abstract

Feed-forward networks (FFNs) dominate a large share of Transformer parameters and inference work. Hidden-FFN (HFFN) was previously introduced as an exact sparse FFN method based on hidden-neuron decompositionand validated on isolated FFN blocks. This paper studies the next question: does HFFN remain useful insidecomplete toy Transformer forward inference?I introduce an evaluation runtime that replaces Transformer FFN sublayers with HFFN while keeping attention,embeddings, normalization, residuals, and the language-model head in the surrounding PyTorch graph. Thereported benchmark path replaces FFN sublayers only; the tighter fused post-attention entry point and full-graphLLM runtime support are left for future work. The system is evaluated end-to-end on a controlled family of 17toy Transformer models covering 3 dense models, 14 sparse models, and 56 sparse model/sequence configurationsover sequence lengths 1, 8, 16, and 32 on an Apple M1 CPU setup.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Maxime Parker (2026) studied this question.

synapsesocial.com/papers/6a3a21f3111626ef22ab6a3chttps://doi.org/10.5281/zenodo.20788035
Ask AI
Helpful
Bookmark
Share
View Full Paper