Исследователи представили рабочий процесс дообучения модели Gemma-3 для решения сложных математических задач из набора GSM8K. Методология опирается на алгоритм GRPO (Group Relative Policy Optimization) в сочетании с адаптерами LoRA, что позволяет эффективно оптимизировать модель для структурированного логического вывода, сохраняя при этом низкие требования к вычислительным ресурсам в процессе тренировки.

Процесс обучения включает подготовку среды и интеграцию с платформой Hugging Face, где модель обучается генерировать ответы через цепочку рассуждений. Ключевым элементом выступает система функций вознаграждения, которая оценивает как строгое соблюдение формата вывода, так и математическую точность итогового результата. Использование LoRA-адаптеров позволяет проводить дообучение без необходимости полной перенастройки весов основной модели.

Данный подход демонстрирует практический метод адаптации предобученных LLM для узкоспециализированных задач, требующих высокой точности вычислений. Внедрение GRPO позволяет модели лучше справляться с многошаговыми логическими цепочками, что критически важно для автоматизации аналитических процессов и работы с данными, требующими верифицируемых ответов.

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

  • Использована архитектура Gemma-3 в качестве базовой модели для дообучения.
  • Применен алгоритм GRPO для оптимизации стратегии генерации ответов.
  • Интегрированы LoRA-адаптеры для минимизации затрат на обучение и сохранения весов.
  • В качестве основного бенчмарка для оценки навыков рассуждения выбран датасет GSM8K.
  • Система вознаграждений настроена на проверку формата вывода и корректности численного результата.