paper-with-me

Papers

Distributed Specialization: Rare-Token Neurons in Large Language Models

2025-09-25 · Jing Liu, Haozheng Wang, Yueheng Li arxiv

Large language models (LLMs) struggle with representing and generating rare tokens despite their importance in specialized domains. We investigate whether LLMs develop internal specialization mechanisms through discrete modular architectures or distributed parameter-level differentiation. Through systematic analysis of final-layer MLP neurons across multiple model families, we discover that rare-token processing emerges via \textit{distributed specialization}: functionally coordinated but spatially distributed subnetworks that exhibit three distinct organizational principles. First, we identify a reproducible three-regime influence hierarchy comprising highly influential plateau neurons(also termed as rare-token neurons), power-law decay neurons, and minimally contributing neurons, which is absent in common-token processing. Second, plateau neurons demonstrate coordinated activation patterns (reduced effective dimensionality) while remaining spatially distributed rather than forming discrete clusters. Third, these specialized mechanisms are universally accessible through standard attention pathways without requiring dedicated routing circuits. Training dynamics reveal that functional specialization emerges gradually through parameter differentiation, with specialized neurons developing increasingly heavy-tailed weight correlation spectra consistent with Heavy-Tailed Self-Regularization signatures. Our findings establish that LLMs process rare-tokens through distributed coordination within shared architectures rather than mixture-of-experts-style modularity. These results provide insights for interpretable model editing, computational efficiency optimization, and understanding emergent functional organization in transformer networks.

📄 PDF Abstract BibTeX arXiv:2509.21163

Code (0)

등록된 구현이 없습니다.

Tasks

Computational Efficiency

Similar Papers 제목 키워드 기반

No Clustering, No Routing: How Transformers Actually Process Rare Tokens

2025-08-30 · Jing Liu arxiv

Large language models struggle with rare token prediction, yet the mechanisms driving their specialization remain unclear. Prior work identified specialized ``plateau'' neurons for rare tokens following distinctive three…

Emergent Specialization: Rare Token Neurons in Language Models

2025-05-19 · Jing Liu, Haozheng Wang, Yueheng Li

Large language models struggle with representing and generating rare tokens despite their importance in specialized domains. In this study, we identify neuron structures with exceptionally strong influence on language mo…

Circuit-Aware Reward Training: A Mechanistic Framework for Longtail Robustness in RLHF

2025-09-29 · Jing Liu arxiv

Reinforcement Learning from Human Feedback (RLHF) reward models exhibit systematic failures on longtail distributions, leading to reward hacking and misalignment. We propose a mechanistic interpretability framework that …

Reinforcement LearningData Augmentation

Switch-Based Multi-Part Neural Network

2025-04-25 · Surajit Majumder, Paritosh Ranjan, Prodip Roy, Bhuban Padhan

This paper introduces decentralized and modular neural network framework designed to enhance the scalability, interpretability, and performance of artificial intelligence (AI) systems. At the heart of this framework is a…

Edge-computingFederated LearningPrivacy Preserving

Beyond Perplexity: UTF-8 Validity in Byte-aware Language Models

2026-06-12 · Sangwhan Moon, Daisuke Oba, Youmi Ma, Tatsuya Hiraoka 외 arxiv

Byte-level tokenization enables language models to handle any Unicode input, but models can generate invalid UTF-8 sequences when encountering rare or unseen characters. We investigate the relationship between training s…