← Search

Dai Shi

10 accepted papers

2026

ATOM: A Pretrained Neural Operator for Multitask Molecular Dynamics

ICLR 2026poster

Molecular dynamics (MD) simulations underpin modern computational drug discovery, materials science, and biochemistry. Recent machine learning models provide high-fidelity MD predictions without the need for repeated quantum-mechanical force calculations, enabling significant speedups over conventio…

Cited by 0SourcecodeScholar
2026

Expanding the Chaos: Neural Operator for Stochastic (Partial) Differential Equations

ICML 2026poster

Stochastic differential equations (SDEs) and stochastic partial differential equations (SPDEs) are fundamental for modeling stochastic dynamics across the natural sciences and modern machine learning. Learning their solution operators with deep learning models promises fast solvers and new perspecti…

Cited by 0SourceScholar
2026

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

ICML 2026poster

Neural rough differential equations (NRDEs) learn continuous-time dynamics from irregularly sampled sequences by encoding the input path with signature features, providing robustness to discretisation and sampling irregularity. However, existing NRDEs implicitly rely on algebraic identities that can…

Cited by 0SourceScholar
2026

S$^3$GNN: Efficient Global Mixing and Local Message Passing for Long-Range Graph Learning

ICML 2026spotlight

Message-passing neural networks (MPNNs) often suffer from an information bottleneck when capturing long-range dependencies, leading to the oversquashing (OSQ) phenomenon. Alongside spatial connectivity enrichment (e.g., rewiring), recent studies have shown that spectral filtering can yield strong lo…

Cited by 0SourceScholar
2025

A Riemannian Approach to Ground Metric Learning for Optimal Transport

ICASSP 2025accepted

Optimal transport (OT) theory has attracted much attention in machine learning and signal processing applications. OT defines a notion of distance between probability distributions of source and target data points. A crucial factor that influences OT-based distances is the ground metric of the embed…

Cited by 0SourceScholar
2025

ACT as Human: Multimodal Large Language Model Data Annotation with Critical Thinking

NeurIPS 2025poster

Supervised learning relies on high-quality labeled data, but obtaining such data through human annotation is both expensive and time-consuming. Recent work explores using large language models (LLMs) for annotation, but LLM-generated labels still fall short of human-level quality. To address this pr…

Cited by 0SourceScholar
2025

Diffusing to the Top: Boost Graph Neural Networks with Minimal Hyperparameter Tuning

ICLR 2025poster

Graph Neural Networks (GNNs) are proficient in graph representation learning and achieve promising performance on versatile tasks such as node classification and link prediction. Usually, a comprehensive hyperparameter tuning is essential for fully unlocking GNN's top performance, especially for com…

2025

When Graph Neural Networks Meet Dynamic Mode Decomposition

ICLR 2025poster

Graph Neural Networks (GNNs) have emerged as fundamental tools for a wide range of prediction tasks on graph-structured data. Recent studies have drawn analogies between GNN feature propagation and diffusion processes, which can be interpreted as dynamical systems. In this paper, we delve deeper int…

Cited by 0SourcePDFScholar