聚焦问题区域微调,让文生图模型精准改进缺陷部位。
Focus-N-Fix: Region-Aware Fine-Tuning for Text-to-Image Generation
- 按图像区域定位问题,只对缺陷区微调
- 安全、合理性和提示对齐提升,整体不变
- 适合需要精准控制生成质量的场景
近年来,文生图(T2I)生成取得显著进展,但仍存在感知伪影、复杂提示对齐不佳和安全性问题。现有方法依赖人工反馈构建奖励模型并进行微调,但可能导致模型行为意外改变——例如提升安全性可能损害提示对齐,或引发奖励欺骗。本文提出 Focus-N-Fix,一种区域感知的微调方法,仅针对原模型表现不佳的图像区域进行修正。微调后模型保持原有高层次结构,但在安全(如过度性化、暴力)、合理性等局部质量方面显著改善。实验表明,该方法在不明显影响其他区域的前提下,有效提升了特定缺陷区域的表现。
原文摘要 · Abstract (English)
Text-to-image (T2I) generation has made significant advances in recent years, but challenges still remain in the generation of perceptual artifacts, misalignment with complex prompts, and safety. The prevailing approach to address these issues involves collecting human feedback on generated images, training reward models to estimate human feedback, and then fine-tuning T2I models based on the reward models to align them with human preferences. However, while existing reward fine-tuning methods can produce images with higher rewards, they may change model behavior in unexpected ways. For example, fine-tuning for one quality aspect (e.g., safety) may degrade other aspects (e.g., prompt alignment), or may lead to reward hacking (e.g., finding a way to increase rewards without having the intended effect). In this paper, we propose Focus-N-Fix, a region-aware fine-tuning method that trains models to correct only previously problematic image regions. The resulting fine-tuned model generates images with the same high-level structure as the original model but shows significant improvements in regions where the original model was deficient in safety (over-sexualization and violence), plausibility, or other criteria. Our experiments demonstrate that Focus-N-Fix improves these localized quality aspects with little or no degradation to others and typically imperceptible changes in the rest of the image. Disclaimer: This paper contains images that may be overly sexual, violent, offensive, or harmful.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。