Splade

🚀 Lightning IR SPLADE

このモデルは、Lightning IR を使用してファインチューニングされたSPLADE[^1]モデルです。

他のモデルとの比較については、Lightning IR Model Zoo を参照してください。

🚀 クイックスタート

モデルの再現

モデルの学習を再現するには、Lightning IRをインストールし、fine-tune.yaml 設定ファイルを使用して以下のコマンドを実行します。

lightning-ir fit --config fine-tune.yaml

MS~MARCOパッセージのインデックス作成

MS~MARCOパッセージをインデックスするには、以下のコマンドと index.yaml 設定ファイルを使用します。

lightning-ir index --config index.yaml

モデルの評価

インデックス作成後、TREC Deep Learning 2019および2020でモデルを評価するには、以下のコマンドと search.yaml 設定ファイルを使用します。

lightning-ir search --config search.yaml

📄 ライセンス

このモデルは、Apache-2.0ライセンスの下で提供されています。

[^1]: Formal et al., SPLADE: Sparse Lexical and Expansion Model for First Stage Ranking