30 Repos
Techniques for transferring capabilities from large teacher models to smaller, more efficient student models.
Distinguishing note: Focuses on model compression and capability transfer rather than initial training.
Explore 30 awesome GitHub repositories matching artificial intelligence & ml · Knowledge Distillation. Refine with filters or upvote what's useful.
DeepSeek-R1 is an open-weights large language model focused on advanced reasoning. It uses chain-of-thought processing and internal monologues to solve complex mathematical and logical problems by breaking tasks into sequential, verifiable thought processes. The model is developed using reinforcement learning to optimize reasoning patterns and verify logical steps. It employs a distillation process to transfer these high-performance logic capabilities from a large teacher model into smaller, computationally efficient versions. The training framework incorporates group relative policy optimiz
Transfers complex reasoning capabilities from large models to smaller versions to reduce computational costs.
This project is a comprehensive framework for the entire lifecycle of transformer-based language models, supporting everything from foundational pretraining to specialized deployment. It provides a modular toolkit for defining neural network architectures, managing data preparation pipelines, and executing training routines across various scales. The framework is designed to handle the full model development process, including supervised fine-tuning, behavioral alignment, and the integration of agentic capabilities. What distinguishes this framework is its focus on efficient training and adva
The framework facilitates knowledge distillation by transferring capabilities from a large teacher model to a smaller student model using teacher-generated outputs to improve efficiency.
Fairseq is a PyTorch toolkit for sequence-to-sequence modeling, specializing in neural machine translation, automatic speech recognition, and large-scale language model training. It provides a framework for processing and aligning diverse data sources, including text, audio, and video, to support tasks such as speech-to-text conversion and multimodal sequence learning. The project is distinguished by its distributed training capabilities, which utilize parameter sharding, mixed-precision training, and CPU offloading to handle models that exceed single-device memory. It also includes specializ
Uses an autoregressive model to simplify training data distributions for non-autoregressive learning targets.
This library provides a comprehensive collection of modular building blocks and research-backed architectures for implementing vision transformers within the PyTorch framework. It serves as a centralized repository for constructing, training, and analyzing attention-based models, offering a wide array of specialized variants designed for image classification and visual representation learning. The project distinguishes itself through a focus on architectural efficiency and flexibility, supporting diverse input formats including non-square images and volumetric data like video. It incorporates
Incorporates knowledge distillation using specialized tokens to transfer capabilities from teacher models to student vision transformers.
This project is a comprehensive framework and toolkit for developing, optimizing, and deploying transformer-based models across multimodal, document intelligence, and natural language processing tasks. It provides a unified neural architecture that processes text, vision, audio, and document layout data through a shared set of weights, enabling researchers and developers to build foundational models that align cross-modal representations. The platform distinguishes itself through advanced training and inference strategies designed for large-scale deep learning. It incorporates specialized mec
Transfers knowledge from teacher models to student retrievers to improve performance and efficiency.
This project is a transformer-based framework for generating dense and sparse vector embeddings of text and multimodal data. It serves as a library for fine-tuning models to perform semantic similarity tasks, retrieval, and reranking. The system is distinguished by its support for diverse architectural patterns, including bi-encoders for fast similarity search and cross-encoders for high-precision reranking. It provides dedicated pipelines for multimodal embeddings, mapping text and images into a shared vector space, and implements knowledge distillation to compress large models into smaller,
Implements knowledge distillation to compress large teacher models into smaller, low-latency student models.
This library provides a comprehensive framework for fine-tuning, aligning, and distilling transformer-based language models. It serves as a toolkit for adapting models to specialized domains through supervised learning, while offering advanced methodologies to improve output quality and reasoning capabilities. The project distinguishes itself through specialized alignment and optimization techniques, including direct preference optimization and reinforcement learning, which allow models to be tuned against human preferences without complex reward modeling. It further supports training efficie
Transfers complex capabilities from large teacher models to smaller architectures to reduce computational requirements.
Swin-Transformer is a deep learning framework designed for training and deploying hierarchical vision transformer models. It serves as a research library and toolkit for computer vision tasks, providing the infrastructure to build models that replace standard convolution operations with sliding window self-attention mechanisms. By utilizing a multi-scale feature hierarchy, the framework enables the processing of visual data at varying resolutions and spatial scales. The project distinguishes itself through its implementation of shifted window partitioning, which facilitates global information
Transfers knowledge from large, high-capacity models into smaller, efficient architectures to maintain high performance while reducing computational resources.
PaddleDetection is an object detection framework designed for the end-to-end development, training, and deployment of computer vision models. It provides a comprehensive library of modular neural network architectures and pipelines that support object detection, instance segmentation, and multi-object tracking tasks. The project distinguishes itself through a configuration-driven approach that decouples model components like backbones and heads, allowing for the flexible assembly of custom vision workflows. It incorporates advanced techniques such as anchor-free detection logic, joint detecti
Transfers learned representations from high-accuracy teacher models to smaller student networks to improve efficiency without sacrificing detection precision.
YOLOv7 is a PyTorch vision library and real-time inference engine designed for object detection, human pose estimation, and instance segmentation. It provides a framework for detecting and locating multiple objects within images or video streams using neural networks. The system includes tools for custom model training and fine-tuning, allowing pre-trained weights to be adapted to specialized datasets via transfer learning. It also supports model weight export and format conversion to facilitate deployment on production servers and embedded edge devices.
Employs a teacher-student framework where a deeper lead model guides a smaller head model to improve performance.
Open CLIP is an open source framework for training and deploying Contrastive Language-Image Pre-training models. It serves as a vision-language training framework and multimodal embedding engine that maps images and text into a shared vector space for similarity searches and zero-shot classification. The project provides a toolkit for distributed training of contrastive models and includes an image-to-text generative model for producing natural language descriptions. It supports custom text encoder integration and utilizes teacher-student model distillation to transfer knowledge from large pr
Transfers knowledge from large pre-trained teacher models to smaller student architectures to maintain accuracy.
This project is a static educational website and comprehensive curriculum focused on computer vision and deep learning. It serves as a public repository of instructional materials, lecture notes, and technical guides specifically detailing convolutional neural networks and visual recognition. The site is developed using static-site generation to host course documentation and student project directories. It provides structured academic resources that guide learners through image classification, generative modeling, and the implementation of various neural network architectures. The curriculum
Explains how to transfer soft labels from teacher to student networks via knowledge distillation to improve resilience.
AutoGluon is an automated machine learning framework and multimodal library designed to automate the end-to-end pipeline from data preprocessing to high-accuracy model training and validation. It functions as an automated model trainer for tabular, image, text, and time series data, as well as a tool for time series forecasting and foundation model finetuning. The project is distinguished by its ability to jointly process and fuse different data types, allowing for the construction of multimodal neural networks that integrate images, text, and structured tables. It supports zero-shot inferenc
Trains small student models to mimic complex teacher ensembles for more efficient production deployment.
Oumi is a comprehensive large language model development platform designed for synthesizing data, fine-tuning models, and running performance evaluations. It serves as a unified environment for the entire model lifecycle, encompassing a training and fine-tuning suite, an evaluation framework, and tools for synthetic data generation and model distillation. The platform is distinguished by its iterative, failure-driven synthesis approach, which analyzes model weaknesses during evaluation to generate targeted training data. It utilizes an LLM-based judge framework to programmatically score respo
Supports the transfer of knowledge from large teacher models to smaller student models through distillation processes.
jetson-inference is a set of libraries and tools for executing optimized deep learning models on embedded GPU hardware. Its primary purpose is to enable real-time computer vision and AI inference at the edge with low latency and high throughput. The project distinguishes itself through high-performance streaming analytics and the ability to execute concurrent AI pipelines on auto-grade silicon. It provides specialized support for multi-sensor stream processing, utilizing zero-copy data transport to load camera frames directly into GPU memory. The codebase covers a broad surface of capabiliti
Uses knowledge distillation to transfer intelligence from large teacher models into smaller student models.
ART is a platform for agentic training, providing a reinforcement learning framework, training environment, and compute orchestrator. It enables the improvement of multi-step agent reasoning and tool usage through group relative policy optimization and a judge-based reward modeling system. The project features tools for model distillation to transfer capabilities from large teacher models to smaller architectures, as well as a system for capturing execution trajectories to generate synthetic training data. It supports specialized training workflows including supervised fine-tuning for baselin
Provides tools for transferring capabilities from large teacher models to smaller, more efficient architectures.
Compresses a large teacher model into a smaller student model via knowledge distillation, preserving accuracy while boosting inference throughput.
Liger-Kernel is a collection of pre-built fused Triton kernels and patching utilities designed to accelerate large language model training. It provides drop-in kernel replacements for common LLM operations such as RMSNorm, cross-entropy loss, and attention, enabling increased throughput and reduced memory usage while preserving bitwise-exact gradients. The project serves as a toolkit for composing custom model architectures from individual optimized kernels and for patching pre-existing models with minimal code changes. The project distinguishes itself through its ability to perform runtime m
Provides optimized kernel implementations for calculating KL divergence and Jensen-Shannon divergence losses for knowledge distillation.
PaddleClas ist ein Toolkit für Bildklassifizierung und -erkennung, das auf PaddlePaddle aufbaut. Es bietet eine Suite von Werkzeugen für das Training von Deep-Learning-Modellen und ein Framework zur Implementierung visueller Such- und Abrufsysteme. Das Projekt enthält eine Suite zur Optimierung von Computer-Vision-Modellen und Werkzeuge für plattformübergreifendes Deployment. Es ermöglicht den Export trainierter Modelle auf Server, Mobilgeräte und Edge-Hardware, um hochperformante Inferenz über verschiedene Programmiersprachen hinweg zu erreichen. Das Toolkit deckt Modellkompression und -optimierung durch Pruning, Quantisierung und Knowledge Distillation ab. Es unterstützt zudem den visuellen Informationsabruf durch die Kombination von Objekterkennung, neuronaler Merkmalsextraktion und Vektorsuche, um ähnliche Bilder innerhalb eines Datensatzes zu identifizieren. Nutzer können End-to-End-Klassifizierungs- und Erkennungs-Workflows mithilfe eines visuellen Workflow-Builders entwerfen.
Employs knowledge distillation to transfer patterns from large teacher models to smaller, efficient student models.
MobileSAM ist ein leichtgewichtiger Bildsegmentierer und ein prompt-basiertes Vision-Modell, das für die schnelle Objektisolierung auf Hardware mit begrenzten Ressourcen entwickelt wurde. Es fungiert als automatisches Bildmaskierungstool, das in der Lage ist, einzelne Objekte in einem gesamten Bild ohne manuelle Eingabe zu erkennen und zu isolieren. Das System ermöglicht eine prompt-basierte Objektmaskierung unter Verwendung von Koordinatenpunkten oder Begrenzungsrahmen, um präzise Masken zu generieren. Es unterstützt zudem die Segmentierung aller Objekte in einem Bild durch objektbewusstes Prompt-Sampling, um jedes einzelne Objekt in einer Szene zu identifizieren. Um das Deployment auf Mobilgeräten und Edge-Geräten zu erleichtern, ist das Modell mit ONNX-Export kompatibel, wodurch das Vision-Modell auf verschiedenen plattformübergreifenden Hardware-Runtimes ausgeführt werden kann.
Uses knowledge distillation to compress a large teacher model into a lightweight student model.