ArrowCanaria-Llama-8B-RL-v0.1

ArrowCanaria-Llama-8B-RL-v0.1

モデル概要

ArrowCanaria-Llama-8B-RL-v0.1 は、ArrowCanaria-Llama-8B-SFT-v0.1 に対して RLHF（Reinforcement Learning from Human Feedback）を適用し、応答品質をさらに向上させたAItuber向け日本語特化モデルです。

SFTモデルは高品質なデータで学習されていますが、モデルの応答が「データに含まれる平均的な応答」に収束してしまう傾向があります。本モデルでは、外部報酬モデル（Reward Model）によるフィードバックを用いた強化学習を適用することで、共感・傾聴の品質や知識応答の正確性・分かりやすさを、SFTの水準からさらに引き上げています。

強化学習アルゴリズムには GRPO（Group Relative Policy Optimization） を採用し、DAPO損失関数による安定した最適化を実現しています。相談応答と知識応答の2フェーズで段階的にRLHFを行うことで、SFTで獲得した雑談・RP・キャラクター対話能力を保持しつつ、応答の質を選択的に向上させています。

想定ユースケース

• AItuber / AI VTuber: 配信中のリスナーとの雑談・コメント対応
• チャットボット: 自然な日本語での日常会話・悩み相談
• ロールプレイ: キャラクターを演じた対話・創作支援
• 汎用アシスタント: 知識応答・推論・ツール呼び出しを含む幅広いタスク

主な特徴

• 🗣️ 自然な日本語応答 — 定型文や翻訳調を排した、人間らしい対話。RLHFによりさらに自然さが向上
• 💬 高い雑談・相談性能 — 共感・傾聴・具体的助言の品質が報酬モデルによるフィードバックで最適化済み
• 🎭 RP・キャラクター対話 — SFTで獲得した一貫した人格・感情表現を保持
• 🧠 推論力・知識応答 — 正確で分かりやすい知識応答をRLHFで強化
• 🔧 Tool Use / RAG — Function Calling・検索拡張生成に対応
• ✍️ クリエイティブ表現 — 文学的な比喩・暗喩・情景描写など、豊かな日本語表現力

モデル仕様

---

推論方法

🤗 Transformers

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "DataPilot/ArrowCanaria-Llama-8B-RL-v0.1"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype="bfloat16",
    device_map="auto",
)

messages = [
    {"role": "system", "content": "あなたは親しみやすいAIアシスタントです。自然な日本語で会話してください。"},
    {"role": "user", "content": "最近ちょっと疲れてるんだよね。何かリフレッシュできる方法ない？"},
]

input_ids = tokenizer.apply_chat_template(
    messages,
    add_generation_prompt=True,
    return_tensors="pt",
).to(model.device)

outputs = model.generate(
    input_ids,
    max_new_tokens=512,
    temperature=0.7,
    top_p=0.9,
    repetition_penalty=1.05,
    do_sample=True,
)

response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
print(response)

⚡ vLLM

# vLLM サーバーの起動
vllm serve DataPilot/ArrowCanaria-Llama-8B-RL-v0.1 \
    --dtype bfloat16 \
    --max-model-len 4096 \
    --host 0.0.0.0 \
    --port 8000

# リクエスト例（curl）
curl http://localhost:8000/v1/chat/completions \
  -H "Content-Type: application/json" \
  -d '{
    "model": "DataPilot/ArrowCanaria-Llama-8B-RL-v0.1",
    "messages": [
      {"role": "system", "content": "あなたは親しみやすいAIアシスタントです。自然な日本語で会話してください。"},
      {"role": "user", "content": "最近ちょっと疲れてるんだよね。何かリフレッシュできる方法ない？"}
    ],
    "temperature": 0.7,
    "max_tokens": 512
  }'

---

データ概要

SFTデータ（ArrowCanaria-Llama-8B-SFT-v0.1 で使用）

SFTモデルは、独自の合成データ生成フレームワーク SDG_LOOM を活用して作成された 175,000件超 の高品質合成データセットで学習されています。

SFTデータの内訳：知識応答・雑談/相談・Tool Use・RAG・推論・RP・AItuber RP・クリエイティブ表現

RLHFデータ

RLHFでは、以下の2つのデータセットを用いて強化学習を実施しました。

RLHFでは学習データの回答（assistant部分）は使用せず、ユーザーの質問（prompt）のみを用いてモデルに応答を生成させ、外部報酬モデルのスコアに基づいて方策を最適化しています。

---

学習概要

モデル作成パイプライン

本モデルは以下の4段階で構築されています：

1. CPT（Continual Pre-Training）
   tokyotech-llm/Llama-3.1-Swallow-8B-Instruct-v0.5
   + 10k フィルタリング済み高品質日本語合成小説データセット
        ↓
2. Chat Vector マージ（Mergekit）
   CPTモデルに元モデルの Chat Vector をマージし対話能力を復元
        ↓
3. SFT（Supervised Fine-Tuning）
   175,000件超の合成データセットで LoRA による
   3フェーズカリキュラム学習 → ArrowCanaria-Llama-8B-SFT-v0.1
        ↓
4. RLHF（Reinforcement Learning from Human Feedback）
   GRPO + 外部報酬モデルによる
   2フェーズ強化学習 → ArrowCanaria-Llama-8B-RL-v0.1 ★本モデル

RLHF 学習設計

アルゴリズム: GRPO + DAPO

本モデルでは GRPO（Group Relative Policy Optimization） を採用しています。GRPOは各プロンプトに対して複数の応答候補を生成し、それらの報酬スコアの相対的な優劣（アドバンテージ）に基づいて方策を更新するアルゴリズムです。PPOと異なりCritic（価値関数）を必要としないため、メモリ効率が高く安定した学習が可能です。

損失関数には DAPO（Direct Advantage Policy Optimization） を使用しています。DAPOはGRPOの改良版で、KLペナルティを用いずクリッピングのみで方策の更新幅を制御するため、ハイパーパラメータのチューニングが簡易で学習が安定します。

2フェーズ構成

Phase 1: 相談 RLHF ──▶ Phase 2: 知識QA RLHF
  1,600件                 2,000件
 「共感・傾聴の最適化」   「正確さ・分かりやすさの最適化」

学習率の設計思想

P1 (5e-6) > P2 (3e-6)

• Phase 1 (5e-6): 相談応答はモデルのEQ中核能力であり、積極的に最適化
• Phase 2 (3e-6): Phase 1 で獲得した相談品質を保護しつつ、知識応答を控えめに改善。RLHFの2段階目は過学習リスクが高いため慎重に設定

報酬設計

生成パラメータ（GRPO候補生成）

主な学習パラメータ

Phase 2 の工夫

• Warmup ratio を 0.08 に増加（Phase 1 は 0.05）：既にRLHF済みのモデルへの追加学習のため、急激なパラメータ変動を抑制
• 学習率を Phase 1 の 60%（3e-6）に低減：Phase 1 で獲得した相談品質を最大限保護しつつ、知識応答品質を向上
• --force_system_prompt の適用：データセット固有のシステムプロンプトを統一し、一貫した応答スタイルを維持

---

引用

本モデルを利用した場合は、以下の情報を引用・参照していただけると幸いです：

DataPilot/ArrowCanaria-Llama-8B-RL-v0.1
SFT model: DataPilot/ArrowCanaria-Llama-8B-SFT-v0.1
Base model: tokyotech-llm/Llama-3.1-Swallow-8B-Instruct-v0.5

謝辞

• tokyotech-llm/Llama-3.1-Swallow-8B-Instruct-v0.5 — ベースモデル
• Unsloth — 高速 LoRA 学習フレームワーク
• TRL — GRPOTrainer
• vLLM — GRPO候補生成エンジン
• SDG_LOOM — 合成データ生成フレームワーク（SFTデータ作成に使用）