← all repositories

lenscloth/RKD

PyTorch implementation of CVPR 2019 Relational Knowledge Distillation for compressing deep learning models via teacher-student training.

418 stars Python ML FrameworksComputer Vision
RKD
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository implements Relational Knowledge Distillation, a technique for transferring structured knowledge from a larger teacher neural network to a smaller student network. The code supports metric learning experiments using architectures like ResNet, with distance-weighted sampling and triplet loss for training. It includes scripts for training teacher embeddings (e.g., ResNet50), evaluating performance, and distilling knowledge to student networks (e.g., ResNet18) for computer vision tasks like CUB-200.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.