单BERT:简单相关性分类的探索
1. 预训练BERT配置与开源贡献
在自然语言处理领域,预训练的BERT模型有多种配置,不同配置在层数、隐藏层大小、注意力头数量和参数数量上存在差异,具体如下表所示:
| Size | Layers | Hidden Size | Attention Heads | Parameters |
| — | — | — | — | — |
| Tiny | 2 | 128 | 2 | 4M |
| Mini | 4 | 256 | 4 | 11M |
| Small | 4 | 512 | 4 | 29M |
| Medium | 8 | 512 | 8 | 42M |
| Base | 12 | 768 | 12 | 110M |
| Large | 24 | 1024 | 16 | 340M |
近年来,开源代码和开放科学已成为常态,但BERT的开源并非理所当然。此前谷歌有许多创新成果仅在学术论文中分享,未发布开源代码,如MapReduce和谷歌文件系统。BERT的开源使得社区无需从头构建实现,避免了重新发现许多细节和技巧,推动了自然语言处理几乎所有方面的创新,包括文本排名应用。
2. 相关性分类任务与检索重排架构
相关性分类任务旨在估计候选文本 $d_i$ 与查询 $q$ 的相关性得分 $s_i$,表示为 $P(Relevant = 1|d_i, q)$。
直接对语料库中的每个文本进行推理在计算上是不切实际的,不仅因为神经网络推理成本高,还因为查询延迟会随语料库大小线性增长。因此,大多数BERT文本排名应用采用检索重排方法
超级会员免费看
订阅专栏 解锁全文
38

被折叠的 条评论
为什么被折叠?



