Papers for

natural language application teams

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Hybrid models learn better with staged training to improve language tasks

Competing Learning in Hybrid Architectures

Abstract: Many modern language models have adopted hybrid architectures that incorporate both linear-time recurrence and softmax attention. Modern recurrence layers provide a highly expressive mixing mechanism while ensuring high throughput by compressing contextual information into a fixed-size state, but underperform softmax attention in contextual information retrieval. In this paper, we reveal a Competing Learning phenomenon where different kinds of mixers compete to learn the same function, and when any layer learns a function, it suppresses the corresponding learning signal for other layers. Layer attribution analysis reveals that under typical data distributions, modern recurrence learns retrieval functions faster than softmax attention, causing attention to fail to learn retrieval in most cases. Stopping weight updates in recurrence in the middle of training helps attention learn retrieval. Based on these results, we propose a staged training procedure for pre-training hybrid architectures on natural language. Empirical results on models up to 1B parameters show that our method results in substantial improvement in language modeling loss, common-sense reasoning, as well as a wide range of retrieval-related tasks across different context lengths.

Mon 19 OctComputation and Language
The gist
Modern language models often combine two ways to understand context: memory that summarizes past input quickly and attention that searches for important details. The authors found that these two methods compete to learn the same task, but the faster learner can block the other from improving. By pausing the training of the faster memory method halfway, attention can better learn the task it usually misses. This staged training leads to better performance in language understanding and reasoning tasks.
Open → 2610.26830v1