View a markdown version of this page

継続的なカスタマイズ - Amazon SageMaker AI

翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。

継続的なカスタマイズ

継続的なカスタマイズでは、別の手法で、または新しいデータで同じ手法で、追加のトレーニングを適用することで、以前にカスタマイズしたモデルに基づいて構築できます。たとえば、SFT から始めてドメインの知識を学習し、DPO を適用してユーザー設定に合わせ、RLVR を適用して事実の精度を向上させます。

重要

継続的なカスタマイズは、MTRL (Multi-Turn Reinforcement Learning) とは異なります。継続的なカスタマイズは、複数の手法にまたがるトレーニングジョブを連鎖させます。MTRL は、1 つのトレーニングジョブ内の複数ステップのツール使用タスクについてエージェントをトレーニングします。

仕組み

  • 以前のトレーニングジョブから LoRA アダプターの重みをロードします

  • オプティマイザ、スケジューラ、ステップカウンターをリセットします (新しい手法またはデータを使用した新しいトレーニング)

  • 前のジョブの出力が新しいトレーニング実行の開始点になります。

注記

現在、継続的なカスタマイズは、MTRL および Python SDK のみを使用した継続的なカスタマイズワークフローを通じて利用できます。LoRA トレーニングタイプでサポートされています。FFT を使用したさまざまな手法での継続的なカスタマイズはサポートされていません。

サポートされている手法の移行

次の表は、継続的なカスタマイズでサポートされているすべての組み合わせを示しています。

手法から テクニックへ サポート
SFTDPO✓
SFTRLVR✓
SFTRLAIF✓
SFTSFT (新しいデータ)✓
SFTMTRL✓
DPODPO (新しいデータ)✓
DPORLVR✓
DPORLAIF✓
DPOSFT✓
RLVRDPO✓
RLVRRLVR (新しいデータ)✓
RLAIFDPO✓
RLAIFRLAIF (新しいデータ)✓

継続的なカスタマイズとトレーニングの再開

継続的なカスタマイズ トレーニングを再開する
テクニック 異なるまたは同じ手法 同じ手法のみ
オプティマイザの状態 リセット (新しい開始) チェックポイントから復元
ユースケース チェーン手法 (SFT → DPO → RLVR) 中断されたトレーニングを継続する
トレーニングタイプ LoRA のみ LoRA のみ

開始方法

継続的なカスタマイズは Python SDK を通じて利用できます。前のトレーニングジョブの出力パスを次の手法resume_from_pathの として指定します。

from sagemaker.modules.train import DPOTrainer # Continue from a previous SFT job with DPO trainer = DPOTrainer( model_id="meta-textgeneration-llama-3-1-8b-instruct", train_data="s3://my-bucket/dpo-preferences.jsonl", resume_from_path="s3://my-bucket/previous-sft-output/", hyperparameters={ "max_epochs": 2, "learning_rate": 1e-5, "lora_rank": 16, } ) trainer.train()

ワークフローの例

SFT → DPO

最初にドメインの知識を教えてから、ユーザー設定に合わせる

SFT → RLVR

最初にタスク形式を教えてから、事実の正確性を最適化する

SFT → DPO → RLVR

フルパイプライン — 知識 → 設定 → 精度

SFT → MTRL

基本機能をトレーニングし、マルチターンエージェントタスク用にトレーニングする

DPO → DPO (新しいデータ)

新しいフィードバックデータを使用して設定を繰り返し絞り込む