SONA: Learning Conditional, Unconditional, and Matching-Aware Discriminator
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 SONA 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.
Cite
Text
Takida et al. "SONA: Learning Conditional, Unconditional, and Matching-Aware Discriminator." International Conference on Learning Representations, 2026.Markdown
[Takida et al. "SONA: Learning Conditional, Unconditional, and Matching-Aware Discriminator." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/takida2026iclr-sona/)BibTeX
@inproceedings{takida2026iclr-sona,
title = {{SONA: Learning Conditional, Unconditional, and Matching-Aware Discriminator}},
author = {Takida, Yuhta and Hayakawa, Satoshi and Shibuya, Takashi and Imaizumi, Masaaki and Murata, Naoki and Nguyen, Bac and Uesaka, Toshimitsu and Lai, Chieh-Hsin and Mitsufuji, Yuki},
booktitle = {International Conference on Learning Representations},
year = {2026},
url = {https://mlanthology.org/iclr/2026/takida2026iclr-sona/}
}