Memory Decoder: A Pretrained, Plug-and-Play Memory for Large Language Models

## Контекст Large Language Models (LLMs) доказали свою эффективность в решении общих языковых задач. Однако применение этих моделей к специализированным дисциплинам, таким как биомедицина, финансы и право, требует индивидуальной адаптации, которая традиционно требует дорогостоящей полного переноса параметров (Domain Adaptive Pretraining, DAPT). Эта процедура затратна из-за сложности обучения и катастрофической забываемости. Кроме того, Retrieval-Augmented Generation (RAG), хотя и улучшает адаптацию, приводит к значительному затрату ресурсов на вычисления и долгому времени ответа из-за необходимости поиска в больших выборках. Эти ограничения подчеркивают необходимость разработки эффективных способов адаптации LLMs к конкретным сферам без негативных побочных эффектов. ## Метод Memory Decoder представляет собой простую, но эффективную архитектуру в виде небольшого transformer-decoder, который настраивается для имитации поведения внешнего непараметрического ретривера. Алгоритм обучения оптимизирует декодер для точного воспроизведения ответов, полученных из внешней памяти. Эта архитектура отличается тем, что она не требует изменения оригинальной модели и может быть легко интегрирована в любую LLM с использованием одного и того же токенизатора. Такое решение обеспечивает высокую эффективность при адаптации к новым сферах, сохраняя всю мощь предварительно обученной модели. ## Результаты Эксперименты проводились на трех специализированных областях: медицина, финансы и право. Для каждой области были использованы специальные данные, а модели Qwen и Llama были адаптированы с помощью Memory Decoder. Результаты показали существенную улучшение в предсказании текста, снижением perplexity на среднем уровне в 6.17 пунктов по сравнению с оригинальными моделями. Это достигное улучшение подтверждает то, что Memory Decoder добавляет мощь и точность в специализированных темах без необходимости изменять оригинальную модель. ## Значимость Memory Decoder представляет собой важное дополнение к возможностям LLMs, позволяя их эффективно адаптировать к конкретным дисциплинам без изменения основной модели. Эта модель может быть применена в различных сферах, где требуется высокая точность и эффективность, таких как системы поддержки решений в биомедицине, финансовые системы и системы судебного толкования. Важное преимущество Memory Decoder заключается в том, что он обеспечивает элегантное решение проблемы затратных вычислений и зависимости от дорогих сетей памяти, обеспечивая быструю и точную адаптацию. ## Выводы Memory Decoder представляет собой перспективное направление в развитии LLMs, предоставляя новый подход к адаптации моделей к специальным дисциплинам. Будущие исследования б