arXiv · 2410.13497
Repetition Neurons: How Do Language Models Produce Repetitions?
Abstract
This paper introduces repetition neurons, regarded as skill neurons responsible for the repetition problem in text generation tasks. These neurons are progressively activated more strongly as repetition continues, indicating that they perceive repetition as a task to copy the previous context repeatedly, similar to in-context learning. We identify these repetition neurons by comparing activation values before and after the onset of repetition in texts generated by recent pre-trained language models. We analyze the repetition neurons in three English and one Japanese pre-trained language models and observe similar patterns across them.
Explore related subjects
Keep this discovery
Tatsuya Hiraoka, Kentaro Inui. 2024-10-17. Repetition Neurons: How Do Language Models Produce Repetitions?. https://arxiv.org/abs/2410.13497
Cite the original work for its findings. Save a collection to share your selection of sources.