Generalization of Diffusion Models Arises with a Balanced Representation Space - Zhang, Zekai; Li, Xiao; Li, Xiang; Shi, Lianghe; Wu, Meng; Tao, Molei; Qu, Qing (International Conference on Learning Representations, 2026)