Everything K-culture — comebacks to K-beauty, straight to your inboxGet it in your inbox

METAL MEDIA

What to Edit Next: Visually Aligned Image-Editing Follow-Up Suggestions in Conversational Systems

arXiv:2608.075652026-08-02

A three-stage training pipeline teaches an AI to suggest the next image edit that users actually click and that matches what's really in the picture

In conversational image-editing apps, follow-up edit suggestions can ask for something already true of the image or something that isn't there at all. The authors first fine-tune a model on human-reviewed appropriate follow-up intents, then use real user clicks to train a reward model and reinforcement learning, and finally add an image-checking verifier as an extra reward to cut these visual mismatches. The full system was deployed on Qwen App and tested in a large-scale randomized online experiment.

METAL MEDIA explanatory visual

Three-stage training pipeline

Evidence statusMeasured results reported

  1. Stage 1: Rule-guided SFTCombines real image/query/intent inputs with a human-reviewed table of 61 follow-up intents to build six-suggestion training targets and fine-tune the multimodal policy
  2. Stage 2: Click-based RLAn 8B reward model trained on position-aware click pairs plus four quality signals (validity, SFT-closeness, length, diversity) optimized via GRPO; visual inconsistency rose from 3.0% to 3.7%
  3. Stage 3: Visual verifier rewardA separate image-first verifier checks each suggestion's required source and target state, adding a sixth grounding reward that cuts inconsistency to 0.9%
  4. Deployment & A/B testOnly the single 8B policy is served (verifier used only in training); a 14-day randomized test with millions of users measured CTR, take-away rate, and conversation turns
An explanatory diagram made by METAL MEDIA, not a figure supplied by the paper's authors.

What they did

  1. Auditing 100,000 real multi-turn image-editing conversations from Qwen App, the authors found 80.1% of follow-up edit requests depend on the current image and cannot be inferred from the prior text alone.
  2. Stage 1 (SFT) fine-tunes a multimodal policy on six-suggestion sets built from real image-query-intent inputs combined with a human-reviewed table of 61 valid follow-up editing intents.
  3. Stage 2 trains an 8B vision-language reward model on position-aware click preference pairs from real users, then uses multi-objective GRPO to optimize five signals: click preference, output validity, closeness to the SFT policy, content-aware length, and within-slate diversity.
  4. Stage 2 alone improved expert-rated quality but increased visual inconsistency (suggestions requiring an absent object or requesting an already-satisfied state) from 3.0% after SFT to 3.7%.
  5. Stage 3 adds a separate image-first structured verifier that checks required sources and target states for each suggestion, feeding a sixth 'grounding' reward into GRPO to fix this regression.
Figure 1: Illustrative multi-turn policy comparison. A click-supervised policy without visual-consistency supervision (left) proposes a slate containing a missing-source edit and an already-satisfied target; the user selects the executable lighting edit. On the updated image, the full framework (right) produces executable, image-consistent, and non-redundant follow-up edits.
Figure 1: Illustrative multi-turn policy comparison. A click-supervised policy without visual-consistency supervision (left) proposes a slate containing a missing-source edit and an already-satisfied target; the user selects the executable lighting edit. On the updated image, the full framework (right) produces executable, image-consistent, and non-redundant follow-up edits.
Table 1: Core end-to-end results. Top: offline metrics. Bottom: PE-relative online lift.
Offline evaluation
PolicyGSB ↑Ground. ↓Redund. ↓
PE (base)+08.6%23.3%
SFT (Stage 1)+3323.0%17.2%
SFT + RL (Stage 2)+4053.7%11.9%
Full framework (Stage 3)+𝟒𝟒𝟔0.9%8.8%
Online lift over PE
PolicyCTR ↑Take-away ↑Turns/user ↑
PE (base)0.00%0.00%0.00%
SFT (Stage 1)+25.33%+13.70%+33.64%
SFT + RL (Stage 2)+33.48%+7.50%+32.56%
Full framework (Stage 3)+32.70%+16.32%+39.90%
Figure 2: Three-stage training framework. Stage 1 constructs validated SFT slates from real contexts. Stage 2 trains a click reward model and optimizes five non-grounding rewards with GRPO. Stage 3 derives an image-first source–target grounding reward and supplies it as a sixth signal to a separate GRPO run, initialized from SFT and otherwise using the Stage 2 pipeline.
Figure 2: Three-stage training framework. Stage 1 constructs validated SFT slates from real contexts. Stage 2 trains a click reward model and optimizes five non-grounding rewards with GRPO. Stage 3 derives an image-first source–target grounding reward and supplies it as a sixth signal to a separate GRPO run, initialized from SFT and otherwise using the Stage 2 pipeline.
Table 2: Cumulative ablation of RL components. All rows start from the SFT policy.
RL configurationGSB vs base ↑Ground. ↓Redund. ↓
SFT initialization+3323.017.2
+ core RL rewards+3744.428.1
+ max-pair diversity+3834.019.1
+ norm./dynamic weights+4053.711.9
+ grounding reward+𝟒𝟒𝟔0.98.8
Figure 3: SFT data-construction pipeline (general track). Pale blocks are the data surviving each stage; grey ribbons show removed records and their filtering reasons.
Figure 3: SFT data-construction pipeline (general track). Pale blocks are the data surviving each stage; grey ribbons show removed records and their filtering reasons.
Table 3: Expert-transfer reward-model study (2,199 within-request pairs; independently trained reward-model checkpoints).
Reward modelPairsExpert agr. ↑Good−Bad ↑Low-tier inv. ↓
Raw-pair (2B)125K0.490+0.0150.090
Position-aware (2B)160K0.515+0.0480.040
Position-aware (8B)160K0.569+0.0860.041
Table 4: Verifier evaluation on the visual-consistency calibration set.
Visual inconsistency recall ↑False
VerifierMissing src.Already sat.Unionrejection ↓
Single-pass VLM61.6%43.4%47.5%22.2%
Source–target92.9%74.5%78.7%0.6%
Table 5: Stage-wise visual inconsistency by error type. Union denotes the rate of suggestions with either error.
PolicyMissing source ↓Already-satisfied target ↓Visual inconsistency (union) ↓
SFT (Stage 1)0.74%2.26%3.0%
SFT + RL (Stage 2)0.70%3.00%3.7%
Full framework (Stage 3)0.42%0.48%0.9%
Table 6: Preference-data funnel. Counts in the first three rows are impression-level unless marked as suggestion-level.
StageFiltering operationOutput
SourceThree displayed suggestions and a non-empty click663,926 requests
Record filteringRemove click not in display (1,861), invalid intent (45,022), invalid clicked type/sub-tag (120,344), fewer than two usable continuation suggestions (15,344), and repeated sessions (249,062)232,293 impressions
Image transferRetain records with a successfully transferred image232,281 impressions
Pair eligibilityRemove unusable query (944), query-copy click (66), missing final intent (30), and no usable suggestion above the click (102,491)128,750 impressions
Suggestion pruningDrop generic refresh-like (39,829), non-continuation (45,420), and below-click options (204,173); these are suggestion-level counts173,071 pairs
Request splitGroup by request ID164,401 train / 8,670 validation pairs
Table 7: Per-candidate dimensions emitted by one verifier call.
DimensionJudged fromUsed for
missing sourcesource existence bits with evidenceRL grounding g^
already satisfiedtarget vs. current stateRL grounding g^
plausibilityfit to image and scenemonitoring only
effective changenon-trivial visible changemonitoring only
content densityglobal appearance vs. object/region editlength budget class
evidence, statespointing phrasesaudits and error analysis
Table 8: Source and target semantics by edit type.
OperationRequired sourceTarget check
Addanchor/context entityrequested addition not satisfied
Removeobject to removeobject not already absent
Modifyobject and current statenew state not already satisfied
Relocateobject and reference regionrequested relation not satisfied
Global styleimagestyle change checkable or uncertain
Table 9: Diversity signal and aggregation on the semantic-redundancy probe.
VariantAdapt. λRedund. ↓Near-dup ↓
SFT×14.8%1.4%
Jaccard×28.0%2.2%
Embed., mean×30.0%2.2%
Embed., max-pair×20.8%0.8%
Embed., max-pair9.6%0.4%
Table 10: RM-only genericization stress test. Cross-image reuse and within-list redundancy are measured at the final checkpoint.
StepsΔ click-RMReuse ↓Redund. ↓
30+0.31
80+1.25
160+2.4699.9%0.1%

Findings

  • In offline evaluation, the full three-stage framework reduced visual inconsistency from 3.7% (Stage 2) to 0.9%, relative to Stage 2, while preserving expert-rated suggestion quality.
  • In a 14-day user-randomized A/B test on Qwen App involving millions of users, the full framework improved recommendation CTR by 32.70%, image take-away rate by 16.32%, and average conversation turns per user by 39.90% relative to the previously deployed prompt-engineered policy, all statistically significant (p<0.05).
  • Training with the click reward model alone caused the policy to collapse into repeating one generic, source-free edit across almost every image (99.9% cross-image reuse after 160 steps), showing that click preference alone is exploitable without the PPL, diversity, and grounding safeguards.
  • With the full reward set, the deployed policy avoided this collapse, showing 54.3% cross-image reuse versus 57.3% for the SFT baseline, with similarly small length variation.
  • Without the content-aware length reward, the click reward model's bias toward longer text pushed the mean suggestion length from 11.9 to 17.0 characters via non-essential wording rather than added content.

Where it can be used

  • Designing follow-up suggestion features for conversational image generation or editing tools
  • Building click-based reward models for recommendation systems while mitigating display-position bias through position-aware preference pairs
  • Using a separate verifier model as a training-time reward signal to enforce factual/visual grounding without adding serving latency
  • Balancing multiple competing objectives (preference, diversity, length, format validity) within a single multi-objective RL training pipeline

Limits and open work

  • The study is specific to Qwen App and the Qwen3-VL model family (8B policy, 30B verifier); transfer to other products or model families is not demonstrated.
  • The visual verifier is used only during training and checkpoint selection, not at serving time, so the deployed policy's visual errors are reduced but not directly checked at inference.
  • The verifier fails open on subjective or unclear target states, meaning some real visual inconsistencies may go undetected.
  • The online A/B test ran for 14 days on a fixed 5% traffic allocation; longer-term effects or behavior under different traffic conditions are not reported.
  • The authors note that position-aware click pairing reduces but does not fully remove display-position bias in the preference data.

Why it matters

As conversational assistants increasingly help users iterate on images, follow-up suggestions that ignore the actual image content waste user effort and erode trust; prior recommendation research mostly targeted text-only dialogue. This work shows, with a live deployment and millions of users, that combining human-defined intents, real click behavior, and explicit visual verification can measurably improve engagement while reducing visually invalid suggestions.

Terms in this paper

  • SFT (supervised fine-tuning) · Training a model further on curated example inputs and target outputs
  • GRPO · A reinforcement learning method that compares multiple sampled outputs against each other to push the policy toward higher-reward ones
  • reward model (RM) · A model trained on signals like user clicks to score how preferable a suggestion is
  • visual inconsistency · When a suggested edit requires an object/state missing from the image, or requests a state that already holds
  • GSB (Good-Same-Bad) · A score representing how much better or worse expert raters judged a policy compared to the reference policy

Figures we cannot republish

  • Figure 4: Source–target visual consistency checks. A suggestion is invalid when it presupposes an absent source or requests a visually checkable target state that already holds.
See the figures in the original paper →

Original abstract (English)

Conversational assistants increasingly recommend follow-up edits to help users continue a task. Existing systems primarily target text-only interactions, leaving image-creation conversations underexplored. In image-creation tasks, useful follow-up edit suggestions must reflect user preferences, offer diverse directions, and remain executable on the current image. We collected 100,000 real multi-turn image-creation conversation samples from Qwen App and found that 80.1% are image-dependent, underscoring the need for multimodal recommendation. We address this setting with a three-stage framework. In Stage 1, we use real online data to build a human-reviewed table of appropriate follow-up editing intents, then create SFT targets and fine-tune a multimodal policy. In Stage 2, to align rule-guided SFT suggestions with actual user choices, we use user click feedback to optimize the policy through multi-objective reinforcement learning. In Stage 3, to reduce visual inconsistencies between suggested edits and the current image, we introduce a visual verifier as additional training supervision. Extensive experiments demonstrate that our framework significantly outperforms baselines on both automatic and human evaluations. In a live user-randomized A/B test with millions of users, our final framework reduces visual inconsistency from 3.7% to 0.9%. Furthermore, it significantly improves recommendation CTR by 32.70%, image take-away rate by 16.32%, and average conversation turns per user by 39.90% (all p<0.05).

Authors · Zhijing Zhang

Read on arXiv

Latest papers

All papers →

Latest from METAL MEDIA

Figures: Zhijing Zhang et al., arXiv:2608.07565, CC BY 4.0