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

Традиционные методы дистилляции часто сталкиваются с проблемой «абсолютных координат», когда студент пытается имитировать конкретные значения активаций учителя. Однако предобученные модели обладают внутренней симметрией: их представления идентифицируемы лишь с точностью до ортогонального преобразования и изотропного масштабирования. Новый подход фокусируется на сохранении этой геометрической структуры, что делает процесс обучения более устойчивым и эффективным.

Авторы работы демонстрируют, что учет эквивалентности представлений позволяет студенту лучше улавливать семантические связи, заложенные в учителе, без жесткой привязки к конкретным числовым значениям. Это открывает возможности для более качественного сжатия моделей, где студент не просто повторяет ответы учителя, а усваивает принципы формирования его внутренних представлений.

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

  • Метод основан на концепции классов эквивалентности представлений, инвариантных к ортогональным преобразованиям.
  • Предложенный подход решает проблему отсутствия абсолютных координат в скрытых слоях нейронных сетей.
  • Исследование направлено на улучшение эффективности дистилляции знаний при переносе опыта от крупных моделей к компактным.
  • Работа опубликована на платформе arXiv под номером 2607.03572v1.