描述
MPNet,由微软开发,代表着用于语言理解的掩码和置换预训练(Masked and Permuted Pre-training for Language Understanding)。这种创新的预训练方法解决了现有模型(如 BERT 的掩码语言模型 MLM 和 XLNet 的置换语言模型 PLM)的局限性,在语言理解基准测试中取得了卓越的准确性。
阅读更多
该项目提供了对包括 BERT、XLNet 和 MPNet 本身在内的多个预训练模型的统一视图和实现。它包含了用于预训练新模型以及针对广泛的语言理解任务微调现有模型的全面代码。支持的任务包括流行的基准测试,如 GLUE、SQuAD 和 RACE,使其成为 NLP 研究人员和开发人员的通用工具。
安装过程很简单,利用了 fairseq 代码库。用户可以通过 pip 安装必要的组件。该工具包需要 PyTorch Transformers、SciPy 和 Scikit-learn。预训练过程涉及数据预处理,包括分词和二值化,然后使用提供的脚本和配置对 MPNet 模型进行实际的预训练。
MPNet 在其预训练方法上提供了灵活性。用户可以分别指定 MLM 或 PLM 等输入模式来训练掩码语言模型或置换语言模型。高级选项包括通过调整模型架构和命令行参数来使用相对位置嵌入和全词掩码。该项目还提供了更新的预训练 MPNet 模型,可直接用于下游任务的微调。
该项目基于 fairseq-0.8.0 构建,并承认其贡献。对于发现该工具包有益的用户,提供了 MPNet 论文的引用详情。还列出了相关工作和替代预训练框架,在 NLP 研究领域提供了更广泛的背景。该存储库是推进语言理解能力的重要资源。
- 类别AI 模型与大语言模型
- 定价模式开源
- 部署云端, 自托管
- 平台Linux, macOS, Windows
MPNet的核心功能
掩码和置换预训练方法
提高了 BERT 和 XLNet 的准确性
预训练模型的统一实现(BERT、XLNet、MPNet)
用于预训练和微调的代码
支持 GLUE、SQuAD、RACE 基准测试
基于 fairseq 代码库
提供更新的预训练模型
支持 MLM 和 PLM 输入模式
相对位置嵌入选项
全词掩码选项
MPNet入门
安装依赖项:pip install fairseq pytorch_transformers==1.0.0 transformers scipy sklearn
预处理数据:使用 encode.py 分词语料库,并使用 fairseq-preprocess 进行二值化
预训练 MPNet:使用指定的参数(如总更新次数、学习率、批次大小等)运行 fairseq-train
加载预训练模型:使用 fairseq.models.masked_permutation_net 中的 MPNet.from_pretrained
在下游任务上微调:为 GLUE 或 SQuAD 等特定 NLP 任务调整预训练模型
MPNet的使用案例
- 语言理解
- 预训练模型
- 微调 NLP 模型
- 基准测试性能
- NLP 研究











