Unified-MoE-Compression

给 MoE 模型瘦身,压缩后推理更快更省内存

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 7 👀 关注 2
GitHub ↗
GitHub 星标 ★ 93
维护状态 维护中
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可证,可自由使用和修改,无付费版本。

项目预览

Unified-MoE-Compression 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队CASE-Lab-UMD
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可证,可自由使用和修改,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型deep-learning,large-language-models,machine-learning,mixture-of-experts,model-compression,natural-language-processing
GitHub 星标★ 93
上线时间2024-02-24 00:00:00
代码更新2026-10-02
仓库体积48.3 MB
本站数据更新2026-11-10
维护状态维护中
收录时间2026-08-09
浏览次数3

项目介绍

Unified-MoE-Compression 是论文《Towards Efficient Mixture of Experts: A Holistic Study of Compression Techniques》的官方实现,聚焦混合专家(MoE)模型的压缩问题。MoE 模型虽在扩展参数时保持高效推理,但巨大的内存占用和部署成本限制了实际应用。该项目系统性地研究了多种压缩技术,包括专家剪枝、专家量化、低秩分解等,并提供了统一的评估框架和工具,帮助研究者和工程师在保持模型性能的前提下,显著减小 MoE 模型的体积和推理开销。核心能力包括:支持主流 MoE 架构(如 Switch Transformer、Mixtral 等),提供模块化的压缩算法库,以及可复现的实验配置。项目代码基于 Python 和 PyTorch,便于集成到现有训练和推理流程中。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 93

核心亮点

  • 官方论文实现,压缩方法经过系统验证,可信度高
  • 覆盖剪枝、量化、低秩分解等多种主流压缩技术,选择灵活
  • 提供统一评估框架,方便对比不同压缩策略的效果

不足之处

  • 早期项目,文档和社区支持尚不完善
  • 依赖特定 MoE 模型结构,迁移到新架构需要额外适配

适用场景

  • 部署 MoE 大模型到资源受限的边缘设备
  • 降低 MoE 模型在云服务上的推理成本和内存占用
  • 研究 MoE 压缩技术,对比不同方法的性能与精度

替代项目

SparseGPT、LLM-Pruner、Mixtral-Quantization

上一篇:Kardia-R1

下一篇:FineTuningLLMs

同类项目推荐

dLLM-RL 开源

给扩散大模型装上强化学习引擎,训练出 SOTA 的 TraDo 系列

[ICLR 2026] Official code for TraceRL: Revolutionizing post-training for Diffusion L···

★ 523 2026-08-09
DiffusionART 开源

给扩散模型做稳定高效的强化学习后训练

Official Implementation of "DiffusionART: Efficient, Stable and Scalable Reinforceme···

★ 229 2026-10-21
stable-dreambooth 开源

用最少代码微调 Stable Diffusion,几张图定制专属模型。

Dreambooth implementation based on Stable Diffusion with minimal code.

★ 146 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 464 2026-08-09
SPPO 开源

让模型自我对弈,无需人工标注即可提升对齐效果

The official implementation of Self-Play Preference Optimization (SPPO)

★ 589 2026-08-09
Prism 开源

动态稀疏注意力,让2K音视频生成模型原生训练更省算力

We propose Prism, a dynamic sparse attention framework for natively training joint v···

★ 1514 2026-10-26
llm-compressor 开源

一键压缩大模型,vLLM 直接跑,省显存提吞吐

State-of-the-art LLM compression, built for production inference with vLLM

★ 3958 2026-09-16
Open-LLaVA-NeXT 开源

开源复现 LLaVA-NeXT,快速训练你的多模态对话模型

An open-source implementation for training LLaVA-NeXT.

★ 438 2026-08-09
ReAlign 开源

换个数据格式,让模型更听话

Reformatted Alignment

★ 111 2026-08-09