5d ago

Can you fine-tune on localized steering of an LLM?

I want to fine tune an LLM to "steer" it in the right direction. I have plenty of training examples in which I stop the generation early and correct the output to go in the right direction, and then resume generation.

Basically, for my dataset doing 100 "steers" on a single task is much cheaper than having to correct 100 full generations completely, and I think each of these "steer" operations has value and could be used for training.

So maybe I'm looking for some kind of localized DPO. Does anyone know if something like this exists?

18 comments

Here's a guide: https://medium.com/@yuxiaojian/fine-tuning-llama3-1-and-deploy-to-ollama-f500a6579090
Not sure what the impact of a few 100 examples will be, though, even if you freeze most parameters.
- Can SFT be used on partial generations? What I mean by a "steer" is a correction to only a portion, and not even the end, of model output.
  For example, a "bad" partial output might be:
  undefined
  
  <assistant> Here are four examples: 1. High-quality example 1 2. Low-quality example 2
  and the "steer" might be:
  undefined
  
  <assistant> Here are four examples: 1. High-quality example 1 2. High-quality example 2
  but the full response will eventually be:
  undefined
  
  <assistant> Here are four examples: 1. High-quality example 1 2. High-quality example 2 3. High-quality example 3 4. High-quality example 4
  The corrections don't include the full output.
  
  I do not know what SFT means. So I can't comment on that, I'm afraid.
  Models only predict the distribution of the next token. So "partial response" vs "full response" is a consequence repeated inference untill the stop token is reached. It's mostly unrelated to the model parameters.
  For training, it makes no difference.

I dont know what you mean with steering?
Do you want a given output structure, like json or toml?
Do you want to align the model, with your dataset of question and answer pairs?
First of all, have you tried giving the model multiple examples of input output pairs in the context, this already helps the model a lot to output the correct format.
Second you can force a specific output structure by using a regex or grammar: https://python.langchain.com/docs/integrations/chat/outlines/#constrained-generation https://github.com/ggerganov/llama.cpp/blob/master/grammars/README.md
And third, in case you want to train a model to respond differently and the previous steps were not good enough, you can fine-tune. I can recommend this project to you, as it teaches how to fine-tune a model: https://github.com/huggingface/smol-course
Depending on the size of the model, that you want to fine-tune and the amount of compute that you have available you can either train by updating all parameters like ORPO or you can train via PEFT (LoRA)

18 comments