Дистилляция знаний

Быстрый старт#

Обучи компактную модель-ученик под руководством более крупной модели-учителя, добавив аргумент distill_model:

from ultralytics import YOLO

model = YOLO("yolo26n.pt")
model.train(data="coco8.yaml", epochs=100, distill_model="yolo26s.pt")

Что такое дистилляция знаний?#

Дистилляция знаний переносит знания от большой и точной модели-учителя к меньшей модели-ученику. Ученик учится имитировать внутренние представления признаков учителя и часто достигает большей точности, чем при обучении с нуля.

Используй дистилляцию, если:

• Тебе нужна компактная и быстрая модель для развёртывания
• У тебя есть точная модель-учитель, обученная на тех же данных
• Ты хочешь добиться большей точности, чем при стандартном обучении

Производительность#

Дистилляция знаний повышает mAP моделей всего семейства YOLO26 на COCO, не увеличивая время инференса. В таблице ниже сравниваются стандартные модели YOLO26 (базовые) с теми же моделями, обученными с дистилляцией от рекомендованных для них учителей.

• Значения mAP^val приведены для одной модели и одного масштаба на наборе данных COCO val2017.
  Чтобы воспроизвести строку с дистилляцией, используй yolo val detect model=yolo26n-distill.pt data=coco.yaml device=0; для столбца e2e добавь nms=False.
• Для значений e2e используется путь инференса без NMS (nms=False); для остальных значений используется стандартная постобработка NMS (nms=None). Подробнее см. в разделе «Сквозное обнаружение объектов».

Предварительные требования#

Перед началом убедись, что выполнены следующие требования:

• Обученная модель-учитель: контрольная точка .pt из того же семейства YOLO, что и модель-ученик.
• Совпадение задач: используй модель-учителя для той же задачи, что и модель-ученик, и обучи её на подходящих данных.
• Ресурсы GPU: памяти должно хватать для обеих моделей; учитель выполняет только прямой проход, без градиентов и состояния оптимизатора.

Рекомендуемые пары моделей#

Дистилляция между разными семействами (например, YOLO11 в роли учителя и YOLO26 в роли ученика) не поддерживается.

Основные параметры#

Как это работает#

1. Модель-учитель остаётся замороженной в режиме eval и выполняет инференс на каждом батче
2. Модель-ученик обучается со стандартными функциями потерь для задачи и подсказками дистилляции
3. Признаки извлекаются из обеих моделей на трёх слоях neck, подающих данные на голову семейства Detect
4. Проектор из двух свёрточных слоёв 1×1 с ReLU приводит карты признаков ученика к числу каналов учителя
5. Функция потерь L2 с весами по оценкам сравнивает проецированные признаки ученика с признаками учителя, взвешивая их по уверенности классификации учителя
6. Функция потерь дистилляции объединяется со стандартными функциями потерь с весом dis

Поддержка задач#

Реализация дистилляции извлекает признаки из трёх слоёв neck, подающих данные на голову модели семейства Detect. Поскольку головы segment, pose и obb наследуют ту же архитектуру Detect, дистилляция технически совместима и с этими задачами.

Классификация, семантическая сегментация, оценка глубины и RT-DETR не используют совместимую голову семейства Detect и не поддерживаются.

from ultralytics import YOLO

# Segment
model = YOLO("yolo26n-seg.pt")
model.train(data="coco8-seg.yaml", epochs=100, distill_model="yolo26s-seg.pt")

# Pose
model = YOLO("yolo26n-pose.pt")
model.train(data="coco8-pose.yaml", epochs=100, distill_model="yolo26s-pose.pt")

# OBB
model = YOLO("yolo26n-obb.pt")
model.train(data="dota8.yaml", epochs=100, distill_model="yolo26s-obb.pt")

Обучение#

Базовое обучение#

Обучение с дистилляцией ничем не отличается от стандартного обучения. Чтобы включить дистилляцию, укажи путь distill_model:

from ultralytics import YOLO

# Загрузи модель-ученика
student = YOLO("yolo26m.pt")

# Обучи модель с дистилляцией знаний от более крупной модели-учителя
results = student.train(data="coco8.yaml", epochs=100, distill_model="yolo26x.pt")

Настройка веса функции потерь дистилляции#

Параметр dis (по умолчанию: 6.0) задаёт вклад функции потерь дистилляции:

from ultralytics import YOLO

student = YOLO("yolo26n.pt")
results = student.train(data="coco8.yaml", epochs=100, distill_model="yolo26s.pt", dis=10.0)

Возобновление обучения с дистилляцией#

Обучение с дистилляцией можно возобновить из контрольных точек. Модель-учитель автоматически создаётся заново по пути distill_model, сохранённому в контрольной точке:

from ultralytics import YOLO

student = YOLO("runs/detect/train/weights/last.pt")
results = student.train(resume=True)

Результаты обучения#

При включённой дистилляции в журналы обучения добавляется столбец dis_loss:

      Epoch    GPU_mem   box_loss   cls_loss    l1_loss   dis_loss  Instances       Size
      1/80      46.2G      1.566      5.404    0.003249      6.658        231        640

Экспортированная модель содержит только веса ученика — размер файла и скорость инференса такие же, как у стандартно обученной модели-ученика.

Часто задаваемые вопросы#