4 papers
NeuraLSP: A Neural Spectral Preconditioner for Accelerating PDE Solvers
Alexander Benanti, Xi Han, Hong Qin
Solving large-scale sparse linear systems originating from partial differential equations (PDEs) is a fundamental topic in high-performance scientific computing, where precondition…
2DMamba: Efficient State Space Model for Image Representation with Applications on Giga-Pixel Whole Slide Image Classification
Jingwei Zhang, Anh Tien Nguyen, Xi Han +4
Efficiently modeling large 2D contexts is essential for various fields including Giga-Pixel Whole Slide Imaging (WSI) and remote sensing. Transformer-based models offer high parall…
LBMamba: Locally Bi-directional Mamba
Jingwei Zhang, Xi Han, Hong Qin +2
Mamba, a State Space Model (SSM) that accelerates training by recasting recurrence as a parallel scan, has recently emerged as a linearly-scaling alternative to self-attention. Bec…
GeoMaNO: Geometric Mamba Neural Operator for Partial Differential Equations
Xi Han, Jingwei Zhang, Dimitris Samaras +2
The neural operator (NO) framework has emerged as a powerful tool for solving partial differential equations (PDEs). Recent NOs are dominated by the Transformer architecture, which…