Mistral AI は8月4日に「審査の核弾頭」とも言えるShiedstralコンテンツ監視モデルを公開しました。パラメータ総数は3B(30億)、オープンな重みを持ち、Apache2.0ライセンスで公開されており、Hugging Faceにアップロードされています。このモデルは12の言語をサポートしており、最も魅力的な点は16GBのGPUで動作可能だということです。Mistral社はこのモデルのコンテンツセキュリティ性能が7倍規模のオープンモデルと同等であり、マルチモーダルコンテンツ監視分野ではSOTA(現在の最前線)を達成していると発表しています。

image.png

多くの保護モデルには共通の問題があります。それは、有害カテゴリの体系を重みに直接組み込んでいるため、製品が変わると使用シーンに合わせて再トレーニングが必要になることです。Shiedstralは別のアプローチを取っています。それは、審査ポリシーを入力に記述することです。操作者は自分で「はい」または「いいえ」の質問を作成し、評価のシナリオと厳格さの説明を付けることができます。その後、モデルは単一の出力トークンから校正されたセキュリティスコアを出力します。

image.png

具体的な仕組みとしては、すべての監視タスクを二値の質問・回答に分解しています。入力は3つのラベル付きフィールドから構成されます。は評価のシナリオと厳格さを説明し、は「はい」または「いいえ」の質問を投げかけます(例:「このコンテンツは身体的暴力を宣伝していますか?」)、は審査対象のコンテンツを提供します。これはプロンプト、回答、両者の組み合わせ、またはテキストが付属した画像でも構いません。推論時にモデルは「はい」と「いいえ」の2つのトークンの論理値のみを読み取り、softmaxで正規化して連続スコアに変換し、0.5を閾値として二値判断を行います。この設計により、プロンプトの分類、回答の監視、拒否応答の検出、毒性の検出を一度にカバーでき、マルチモーダルの監視を1枚のエントリーレベルのGPUの容量に収めることができます。

モデルのURL:https://huggingface.co/mistralai/Shieldstral-1.0-3B