← Search

Ze Cheng

8 accepted papers

2026

BridgeDrive: Diffusion Bridge Policy for Closed-Loop Trajectory Planning in Autonomous Driving

ICLR 2026poster

Diffusion-based planners have shown great promise for autonomous driving due to their ability to capture multi-modal driving behaviors. However, guiding these models effectively in reactive, closed-loop environments remains a significant challenge. Simple conditioning often fails to provide sufficie…

Cited by 0SourcecodeScholar
2026

Operator Learning with Domain Decomposition for Geometry Generalization in PDE Solving

ICLR 2026poster

Neural operators have become increasingly popular in solving partial differential equations (PDEs) due to their superior capability to capture intricate mappings between function spaces over complex domains. However, the data-hungry nature of operator learning inevitably poses a bottleneck for their…

Cited by 0SourcecodeScholar
2025

Accelerating PDE-Constrained Optimization by the Derivative of Neural Operators

ICML 2025poster

PDE-Constrained Optimization (PDECO) problems can be accelerated significantly by employing gradient-based methods with surrogate models like neural operators compared to traditional numerical solvers. However, this approach faces two key challenges: (1) **Data inefficiency**: Lack of efficient dat…

Cited by 0SourcePDFScholar
2024

Reference Neural Operators: Learning the Smooth Dependence of Solutions of PDEs on Geometric Deformations

ICML 2024poster

For partial differential equations on domains of arbitrary shapes, existing works of neural operators attempt to learn a mapping from geometries to solutions. It often requires a large dataset of geometry-solution pairs in order to obtain a sufficiently accurate neural operator. However, for many in…

Cited by 2SourcePDFScholar
2023

Bi-level Physics-Informed Neural Networks for PDE Constrained Optimization using Broyden's Hypergradients

ICLR 2023poster

Deep learning based approaches like Physics-informed neural networks (PINNs) and DeepONets have shown promise on solving PDE constrained optimization (PDECO) problems. However, existing methods are insufficient to handle those PDE constraints that have a complicated or nonlinear dependency on optim…

Cited by 19SourcePDFScholar
2023

GNOT: A General Neural Operator Transformer for Operator Learning

ICML 2023poster

Learning partial differential equations' (PDEs) solution operators is an essential problem in machine learning. However, there are several challenges for learning operators in practical applications like the irregular mesh, multiple input functions, and complexity of the PDEs' solution. To address t…

2023

NUNO: A General Framework for Learning Parametric PDEs with Non-Uniform Data

ICML 2023poster

The neural operator has emerged as a powerful tool in learning mappings between function spaces in PDEs. However, when faced with real-world physical data, which are often highly non-uniformly distributed, it is challenging to use mesh-based techniques such as the FFT. To address this, we introduce…

2022

A Unified Hard-Constraint Framework for Solving Geometrically Complex PDEs

NeurIPS 2022accept

We present a unified hard-constraint framework for solving geometrically complex PDEs with neural networks, where the most commonly used Dirichlet, Neumann, and Robin boundary conditions (BCs) are considered. Specifically, we first introduce the "extra fields'' from the mixed finite element method t…