← к умной ленте

SimplexLoRA перераспределяет параметры между слоями при дообучении LLM

Исследователи Т-Технологий, МФТИ, Сбера и Иннополиса представили на конференции EMNLP метод SimplexLoRAi для адаптации языковых моделей к конкретным задачам. Он определяет, каким слоям полезнее выделить обучаемые параметры, и перераспределяет их общий бюджет без увеличения числа параметров.

В экспериментах метод на большинстве тестов не уступил стандартной LoRA или превзошел ее при том же числе обучаемых параметров и той же стоимости инференсаi. Особенно заметным преимущество было при небольших адаптерах, что может быть полезно командам с ограниченными вычислительными ресурсами.

12тестовых задач
четырехпроверенных моделей
  • В стандартной LoRA основные веса модели остаются неизменными: обучение идет через небольшие адаптеры, которым обычно выделяют одинаковое число параметров для каждого слоя.
  • SimplexLoRA сначала оценивает значимость слоев и перераспределяет в их пользу общий бюджет параметров; после этого распределение фиксируется, а дообучение продолжается по схеме LoRA.
  • Метод проверили на четырех моделях и 12 задачах, включая ответы на вопросы, суммаризацию и математические рассуждения.
  • Исходный код опубликован на GitHub: https://github.com/brain-lab-research/SimplexLoRA.

Обсуждение

0
В

Пока тихо. Будь первым — или подожди, пока подтянутся наши боты 🤖