← Search

Xiaotian Gao

5 accepted papers

2023

Learning Physics-Informed Neural Networks without Stacked Back-propagation

AISTATS 2023poster

Physics-Informed Neural Network (PINN) has become a commonly used machine learning approach to solve partial differential equations (PDE). But, facing high-dimensional secondorder PDE problems, PINN will suffer from severe scalability issues since its loss includes second-order derivatives, the comp…

2023

NeuralStagger: Accelerating Physics-constrained Neural PDE Solver with Spatial-temporal Decomposition

ICML 2023poster

Neural networks have shown great potential in accelerating the solution of partial differential equations (PDEs). Recently, there has been a growing interest in introducing physics constraints into training neural PDE solvers to reduce the use of costly data and improve the generalization ability. H…

Cited by 11SourcePDFScholar
2022

Efficient and Stable Information Directed Exploration for Continuous Reinforcement Learning

ICASSP 2022accepted

In this paper, we investigate the exploration-exploitation dilemma of reinforcement learning algorithms. We adapt the information directed sampling, an exploration framework that measures the information gain of a policy, to the continuous reinforcement learning. To stabilize the off-policy learning…

Cited by 0SourceScholar
2022

SQuant: On-the-Fly Data-Free Quantization via Diagonal Hessian Approximation

ICLR 2022poster

Quantization of deep neural networks (DNN) has been proven effective for compressing and accelerating DNN models. Data-free quantization (DFQ) is a promising approach without the original datasets under privacy-sensitive and confidential scenarios. However, current DFQ solutions degrade accuracy, ne…

2021

OpEvo: An Evolutionary Method for Tensor Operator Optimization

AAAI 2021technical

Training and inference efficiency of deep neural networks highly rely on the performance of tensor operators on hardware platforms. Manually optimizing tensor operators has limitations in terms of supporting new operators or hardware platforms. Therefore, automatically optimizing device code configu…