TextBox：一个统一、模块化且可扩展的文本生成框架
人工智能
2021-04-20 v3
摘要
在本文中，我们发布一个开源库，称为TextBox，以提供一个统一、模块化且可扩展的文本生成框架。TextBox旨在支持广泛的文本生成任务和模型。在我们的库中，我们在9个基准数据集上实现了21个文本生成模型，涵盖VAE、GAN和预训练语言模型的类别。同时，我们的库通过将模型架构、推断和学习过程恰当地分解为高度可复用的模块，保持了充分的模块化和可扩展性，使用户能够轻松将新模型纳入我们的框架。上述特性使TextBox特别适合研究人员和从业者快速复现基线模型并开发新模型。TextBox基于PyTorch实现，并在Apache License 2.0下发布于 https://github.com/RUCAIBox/TextBox。
引用
@article{arxiv.2101.02046,
title = {TextBox: A Unified, Modularized, and Extensible Framework for Text Generation},
author = {Junyi Li and Tianyi Tang and Gaole He and Jinhao Jiang and Xiaoxuan Hu and Puzhao Xie and Zhipeng Chen and Zhuohao Yu and Wayne Xin Zhao and Ji-Rong Wen},
journal= {arXiv preprint arXiv:2101.02046},
year = {2021}
}
备注
9 pages, 2 figures, 4 tables. For our GitHub page, see https://github.com/RUCAIBox/TextBox