← Search

Yiheng Du

7 accepted papers

2025

AlignedGen: Aligning Style Across Generated Images

NeurIPS 2025poster

Diffusion-based generative models struggle to maintain high style consistency across generated images via text description. Although several style-aligned image generation methods have been proposed to address this issue, they exhibit suboptimal performance and are primarily built upon the U-Net arc…

Cited by 0SourcecodeScholar
2025

EddyFormer: Accelerated Neural Simulations of Three-Dimensional Turbulence at Scale

NeurIPS 2025poster

Computationally resolving turbulence remains a central challenge in fluid dynamics due to its multi-scale interactions. Fully resolving large-scale turbulence through direct numerical simulation (DNS) is computationally prohibitive, motivating data-driven machine learning alternatives. In this work,…

Cited by 0SourceScholar
2025

Homomorphism Expressivity of Spectral Invariant Graph Neural Networks

ICLR 2025oral

Graph spectra are an important class of structural features on graphs that have shown promising results in enhancing Graph Neural Networks (GNNs). Despite their widespread practical use, the theoretical understanding of the power of spectral invariants --- particularly their contribution to GNNs ---…

Cited by 0SourcePDFScholar
2024

Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness

ICLR 2024oral

Designing expressive Graph Neural Networks (GNNs) is a fundamental topic in the graph learning community. So far, GNN expressiveness has been primarily assessed via the Weisfeiler-Lehman (WL) hierarchy. However, such an expressivity measure has notable limitations: it is inherently coarse, qualitati…

2024

Neural Spectral Methods: Self-supervised learning in the spectral domain

ICLR 2024poster

We present Neural Spectral Methods, a technique to solve parametric Partial Differential Equations (PDEs), grounded in classical spectral methods. Our method uses orthogonal bases to learn PDE solutions as mappings between spectral coefficients, instantiating a spectral-based neural operator. In con…

2024

Scaling physics-informed hard constraints with mixture-of-experts

ICLR 2024poster

Imposing known physical constraints, such as conservation laws, during neural network training introduces an inductive bias that can improve accuracy, reliability, convergence, and data efficiency for modeling physical dynamics. While such constraints can be softly imposed via loss function penaltie…

2023

A Complete Expressiveness Hierarchy for Subgraph GNNs via Subgraph Weisfeiler-Lehman Tests

ICML 2023poster

Recently, subgraph GNNs have emerged as an important direction for developing expressive graph neural networks (GNNs). While numerous architectures have been proposed, so far there is still a limited understanding of how various design paradigms differ in terms of expressive power, nor is it clear w…