Yi He

Undergraduate Student · Mathematics and Applied Mathematics

Cuiying Honors College, Lanzhou University
National Program for Top Talents in Basic Sciences

Yi He

About Me

I am currently a third-year undergraduate student pursuing Mathematics and Applied Mathematics at Cuiying Honors College, Lanzhou University. My primary research interests lie in AI × Science. I have led or participated in multiple interdisciplinary research projects, including the development of spherical BSDE algorithms, the construction of the DNA methylation prediction model and interpretability method CAD and CWGA, and the application of point cloud equivariant networks in part representation recognition.

Future Directions: I aim to focus on AI, and I plan to start up during or after my PhD, taking on a core leadership role..

Email: heyi2023@lzu.edu.cn

Wechat: HEYI050410

Education

University of California, Berkeley

Jan. 2026 - May. 2026
Visiting International Student (Semester Study Abroad)
  • Selected Courses: CS 61B (Data Structures), Introduction to Neurotechnology, Linux System Administration
  • Focus: Computer Science, Neurotechnology, and Entrepreneurship
  • Affiliation: Member of Open Computing Facility (OCF)

Lanzhou University

Sept. 2023 - Jun. 2027
B.S. in Mathematics and Applied Mathematics, Cuiying Honors College
  • GPA: 4.02 / 5.0
  • Rank: 9 / 126
  • Awards: 2025 MCM/ICM Finalist (Top 2%)
  • Extracurricular: Captain of the Swimming Team (School of Mathematics and Statistics); Captain of the Track and Field Team (Cuiying Honors College)

Leading Learning Projects

RT-2, SmolVLA and Feedback World Model

Nanjing University

Working on Embodied AI projects. Currently studied: RT-2's action tokenization, SmolVLA's lightweight design & Flow Matching, Feedback WM's training‑inference decoupling & heterogeneous architecture, feedback world model and action‑aware weighting. Also performed local deployment, training, and evaluation of SmolVLA.

Lead Learner VLA World Model Flow Matching
EAI

Leading Research Experience

MEDNA-DFM Model and XAI method: CAD & CWGA

City University of Hong Kong · Dr. Tianchi Lu's Group

Developed a DNA methylation prediction framework based on Dual-View FiLM-MoE architectures, and proposed the CAD & CWGA interpretability method, emphasizing internal mechanism analysis as a necessary condition for interpretability.

Paper: arXiv:2602.22850 Web: MEDNA-DFM-Web Status: Under Review at Genome Research
Lead Researcher (1st Author) Bioinformatics MEDNA-DFM CAG & CWGA Explainable AI
Bio Research Cover

Deep BSDEs: Solving Spherical Fokker-Planck and Feynman-Kac Equations

Lanzhou University · Prof. Weihua Deng's Group (Distinguished Young Scholar); Collaborative Advisor: Dr. Heng Wang

Developed a numerical solution method for high-dimensional partial differential equations under spherical geometric constraints, combining deep learning with backward stochastic differential equations (BSDEs) to address the curse of dimensionality in traditional methods.

Lead Researcher AI for PDE Deep BSDE Spherical Geometry
BSDE Research Cover

Collaborative Research Experience

Fixed points of orientation-preserving full transformation

Lanzhou University · Prof. Wenting Zhang's Group; Collaborative Advisor: Yang An, M.S.

Paper: arXiv:2604.26661 Status: Under Consideration at Discrete Applied Mathematics (CCF-C)
Code Member Computational Group Theory Combinatorics
Computational Algebra Cover

PathMoG : A Pathway-Centric Modular Graph Network for Multi-Omics Cancer Survival Prediction

City University of Hong Kong · Dr. Tianchi Lu's Group

Github: PathMoG Status: Under Consideration at Bioinformatics (CCF-A)
Model Design Member Survival Prediction Multi-Omics Fusion
PathMoG

HBGSA: Hydrogen Bond Graph with Self-Attention for Drug-Target Binding Affinity Prediction

City University of Hong Kong · Dr. Tianchi Lu's Group

Paper: arXiv:2604.23115 Status: Accepted for publication in Knowledge-Based Systems
Core Member GNN Hydrogen Bonds Drug-Target Affinity
HBGSA

Industrial Part Representation and Assembly

Beijing Normal University-Hong Kong Baptist University United International College · Prof. Tieyong Zeng's UIC Path Group; Provincial Key Project

Adopted point cloud-level contrastive learning and equivariant neural network approaches to transfer PointNet for part representation learning.

Core Member PointNet Contrastive Learning REQNN
3D Research Cover

Skills

Programming Languages

Python (PyTorch)

C & C++

MATLAB

Development Tools

VS Code & SSH

Linux & Conda

Algorithm Foundations

Numerical Computation

Monte Carlo Methods

FFT & IFFT

Authoring Tools

LaTeX

Markdown

HTML & CSS

Other Skills

AutoDL Cloud Computing

Web Deployment