avatar

Yiheng Du

Ph.D. Student
BAIR, UC Berkeley
yihengdu (at) berkeley.edu


About Me

I am a new Ph.D. student at BAIR, UC Berkeley, advised by Prof. Aditi Krishnapriyan. I graduated from Peking University, Yuanpei College, where I was advised by Prof. Liwei Wang and Prof. Di He.

My research focuses on AI for physical and chemical sciences.

Publications

  1. Neural Spectral Methods: Self-supervised learning in the spectral domain
    Yiheng Du, Nithin Chalapathi, Aditi Krishnapriyan
    International Conference on Learning Representations 2024

  2. Scaling physics-informed hard constraints with mixture-of-experts
    Nithin Chalapathi, Yiheng Du, Aditi S. Krishnapriyan
    International Conference on Learning Representations 2024

  3. Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness
    Bohang Zhang, Jingchu Gai, Yiheng Du, Qiwei Ye, Di He, Liwei Wang
    International Conference on Learning Representations 2024
    OUTSTANDING PAPER HONORABLE MENTION

  4. A Complete Expressiveness Hierarchy for Subgraph GNNs via Subgraph Weisfeiler-Lehman Tests
    Bohang Zhang, Guhao Feng*, Yiheng Du*, Di He, Liwei Wang
    International Conference on Machine Learning 2023


Powered by Jekyll and Minimal Light theme.