← Search

Junxuan Wang

3 accepted papers

2026

Dimensional Collapse in Transformer Attention Outputs: A Challenge for Sparse Dictionary Learning

ICML 2026poster

Transformer architectures, and their attention mechanisms in particular, form the foundation of modern large language models. While transformer models are widely believed to operate in high-dimensional hidden spaces, we show that attention outputs are confined to a surprisingly low-dimensional subsp…

Cited by 0SourceScholar
2026

Towards Understanding the Nature of Attention with Low-Rank Sparse Decomposition

ICLR 2026poster

We propose Low-Rank Sparse Attention (Lorsa), a sparse replacement model of Transformer attention layers to disentangle original Multi Head Self Attention (MHSA) into individually comprehensible components. Lorsa is designed to address the challenge of \textit{attention superposition} to understand…

Cited by 0SourcecodeScholar
2025

Towards Universality: Studying Mechanistic Similarity Across Language Model Architectures

ICLR 2025poster

The hypothesis of \textit{Universality} in interpretability suggests that different neural networks may converge to implement similar algorithms on similar tasks. In this work, we investigate two mainstream architectures for language modeling, namely Transformers and Mambas, to explore the extent of…

Cited by 3SourcePDFScholar