Nathanael Carraz Rakotonirina

MemoryPrompt: A Light Wrapper to Improve Context Tracking in Pre-trained Language Models

Transformer-based language models (LMs) track contextual information through large, hard-coded input windows. We introduce MemoryPrompt, a leaner approach in which the LM is complemented by a small auxiliary recurrent network that passes information to the LM by prefixing its regular input with a se…