Mechanism

Generalizes STaR from curated QA datasets to arbitrary unstructured text: at every token position tt in pretraining/post-training text, the model generates parallel internal "thought" continuations (<|startthought|>...<|endthought|>) in a custom attention mask and receives a REINFORCE reward based on whether the thought improves the log-likelihood of the next ntruen_{\text{true}} future ground-truth tokens relative to no thought.

References

  1. Quiet-STaR: Language Models Can Teach Themselves to Think Before Speaking ↗

    Eric Zelikman, Georges Harik, Yijia Shao, Varuna Jayasiri, Nick Haber, Noah D. Goodman · 2024