3 citations · 4 across the 10 of their papers we have counts for
4 papers · 1 filter
Breaking the Memory Wall: Exact Analytical Differentiation via Tiled Operator-Space Evolution
Shuhuan Wang, Yuzhen Xie, Jiayi Li +1
Selective State Space Models (SSMs) achieve linear-time inference, yet their gradient-based sensitivity analysis remains bottlenecked by O(L) memory scaling during backpropagation.…
DRIFT-Net: A Spectral--Coupled Neural Operator for PDEs Learning
Jiayi Li, Flora D. Salim
Learning PDE dynamics with neural solvers can significantly improve wall-clock efficiency and accuracy compared with classical numerical solvers. In recent years, foundation models…
Federated Multi-Objective Learning with Controlled Pareto Frontiers
Jiansheng Rao, Jiayi Li, Zhizhi Gong +2
Federated learning (FL) is a widely adopted paradigm for privacy-preserving model training, but FedAvg optimise for the majority while under-serving minority clients. Existing meth…
Benchmarking Multi-Domain Active Learning on Image Classification
Jiayi Li, Rohan Taori, Tatsunori B. Hashimoto
Active learning aims to enhance model performance by strategically labeling informative data points. While extensively studied, its effectiveness on large-scale, real-world dataset…