Dense Encoder Distilbert Frozen Emb
D
Dense Encoder Distilbert Frozen Emb
vocab-transformersによって開発
DistilBERTアーキテクチャに基づく高密度検索モデルで、MS MARCOデータセットでトレーニングされ、単語埋め込み層は凍結状態を維持
ダウンロード数 26
リリース時間 : 4/5/2022
モデル概要
このモデルはDistilBERTのバリエーションで、情報検索タスクに特化して最適化されており、MarginMSE損失関数を使用してトレーニングされ、ドキュメントとクエリの高密度ベクトル表現生成に適しています
モデル特徴
凍結単語埋め込みトレーニング
トレーニング中に事前トレーニングされた単語埋め込み層のパラメータを変更せず、モデルの安定性を向上させる可能性があります
MarginMSE最適化
MarginMSE損失関数を使用してトレーニングを行い、検索タスクのランキング性能を最適化
軽量アーキテクチャ
DistilBERTアーキテクチャに基づいており、元のBERTモデルよりも小さく高速で、良好な性能を維持
モデル能力
テキストベクトル表現
意味的類似度計算
情報検索
ドキュメントランキング
使用事例
検索エンジン
ウェブ検索結果ランキング
検索エンジン用にクエリとドキュメントの高密度ベクトル表現を生成し、関連性ランキングに使用
TREC-DLなどの標準検索評価で良好なパフォーマンス
質問応答システム
回答段落検索
大量のドキュメントから質問に関連する段落を迅速に検索
FiQAなどの金融質問応答データセットで安定したパフォーマンス
おすすめAIモデル
Llama 3 Typhoon V1.5x 8b Instruct
タイ語専用に設計された80億パラメータの命令モデルで、GPT-3.5-turboに匹敵する性能を持ち、アプリケーションシナリオ、検索拡張生成、制限付き生成、推論タスクを最適化
大規模言語モデル
Transformers 複数言語対応

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-TinyはSODAデータセットでトレーニングされた超小型対話モデルで、エッジデバイス推論向けに設計されており、体積はCosmo-3Bモデルの約2%です。
対話システム
Transformers 英語

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
RoBERTaアーキテクチャに基づく中国語抽出型QAモデルで、与えられたテキストから回答を抽出するタスクに適しています。
質問応答システム 中国語
R
uer
2,694
98