Skip to content
01

About Me

I am a Ph.D. student at Westlake University (Fall 2025), advised by Prof. Tailin Wu. I am deeply interested in the physics of intelligence. I do research to reduce my perplexity about the world we live in.

I study the interplay between representation learning and generative modeling, drawing on statistical physics and information theory. Some questions that guide my research are:

These questions guide my current focus on end-to-end (single-step) latent generative models: jointly learning representations and generators to achieve both efficient learning and few-step generation.

I am actively seeking internships with leading academic and industry research groups. Please get in touch if our interests overlap.

Previously, I studied astronomy at Nanjing University and the Australian National University, working with Prof. Yuan-Sen Ting.

02

Research Highlight

NeurIPS 2026

Feature Information Dynamics in Diffusion

Using the I-MMSE relation, we define feature information density during diffusion for any feature in any representation space, quantifying the order in which different features emerge.

Across the four representation spaces we study, only RAE exhibits the class → mask → Canny order, and diffusion models converge fastest in this space. We hypothesize that these ordered feature dynamics help explain its faster convergence.

Feature information dynamics for class, mask, and Canny in pixel space
Feature information dynamics in pixel space.
Explore the definition and compare four representation spaces Collapse details

Diffusion models generate data through a continuum of denoising problems, and are widely observed to reveal coarse structure before fine detail. Yet, this intuition is mostly empirical and qualitative—and, crucially, it need not hold after data are mapped into a representation space.

Feature information density

Let \(X\in\mathbb{R}^d\) be clean data (e.g., an image or latent), \(Y\) a feature of \(X\) (e.g., class, mask, or Canny), and \(X_\gamma=\sqrt{\gamma}X+N\) the Gaussian-corrupted data at signal-to-noise ratio \(\gamma\), where \(N\sim\mathcal{N}(0,I)\) is independent noise. We define the feature information density as

\[D_Y(\gamma):=\frac{\mathrm d}{\mathrm d\gamma}I(Y;X_\gamma).\]

Intuitively, \(D_Y(\gamma)\) distributes the total information about \(Y\) along the SNR axis: it measures how much additional feature information becomes accessible from an infinitesimal increase in SNR. A peak therefore identifies the noise level at which that feature is revealed most rapidly during denoising.

Let \(m_\varnothing(\gamma)\) and \(m_Y(\gamma)\) denote the optimal unconditional and feature-conditional denoising losses (MMSE). Conditioning cannot increase this loss. I-MMSE connects feature information density to their difference:

\[D_Y(\gamma)=\frac{1}{2}\left[m_\varnothing(\gamma)-m_Y(\gamma)\right].\]

Its trajectory across noise levels describes how that feature's information is distributed over the generation process. Empirically, we find that class, mask, and Canny information exhibit markedly different dynamics across pixel, SDVAE, VAVAE, and RAE spaces. Only RAE exhibits the class → mask → Canny order. We hypothesize that these ordered feature dynamics may explain why diffusion models converge fastest in the RAE space.

Feature information densities across pixel, SDVAE, VAVAE, and RAE spaces
Feature information dynamics across pixel, SDVAE, VAVAE, and RAE spaces. Curves show normalized information densities for successive class, mask, and Canny conditions along log-SNR. Only RAE exhibits the class → mask → Canny order.
03

Selected Publications

NeurIPS 2026

Feature Information Dynamics in Diffusion

Jia-Shu Pan, Tao Zhang, Yufei Huang, Yanjun Sheng, Tailin Wu

A quantitative framework for locating hierarchical features along diffusion trajectories and relating their temporal organization to representation-dependent convergence.

ICLR 2026

VFScale: Intrinsic Reasoning through Verifier-Free Test-time Scalable Diffusion Model

Tao Zhang*, Jia-Shu Pan*, Ruiqi Feng, Tailin Wu

* Equal contribution

VFScale trains a diffusion model's own energy to serve as a verifier and combines it with hybrid Monte Carlo Tree Search, enabling verifier-free test-time scaling on Maze and Sudoku.

04

Hobbies

🏃 Running 🎞️ Anime
05

Contact

Interesting questions, disagreements, and half-formed ideas are always welcome. If our research interests overlap, I would be happy to talk.

Email: panjiashu@westlake.edu.cn

Last updated: © 2026 Jia-Shu Pan · Built with Jekyll and al-folio · Website design inspired by Huanran Chen · Milky Way photographed by the author in the Tengger Desert on the night of August 12, 2026.