D

Dense Encoder Distilbert Frozen Emb

vocab-transformersによって開発
DistilBERTアーキテクチャに基づく高密度検索モデルで、MS MARCOデータセットでトレーニングされ、単語埋め込み層は凍結状態を維持
ダウンロード数 26
リリース時間 : 4/5/2022

モデル概要

このモデルはDistilBERTのバリエーションで、情報検索タスクに特化して最適化されており、MarginMSE損失関数を使用してトレーニングされ、ドキュメントとクエリの高密度ベクトル表現生成に適しています

モデル特徴

凍結単語埋め込みトレーニング
トレーニング中に事前トレーニングされた単語埋め込み層のパラメータを変更せず、モデルの安定性を向上させる可能性があります
MarginMSE最適化
MarginMSE損失関数を使用してトレーニングを行い、検索タスクのランキング性能を最適化
軽量アーキテクチャ
DistilBERTアーキテクチャに基づいており、元のBERTモデルよりも小さく高速で、良好な性能を維持

モデル能力

テキストベクトル表現
意味的類似度計算
情報検索
ドキュメントランキング

使用事例

検索エンジン
ウェブ検索結果ランキング
検索エンジン用にクエリとドキュメントの高密度ベクトル表現を生成し、関連性ランキングに使用
TREC-DLなどの標準検索評価で良好なパフォーマンス
質問応答システム
回答段落検索
大量のドキュメントから質問に関連する段落を迅速に検索
FiQAなどの金融質問応答データセットで安定したパフォーマンス
AIbase
未来を切り開く、あなたのAIソリューション知識ベース
© 2025AIbase