← Search

Haixu Wu

26 accepted papers

2026

Brep2Shape: Boundary and Shape Representation Alignment via Self-supervised Transformers

ICML 2026poster

Boundary representation (B-rep) is the industry standard for computer-aided design (CAD). While deep learning shows promise in processing B-rep models, existing methods suffer from a representation gap: continuous approaches offer analytical precision but are visually abstract, whereas discrete meth…

Cited by 0SourceScholar
2026

GeoPT: Scaling Physics Simulation via Lifted Geometric Pre-Training

ICML 2026poster

Neural simulators promise efficient surrogates for physics simulation, but scaling them is bottlenecked by the prohibitive cost of generating high-fidelity training data. Pre-training on abundant off-the-shelf geometries offers a natural alternative, yet faces a fundamental gap: supervision on stati…

Cited by 0SourceScholar
2026

Transolver-3: Scaling Up Transformer Solvers to Industrial-Scale Geometries

ICML 2026poster

Deep learning has emerged as a transformative tool for the neural surrogate modeling of partial differential equations (PDEs), known as neural PDE solvers. However, scaling these solvers to industrial-scale geometries with over $10^8$ cells remains a fundamental challenge due to the prohibitive memo…

Cited by 0SourceScholar
2025

FlashBias: Fast Computation of Attention with Bias

NeurIPS 2025poster

Attention with bias, which extends standard attention by introducing prior knowledge as an additive bias matrix to the query-key scores, has been widely deployed in vision, language, protein-folding and other advanced scientific models, underscoring its status as a key evolution of this foundational…

Cited by 0SourcecodeScholar
2025

PhySense: Sensor Placement Optimization for Accurate Physics Sensing

NeurIPS 2025oral

Physics sensing plays a central role in many scientific and engineering domains, which inherently involves two coupled tasks: reconstructing dense physical fields from sparse observations and optimizing scattered sensor placements to observe maximum information. While deep learning has made rapid ad…

Cited by 0SourcecodeScholar
2025

Transolver++: An Accurate Neural Solver for PDEs on Million-Scale Geometries

ICML 2025poster

Although deep models have been widely explored in solving partial differential equations (PDEs), previous works are primarily limited to data only with up to tens of thousands of mesh points, far from the million-point scale required by industrial simulations that involve complex geometries. In the…

Cited by 1SourcePDFScholar
2025

Unisolver: PDE-Conditional Transformers Towards Universal Neural PDE Solvers

ICML 2025poster

Deep models have recently emerged as promising tools to solve partial differential equations (PDEs), known as neural PDE solvers. While neural solvers trained from either simulation data or physics-informed loss can solve PDEs reasonably well, they are mainly restricted to a few instances of PDEs, e…

2024

DeepLag: Discovering Deep Lagrangian Dynamics for Intuitive Fluid Prediction

NeurIPS 2024poster

Accurately predicting the future fluid is vital to extensive areas such as meteorology, oceanology, and aerodynamics. However, since the fluid is usually observed from the Eulerian perspective, its moving and intricate dynamics are seriously obscured and confounded in static grids, bringing thorny c…

2024

HelmFluid: Learning Helmholtz Dynamics for Interpretable Fluid Prediction

ICML 2024poster

Fluid prediction is a long-standing challenge due to the intrinsic high-dimensional non-linear dynamics. Previous methods usually utilize the non-linear modeling capability of deep models to directly estimate velocity fields for future prediction. However, skipping over inherent physical properties…

2024

Mobile Attention: Mobile-Friendly Linear-Attention for Vision Transformers

ICML 2024poster

Vision Transformers (ViTs) excel in computer vision tasks due to their ability to capture global context among tokens. However, their quadratic complexity $\mathcal{O}(N^2D)$ in terms of token number $N$ and feature dimension $D$ limits practical use on mobile devices, necessitating more mobile-frie…

2024

RoPINN: Region Optimized Physics-Informed Neural Networks

NeurIPS 2024poster

Physics-informed neural networks (PINNs) have been widely applied to solve partial differential equations (PDEs) by enforcing outputs and gradients of deep models to satisfy target equations. Due to the limitation of numerical computation, PINNs are conventionally optimized on finite selected points…

2024

TimeMixer: Decomposable Multiscale Mixing for Time Series Forecasting

ICLR 2024poster

Time series forecasting is widely used in extensive applications, such as traffic planning and weather forecasting. However, real-world time series usually present intricate temporal variations, making forecasting extremely challenging. Going beyond the mainstream paradigms of plain decomposition an…

2024

TimeSiam: A Pre-Training Framework for Siamese Time-Series Modeling

ICML 2024poster

Time series pre-training has recently garnered wide attention for its potential to reduce labeling expenses and benefit various downstream tasks. Prior methods are mainly based on pre-training techniques well-acknowledged in vision or language, such as masked modeling and contrastive learning. Howev…

2024

TimeXer: Empowering Transformers for Time Series Forecasting with Exogenous Variables

NeurIPS 2024poster

Deep models have demonstrated remarkable performance in time series forecasting. However, due to the partially-observed nature of real-world applications, solely focusing on the target of interest, so-called endogenous variables, is usually insufficient to guarantee accurate forecasting. Notably, a…

2024

Transolver: A Fast Transformer Solver for PDEs on General Geometries

ICML 2024spotlight

Transformers have empowered many milestones across various fields and have recently been applied to solve partial differential equations (PDEs). However, since PDEs are typically discretized into large-scale meshes with complex geometries, it is challenging for Transformers to capture intricate phys…

2024

iTransformer: Inverted Transformers Are Effective for Time Series Forecasting

ICLR 2024spotlight

The recent boom of linear forecasting models questions the ongoing passion for architectural modifications of Transformer-based forecasters. These forecasters leverage Transformers to model the global dependencies over temporal tokens of time series, with each token formed by multiple variates of th…

2023

SimMTM: A Simple Pre-Training Framework for Masked Time-Series Modeling

NeurIPS 2023spotlight

Time series analysis is widely used in extensive areas. Recently, to reduce labeling expenses and benefit various tasks, self-supervised pre-training has attracted immense interest. One mainstream paradigm is masked modeling, which successfully pre-trains deep models by learning to reconstruct the m…

2023

Solving High-Dimensional PDEs with Latent Spectral Models

ICML 2023poster

Deep models have achieved impressive progress in solving partial differential equations (PDEs). A burgeoning paradigm is learning neural operators to approximate the input-output mappings of PDEs. While previous deep models have explored the multiscale architectures and various operator designs, the…

2023

TimesNet: Temporal 2D-Variation Modeling for General Time Series Analysis

ICLR 2023poster

Time series analysis is of immense importance in extensive applications, such as weather forecasting, anomaly detection, and action recognition. This paper focuses on temporal variation modeling, which is the common key problem of extensive analysis tasks. Previous methods attempt to accomplish this…

2022

Anomaly Transformer: Time Series Anomaly Detection with Association Discrepancy

ICLR 2022spotlight

Unsupervised detection of anomaly points in time series is a challenging problem, which requires the model to derive a distinguishable criterion. Previous methods tackle the problem mainly through learning pointwise representation or pairwise association, however, neither is sufficient to reason abo…

2022

Flowformer: Linearizing Transformers with Conservation Flows

ICML 2022spotlight

Transformers based on the attention mechanism have achieved impressive success in various areas. However, the attention mechanism has a quadratic complexity, significantly impeding Transformers from dealing with numerous tokens and scaling up to bigger models. Previous methods mainly utilize the sim…

2022

Non-stationary Transformers: Exploring the Stationarity in Time Series Forecasting

NeurIPS 2022accept

Transformers have shown great power in time series forecasting due to their global-range modeling ability. However, their performance can degenerate terribly on non-stationary real-world data in which the joint distribution changes over time. Previous studies primarily adopt stationarization to atte…

2022

Supported Policy Optimization for Offline Reinforcement Learning

NeurIPS 2022accept

Policy constraint methods to offline reinforcement learning (RL) typically utilize parameterization or regularization that constrains the policy to perform actions within the support set of the behavior policy. The elaborative designs of parameterization methods usually intrude into the policy netwo…

2021

Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting

NeurIPS 2021poster

Extending the forecasting time is a critical demand for real applications, such as extreme weather early warning and long-term energy consumption planning. This paper studies the long-term forecasting problem of time series. Prior Transformer-based models adopt various self-attention mechanisms to d…

2021

MotionRNN: A Flexible Model for Video Prediction With Spacetime-Varying Motions

CVPR 2021poster

This paper tackles video prediction from a new dimension of predicting spacetime-varying motions that are incessantly changing across both space and time. Prior methods mainly capture the temporal state transitions but overlook the complex spatiotemporal variations of the motion itself, making them…

Cited by 197PDFcodeScholar