4 ms·
This is interesting theoretically, but in practical terms it’s hard to apply. RL is already hard. There are many things which can go wrong. You have all of the
by janalsncm 4mo ago
This is interesting theoretically, but in practical terms it’s hard to apply.
RL is already hard. There are many things which can go wrong. You have all of the problems with regular LLM SFT, plus now you have a reward model which can be hacked or too hard. Or KL collapse because the outputs are repetitive. Or maybe your groups in GRPO aren’t producing advantages. Or the rollouts are OOD for your reward model. Or maybe you’re running the rollout at a different precision as the trained weights. Or maybe your importance sampling should be clipping when it’s not, or should be clipping at the token level rather than sequence level.
Maybe after reading the above you think that the above are not problems because smart people wouldn’t make those mistakes. Fair enough. But I would prefer RL people like myself who are not geniuses.
Now, this is adding another variable into the mix: choosing a single layer to train. If it doesn’t work is it because there’s a problem with your RL setup? Or did you just choose the wrong layer? Or maybe there’s no problem with your setup but you chose a suboptimal layer to train.
Also note that we already have LoRA, which is a more established method for low memory parameter updates.
- redpig_hongzhou 4mo agoI guess the practical implication is probably just train on middle layers or increase the corresponding learning rate