In Search of Needles in a 10M Haystack
Recurrent Memory Finds What LLMs Miss
paper addresses the challenge of processing long documents using generative transformer models. To evaluate different approaches, we introduce BABILong, a new benchmark designed to assess model capabilities in extracting and processing distributed facts within extensive texts. Our evaluation, which includes benchmarks for GPT-4 and RAG, reveals that common methods are effective only for sequences up to 10^4 elements. In contrast, fine-tuning GPT-2 with recurrent memory augmentations enables it to handle tasks involving up to 10^7 elements. This achievement marks a substantial leap, as it is by far the longest input processed by any open neural network model to date, demonstrating a significant improvement in the processing capabilities for long sequences
In Search of Needles in a 10M Haystack: Recurrent Memory Finds What LLMs Miss
Fine-tuning GPT-2 with recurrent memory augmentations enables it to handle tasks involving up to 10M timesteps
https://t.co/mrBij6JGzY
@dooraven @MikhailBurtsev@yurakuratov Here we show that RMT shines on specific memory-intensive tasks. Stay tuned for our future work, where we will outline more applied cases with different Transformers on real-world tasks!
@vrungta@MikhailBurtsev@_akhaliq@yurakuratov No, here we do create a database since the memory is small (fixed to 10 vectors) and updated every timestep. To retrieve from memory we use attention rather than similarity.
@generatorman_ai@MikhailBurtsev@yurakuratov It is a separate line of work but there are similarities. RMT also uses the recurrent approach (but on level of segments) and memory can be viewed as the dynamic state of a turing machine
Две вещи наполняют душу всегда новым и всё более сильным удивлением и благоговением, чем чаще и продолжительнее мы размышляем о них — это звёздное небо надо мной и моральный закон во мне. И.Кант "Критика практического разума"
надеюсь, никогда не забуду лучшего диджейского перформанса, который мне довелось увидеть. публика кричала от восторга, даже не знал, что так бывает. жаль, что я так и не смог понять, кто там диджей