Skip to content

Author

Jun-Jie Hu

We have 3 of 21 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#machine learning Preprint Oct 2026

Universal Test-Time Training

Recent Test-Time Training (TTT) architectures compress context into fast weights that are updated online and queried as memory. Existing TTT designs keep this memory private to each layer: it recurs only over time, and depth merely indexes L separate memories. We argue that memory ownership need not be tied to depth, a...

Ze-Fan Cai, Qin-Zhe Hu, Ziqiao Ma et al. · 1 citation
Jul 2026

Multi-Head Attention Residuals

Multi-Head Attention Residuals (MHAR) is introduced: the routing query is reshaped into H per-subspace heads, each with its own softmax over the depth history, and a direct probe of the trained queries confirms that learned subspace disagreement is the underlying driver.

Cheng Luo, Zefan Cai, Jun-Jie Hu · 1 citation

Test-Time Training with Next-Token Prediction

Test-Time Training with Next-Token Prediction (TTT-NTP), a drop-in fast-weight adaptation method for pretrained LLMs that instead supervises updates using the model's own next contextual hidden state, while preserving commonsense and knowledge performance.

Xuan Ouyang, Zefan Cai, Junjie Hu · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.