← Search

Lianghe Shi

5 accepted papers

2026

Evaluating the Representation Space of Diffusion Models via Self-Supervised Principles

ICML 2026poster

Diffusion models are effective generative frameworks with strong representation learning capabilities, yet the intrinsic properties that govern their semantic structure and generalization remain poorly understood. Drawing inspiration from self-supervised representation learning (SSL), we introduce a…

Cited by 0SourceScholar
2026

Generalization of Diffusion Models Arises with a Balanced Representation Space

ICLR 2026poster

Diffusion models generate high-quality, diverse images with great generalizability, yet when overfit to the training objective, they may memorize training samples. We analyze memorization and generalization of diffusion models through the lens of representation learning. Using a two-layer ReLU denoi…

Cited by 0SourcecodeScholar
2025

A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective

NeurIPS 2025spotlight

The widespread use of diffusion models has led to an abundance of AI-generated data, raising concerns about model collapse---a phenomenon in which recursive iterations of training on synthetic data lead to performance degradation. Prior work primarily characterizes this collapse via variance shrinka…

Cited by 0SourceScholar
2023

Adversarial Self-Training Improves Robustness and Generalization for Gradual Domain Adaptation

NeurIPS 2023poster

Gradual Domain Adaptation (GDA), in which the learner is provided with additional intermediate domains, has been theoretically and empirically studied in many contexts. Despite its vital role in security-critical scenarios, the adversarial robustness of the GDA model remains unexplored. In this pape…