← Search

Thomas X Wang

4 accepted papers

2025

Zebra: In-Context Generative Pretraining for Solving Parametric PDEs

ICML 2025poster

Solving time-dependent parametric partial differential equations (PDEs) is challenging for data-driven methods, as these models must adapt to variations in parameters such as coefficients, forcing terms, and initial conditions. State-of-the-art neural surrogates perform adaptation through gradient-b…

Cited by 2SourcePDFScholar
2024

AROMA: Preserving Spatial Structure for Latent PDE Modeling with Local Neural Fields

NeurIPS 2024poster

We present AROMA (Attentive Reduced Order Model with Attention), a framework designed to enhance the modeling of partial differential equations (PDEs) using local neural fields. Our flexible encoder-decoder architecture can obtain smooth latent representations of spatial physical fields from a varie…

2024

Weight Conditioning for Smooth Optimization of Neural Networks

ECCV 2024poster

"In this article, we introduce a novel normalization technique for neural network weight matrices, which we term weight conditioning. This approach aims to narrow the gap between the smallest and largest singular values of the weight matrices, resulting in better-conditioned matrices. The inspiratio…

Cited by 1SourcePDFScholar
2023

Operator Learning with Neural Fields: Tackling PDEs on General Geometries

NeurIPS 2023poster

Machine learning approaches for solving partial differential equations require learning mappings between function spaces. While convolutional or graph neural networks are constrained to discretized functions, neural operators present a promising milestone toward mapping functions directly. Despite i…