← Search

Sizhuang He

4 accepted papers

2026

Learning Permutation Distributions via Reflected Diffusion on Ranks

ICML 2026poster

The finite symmetric group $S_n$ provides a natural domain for permutations, yet learning probability distributions on $S_n$ is challenging due to its factorially growing size and discrete, non-Euclidean structure. Recent permutation diffusion methods define forward noising via shuffle-based random …

Cited by 0Source ↗Scholar ↗
2026

STRIDE: Post-Training LLMs to Reason and Refine Bio-Sequences via Edit Trajectories

ICML 2026poster

Discrete biological sequence optimization demands iterative refinement while satisfying strict syntactic constraints. Diffusion-based approaches provide strong progressive refinement but are not naturally aligned with discrete, grammar-constrained edit operations, whereas autoregressive LLMs readily…

Cited by 0Source ↗Scholar ↗
2025

Intelligence at the Edge of Chaos

ICLR 2025poster

We explore the emergence of intelligent behavior in artificial systems by investigating how the complexity of rule-based systems influences the capabilities of models trained to predict these rules. Our study focuses on elementary cellular automata (ECA), simple yet powerful one-dimensional systems…

Cited by 2Source ↗PDF ↗Scholar ↗
2025

Non-Markovian Discrete Diffusion with Causal Language Models

NeurIPS 2025poster

Discrete diffusion models offer a flexible, controllable approach to structured sequence generation, yet they still lag behind causal language models in expressive power. A key limitation lies in their reliance on the Markovian assumption, which restricts each step to condition only on the current s…

Cited by 0Source ↗Scholar ↗