Torch-Struct：深度结构化预测库
计算与语言
2020-02-04 v1 神经与进化计算
机器学习
摘要
NLP 中结构化预测的文献描述了关于序列、分割、对齐和树的丰富分布与算法集合；然而，这些算法难以在深度学习框架中利用。我们介绍 Torch-Struct，一个为利用并集成基于向量化、自动微分的框架而设计的结构化预测库。Torch-Struct 包含通过简单灵活的基于分布的 API 访问的广泛概率结构集合，可连接任意深度学习模型。该库利用批处理、向量化操作并发挥自动微分优势，以生成可读、快速且可测试的代碼。在内部，我们还包含若干通用优化以提供跨算法效率。实验显示相较快速基线有显著性能提升，案例研究展示了该库的优势。Torch-Struct 可在 https://github.com/harvardnlp/pytorch-struct 获取。
引用
@article{arxiv.2002.00876,
title = {Torch-Struct: Deep Structured Prediction Library},
author = {Alexander M. Rush},
journal= {arXiv preprint arXiv:2002.00876},
year = {2020}
}