中文

minicons:支持 Transformer 语言模型灵活行为与表征分析的工具库

计算与语言 2022-03-25 v1

摘要

我们推出 minicons,一个开源库,为有意对基于 transformer 的语言模型(LMs)进行行为与表征分析的研究者提供标准 API。具体而言,minicons 使研究者能在两个层面应用分析方法:(1)在预测层面——通过提供高效提取词/句级概率的函数;以及(2)在表征层面——通过促进从一层或多层高效提取词/短语级向量。在本文中,我们描述该库并将其应用于两个启发性案例研究:一个聚焦 BERT 架构在相对语法判断上的学习动态,另一个在零样本溯因推理上对标 23 个不同 LMs。minicons 可在 https://github.com/kanishkamisra/minicons 获取。

关键词

引用

@article{arxiv.2203.13112,
  title  = {minicons: Enabling Flexible Behavioral and Representational Analyses of Transformer Language Models},
  author = {Kanishka Misra},
  journal= {arXiv preprint arXiv:2203.13112},
  year   = {2022}
}

备注

To be submitted; Code to reproduce experiments can be found on https://github.com/kanishkamisra/minicons-experiments