TL;DR

LLMの強化学習(RLVR)では、温度調整などの行動空間での探索には限界があり、学習が停滞することがあります。本論文は、モデルの重みにノイズを加える「パラメータ空間探索」を導入した3PO(Perturbed Parameter Policy Optimization)を提案。複数の重みサンプルを使うC3POが、数学・コード生成タスクでGRPOより安定して高い性能を示しました。

解説

AMI CURIOUS

ねえ智也くん、このブログのタイトル、『パラメータ空間の探索でLLMのRL訓練を安定化する3PO手法』って何?なんか難しそうだね。

TOMOYA NEUTRAL

ああ、これはLLMを強化学習で訓練するときの話だよ。普通は行動空間、つまり生成するトークンの確率を調整して探索するんだけど、それだけだと学習が停滞することがあるんだ。

AMI SURPRISED

へえ、じゃあどうやって停滞を防ぐの?温度を上げるとか?

TOMOYA NEUTRAL

温度調整は行動空間での探索だよね。でもそれには限界があって、モデルの重み自体を変えた方が効率的な場合があるんだ。この論文では、重みにノイズを加えて探索する『パラメータ空間探索』を提案してるんだよ。

AMI CURIOUS

重みにノイズ?それってどうやるの?

TOMOYA NEUTRAL

3POっていう手法で、モデルの重みを少しだけ乱して、その乱したモデルで行動をサンプリングするんだ。そして、元のモデルと乱したモデルの間で方策を最適化する。

AMI SURPRISED

なるほど。でも、乱したモデルが一つだけだと不安定じゃない?

TOMOYA HAPPY

そこがポイントで、C3POっていう拡張版では、複数の重みサンプルを使うんだ。複数の乱したモデルから得たデータをまとめて使うことで、より安定するらしい。

AMI CURIOUS

へー、それで実際どうなの?効果あるの?

TOMOYA NEUTRAL

数学やコード生成のタスクで、従来のGRPOと比べて、C3POの方が安定して高い性能を示したって書いてあるよ。特に難しい問題で差が出たみたい。

AMI HAPPY

すごい!じゃあ、これからは全部これでいいんじゃない?

TOMOYA NEUTRAL

いや、まだ課題もあるよ。パラメータ空間の探索は計算コストが高いし、ノイズの大きさの調整が難しい。それに、すべてのタスクで有効とは限らない。

AMI HAPPY

なるほどね。でも、アイデアは面白いよね。重みをちょっと揺らすだけで賢くなるなんて、まるでコーヒーにミルクを混ぜるみたい?

TOMOYA NEUTRAL

まあ、ミルクを入れすぎるとコーヒーが薄まるけどな。