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
BalancedDPO teaser: multi-metric alignment vs single-metric models
Single-metric alignment tends to win on one axis while failing another (e.g. aesthetics vs. prompt-alignment). BalancedDPO, which combines multiple preference metrics, stays strong across all of them.

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 over-fits whichever metric you pick and neglects 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 separate reward model to train 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

  • Consistently higher preference win rates over baselines on Pick-a-Pic, PartiPrompt, and HPD.
  • 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