
BERT-Whitening-Pytorch: Pytorch下的BERT白化实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
简介:本项目为基于PyTorch框架实现的BERT模型白化技术,旨在优化预训练语言模型在特定任务上的表现,提升特征提取效率与准确性。
BERT增白是一种在文本语义搜索中有广泛应用的技术。该技术不仅提升了无监督语义矢量匹配的性能,还通过减少向量维度来降低内存使用,并提高如FAISS等矢量搜索引擎的检索效率。
这种方法最早由苏建林在其博客中提出。
重现实验结果所需准备如下:
1. 下载数据集:
```
$ cd data
$ .download_datasets.sh
$ cd ..
```
2. 下载模型:
```
$ cd model
$ .download_models.sh
$ cd ..
```
在完成下载后,`data`和`model`目录的结构如下:
```
├── data
│ ├── AllNLI.tsv
│ ├── download_datasets.sh
│ └── downstream
│ ├── COCO
│ ├── CR
```
全部评论 (0)
还没有任何评论哟~


