activity
20222026
most citedCCDSReFormer: Traffic Flow Prediction with a Criss-Crossed Dual-Stream Enhanced Rectified Transformer Model

13 citations · 28 across the 28 of their papers we have counts for

collaborators

31 papers

cs.LG2026

Stochastic Adaptive Fourier Decomposition for Operator Learning

Pengqing Shi, Liming Zhang, Tao Qian +3

Fourier Neural Operators (FNOs) offer an efficient paradigm for solving partial differential equations (PDEs). However, FNOs rely on a fixed Fourier basis and hard frequency trunca…

cs.FL2026

Characterizing Language Generation in the Limit: Finite Witnesses and a Separation-Width Hierarchy

Xiaoyu Li, Andi Han, Jiaojiao Jiang +1

Language generation in the limit asks for valid unseen elements from every exhaustive positive presentation of an unknown infinite language. We characterize this task for arbitrary…

cs.LG2026

Nearly Tight Rademacher Bounds for Sparsely Activated Neural Networks

Xiaoyu Li, Zhizhou Sha, Jiaojiao Jiang +2

An input may activate few hidden units even when different inputs collectively use an entire network. We study the statistical complexity of this input-dependent sparsity in the on…

cs.AI2026

Matched Excess-Outranker Regularization for Candidate-Set Interference in Continual Knowledge Graph Embedding

Hao Ren, Junbin Gao, Jiaojiao Jiang

Continual knowledge graph embedding updates entity and relation representations as a graph grows. Existing methods primarily address catastrophic forgetting, but entity admission a…

cs.LG2026

Optimistic Rates for Multiclass PAC Learning

Xiaoyu Li, Andi Han, Jiaojiao Jiang +1

Worst-case multiclass bounds do not become smaller when the best classifier is already nearly correct: what is missing is an optimistic rate, a guarantee whose fluctuation scales w…

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…

math.PR2026

The Exact Worst-Case Tail Probability under Bounded Kurtosis

Xiaoyu Li, Andi Han, Jiaojiao Jiang +1

We determine exactly what a kurtosis bound buys for one-sided tail control. For the class of real random variables with mean , variance , and fourth moment a…

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

SirenFNO: Efficient and Full Frequency Learning of Fourier Neural Operators

Pengqing Shi, Jie Yin, Stephen Tierney +1

Fourier neural operators (FNOs) are effective and efficient surrogates for approximating solutions of PDEs and generalize across discretizations. However, owing to the reliance on…

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

Contrastive Identification and Generation in the Limit

Xiaoyu Li, Andi Han, Jiaojiao Jiang +1

In the classical identification in the limit model of Gold [1967], a stream of positive examples is presented round by round, and the learner must eventually recover the target hyp…

cs.LG2026

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

Dai Shi, Lequan Lin, Andi Han +4

Stochastic differential equations (SDEs) and stochastic partial differential equations (SPDEs) are fundamental for modeling stochastic dynamics across the natural sciences and mode…

cs.LG2025

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

Lequan Lin, Dai Shi, Andi Han +7

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 lan…

cs.LG2025

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…

cs.LG2025

Graph Pseudotime Analysis and Neural Stochastic Differential Equations for Analyzing Retinal Degeneration Dynamics and Beyond

Dai Shi, Kuan Yan, Lequan Lin +6

Understanding disease progression at the molecular pathway level usually requires capturing both structural dependencies between pathways and the temporal dynamics of disease evolu…

cs.LG2024

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

Lequan Lin, Dai Shi, Andi Han +2

Graph Neural Networks (GNNs) are proficient in graph representation learning and achieve promising performance on versatile tasks such as node classification and link prediction. U…

cs.LG2024

When Graph Neural Networks Meet Dynamic Mode Decomposition

Dai Shi, Lequan Lin, Andi Han +3

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 featur…

cs.LG2024★ 1 cited

A Riemannian Approach to Ground Metric Learning for Optimal Transport

Pratik Jawanpuria, Dai Shi, Bamdev Mishra +1

Optimal transport (OT) theory has attracted much attention in machine learning and signal processing applications. OT defines a notion of distance between probability distributions…

q-bio.GN2024

Machine Learning-Based Prediction of Key Genes Correlated to the Subretinal Lesion Severity in a Mouse Model of Age-Related Macular Degeneration

Kuan Yan, Yue Zeng, Dai Shi +5

Age-related macular degeneration (AMD) is a major cause of blindness in older adults, severely affecting vision and quality of life. Despite advances in understanding AMD, the mole…

cs.LG2024

Unleash Graph Neural Networks from Heavy Tuning

Lequan Lin, Dai Shi, Andi Han +2

Graph Neural Networks (GNNs) are deep-learning architectures designed for graph-type data, where understanding relationships among individual observations is crucial. However, achi…

cs.LG2024★ 13 cited

CCDSReFormer: Traffic Flow Prediction with a Criss-Crossed Dual-Stream Enhanced Rectified Transformer Model

Zhiqi Shao, Michael G. H. Bell, Ze Wang +3

Accurate, and effective traffic forecasting is vital for smart traffic systems, crucial in urban traffic planning and management. Current Spatio-Temporal Transformer models, despit…

cs.LG2024

Design Your Own Universe: A Physics-Informed Agnostic Method for Enhancing Graph Neural Networks

Dai Shi, Andi Han, Lequan Lin +3

Physics-informed Graph Neural Networks have achieved remarkable performance in learning through graph-structured data by mitigating common GNN challenges such as over-smoothing, ov…

cs.LG2024★ 3 cited

SpecSTG: A Fast Spectral Diffusion Framework for Probabilistic Spatio-Temporal Traffic Forecasting

Lequan Lin, Dai Shi, Andi Han +1

Traffic forecasting, a crucial application of spatio-temporal graph (STG) learning, has traditionally relied on deterministic models for accurate point estimations. Yet, these mode…

cs.LG2023

Exposition on over-squashing problem on GNNs: Current Methods, Benchmarks and Challenges

Dai Shi, Andi Han, Lequan Lin +2

Graph-based message-passing neural networks (MPNNs) have achieved remarkable success in both node and graph-level learning tasks. However, several identified problems, including ov…