解决条件生成中标签分布不均与采样慢的问题,提升图像生成质量与效率。
Imbalance-Robust and Sampling-Efficient Continuous Conditional GANs via Adaptive Vicinal Learning and Auxiliary Regularization
- 自适应邻域机制根据标签密度动态调整训练范围
- 多任务判别器同时优化标签一致性和分布匹配,生成效果更好
- 相比扩散模型快300至2000倍,适合实时应用
连续条件生成模型如CcGAN和CCDM用于在标量标签(如角度、年龄、温度)条件下建模高维数据分布。但固定邻域训练对标签密度不均敏感,而CCDM依赖计算昂贵的迭代采样。本文提出CcGAN-AVAR,结合软/混合自适应邻域与辅助判别器引导正则化。自适应邻域根据目标条件周围样本分布动态构建局部半径,多任务判别器提供回归信号以保证标签一致性,并通过密度比估计实现分布匹配。理论分析揭示了自适应加权如何影响判别器的局部偏差-方差行为,混合截断减少跨条件混淆,密度比惩罚项近似皮尔逊卡方差异。在四个数据集(含新构建的不平衡数据集RC-49-I)上,分辨率从64x64到256x256共11个设置的实验表明,CcGAN-AVAR在保持单步采样效率的同时,生成质量与标签一致性显著优于基线,推理速度比CCDM快300至2000倍。
原文摘要 · Abstract (English)
Recent advances in continuous conditional generative modeling, including Continuous conditional Generative Adversarial Network (CcGAN) and Continuous Conditional Diffusion Model (CCDM), estimate high-dimensional data distributions conditioned on scalar regression labels such as angles, ages, or temperatures. However, fixed-size vicinal training in CcGAN can be sensitive to non-uniform label densities, whereas CCDM relies on computationally expensive iterative sampling. To address these issues, we propose CcGAN-AVAR, an imbalance-aware extension of CcGAN that combines soft/hybrid adaptive vicinity with auxiliary discriminator-guided regularization. The adaptive vicinity constructs a label-dependent local radius according to the available samples around each target condition, and the multi-task discriminator supplies both a regression signal for label consistency and a density-ratio-estimation signal for distribution matching. We further provide a theoretical interpretation characterizing how adaptive vicinal weighting affects the local bias-variance behavior of the discriminator target, how hybrid truncation reduces objective-level cross-condition mixing, and how the density-ratio-based generator penalty approximates a Pearson Chi-square discrepancy up to the estimation error of the density-ratio branch. Extensive experiments on four datasets, including the newly constructed imbalanced RC-49-I, covering resolutions from 64x64 to 256x256 across eleven settings, demonstrate that CcGAN-AVAR obtains strong generation quality and label consistency while preserving the one-step sampling efficiency of GANs, achieving 300x--2000x faster inference than CCDM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。