← Search

Shandian Zhe

46 accepted papers

2026

ElastoGen: 4D Generative Elastodynamics

AAAI 2026technical

We present ElastoGen, a knowledge-driven AI model that generates physically accurate 4D elastodynamics. Unlike deep models that learn from video- or image-based observations, ElastoGen leverages the principles of physics and learns from established mathematical and optimization procedures. The core

Cited by 0SourcePDFScholar
2025

Arbitrarily-Conditioned Multi-Functional Diffusion for Multi-Physics Emulation

ICML 2025poster

Modern physics simulation often involves multiple functions of interests, and traditional numerical approaches are known to be complex and computationally costly. While machine learning-based surrogate models can offer significant cost reductions, most focus on a single task, such as forward predict…

Cited by 1SourcePDFScholar
2025

Invertible Fourier Neural Operators for Tackling Both Forward and Inverse Problems

AISTATS 2025poster

Fourier Neural Operator (FNO) is a powerful and popular operator learning method. However, FNO is mainly used in forward prediction, yet a great many applications rely on solving inverse problems. In this paper, we propose an invertible Fourier Neural Operator (iFNO) for jointly tackling the forwar…

Cited by 0SourcecodeScholar
2025

Standard Gaussian Process is All You Need for High-Dimensional Bayesian Optimization

ICLR 2025oral

A long-standing belief holds that Bayesian Optimization (BO) with standard Gaussian processes (GP) --- referred to as standard BO --- underperforms in high-dimensional optimization problems. While this belief seems plausible, it lacks both robust empirical evidence and theoretical justification. To…

Cited by 4SourcePDFScholar
2025

Toward Efficient Kernel-Based Solvers for Nonlinear PDEs

ICML 2025poster

We introduce a novel kernel learning framework toward efficiently solving nonlinear partial differential equations (PDEs). In contrast to the state-of-the-art kernel solver that embeds differential operators within kernels, posing challenges with a large number of collocation points, our approach el…

Cited by 1SourcePDFScholar
2024

BayOTIDE: Bayesian Online Multivariate Time Series Imputation with Functional Decomposition

ICML 2024spotlight

In real-world scenarios such as traffic and energy management, we frequently encounter large volumes of time-series data characterized by missing values, noise, and irregular sampling patterns. While numerous imputation methods have been proposed, the majority tend to operate within a local horizon,…

2024

Equation Discovery with Bayesian Spike-and-Slab Priors and Efficient Kernels

AISTATS 2024poster

Discovering governing equations from data is important to many scientific and engineering applications. Despite promising successes, existing methods are still challenged by data sparsity and noise issues, both of which are ubiquitous in practice. Moreover, state-of-the-art methods lack uncertainty…

2024

Functional Bayesian Tucker Decomposition for Continuous-indexed Tensor Data

ICLR 2024poster

Tucker decomposition is a powerful tensor model to handle multi-aspect data. It demonstrates the low-rank property by decomposing the grid-structured data as interactions between a core tensor and a set of object representations (factors). A fundamental assumption of such decomposition is that ther…

2024

Multi-Resolution Active Learning of Fourier Neural Operators

AISTATS 2024poster

Fourier Neural Operator (FNO) is a popular operator learning framework. It not only achieves the state-of-the-art performance in many tasks, but also is efficient in training and prediction. However, collecting training data for the FNO can be a costly bottleneck in practice, because it often demand…

2024

Solving High Frequency and Multi-Scale PDEs with Gaussian Processes

ICLR 2024poster

Machine learning based solvers have garnered much attention in physical simulation and scientific computing, with a prominent example, physics-informed neural networks (PINNs). However, PINNs often struggle to solve high-frequency and multi-scale PDEs, which can be due to spectral bias during neural…

2023

Dynamic Tensor Decomposition via Neural Diffusion-Reaction Processes

NeurIPS 2023spotlight

Tensor decomposition is an important tool for multiway data analysis. In practice, the data is often sparse yet associated with rich temporal information. Existing methods, however, often under-use the time information and ignore the structural knowledge within the sparsely observed tensor entries.…

2023

Meta Learning of Interface Conditions for Multi-Domain Physics-Informed Neural Networks

ICML 2023poster

Physics-informed neural networks (PINNs) are emerging as popular mesh-free solvers for partial differential equations (PDEs). Recent extensions decompose the domain, apply different PINNs to solve the problem in each subdomain, and stitch the subdomains at the interface. Thereby, they can further al…

Cited by 7SourcePDFScholar
2023

Provably Convergent Schrödinger Bridge with Applications to Probabilistic Time Series Imputation

ICML 2023poster

The Schrödinger bridge problem (SBP) is gaining increasing attention in generative modeling and showing promising potential even in comparison with the score-based generative models (SGMs). SBP can be interpreted as an entropy-regularized optimal transport problem, which conducts projections onto ev…

2023

Streaming Factor Trajectory Learning for Temporal Tensor Decomposition

NeurIPS 2023poster

Practical tensor data is often along with time information. Most existing temporal decomposition approaches estimate a set of fixed factors for the objects in each tensor mode, and hence cannot capture the temporal evolution of the objects' representation. More important, we lack an effective approa…

2022

AutoIP: A United Framework to Integrate Physics into Gaussian Processes

ICML 2022spotlight

Physical modeling is critical for many modern science and engineering applications. From a data science or machine learning perspective, where more domain-agnostic, data-driven models are pervasive, physical knowledge {—} often expressed as differential equations {—} is valuable in that it is comple…

2022

Batch Multi-Fidelity Active Learning with Budget Constraints

NeurIPS 2022accept

Learning functions with high-dimensional outputs is critical in many applications, such as physical simulation and engineering design. However, collecting training examples for these applications is often costly, e.g., by running numerical solvers. The recent work (Li et al., 2022) proposes the firs…

Cited by 20SourcePDFScholar
2022

Deep Multi-Fidelity Active Learning of High-Dimensional Outputs

AISTATS 2022poster

Many applications, such as in physical simulation and engineering design, demand we estimate functions with high-dimensional outputs. To reduce the expensive cost of generating training examples, we usually choose several fidelities to enable a cost/quality trade-off. In this paper, we consider the…

2022

Nonparametric Embeddings of Sparse High-Order Interaction Events

ICML 2022spotlight

High-order interaction events are common in real-world applications. Learning embeddings that encode the complex relationships of the participants from these events is of great importance in knowledge mining and predictive tasks. Despite the success of existing approaches, e.g. Poisson tensor factor…

Cited by 2SourcePDFScholar
2022

Nonparametric Sparse Tensor Factorization with Hierarchical Gamma Processes

ICML 2022spotlight

We propose a nonparametric factorization approach for sparsely observed tensors. The sparsity does not mean zero-valued entries are massive or dominated. Rather, it implies the observed entries are very few, and even fewer with the growth of the tensor; this is ubiquitous in practice. Compared with…

Cited by 8SourcePDFScholar
2022

Recall Distortion in Neural Network Pruning and the Undecayed Pruning Algorithm

NeurIPS 2022accept

Pruning techniques have been successfully used in neural networks to trade accuracy for sparsity. However, the impact of network pruning is not uniform: prior work has shown that the recall for underrepresented classes in a dataset may be more negatively affected. In this work, we study such relati…

Cited by 14SourcePDFScholar
2022

The Combinatorial Brain Surgeon: Pruning Weights That Cancel One Another in Neural Networks

ICML 2022spotlight

Neural networks tend to achieve better accuracy with training if they are larger {—} even if the resulting models are overparameterized. Nevertheless, carefully removing such excess of parameters before, during, or after training may also produce models with similar or even improved accuracy. In man…

2021

Batch Multi-Fidelity Bayesian Optimization with Deep Auto-Regressive Networks

NeurIPS 2021poster

Bayesian optimization (BO) is a powerful approach for optimizing black-box, expensive-to-evaluate functions. To enable a flexible trade-off between the cost and accuracy, many applications allow the function to be evaluated at different fidelities. In order to reduce the optimization cost while max…

Cited by 16SourcePDFScholar
2021

Characterizing possible failure modes in physics-informed neural networks

NeurIPS 2021poster

Recent work in scientific machine learning has developed so-called physics-informed neural network (PINN) models. The typical approach is to incorporate physical domain knowledge as soft constraints on an empirical loss function and use existing machine learning methodologies to train the model. We…

2021

Multi-Fidelity High-Order Gaussian Processes for Physical Simulation

AISTATS 2021poster

The key task of physical simulation is to solve partial differential equations (PDEs) on discretized domains, which is known to be costly. In particular, high-fidelity solutions are much more expensive than low-fidelity ones. To reduce the cost, we consider novel Gaussian process (GP) models that le…

Cited by 17SourcePDFScholar
2021

Self-Adaptable Point Processes with Nonparametric Time Decays

NeurIPS 2021poster

Many applications involve multi-type event data. Understanding the complex influences of the events on each other is critical to discover useful knowledge and to predict future events and their types. Existing methods either ignore or partially account for these influences. Recent works use recurren…

Cited by 13SourcePDFScholar
2020

Multi-Fidelity Bayesian Optimization via Deep Neural Networks

NeurIPS 2020poster

Bayesian optimization (BO) is a popular framework for optimizing black-box functions. In many applications, the objective function can be evaluated at multiple fidelities to enable a trade-off between the cost and accuracy. To reduce the optimization cost, many multi-fidelity BO methods have been p…

2020

Scalable Nonparametric Factorization for High-Order Interaction Events

AISTATS 2020poster

Interaction events among multiple entities are ubiquitous in real-world applications. Although these interactions can be naturally represented by tensors and analyzed by tensor decomposition, most existing approaches are limited to multilinear decomposition forms, and cannot estimate complex, nonlin…

2018

Learning Compact Recurrent Neural Networks With Block-Term Tensor Decomposition

CVPR 2018poster

Recurrent Neural Networks (RNNs) are powerful sequence modeling tools. However, when dealing with high dimensional inputs, the training of RNNs becomes computational expensive due to the large number of model parameters. This hinders RNNs from solving many important computer vision tasks, such as Ac…

Cited by 171SourcePDFScholar
2017

Asynchronous Distributed Variational Gaussian Process for Regression

ICML 2017poster

Gaussian processes (GPs) are powerful non-parametric function estimators. However, their applications are largely limited by the expensive computational cost of the inference procedures. Existing stochastic or distributed synchronous variational inferences, although have alleviated this issue by sca…

Cited by 30SourcePDFScholar
2016

Distributed Flexible Nonlinear Tensor Factorization

NeurIPS 2016poster

Tensor factorization is a powerful tool to analyse multi-way data. Recently proposed nonlinear factorization methods, although capable of capturing complex relationships, are computationally quite expensive and may suffer a severe learning bias in case of extreme data sparsity. Therefore, we propose…

Cited by 78SourcePDFScholar