Improve MiniPLM-Mamba-130M model card (#1)

- Improve MiniPLM-Mamba-130M model card (a111d62001f6ec263b850a99873c4cb151de3e9a)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -1,11 +1,11 @@
 ---
-library_name: transformers
-license: apache-2.0
 datasets:
 - monology/pile-uncopyrighted
 - MiniLLM/pile-diff_samp-qwen_1.8B-qwen_104M-r0.5
 language:
 - en
+library_name: transformers
+license: apache-2.0
 metrics:
 - accuracy
 pipeline_tag: text-generation
@@ -15,10 +15,12 @@ pipeline_tag: text-generation
 
 [paper](https://arxiv.org/abs/2410.17215) | [code](https://github.com/thu-coai/MiniPLM)
 
-**MiniPLM-Mamba-130M** is a 130M model with the [Mamba achitecture](https://github.com/state-spaces/mamba) pre-trained from scratch on [the Pile](https://huggingface.co/datasets/monology/pile-uncopyrighted) using the MiniPLM knowledge distillation framework with the [offcial Qwen1.5-1.8B](https://huggingface.co/Qwen/Qwen1.5-1.8B) as the teacher model.
-This model shows the flexibility of the MiniPLM framework in conducting knowledge distillation across model families.
+**MiniPLM-Mamba-130M** is a 130M parameter language model with the [Mamba architecture](https://github.com/state-spaces/mamba) pre-trained from scratch on [the Pile](https://huggingface.co/datasets/monology/pile-uncopyrighted) using the MiniPLM knowledge distillation framework. It uses the [official Qwen1.5-1.8B](https://huggingface.co/Qwen/Qwen1.5-1.8B) as the teacher model.
+This model demonstrates the flexibility of the MiniPLM framework in conducting knowledge distillation across model families.  The [pre-training corpus](https://huggingface.co/datasets/MiniLLM/pile-diff_samp-qwen_1.8B-qwen_104M-r0.5) refined by Difference Sampling in MiniPLM is open-sourced for reproducibility.
+
+## MiniPLM: Knowledge Distillation for Pre-Training Language Models
 
-We also open-source the [pre-training corpus](https://huggingface.co/datasets/MiniLLM/pile-diff_samp-qwen_1.8B-qwen_104M-r0.5) refined by Difference Sampling in MiniPLM for reproducibility.
+Knowledge distillation (KD) is widely used to train small, high-performing student language models (LMs) using large teacher LMs. While effective in fine-tuning, KD during pre-training faces challenges in efficiency, flexibility, and effectiveness. Existing methods either incur high computational costs due to online teacher inference, require tokenization matching between teacher and student LMs, or risk losing the difficulty and diversity of the teacher-generated training data. To address these issues, MiniPLM is proposed, a KD framework for pre-training LMs by refining the training data distribution with the teacher's knowledge. For efficiency, MiniPLM performs offline teacher LM inference, allowing KD for multiple student LMs without adding training-time costs. For flexibility, MiniPLM operates solely on the training corpus, enabling KD across model families. For effectiveness, MiniPLM leverages the differences between large and small LMs to enhance the difficulty and diversity of the training data, helping student LMs acquire versatile and sophisticated knowledge.
 
 <p align='left'>
     <img src="https://cdn-uploads.huggingface.co/production/uploads/624ac662102fcdff87be51b9/2BqT0NgkmIXYlktovw9kG.png" width="1000">
@@ -26,7 +28,7 @@ We also open-source the [pre-training corpus](https://huggingface.co/datasets/Mi
 
 ## Evaluation
 
-MiniPLM models achieves better performance given the same computation and scales well across model sizes:
+MiniPLM models achieve better performance given the same computation and scale well across model sizes:
 
 <p align='left'>
     <img src="https://cdn-uploads.huggingface.co/production/uploads/624ac662102fcdff87be51b9/EOYzajQcwQFT5PobqL3j0.png" width="1000">