翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。
取得拡張生成
基盤モデルは通常オフラインでトレーニングされるため、モデルはモデルのトレーニング後に作成されたどのデータにも依存しません。さらに、基盤モデルは非常に一般的なドメインコーパスでトレーニングされるため、ドメイン固有のタスクにはあまり効果的ではありません。Retrieval Augmented Generation (RAG) を使用すると、基盤モデルの外側からデータを取得し、コンテキスト内で取得した関連するデータを追加してプロンプトを補強できます。RAG モデルアーキテクチャの詳細については、「Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks
ではRAG、プロンプトの拡張に使用される外部データは、ドキュメントリポジトリ、データベース、 などの複数のデータソースから取得できますAPIs。最初のステップは、関連性検索を実行するために、ドキュメントとユーザークエリを互換性のある形式に変換することです。形式を互換性のあるものにするために、ドキュメントコレクションまたはナレッジライブラリ、およびユーザーが送信したクエリを、埋め込み言語モデルを使用して数値表現に変換します。埋め込みとは、ベクトル空間でテキストに数値表現を与えるプロセスです。RAG モデルアーキテクチャは、ナレッジライブラリのベクトル内のユーザークエリの埋め込みを比較します。その後、元のユーザープロンプトに、ナレッジライブラリ内の類似ドキュメントからの関連コンテキストが追加されます。次に、この拡張プロンプトが基盤モデルに送信されます。ナレッジライブラリと関連する埋め込みは非同期で更新できます。
取得したドキュメントは、プロンプトを補強するのに役立つコンテキストを含むのに十分な大きさで、プロンプトの最大シーケンス長に収まる大きさである必要があります。の一般的なテキスト埋め込み (GTE) JumpStart モデルなどのタスク固有のモデルを使用してHugging Face、プロンプトやナレッジライブラリドキュメントの埋め込みを提供できます。プロンプトとドキュメントの埋め込みを比較して最も関連性の高いドキュメントを見つけたら、補足コンテキストを使用して新しいプロンプトを作成します。次に、拡張プロンプトを選択したテキスト生成モデルに渡します。
サンプルノートブックの例
RAG 基盤モデルソリューションの詳細については、次のサンプルノートブックを参照してください。
Amazon SageMaker サンプルリポジトリ