
UniSD
Ahren09
Official implementation for "Towards a Unified Self-Distillation Framework for Large Language Models" (https://arxiv.org/abs/2605.06597).
AI 简介
UniSD 是一个面向大语言模型的统一自蒸馏框架,旨在不依赖外部教师模型的情况下提升模型性能。它整合了多教师一致性、EMA教师稳定、令牌级对比学习、特征匹配与发散裁剪等互补机制,系统性解决自蒸馏中监督信号不可靠、表征对齐困难和训练不稳定等核心问题。项目基于 PyTorch 和 Hugging Face Transformers 实现,支持主流开源 LLM,并在六项基准任务上验证了有效性。适用于需要轻量级模型优化、无监督/弱监督微调、或受限于教师模型获取能力的研究与工程场景。
Python
Apache License 2.0326
Stars
0
Forks
2
Watchers
1
Issues
Star 增长
今日0
近 7 天0
近 30 天0
综合评分40
默认分支main