English

SONA: Learning Conditional, Unconditional, and Mismatching-Aware Discriminator

Machine Learning 2025-10-07 v1 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Deep generative models have made significant advances in generating complex content, yet conditional generation remains a fundamental challenge. Existing conditional generative adversarial networks often struggle to balance the dual objectives of assessing authenticity and conditional alignment of input samples within their conditional discriminators. To address this, we propose a novel discriminator design that integrates three key capabilities: unconditional discrimination, matching-aware supervision to enhance alignment sensitivity, and adaptive weighting to dynamically balance all objectives. Specifically, we introduce Sum of Naturalness and Alignment (SONA), which employs separate projections for naturalness (authenticity) and alignment in the final layer with an inductive bias, supported by dedicated objective functions and an adaptive weighting mechanism. Extensive experiments on class-conditional generation tasks show that \ours achieves superior sample quality and conditional alignment compared to state-of-the-art methods. Furthermore, we demonstrate its effectiveness in text-to-image generation, confirming the versatility and robustness of our approach.

Keywords

Cite

@article{arxiv.2510.04576,
  title  = {SONA: Learning Conditional, Unconditional, and Mismatching-Aware Discriminator},
  author = {Yuhta Takida and Satoshi Hayakawa and Takashi Shibuya and Masaaki Imaizumi and Naoki Murata and Bac Nguyen and Toshimitsu Uesaka and Chieh-Hsin Lai and Yuki Mitsufuji},
  journal= {arXiv preprint arXiv:2510.04576},
  year   = {2025}
}

Comments

24 pages with 9 figures

R2 v1 2026-07-01T06:18:40.631Z