arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.13344cs.CVeess.IV

FireRed-Image-Edit-1.0 技术报告

FireRed-Image-Edit-1.0 Technical Report

Super Intelligence Team(超级智能团队);Xiaohongshu Inc.(小红书公司)
Super Intelligence Team, Changhao Qiao, Chao Hui, Chen Li, Cunzheng Wang, Dejia Song, Jiale Zhang, Jing Li, Qiang Xiang, Runqi Wang, Shuang Sun, Wei Zhu, Xu Tan… 展开作者

Super Intelligence Team, Changhao Qiao, Chao Hui, Chen Li, Cunzheng Wang, Dejia Song, Jiale Zhang, Jing Li, Qiang Xiang, Runqi Wang, Shuang Sun, Wei Zhu, Xu Tang, Yao Hu, Yibo Chen, Yuhao Huang, Yuxuan Duan, Zhiyi Chen, Ziyuan Guo

AI总结:

提出FireRed-Image-Edit扩散变换器,通过数据整理、多阶段训练和评估优化,在指令图像编辑上达到最先进性能,并开源代码、模型和基准。

摘要:

我们提出FireRed-Image-Edit,一种基于指令的图像编辑扩散变换器,通过系统优化数据整理、训练方法和评估设计,实现了最先进的性能。我们构建了一个16亿样本的训练语料库,包含来自不同来源的9亿文本到图像和7亿图像编辑对。经过严格清洗、分层、自动标注和两阶段过滤,我们保留了超过1亿高质量样本,在生成和编辑之间取得平衡,确保强大的语义覆盖和指令对齐。我们的多阶段训练流程通过预训练、监督微调和强化学习逐步构建编辑能力。为了提高数据效率,我们引入了多条件感知桶采样器用于可变分辨率批处理,以及带有动态提示重索引的随机指令对齐。为了稳定优化并增强可控性,我们提出了用于DPO的非对称梯度优化、用于文本编辑的具有布局感知OCR奖励的DiffusionNFT,以及用于身份保持的可微一致性损失。我们进一步建立了REDEdit-Bench,一个涵盖15个编辑类别的综合基准,包括新引入的美化和低级增强任务。在REDEdit-Bench和公共基准(ImgEdit和GEdit)上的大量实验表明,与开源和专有系统相比,我们的性能具有竞争力或更优。为了支持未来研究,我们的代码、模型和基准套件在此https URL公开提供。

英文摘要:

We present FireRed-Image-Edit, a diffusion transformer for instruction-based image editing that achieves state-of-the-art performance through systematic optimization of data curation, training methodology, and evaluation design. We construct a 1.6B-sample training corpus, comprising 900M text-to-image and 700M image editing pairs from diverse sources. After rigorous cleaning, stratification, auto-labeling, and two-stage filtering, we retain over 100M high-quality samples balanced between generation and editing, ensuring strong semantic coverage and instruction alignment. Our multi-stage training pipeline progressively builds editing capability via pre-training, supervised fine-tuning, and reinforcement learning. To improve data efficiency, we introduce a Multi-Condition Aware Bucket Sampler for variable-resolution batching and Stochastic Instruction Alignment with dynamic prompt re-indexing. To stabilize optimization and enhance controllability, we propose Asymmetric Gradient Optimization for DPO, DiffusionNFT with layout-aware OCR rewards for text editing, and a differentiable Consistency Loss for identity preservation. We further establish REDEdit-Bench, a comprehensive benchmark spanning 15 editing categories, including newly introduced beautification and low-level enhancement tasks. Extensive experiments on REDEdit-Bench and public benchmarks (ImgEdit and GEdit) demonstrate competitive or superior performance against both open-source and proprietary systems. To support future research, our code, models, and benchmark suite are publicly available at https://github.com/FireRedTeam/FireRed-Image-Edit/ .

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑