D

Distilbert Word2vec 256k MLM 500k

由vocab-transformers開發
該模型結合了word2vec詞嵌入和DistilBERT架構,通過大規模數據訓練,適用於多種自然語言處理任務。
下載量 15
發布時間 : 4/7/2022

模型概述

該模型使用256k詞條的word2vec詞嵌入矩陣,基於C4、MSMARCO、新聞、維基百科和S2ORC等共計100GB數據訓練。隨後通過掩碼語言建模(MLM)進行了50萬步訓練,詞嵌入層在訓練過程中保持凍結狀態。

模型特點

大規模詞嵌入
使用256k詞條的word2vec詞嵌入矩陣,基於100GB多樣化數據訓練。
高效訓練
通過掩碼語言建模進行了50萬步訓練,批量大小為64。
凍結詞嵌入
訓練過程中詞嵌入層保持凍結狀態,保留原始word2vec特徵。
輕量架構
基於DistilBERT架構,在保持性能的同時減少模型複雜度。

模型能力

文本理解
語言建模
文本特徵提取

使用案例

自然語言處理
文本分類
可用於情感分析、主題分類等任務。
信息檢索
適用於搜索引擎和問答系統中的文本匹配。
語言模型微調
可作為下游任務的預訓練模型進行微調。
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase