Yu He

swan.PNG

Hi, I am Heyu, a third-year CS PhD student at Stanford, where I am advised by Ellen Vitercik. Previously, I graduated from University of Cambridge with BA+MEng in Computer Science, where I was supervised by Pietro Liò.

I am interested in understanding, evaluating, and improving the algorithmic and structured reasoning abilities in neural networks, such as large language models and graph neural networks.

My research is generously supported by the Cubist PhD Fellowship and the Stanford School of Engineering Fellowship.

publications

  1. dsr.png
    Can LLMs Reason Structurally? An Evaluation via the Lens of Data Structures
    Yu He*, Yingxi Li*, Colin White, Ellen Vitercik
    In Preprint, 2025
    Paper Code
  2. rewire.png
    Overcoming Information Bottlenecks in Directed Graph Neural Networks through Rewiring
    Yu He, Ishani Karmarkar, and Ellen Vitercik
    In Learning on Graph Conference (LoG), 2025
    Paper
  3. primal-dual.png
    Primal-Dual Neural Algorithmic Reasoning
    Yu He and Ellen Vitercik
    In International Conference on Machine Learning (ICML), 2025
    Spotlight (top 2.6%)
    Paper Code Poster
  4. dem.png
    Deep Equilibrium Models For Algorithmic Reasoning
    Sophie Xhonneux, Yu He, Andreea Deac, Jian Tang, Gauthier Gidel
    In The Third Blogpost Track at International Conference on Learning Representations (ICLR), 2024
    Paper Blog
  5. higher-egp.png
    Higher-Order Expander Graph Propagation
    Thomas Christie* and Yu He*
    In NeurIPS 2023 Workshop: New Frontiers in Graph Learning, 2023
    Paper Code Poster
  6. sheaf-pe.png
    Sheaf-based Positional Encodings for Graph Neural Networks
    Yu He, Cristian Bodnar, and Pietro Liò
    In NeurIPS 2023 Workshop on Symmetry and Geometry in Neural Representations, 2023
    Paper Poster
  7. cnap.png
    Continuous Neural Algorithmic Planners
    Yu He, Petar Veličković, Pietro Liò, Andreea Deac
    In Proceedings of the First Learning on Graphs Conference, 2022
    Paper Code Poster
  8. algo-selection.png
    Algorithm Selection for Classification Problems via Cluster-based Meta-features
    Daren Ler, Hongyu Teng, Yu He, Rahul Gidijala
    In 2018 IEEE International Conference on Big Data (Big Data), 2018
    Paper