2 papers
cs.LG2025
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.…
cs.CV2025
Compact neural networks for astronomy with optimal transport bias correction
Shuhuan Wang, Yuzhen Xie, Jiayi Li
Astronomical imaging confronts an efficiency-resolution tradeoff that limits large-scale morphological classification and redshift prediction. We introduce WaveletMamba, a theory-d…