コーヒー1杯の値段で自分専用の埋め込みモデルをファインチューニングする

・ Fireworks AI その他

Fine-Tune Your Own Embedding Model for the Price of a Coffee

Fireworks AIは、対照学習を用いてドメイン特化データで埋め込みモデルをファインチューニングする手法を紹介し、費用は10ドル未満で済むとしている。検索精度は大きく改善し、法令引用タスクでnDCG@100が39%向上、EU判例のマッチングでは61%向上した。柔軟性と効率のバランスが異なる3つの学習モードも用意されている。InfoNCE損失とバッチ内ネガティブを使うことで、ベースモデルの汎化性能を保ちながらドメインへの特化を実現するという。

Fireworks AI 公式ブログで全文を読む ↗

Fireworks AI の他のリリース