English

Selective Neuron Amplification in Transformer Language Models

Machine Learning 2026-05-12 v2 Computation and Language

Abstract

Large language models often fail on tasks they seem to already understand. In our experiments, this appears to be less about missing knowledge and more about certain internal circuits not being strongly activated during inference. We explore Selective Neuron Amplification, which increases the influence of task relevant neurons without changing the model's parameters. The method works at inference time and does not permanently alter the model. SNA helps mainly when the model is uncertain, while having low effect when the model is already confident. This suggests that some model failures are due to weak activation rather than lack of capability.

Keywords

Cite

@article{arxiv.2604.07098,
  title  = {Selective Neuron Amplification in Transformer Language Models},
  author = {Ryyan Akhtar and Payal Pahwa and Monika Arora},
  journal= {arXiv preprint arXiv:2604.07098},
  year   = {2026}
}

Comments

11 pages, 3 figures. Preprint. Code and experiments conducted independently

R2 v1 2026-07-01T11:59:20.439Z