Fine-tuning of diffusion models via stochastic control: entropy regularization and beyond
<|reference_start|>Fine-tuning of diffusion models via stochastic control: entropy regularization and beyond: This paper aims to develop and provide a rigorous treatment to the problem of entropy regularized fine-tuning in the context of continuous-time diffusion models, which was recently proposed by Uehara et al. (arXiv:2402.15194, 2024). The idea is to use stochastic control for sample generation, where the entropy regularizer is introduced to mitigate reward collapse. We also show how the analysis can be extended to fine-tuning involving a general $f$-divergence regularizer.<|reference_end|>
@article{tang2024fine-tuning,
    title={Fine-tuning of diffusion models via stochastic control: entropy
  regularization and beyond},
    author={Wenpin Tang},
    journal={arXiv preprint arXiv:2403.06279},
    year={2024},
    archivePrefix={arXiv},
    eprint={2403.06279},
    primaryClass={math.OC cs.LG}
}