avatar

Yiheng Du

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


About Me

I am a 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 lies at the intersection of AI and computational physics, where I develop learning-based approaches for accurate and efficient large-scale fluid simulation.

Publications

  1. EddyFormer: Accelerated Neural Simulations of Three-Dimensional Turbulence at Scale
    Yiheng Du, Aditi Krishnapriyan
    Conference on Neural Information Processing Systems (2025)

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

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

  4. 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

  5. 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.