Haimeng Zhao

Haimeng Zhao

Undergraduate of Physics and Maths

Tsinghua University

Welcome!

Hello ~ I am Haimeng Zhao (赵海萌), an undergraduate of Physics and Mathematics at Zhili College, Tsinghua University. I’m deeply fascinated by how the universe works and how we can possibly understand it. So I started working in the intersection of Physics and Artificial Intelligence.

Specifically, my interest comes in two folds:

  • the foundation, application and interplay of quantum physics, information theory and statistics;
  • (quantum) machine learning for physics and physics for machine learning.

I’m currently an undergrad research fellow in John Preskill’s group at IQIM, Caltech. Before that I was an exchange student at EPFL, Switzerland, where I worked with Giuseppe Carleo and Filippo Vicentini on neural quantum states. In my sophomore year, I worked on AI for Astronomy with Wei Zhu at Tsinghua.

I am joining Caltech as a Physics graduate student in 2024 Fall!

Interests

  • Quantum Information, Statistics & Learning Theory
  • AI for Science, especially Physics & Astrophysics
  • Quantum Many-body Physics
  • Generative Learning

Education

  • Undergraduate, 2020 - 2024(expected)

    Physics, Tsinghua University

  • Outstanding Graduate, 2020

    Shanghai High School

Skills

Programming

High performance scientific computing with Python & C++.

Differentiable programming with JAX, PyTorch (6 years) & TensorFlow.

(Quantum) Machine Learning

(Quantum) Learning Theory,

Variational Quantum Algorithms,

Generaitive Learning,

Neural Differential Equations,

Computational Physics

(Quantum) Monte Carlo,

Tensor Network, DFT,

Neural Quantum State

Publications

Haimeng Zhao, Laura Lewis, Ishaan Kannan, Yihui Quek, Hsin-Yuan Huang, Matthias C. Caro (2023). Learning quantum states and unitaries of bounded gate complexity.

PDF

Haimeng Zhao, Giuseppe Carleo, Filippo Vicentini (2023). Empirical Sample Complexity of Neural Network Mixed State Reconstruction.

PDF Code

Haimeng Zhao (2023). Non-IID Quantum Federated Learning with One-shot Communication Complexity. Quantum Machine Intelligence, 5(1), 3.

PDF Code Conf Talk

Haimeng Zhao, Wei Zhu (2022). MAGIC: Microlensing Analysis Guided by Intelligent Computation. The Astronomical Journal, 164(5), 192.

PDF Code Conf Talk

Junyi Liu, Yifu Tang, Haimeng Zhao, Xieheng Wang, Fangyu Li, Jingyi Zhang (2022). CPS Attack Detection under Limited Local Information in Cyber Security: A Multi-node Multi-class Classification Ensemble Approach.

PDF

See all publications

Experience

 
 
 
 
 

AI for Quantum: Sample Complexity of Neural Quantum State Tomography

Advisor: Prof. Giuseppe Carleo & Filippo Vicentini, Computational Quantum Science Lab @ EPFL.

Aug 2022 – Jul 2023 First Author. Lausanne, Switzerland
Introduced control variates to control gradient variance and significantly reduce sample complexity. Conducted extensive numerical & theoretical studies to understand different sample complexity behavior. Benchmarked different tomography methods and propose to design quantum-resource-efficient NQSs.
 
 
 
 
 

Federated Learning in Multi-class Classification

In collaboration with Prof. Jingyi Zhang, Center for Statistical Science @ Tsinghua, and my friends Junyi & Yifu.

Apr 2022 – Apr 2022 Third Author. Beijing, China
Proved the key theorem in the paper, which enables one to merge partial classifiers trained in different nodes into a global one without leaking private data.
 
 
 
 
 

AI for Astro: ML Framework for Realistic Microlensing Event Analysis

Advisor: Prof. Wei Zhu, Department of Astronomy @ Tsinghua.

Oct 2021 – Jun 2022 First Author. Beijing, China
Introduced U-Net and neural controlled differential equations to parameter estimation of microlensing. Developed a machine learning framework for irregular astronomical time series, listed on NASA EMAC. Accelerate microlensing analysis by 5 orders of magnitude and successfully applied to real events for the first time.
 
 
 
 
 

Quantum AI: Non-IID Quantum Federated Learning

Single authored work. Extending Liu et al. to the quantum regime.

Jul 2021 – Sep 2022 Single Author. Beijing, China
Proposed and studied the non-IID quagmire in quantum federated learning, theoretically and numerically. Extended Liu et al. to a quantum algorithm. Conducted extensive numerics to show its robustness and efficiency.
 
 
 
 
 

AI for HEP-Ex: A Neutrino Data Analysis Tournament

Advisor: Prof. Benda Xu, Department of Engineering Physics @ Tsinghua.

Jan 2021 – Jun 2021 First Prize & Most Innovative Algorithm. Beijing, China
Led a team that developed a simulation & machine learning pipeline to promote neutrino energy detection precision, a key step towards understanding the neutrino mass ordering problem.
 
 
 
 
 

AI for Vision: Learned Lossy Image Compression

Advisor: the Internet. In collaboration with a friend Peiyuan back in high school.

Sep 2018 – Dec 2019 First Author. Shanghai, China
Introduced a pruning method originally used in neural architecture search to the field of lossy image compression. Achieved the state-of-the-art performance with much simpler training procedure.

Recent Articles

数理教材阅读进度及读后简评（大二秋冬）

Maths & Physics Book Reviews (2022 Fall)

Maths and Physics, in Chinese Aug 2022

数理教材阅读进度及读后简评（大一春夏）

Maths & Physics Book Reviews (2021 Spring)

Maths and Physics, in Chinese Sep 2021

数理教材阅读进度及读后简评（大一秋冬）

Maths & Physics Book Reviews (2020 Fall)

Maths and Physics, in Chinese Feb 2021

不确定性关系的信息熵

Entropic Uncertainty Relation

Physics, in Chinese Jan 2021

数列、矩阵与微分方程的特征值

Eigenvalues of Sequences, Matrices and ODEs

Mathematics, in Chinese Feb 2020

利用定量时空图解决狭义相对论运动学问题

Solving Kinematic Problems in Special Relativity via Quantitative Spacetime Diagrams

Physics, in Chinese Dec 2019

利用保角变换解决一类平面静电场问题

Using Conformal Transformation to Solve Electrostatic Problems

Physics, in Chinese Feb 2019

Contact