← all repositories

AberHu/Knowledge-Distillation-Zoo

A PyTorch library of knowledge distillation methods for model compression and teacher-student knowledge transfer.

1.7k stars Python ML Frameworks
Knowledge-Distillation-Zoo
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of various knowledge distillation techniques used to compress neural networks. It includes classic methods like soft targets, attention transfer, Fitnet hints, and probabilistic knowledge transfer. The library enables training smaller student models to mimic larger teacher models, reducing model size and computational cost while preserving predictive performance.

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