← Search

Zhong Yi Wan

6 accepted papers

2026

Efficient, Property-Aligned Fan-Out Retrieval via RL-Amortized Diffusion

ICML 2026poster

Many modern retrieval problems are \emph{set-valued}: given a broad intent, the system must return a \emph{collection} of results that optimizes higher-order properties (e.g., diversity, coverage, complementarity, coherence) while staying grounded to a fixed database. Set-valued objectives are inher…

Cited by 0SourceScholar
2025

Diff4Steer: Steerable Diffusion Prior for Generative Music Retrieval with Semantic Guidance

ICASSP 2025accepted

Modern music retrieval systems often rely on fixed representations of user preferences, limiting their ability to capture users’ diverse and uncertain retrieval needs. To address this limitation, we introduce Diff4Steer, a novel generative retrieval framework that employs lightweight diffusion model…

Cited by 0SourceScholar
2024

DySLIM: Dynamics Stable Learning by Invariant Measure for Chaotic Systems

ICML 2024poster

Learning dynamics from dissipative chaotic systems is notoriously difficult due to their inherent instability, as formalized by their positive Lyapunov exponents, which exponentially amplify errors in the learned dynamics. However, many of these systems exhibit ergodicity and an attractor: a compact…

2023

Debias Coarsely, Sample Conditionally: Statistical Downscaling through Optimal Transport and Probabilistic Diffusion Models

NeurIPS 2023spotlight

We introduce a two-stage probabilistic framework for statistical downscaling using unpaired data. Statistical downscaling seeks a probabilistic map to transform low-resolution data from a biased coarse-grained numerical scheme to high-resolution data that is consistent with a high-fidelity scheme. O…

2023

Evolve Smoothly, Fit Consistently: Learning Smooth Latent Dynamics For Advection-Dominated Systems

ICLR 2023top-25%

We present a data-driven, space-time continuous framework to learn surrogate models for complex physical systems described by advection-dominated partial differential equations. Those systems have slow-decaying Kolmogorov n-width that hinders standard methods, including reduced order modeling, from…

Cited by 18SourcePDFScholar
2023

Neural Ideal Large Eddy Simulation: Modeling Turbulence with Neural Stochastic Differential Equations

NeurIPS 2023poster

We introduce a data-driven learning framework that assimilates two powerful ideas: ideal large eddy simulation (LES) from turbulence closure modeling and neural stochastic differential equations (SDE) for stochastic modeling. The ideal LES models the LES flow by treating each full-order trajectory a…

Cited by 8SourcePDFScholar