{"id":"arxiv/2602.16796","url":"https://arxiv.org/abs/2602.16796","tags":["cs.LG","math.OC"],"score":0,"title":"Efficient Tail-Aware Generative Optimization via Flow Model Fine-Tuning","domain":"cs.LG","source":"arxiv","authors":["Zifan Wang","Riccardo De Santi","Xiaoyu Mo","Michael M. Zavlanos","Andreas Krause","Karl H. Johansson"],"reviews":0,"subject":"cs.LG","summary":"Fine-tuning pre-trained diffusion and flow models to optimize downstream utilities is central to real-world deployment. Existing entropy-regularized methods primarily maximize expected reward, providing no mechanism to shape tail behavior. However, tail control is often essential: the lower tail determines reliability by limiting low-reward failures, while the upper tail enables discovery by prioritizing rare, high-reward outcomes. In this work, we present Tail-aware Flow Fine-Tuning (TFFT), a principled and efficient distributional fine-tuning algorithm based on the Conditional…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-02-18","reactions":{}}