← Search

Xiaojun Duan

3 accepted papers

2026

The Geometric Origin of Grokking: Accelerating Generalization via Active Structural Reorganization

ICML 2026poster

Grokking, the phenomenon where models suddenly generalize long after overfitting training data, remains a puzzling challenge in neural network dynamics. Through mechanistic analysis, we find that this transition is fundamentally driven by a structural reorganization of token embeddings, with the ons…

Cited by 0SourceScholar
2024

From Fourier to Neural ODEs: Flow Matching for Modeling Complex Systems

ICML 2024poster

Modeling complex systems using standard neural ordinary differential equations (NODEs) often faces some essential challenges, including high computational costs and susceptibility to local optima. To address these challenges, we propose a simulation-free framework, called Fourier NODEs (FNODEs), tha…

Cited by 6SourcePDFScholar