PreprintCited by 6Topic ModelingNeural Networks and ApplicationsNatural Language Processing Techniques

The DOI matches its DataCite record; 2 of the 2 references it cites were found in a registry or an index.

HGRN2: Gated Linear RNNs with State Expansion

Open access

Abstract

Hierarchically gated linear RNN (HGRN, \citealt{HGRN}) has demonstrated competitive training speed and performance in language modeling while offering efficient inference. However, the recurrent state size of HGRN remains relatively small, limiting its expressiveness. To address this issue, we introduce a simple outer product-based state expansion mechanism, which significantly enlarges the recurrent state size without introducing any additional parameters. This enhancement also provides a linear attention interpretation for HGRN2, enabling hardware-efficient training. Our extensive experiments verify the advantage of HGRN2 over HGRN consistently across different settings and competitive with other recurrent models.

Overview

Integrity signals

Registry

DataCite · 10.48550/arxiv.2404.07904

Cites

0 citation errors · 0 not checkable

Statistics

Recomputing p-values needs the paper’s full text.

Verification

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • To correct biased rounding error in flash attention, Kimi K3 adopts the method of [98] and keeps the attention output in FP32 during training.

    “To correct the biased rounding error that arises in flash attention, we adopt the method of [98] and keep the attention output in FP32 during training.”

    From Kimi K3: Open Frontier Intelligence (2026) · cites this paper as [98]

  • Conceptually the closest models to xLSTM are Retention, RWKV, and HGRN2.

    “Conceptually the closest models to xLSTM are Retention (Sun et al., 2023), RWKV (Peng et al., 2023(Peng et al., , 2024)), and HGRN2 (Qin et al., 2024).”

    From xLSTM: Extended Long Short-Term Memory (2024) · cites this paper as Qin et al., 2024

  • All presented sequence models in this work (Moneta, Memora, Yaad) as well as HGRN2 are placed outside the class of models that internally optimize regression loss, due to their different attentional bias.

    “Furthermore, all presented sequence models in this work (e.g., Moneta, Memora, Yaad) as well as models like HGRN2 (Qin et al. 2024) are placed outside of this class of models, due to their different attentional bias.”

Citations

Cites

The publisher hasn’t deposited this paper’s reference list.

Cited by

Sorted by citations

Metadata

Authors
Qin, Z., Yang, S., Sun, W., Shen, X., Li, D., Sun, W., & Zhong, Y.
Affiliations
MIT Computer Science and Artificial Intelligence Laboratory (US)
Venue
arXiv
Publisher
arXiv
Published
2024
ISSN
2331-8422 (ISSN-L)
OpenAlex
W4394782164
Cited by
1
References
0
Citations by year
2025: 1
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion, other-oa)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Neural Networks and Applications
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
Keywords
state expansion, outer product, linear attention, language modeling, parameter-efficient inference

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review