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

Авторы предложили обучаемое низкоранговое сжатие нейросетей

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

Авторы сообщают о результатах на OPT, Llama-2, Qwen3 и ViT: при сжатии до 50–70% качество оказалось лучше, чем у рассмотренных бейзлайнов, а ускорение инференса достигало 1,5×. При этом источник отмечает, что заметное ускорение всё ещё может сопровождаться серьёзным снижением качества.

50–70%уровень сжатия моделей
1,5×максимальное ускорение инференса
  • Для слоя обучают ортогональную матрицуi U; проектор задаётся как I − U Uᵀ.
  • Ранги распределяют по KL-расхождениюi с исходной моделью: слоям с большим KL выделяют больший ранг. После распределения ранги фиксируются и в ходе обучения не меняются.
  • Для улучшения сходимости матрицы инициализируют с учётом послойной ошибки, а проектор вводят постепенно, чтобы избежать резкого ухудшения модели в начале обучения.
  • Среди сравниваемых методов не хватало GFW-SWD и Basis Sharing; авторы также проверили сжатие ViT.
  • Q, K и V-проекции сжимаются заметно лучше, чем MLP-слоиi.

Обсуждение

0
В

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