collaborators

10 papers

cs.CC2026

On Computational Hardness of Mistake-Bounded Language Generation: A Random-Oracle Query Separation

Xiaoyu Li, Andi Han, Dai Shi +2

Generation in the limit guarantees eventual generation for every countable collection of infinite languages in the model of Kleinberg and Mullainathan [KM24], while closure dimensi…

cs.LG2026

Flood and Harvest: The Provable Necessity of Trivia for Generating Valuable Mathematics via the Lens of Language Generation in the Limit

Xiaoyu Li, Andi Han, Dai Shi +3

AI systems coupled to proof assistants now generate formal mathematics at scale, and the gap between what a checker can verify and what a mathematician would value has become the b…

cs.LG2026

Learning Manifold and Itô Dynamics with Branched Neural Rough Differential Equations

Luke Thompson, Dai Shi, Lequan Lin +2

Neural rough differential equations (NRDEs) stay accurate under irregular sampling while taking far fewer integration steps than standard neural differential equations, summarising…

cs.LG2026

SGNN: Efficient Global Mixing and Local Message Passing for Long-Range Graph Learning

Dai Shi, Luke Thompson, Linhan Luo +4

Message-passing neural networks (MPNNs) often suffer from an information bottleneck when capturing long-range dependencies, leading to the oversquashing (OSQ) phenomenon. Alongside…

cs.LG2026

SPDEBench: An Extensive Benchmark for Learning Stochastic PDEs

Yuantu Zhu, Zheyan Li, Dai Shi +8

Stochastic Partial Differential Equations (SPDEs) driven by random noise play a central role in modeling physical processes with rough spatio-temporal dynamics, such as turbulence…

cs.LG2026

ATOM: A Pretrained Neural Operator for Multitask Molecular Dynamics

Luke Thompson, Davy Guan, Dai Shi +3

Molecular dynamics (MD) simulations underpin modern computational drug discovery, materials science, and biochemistry. Recent machine learning models provide high-fidelity MD predi…