Publication · TMLR 2026

BalancedDPO: Adaptive Multi-Metric Alignment

Dipesh Tamboli, Souradip Chakraborty, Aditya Malusare, Biplab Banerjee, Amrit Singh Bedi, Vaneet Aggarwal

Transactions on Machine Learning Research (TMLR), 2026 · First author

← All publications
Grid of images generated from prompts sampled from the Pick-a-Pic dataset
Examples generated from randomly sampled prompts in the Pick-a-Pic dataset.

TL;DR

Aligning text-to-image diffusion models with human preferences usually means optimizing a single reward — aesthetics, or a CLIP/semantic score, or a human-preference model. That can bias the model toward the chosen metric and neglect the others. BalancedDPO aligns to several preference metrics at once inside the Direct Preference Optimization (DPO) framework: instead of averaging rewards (where scale differences let one metric dominate), it takes a majority vote among multiple preference scorers to decide the preferred image, and feeds that consensus into DPO with dynamically updated reference models for more stable training.

Key ideas

  • The problem: multiple, sometimes conflicting metrics (semantic consistency, aesthetics, human preference). Single-metric or scalarized (weighted-sum) rewards bias the model toward one criterion.
  • Majority-vote consensus: aggregate agreement across diverse scorers in the preference space rather than mixing raw reward values — sidestepping reward-scale conflicts.
  • Consensus-in-the-loop DPO: the vote is integrated directly into the DPO objective, with dynamic reference-model updates, yielding stabler gradient directions across heterogeneous metrics.
  • Simple & scalable: keeps the standard DPO pipeline — no reward model is needed at inference.
BalancedDPO method overview: majority-vote consensus across preference metrics feeding DPO
Method overview. Multiple preference scorers vote on the preferred image; the majority-vote consensus is fed into the DPO objective with dynamically updated reference models.

Results

  • Higher pairwise win rates than DiffusionDPO across all four metrics on Pick-a-Pic, PartiPrompt, and HPD in the SD 1.5 and SDXL evaluations.
  • Holds across backbones: Stable Diffusion 1.5, SD 2.1, and SDXL.
  • Ablations confirm both the majority-vote aggregation and dynamic reference updating drive the gains.

Cite

@article{tamboli2026balanceddpo,
  title   = {BalancedDPO: Adaptive Multi-Metric Alignment},
  author  = {Tamboli, Dipesh and Chakraborty, Souradip and Malusare, Aditya and
             Banerjee, Biplab and Bedi, Amrit Singh and Aggarwal, Vaneet},
  journal = {Transactions on Machine Learning Research},
  year    = {2026},
  url     = {https://openreview.net/forum?id=8HRID5VLQw}
}
← Back to home