Imbalance in Balance: Online Concept Balancing in Generation Models

  • 2025-07-17 17:59:47
  • Yukai Shi, Jiarong Ou, Rui Chen, Haotian Yang, Jiahao Wang, Xin Tao, Pengfei Wan, Di Zhang, Kun Gai
  • 0

Abstract

In visual generation tasks, the responses and combinations of complexconcepts often lack stability and are error-prone, which remains anunder-explored area. In this paper, we attempt to explore the causal factorsfor poor concept responses through elaborately designed experiments. We alsodesign a concept-wise equalization loss function (IMBA loss) to address thisissue. Our proposed method is online, eliminating the need for offline datasetprocessing, and requires minimal code changes. In our newly proposed complexconcept benchmark Inert-CompBench and two other public test sets, our methodsignificantly enhances the concept response capability of baseline models andyields highly competitive results with only a few codes.

 

Quick Read (beta)

loading the full paper ...