REVIEW 4 major objections 5 minor 2 cited by
ImageReFL: Balancing Quality and Diversity in Human-Aligned Diffusion Models
T0 review · 4 major / 5 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read Reward-tuned diffusion models can keep image diversity when the base model handles early denoising steps and the tuned model only refines the final steps, trained on real images.
desk verdict The combined generation idea is simple and sensible, but the head-to-head evidence for ImageReFL rests on per-method test-set-selected handover points, so the main claim is not yet established. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The machinery is a piecewise noise-prediction switch, $\epsilon_{\text{combined}}(x_t,t,c) = \epsilon_{\theta_0}(x_t,t,c)$ for $t>T'$ and $\epsilon_\theta(x_t,t,c)$ for $t \le T'$, which lets the original model set global structure before the reward-tuned model takes over. ImageReFL is the matching training objective: start from a noised real image $x_{t'} = \sqrt{\alpha_{t'}}x^{\text{real}}_0 + \sqrt{1-\alpha_{t'}}\epsilon$, denoise with the trainable model, and backpropagate the reward through the final step as $\mathcal{L}'_{\text{reward}}=-R(F_\theta(x_{t'},t',c),c)$, regularized by the standard diffusion loss $\|\epsilon-\epsilon_\theta(x_{t'},t',c)\|_2^2$ and by periodic full ReFL steps. ReFL is a reward fine-tuning scheme that backpropagates through only the final denoising step. The combination keeps early steps untouched and forces the fine-tuned model to make its quality changes late.
What would settle it
Generate the same prompt from the same initial noise with the base model alone and with combined generation switching at $T'$, then compare the final images' global structure, for example by DINO embedding distance; if composition or pose changes substantially, the early-step invariance premise fails.
Extended reading notes
Core claim
The central claim is a division of labor across the denoising trajectory: early steps decide global structure, so a reward-tuned model should only be trusted later. The paper implements this as combined generation, which applies the base model for steps $t > T'$ and the fine-tuned model for $t \le T'$, and as ImageReFL, which starts training from a noised real image $x_{t'}$ and backpropagates the reward through the final step while adding a diffusion-loss regularizer and periodic ReFL updates. On quality-diversity trade-off curves, ImageReFL sits at a more favorable point than ReFL and ReFL AIG, and the user study gives it higher diversity and alignment votes.
Load-bearing premise
The load-bearing premise is that global features such as composition and pose are fixed during early denoising steps and stay stable, so a model switched in later can refine local quality without disturbing them.
Editorial extensions
If this is right
- Fine-tuned diffusion models can be deployed with a tunable quality-diversity knob: shifting the switch step $T'$ trades reward score against sample variety.
- Training directly on real images with the diffusion-loss regularizer reduces reward hacking and lets the model make its main changes in the final denoising steps.
- The two-model sampler works across base models (SD1.5, SDXL) and rewards (HPSv2.1, PickScore), so it is not tied to one architecture.
- The regularization recipe can be grafted onto other reward-tuning methods such as DRTune and DraftK, as the paper suggests, potentially improving their diversity as well.
Reading between the lines
- If early-step invariance is the mechanism, the optimal switch step $T'$ should depend on the prompt: scenes with complex geometry may need more late-stage refinement than simple ones, and the paper only sweeps a global $T'$.
- A natural extension is to apply combined generation as a wrapper at inference time around any reward-tuned diffusion model, even one not trained with ImageReFL, since the paper's evidence suggests the switch alone restores much of the diversity.
- The reported DinoDiversity metric captures intra-prompt variety only; a fuller test of the method would track cross-prompt and distribution-level diversity together, which FID and covariance distances partially cover.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes two mechanisms to mitigate diversity collapse in reward-fine-tuned text-to-image diffusion models: (i) combined generation, which runs the base model for early denoising steps and switches to the reward-tuned model for later steps; and (ii) ImageReFL, a training procedure that starts from noised real images, applies the reward loss only through the last denoising step, and regularizes with the standard diffusion loss and intermittent ReFL steps. The authors evaluate on SD1.5 and SDXL with HPSv2.1 and PickScore rewards, reporting quality metrics (HPSv2.1, MPS, PickScore, ImageReward), diversity metrics (FID, CovDist, LogCovDist, DinoDiversity), and a user study. They report improvements over ReFL and a ReFL AIG baseline.
Significance. If the claimed trade-off improvements were established on a fair comparison, the paper would make a useful practical contribution: it offers a simple inference-time knob (combined generation) and a training regularizer (ImageReFL) that could help practitioners retain diversity in reward-aligned diffusion models. The idea of training a model to refine images from intermediate latents of real images, rather than from pure noise, is interesting and plausibly complementary to existing reward fine-tuning. The paper also ships source code and evaluates across two base models and two reward models, which is a strength. However, the central quantitative claim—that ImageReFL improves the quality–diversity frontier relative to ReFL-based baselines—is currently not supported with adequate experimental controls, for the reasons detailed below.
major comments (4)
- [§5.4 and Appendix A.2 (Table 3)] The comparison is made at different handover points for ImageReFL and the ReFL baselines: for SD1.5, ReFL uses 10 base + 30 fine-tuned steps while ImageReFL uses 30 base + 10 fine-tuned steps; for SDXL the numbers are 20/20 versus 35/5. Since the handover point is precisely the knob that trades diversity against quality in combined generation, ImageReFL’s larger FID and DinoDiversity gains may simply reflect that fewer fine-tuned steps are being used, rather than a property of the training method. Table 3’s caption states these are the “best-performing configurations,” selected on the same test used for evaluation, which compounds the issue. The authors should report results at matched handover points (e.g., equal numbers of fine-tuned steps across methods) and, if different points are chosen per method, they should be selected on a separate validation split with the selection rule stated in advance. Without this, the central claim that ImageReFL “outperforms conventional reward tuning methods on the quality–diversity balance” is not established.
- [§5.5 and Table 1] The baseline “ReFL AIG” is used as a key comparison throughout the experiments, but it is never defined in the main text, the appendix, or the references. The abbreviation “AIG” (“Artificially-Inspired Guidance”) appears only in Table 1 and Appendix A.2, where a parameter λ is set to 1 without explanation. The paper must either provide a full definition of this baseline (including the guiding signal and the source paper, or specify that this is a new baseline introduced here) or replace it with a properly documented baseline. As written, the reader cannot verify what ReFL AIG is or why its parameter is set to that value.
- [§5.5 and Table 1] The paper states results “consistently and significantly outperform” the baselines, but no statistical significance tests are reported. The ± values in the table headers appear to be metric-level uncertainties (possibly bootstrap confidence intervals over sampled images) rather than variability due to training seeds or model initialization. With only one training run per configuration, the reported differences (e.g., ImageReFL versus ReFL Combined on DinoDiversity: 0.220 vs 0.220 in the SD1.5 HPS row, or 0.2116 vs 0.2317 on SDXL, where the latter actually favors ReFL Combined) may not be reproducible. The authors should report multiple seeds or otherwise quantify the variance of the training procedure and use an appropriate significance test, especially since some of the apparent gaps are small relative to the reported error bars.
- [§5.6 and Table 2] The user study compares ReFL and ImageReFL, but it does not appear to control the handover point. Given that the quantitative comparison at different handover points casts doubt on the fairness of the comparison, the user study inherits the same problem. The study also compresses three annotators per task into a single aggregated vote; the number of pairwise comparisons (4,800 votes over 20 participants) should be reported with per-criterion counts and a measure of inter-annotator agreement. As it stands, the user study does not provide independent evidence for the method-specific claim.
minor comments (5)
- [§5.4 and Table 1] The abbreviation “ReFL AIG” is not expanded at its first use in Table 1; consider spelling it out and citing the origin or explicitly introducing it as a new baseline.
- [§6 (Conclusion)] “DraftK” appears to be a typo for “DRAFT” (the method of Clark et al., reference [5]); the sentence mentions DRTune and DraftK with no citations or definitions and should be corrected.
- [§5.1 and References] Reference [25] is cited for DINO embeddings, but the cited paper is “DETR with Improved DeNoising anchor boxes,” an object detection model; the actual DINOv2 paper is reference [14], and the metric DinoDiversity likely uses DINOv2. The reference list should be reconciled with the model actually used.
- [Figure 5 caption] The caption says “using combined generation,” but the figure appears to show curves for both ReFL and ImageReFL under combined generation; please clarify which methods and which switch points are included.
- [Equation (3)] The condition “t=T,...,0” is redundant and should be simply “t≥T′”; also note that in standard DDIM sampling the last step index is often 0, but the set description should be consistent.
Circularity Check
Reported ImageReFL-vs-ReFL comparison uses per-method, test-selected handover steps, so the diversity/quality advantage is partly an artifact of the chosen operating point rather than the training method alone.
-
fitted input called prediction
[Appendix A.2, Table 3 caption (cf. Section 5.4)]
"Table 3 presents the evaluation metrics for the best-performing configuration of each method. For SD1.5, we use 10 steps with the base model and 30 steps with the fine-tuned model in the ReFL setup, and 30 steps with the base model and 10 with the fine-tuned model in ImageReFL. For SDXL, the corresponding step configurations are 20/20 for ReFL and 35/5 for ImageReFL."
The handover step T' is the paper's own quality-diversity control knob: Figure 5 shows that increasing fine-tuned steps improves quality but reduces diversity. The headline comparisons in Table 3 use different, 'best-performing' T' values per method, selected on the same HPDv2 test split from which the metrics are computed. ReFL is measured at 30 (or 20) fine-tuned steps while ImageReFL is measured at 10 (or 5), i.e., at the diversity-favorable end of the curve. Thus the larger DinoDiv/FID gains and comparable HPSv2.1 partly reflect the chosen operating point rather than a property of the training loss; the claimed superiority is not tested at matched step counts in the main tables.
full rationale
Apart from the unequal, test-selected operating points, the method derivation is self-contained: combined generation is defined by Eq. 3 and ImageReFL by Eqs. 4-5, with the early-stability premise cited to external works [6,3]. The DinoDiversity metric is introduced by the authors and the CLIPDiversity reference [1] includes two of the same authors, but the main diversity conclusions also hold for external FID, CovDist, and LogCovDist, so the self-citation is not load-bearing. The ReFL training loss and reward models are external. The reported advantage of ImageReFL, however, is not independent of the evaluation data because the switch points are selected per method on the same test split; this is an evaluation-leakage/confounded-comparison issue rather than a fully circular derivation, hence score 4.
Assumptions & free parameters
free parameters (5)
- Handover step T' for combined generation =
10/30, 20/20, 30/10, 35/5 per setup
- Noise level t' for ImageReFL training =
not reported explicitly
- Diffusion loss scale =
1e-5
- Reward scaling =
1e-3
- ReFL interleave frequency =
one ReFL step per 4 training steps
assumptions (3)
- domain assumption Global image features are formed mainly in early diffusion steps and remain stable in later steps.
- domain assumption Reward models HPSv2.1, PickScore, ImageReward, MPS are valid proxies for human preference.
- domain assumption COCO is an appropriate reference distribution for FID and covariance distance metrics.
Cite this review
Pith. "Pith review of ImageReFL: Balancing Quality and Diversity in Human-Aligned Diffusion Models." pith.science (2026). https://pith.science/paper/V6MLAKHA
@misc{pith2026250522569,
author = {Pith},
title = {Pith review of: ImageReFL: Balancing Quality and Diversity in Human-Aligned Diffusion Models},
year = {2026},
howpublished = {\url{https://pith.science/paper/V6MLAKHA}},
note = {Machine review of arXiv:2505.22569}
}
read the original abstract
Recent advances in diffusion models have led to impressive image generation capabilities, but aligning these models with human preferences remains challenging. Reward-based fine-tuning using models trained on human feedback improves alignment but often harms diversity, producing less varied outputs. In this work, we address this trade-off with two contributions. First, we introduce \textit{combined generation}, a novel sampling strategy that applies a reward-tuned diffusion model only in the later stages of the generation process, while preserving the base model for earlier steps. This approach mitigates early-stage overfitting and helps retain global structure and diversity. Second, we propose \textit{ImageReFL}, a fine-tuning method that improves image diversity with minimal loss in quality by training on real images and incorporating multiple regularizers, including diffusion and ReFL losses. Our approach outperforms conventional reward tuning methods on standard quality and diversity metrics. A user study further confirms that our method better balances human preference alignment and visual diversity. The source code can be found at https://github.com/ControlGenAI/ImageReFL .
Figures
Figures from the paper (16 more)
Forward citations
Cited by 2 Pith papers
-
Calibri: Enhancing Diffusion Transformers via Parameter-Efficient Calibration
Post-hoc scalar scaling of individual DiT block outputs, found by CMA-ES, improves text-to-image quality and cuts inference steps from 30-100 to 10-15.
-
When Models Learn to Ask Why: Adaptive Causal Reasoning for Trustworthy Medical Vision-Language Models
MedCausalX adds adaptive causal reflection tokens and trajectory-level RL on a new CRMed dataset to cut medical VLM hallucinations and raise diagnostic consistency.
Reference graph
Works this paper leans on
-
[1]
Aibek Alanov, Vadim Titov, Maksim Nakhodnov, and Dmitry Vetrov. Styledomain: Efficient and lightweight parameterizations of stylegan for one-shot and few-shot domain adaptation. https://arxiv.org/abs/2212.10229, 2022
work page Pith review arXiv 2022
-
[2]
Training diffusion models with reinforcement learning.arXiv preprint arXiv:2305.13301, 2023
Kevin Black, Michael Janner, Yilun Du, Ilya Kostrikov, and Sergey Levine. Training diffusion models with reinforcement learning.arXiv preprint arXiv:2305.13301, 2023
arXiv 2023
-
[3]
Perception prioritized training of diffusion models.https://arxiv.org/abs/2204.00227, 2022
Jooyoung Choi, Jungbeom Lee, Chaehun Shin, Sungwon Kim, Hyunwoo Kim, and Sungroh Yoon. Perception prioritized training of diffusion models.https://arxiv.org/abs/2204.00227, 2022
arXiv 2022
-
[4]
Brown, Miljan Martic, Shane Legg, and Dario Amodei
Paul Christiano, Jan Leike, Tom B. Brown, Miljan Martic, Shane Legg, and Dario Amodei. Deep reinforcement learning from human preferences.arXiv preprint arXiv:1706.03741, 2017
arXiv 2017
-
[5]
Draft: Diffusion reward fine-tuning
Kevin Clark, Paul Vicol, Kevin Swersky, and David J Fleet. Draft: Diffusion reward fine-tuning. arXiv preprint arXiv:2309.17400, 2023
arXiv 2023
-
[6]
Kamil Deja, Anna Kuzina, Tomasz Trzci ´nski, and Jakub M. Tomczak. On analyz- ing generative and denoising capabilities of diffusion-based deep generative models. https://arxiv.org/abs/2206.00070, 2022
work page Pith review arXiv 2022
-
[7]
Ying Fan, Olivia Watkins, Yuqing Du, Hao Liu, Moonkyung Ryu, Craig Boutilier, Pieter Abbeel, Mohammad Ghavamzadeh, Kangwook Lee, and Kimin Lee. Dpok: Reinforcement learning for fine-tuning text-to-image diffusion models.arXiv preprint arXiv:2305.16381, 2023
arXiv 2023
-
[8]
Gans trained by a two time-scale update rule converge to a local nash equilibrium
Martin Heusel, Hubert Ramsauer, Thomas Unterthiner, Bernhard Nessler, and Sepp Hochre- iter. Gans trained by a two time-scale update rule converge to a local nash equilibrium. https://arxiv.org/abs/1706.08500, 2017
arXiv 2017
Show all 28 references
-
[9]
Denoising diffusion probabilistic models.arXiv preprint arXiv:2006.11239, 2020
Jonathan Ho, Ajay Jain, and Pieter Abbeel. Denoising diffusion probabilistic models.arXiv preprint arXiv:2006.11239, 2020
2006 arXiv
-
[10]
Elucidating optimal reward-diversity tradeoffs in text-to-image diffusion models
Rohit Jena, Ali Taghibakhshi, Sahil Jain, Gerald Shen, Nima Tajbakhsh, and Arash Vah- dat. Elucidating optimal reward-diversity tradeoffs in text-to-image diffusion models. https://arxiv.org/abs/2409.06493, 2024
2024 arXiv
-
[11]
Pick-a-pic: An open dataset of user preferences for text-to-image generation.arXiv preprint arXiv:2305.01569, 2023
Yuval Kirstain, Adam Polyak, Uriel Singer, Shahbuland Matiana andJoe Penna, and Omer Levy. Pick-a-pic: An open dataset of user preferences for text-to-image generation.arXiv preprint arXiv:2305.01569, 2023
2023 arXiv
-
[12]
Aesthetic post-training diffusion models from generic preferences with step-by-step preference optimization.arXiv preprint arXiv:2406.04314, 2024
Zhanhao Liang, Yuhui Yuan, Shuyang Gu, Bohan Chen, Tiankai Hang, Mingxi Cheng, Ji Li, and Liang Zheng. Aesthetic post-training diffusion models from generic preferences with step-by-step preference optimization.arXiv preprint arXiv:2406.04314, 2024
2024 arXiv
-
[13]
Lawrence Zitnick, and Piotr Dollár
Tsung-Yi Lin, Michael Maire, Serge Belongie, Lubomir Bourdev, Ross Girshick, James Hays, Pietro Perona, Deva Ramanan, C. Lawrence Zitnick, and Piotr Dollár. Microsoft coco: Common objects in context.https://arxiv.org/abs/1405.0312, 2014
2014 arXiv
-
[15]
Long Ouyang, Jeff Wu, Xu Jiang, Diogo Almeida, Carroll L. Wainwright, Pamela Mishkin, Chong Zhang, Sandhini Agarwal, Katarina Slama, Alex Ray, John Schulman, Jacob Hilton, Fraser Kelton, Luke Miller, Maddie Simens, Amanda Askell, Peter Welinder, Paul Christiano, Jan Leike, and...
2022 arXiv
-
[16]
Aligning text- to-image diffusion models with reward backpropagation.arXiv preprint arXiv:2310.03739, 2023
Mihir Prabhudesai, Anirudh Goyal, Deepak Pathak, and Katerina Fragkiadaki. Aligning text- to-image diffusion models with reward backpropagation.arXiv preprint arXiv:2310.03739, 2023
2023 arXiv
-
[17]
Learning transferable visual models from natural language supervision
Alec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh, Gabriel Goh, Sandhini Agarwal, Girish Sastry, Amanda Askell, Pamela Mishkin, Jack Clark, et al. Learning transferable visual models from natural language supervision. InInternational conference on machine learning, pag...
2021
-
[18]
Manning, and Chelsea Finn
Rafael Rafailov, Archit Sharma, Eric Mitchell, Stefano Ermon, Christopher D. Manning, and Chelsea Finn. Direct preference optimization: Your language model is secretly a reward model. arXiv preprint arXiv:2305.18290, 2023
2023 arXiv
-
[19]
High- resolution image synthesis with latent diffusion models
Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, and Björn Ommer. High- resolution image synthesis with latent diffusion models. InProceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pages 10684–10695, June 2022
2022
-
[20]
High- resolution image synthesis with latent diffusion models.arXiv preprint arXiv:2112.10752, 2021
Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, and Björn Ommer. High- resolution image synthesis with latent diffusion models.arXiv preprint arXiv:2112.10752, 2021
2021 arXiv
-
[21]
Diffusion model alignment using direct preference optimization.arXiv preprint arXiv:2311.12908, 2023
Bram Wallace, Meihua Dang, Rafael Rafailov, Linqi Zhou, Aaron Lou, Senthil Purushwalkam, Stefano Ermon, Caiming Xiong, Shafiq Joty, and Nikhil Naik. Diffusion model alignment using direct preference optimization.arXiv preprint arXiv:2311.12908, 2023
2023 arXiv
-
[22]
Human preference score v2: A solid benchmark for evaluating human preferences of text-to-image synthesis.arXiv preprint arXiv:2306.09341, 2023
Xiaoshi Wu, Yiming Hao, Keqiang Sun, Yixiong Chen, Feng Zhu, Rui Zhao, and Hongsheng Li. Human preference score v2: A solid benchmark for evaluating human preferences of text-to-image synthesis.arXiv preprint arXiv:2306.09341, 2023
2023 arXiv
-
[23]
Deep reward supervisions for tuning text-to-image diffusion models
Xiaoshi Wu, Yiming Hao, Manyuan Zhang, Keqiang Sun, Zhaoyang Huang, Guanglu Song, Yu Liu, and Hongsheng Li. Deep reward supervisions for tuning text-to-image diffusion models. arXiv preprint arXiv:2405.00760, 2024
2024 arXiv
-
[24]
Imagereward: Learning and evaluating human preferences for text-to-image generation
Jiazheng Xu, Xiao Liu, Yuchen Wu, Yuxuan Tong, Qinkai Li, Ming Ding, Jie Tang, and Yuxiao Dong. Imagereward: Learning and evaluating human preferences for text-to-image generation. arXiv preprint arXiv:2304.05977, 2023
2023 arXiv
-
[25]
Ni, and Heung- Yeung Shum
Hao Zhang, Feng Li, Shilong Liu, Lei Zhang, Hang Su, Jun Zhu, Lionel M. Ni, and Heung- Yeung Shum. Dino: Detr with improved denoising anchor boxes for end-to-end object detection. https://arxiv.org/abs/2203.03605, 2022
2022 arXiv
-
[26]
Learning multi-dimensional human preference for text-to-image gener- ation.https://arxiv.org/abs/2405.14705, 2024
Sixian Zhang, Bohan Wang, Junqiang Wu, Yan Li, Tingting Gao, Di Zhang, and Zhongyuan Wang. Learning multi-dimensional human preference for text-to-image gener- ation.https://arxiv.org/abs/2405.14705, 2024. 11 A Appendix A.1 User study settings In the user study, annotators wer...
2024 arXiv
-
[27]
Which row of images is more diverse? To what extent do the images within a single row differ from each other in terms of content, colors, composition, and other visual elements?
-
[28]
Which row of images corresponds better to the textual description? To what degree do the images reflect the content described in the textual prompt above?
-
[29]
For each question, annotators could choose between three options: preference for the first row, preference for the second row, or no preference
Which row of images is more visually appealing? Which images are more aesthetically pleasing in terms of visual quality, design, and overall presentation? Figure 11 shows an example task from the study. For each question, annotators could choose between three options: preferen...
1923
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.