Apple researchers introduce probe guidance, a method to guide flow matching models that uses frozen internal states from existing diffusion models to construct guidance signals.

Applied to a 1.7B diffusion language model, probe guidance achieves state-of-the-art performance on unconditional generation and improves multiple choice question answering benchmarks while eliminating the need for additional forward passes at inference time.