Дистилляция памяти учителя подняла точность маленьких ИИ-агентов на 27,2 п.п.

Дистилляция памяти учителя подняла точность маленьких ИИ-агентов на 27,2 п.п.

Авторы предложили Agent Memory Distillation (AMD), тренинг-фри (не требующий дообучения) фреймворк, который передаёт структурированные знания от большого агента-учителя маленькому агенту-ученику через иерархическую память. Идея отвечает на конкретную проблему: системы памяти уже показали пользу для ИИ-агентов, но для маленьких языковых моделей это почти не исследовано, потому что такие модели сами по себе плохо генерируют достаточно успешных траекторий действий, из которых можно было бы строить память.

AMD строит три взаимодополняющих типа памяти из успешных траекторий учителя. Workflow-память кодирует стратегии на уровне всей задачи. Subtask-память даёт конкретные примеры поведения на промежуточном уровне детализации, отдельных подзадач. Function-память фиксирует конвенции вызова конкретных функций и типичные ошибки при их использовании. Workflow- и Subtask-память подгружаются заранее, в начале выполнения задачи, а Function-память подключается реактивно, только когда агент уже ошибся при вызове инструмента.

Авторы протестировали AMD на трёх бенчмарках использования инструментов, AppWorld, BFCL V3 и ToolSandbox, используя четыре модели-ученика с 4, 8 млрд параметров и GPT-5-mini в роли учителя. Средний прирост точности составил 27,2 процентного пункта на AppWorld, 11,2 процентного пункта на BFCL V3 и 3,4 процентного пункта на ToolSandbox; при этом AMD стабильно превосходил существующие методы на основе памяти. Дополнительный анализ показал: наибольший вклад в прирост даёт именно Subtask-память; эффективность учителя зависит и от возможностей самого учителя, и от того, насколько его память совместима с учеником; а больше всего от AMD выигрывают модели-ученики с 4 млрд параметров.

Ключевые факты

  • AMD, не требующий дообучения (training-free) способ передать маленькому ИИ-агенту структурированную память большого агента-учителя
  • Три типа памяти: Workflow (стратегии на уровне задачи), Subtask (конкретные примеры поведения), Function (правила вызова функций и типичные ошибки)
  • Workflow и Subtask подгружаются заранее в начале задачи, Function, реактивно, только при ошибке вызова инструмента
  • На трёх бенчмарках (AppWorld, BFCL V3, ToolSandbox) с моделями-учениками 4, 8 млрд параметров и GPT-5-mini в роли учителя прирост точности составил 27,2, 11,2 и 3,4 процентного пункта соответственно, обойдя существующие методы на основе памяти
  • Наибольший вклад даёт Subtask-память; больше всего от AMD выигрывают модели-ученики с 4 млрд параметров

Почему это важно

Память как способ улучшить агента уже доказала пользу, но почти всегда её изучали на больших моделях. Маленькие модели оставались в стороне по конкретной причине: чтобы построить полезную память, обычно нужны собственные успешные траектории действий агента, а маленькая модель их генерирует плохо и редко. AMD обходит это ограничение, вместо того чтобы требовать от маленькой модели самой нарабатывать опыт, метод переносит уже готовую, структурированную память от большого агента-учителя. И делает это без дообучения: сам ученик не меняется, меняется лишь то, что ему подкладывают в контекст и что он может подгрузить при ошибке.

Кому это важно

В первую очередь, тем, кто строит агентов на маленьких моделях (4, 8 млрд параметров) ради экономии на инференсе, но упирается в их слабую надёжность при работе с инструментами. Метод даёт способ поднять точность такого агента без затрат на дообучение, то есть без отдельного тренировочного прогона и связанной с ним инфраструктуры. Вторая аудитория, исследователи систем памяти для агентов: работа показывает, что не все виды памяти одинаково полезны и что подбирать учителя нужно с оглядкой на конкретного ученика, а не по принципу «чем больше учитель, тем лучше».

Как это применить

Практическая схема выглядит так: сначала большой агент-учитель (в эксперименте, GPT-5-mini) прогоняет задачи и оставляет успешные траектории; из них строятся три слоя памяти. Workflow- и Subtask-память подгружаются агенту-ученику сразу, до начала работы над задачей, как заранее известная стратегия и примеры поведения. Function-память подключается только по необходимости: когда ученик уже ошибся при вызове конкретной функции, ему реактивно подсказывают конвенцию вызова и типичные подводные камни именно для этой функции. Дообучать модель-ученика не нужно, вся работа сводится к формированию и подгрузке структурированной памяти.

Можно ли доверять

Это исследовательская работа (научная статья на HuggingFace Papers) с прозрачно описанной методологией: три типа памяти, механизм подгрузки для каждого, три отдельных бенчмарка использования инструментов и четыре модели-ученика разного масштаба. Результаты, собственные измерения авторов, независимой проверки или рецензирования в тексте не упомянуто. В самом тексте не названы ни авторы, ни организация, стоящая за работой, поэтому по составу и институциональной принадлежности судить не по чему.

Риски и подводные камни

Приведены только относительные приросты точности в процентных пунктах, абсолютных цифр точности до и после в источнике нет, так что судить о реальном уровне качества агентов напрямую нельзя. Разброс результата велик: от 27,2 процентного пункта на одном бенчмарке до 3,4 на другом, то есть выигрыш сильно зависит от конкретной задачи. Данных о стоимости и задержке, которые добавляет хранение и подгрузка памяти на инференсе, а также о времени построения самой памяти на этапе обучения, в тексте нет.