DavidNguyen commited on
Commit
c38c17a
1 Parent(s): 733803d

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ base_model:
6
+ - microsoft/Phi-3.5-mini-instruct
7
+ - google/siglip-so400m-patch14-224
8
+ pipeline_tag: image-text-to-text
9
+ ---
10
+
11
+
12
+ ## LibMoE: A Library for Comprehensive Benchmarking of Mixture of Experts in Large Language Models
13
+
14
+ ### Introduction
15
+ Mixture of Experts (MoEs) plays an essential role in the development of more efficient and effective large language models (LLMs). Due to the enormous resource requirements, studying large-scale MoE algorithms remains inaccessible to many researchers. This work introduces LibMoE, a comprehensive and modular framework designed to streamline the research, training, and evaluation of MoE algorithms. Built upon three core principles: (i) modular design, (ii) efficient training, and (iii) comprehensive evaluation, LibMoE makes MoEs in LLMs more accessible to a wider range of researchers by standardizing the training and evaluation pipelines. Using LibMoE, we extensively benchmarked five state-of-the-art MoE algorithms across three different LLMs and 11 datasets under a zero-shot setting. The results show that, despite unique characteristics, all MoE algorithms perform similarly when averaged across a broad range of tasks. With its modular design and extensive evaluation capabilities, we believe LibMoE will be invaluable for researchers striving to make meaningful progress toward the next generation of MoE and LLMs.
16
+
17
+ ### Model and Evaluation Benchmarks
18
+
19
+ We have released five MoE algorithms trained based on `microsoft/Phi-3.5-mini-instruct` for LLMs and `SigLIP` for vision encoding. These models were trained on the [LLAVA-665K dataset](https://huggingface.co/datasets/liuhaotian/LLaVA-Instruct-150K). We evaluated these state-of-the-art algorithms on 11 benchmarks, examining various aspects of MoE algorithm performance.
20
+
21
+ | Model | MoE Method | AI2D | Text VQA | GQA | Hallusion<br>Benchmark | MathVista<br>Validation | MMBenchEN<br>/ dev | MMMU<br>Validation | MMStar | POPE | SQA IMG<br>Full | MME | AVG |
22
+ |----------------------|---------------------|-------|----------|-------|-------------------------|-------------------------|---------------------|---------------------|--------|-------|------------------|-----------|-------|
23
+ | SigLIP 224 + Phi3.5 | SMoE-R | **65.71** | 41.21 | 60.78 | **42.38** | 29.70 | 70.79 | 42.11 | 41.67 | 85.19 | 79.13 | 1,665.15 | 55.87 |
24
+ | | Cosine-R | 65.16 | 41.92 | **61.15** | 41.96 | 29.20 | 70.79 | 42.78 | 41.57 | 86.39 | 79.42 | 1,748.67 | 56.03 |
25
+ | | Sigmoid-R | 64.60 | 40.49 | 60.23 | 41.01 | 30.40 | 70.45 | 41.44 | 42.05 | 86.03 | 78.54 | 1,684.87 | 55.52 |
26
+ | | Hyper-R | 65.06 | 41.68 | 60.97 | 41.01 | **31.40** | 70.88 | 42.22 | **42.60** | 86.78 | **79.43** | 1,707.86 | 56.20 |
27
+ | | Perturbed Cosine-R | 65.54 | **41.85** | 61.04 | 41.75 | 30.50 | **71.65** | **43.00** | 41.72 | **86.73** | 78.88 | **1,688.82** | **56.27** |
28
+ -
29
+ ### Run LibMoE
30
+
31
+ We provide detailed instructions for setting up and running experiments in this repository: [https://github.com/Fsoft-AIC/LibMoE](https://github.com/Fsoft-AIC/LibMoE)
32
+
33
+ ### Hardware Resources
34
+
35
+ | Stage | MoE Method | Hardware |
36
+ |-------------------|----------------------|-----------|
37
+ | Pre-Training | | 4xA100 |
38
+ | Pre-FineTuning | | 4xA100 |
39
+ | VIT | SMoE-R | 6xA100 |
40
+ | | Cosine-R | 6xA100 |
41
+ | | Sigmoid-R | 6xA100 |
42
+ | | Hyper-R | 6xA100 |
43
+ | | Perturbed Cosine-R | 6xA100 |
44
+
45
+ ---
46
+
47
+
48
+ ### Citation Information
49
+ More details can be found in our paper.
50
+
51
+ If you use LibMoE, please cite it using this BibTeX:
52
+
53
+ ```
54
+ ```