Исследователи предложили метод дистилляции знаний, который отказывается от прямого копирования логитов или скрытых признаков учителя. Вместо этого предлагается обучать студента распознавать классы эквивалентности представлений учителя, учитывая их инвариантность к ортогональным преобразованиям и масштабированию. Это позволяет более гибко передавать знания между моделями, игнорируя произвольные координаты в пространстве признаков.
Традиционные методы дистилляции часто сталкиваются с проблемой «абсолютных координат», когда студент пытается имитировать конкретные значения активаций учителя. Однако предобученные модели обладают внутренней симметрией: их представления идентифицируемы лишь с точностью до ортогонального преобразования и изотропного масштабирования. Новый подход фокусируется на сохранении этой геометрической структуры, что делает процесс обучения более устойчивым и эффективным.
Авторы работы демонстрируют, что учет эквивалентности представлений позволяет студенту лучше улавливать семантические связи, заложенные в учителе, без жесткой привязки к конкретным числовым значениям. Это открывает возможности для более качественного сжатия моделей, где студент не просто повторяет ответы учителя, а усваивает принципы формирования его внутренних представлений.
Ключевые факты
- Метод основан на концепции классов эквивалентности представлений, инвариантных к ортогональным преобразованиям.
- Предложенный подход решает проблему отсутствия абсолютных координат в скрытых слоях нейронных сетей.
- Исследование направлено на улучшение эффективности дистилляции знаний при переносе опыта от крупных моделей к компактным.
- Работа опубликована на платформе arXiv под номером 2607.03572v1.