arXiv · 2404.11870
Enhancing Length Extrapolation in Sequential Models with Pointer-Augmented Neural Memory
Abstract
We propose Pointer-Augmented Neural Memory (PANM) to help neural networks understand and apply symbol processing to new, longer sequences of data. PANM integrates an external neural memory that uses novel physical addresses and pointer manipulation techniques to mimic human and computer symbol processing abilities. PANM facilitates pointer assignment, dereference, and arithmetic by explicitly using physical pointers to access memory content. Remarkably, it can learn to perform these operations through end-to-end training on sequence data, powering various sequential models. Our experiments demonstrate PANM's exceptional length extrapolating capabilities and improved performance in tasks that require symbol processing, such as algorithmic reasoning and Dyck language recognition. PANM helps Transformer achieve up to 100% generalization accuracy in compositional learning tasks and significantly better results in mathematical reasoning, question answering and machine translation tasks.
Explore related subjects
Keep this discovery
Hung Le, Dung Nguyen, Kien Do, Svetha Venkatesh, Truyen Tran. 2024-04-18. Enhancing Length Extrapolation in Sequential Models with Pointer-Augmented Neural Memory. https://arxiv.org/abs/2404.11870
Cite the original work for its findings. Save a collection to share your selection of sources.