About Me
I’m currently a Master’s student at Electrical and Computer Engineering, UC San Diego, majoring in Machine Learning and Data Science.
I received my Bachelor’s Degree from Peking University last year. At PKU, I’m a member of the first Tong Class (an honorary pilot class in AI) at Yuanpei College, and was also affiliated with Institute for Artificial Intelligence.
My research and career interests include but are not limited to:
- Research: Machine Learning, (Machine and Human) Cognitive Reasoning, LLMs, (Robotic) Planning.
- Industry: Self-driving, Recommendation, Cloud/Distributed Computation, Data-driven Learning, Software Development, especially related to AI and AI products.
Feel free to contact me through [Github] [LinkedIn] [Email], I’m happy to share ideas and infos.
Education
University of California, San Diego
Master of Science in Machine Learning and Data Science | Sep 2024 - Spring 2026 (Expected)
- Department of Electrical and Computer Engineering
Peking University
Bachelor of Engineering in Artificial Intelligence (Computer Science) | Sep 2020 - Jul 2024
- Yuanpei College
- Member of the first Tong Class
Professional Experience
WeRide
Software Development Engineer Intern (Perception Group) | San Jose, CA | Summer 2025
- Building and optimizing data pipelines for data mining, labeling, training, and evaluation to support end-to-end onboard development
- Conducting advanced research to solve complex real-world challenges in object detection, segmentation, and scene understanding
- Leveraging multi-modal sensor fusion techniques to design robust and scalable architectures for autonomous vehicles
- Engaging in cross-functional collaboration to support model enhancement and refinement for platform deployment
Megvii Technology Limited
Software Development Intern (Self-Driving) | Beijing, CN | Jun 2023 - Dec 2023
- Models and tools development for self-driving vehicles
- Proposed Transformer neural networks to simulate vehicle trajectories
- Developed visualization tools to better understand vehicle behaviors
- Built benchmarks to evaluate trajectories
Beijing Institute for General Artificial Intelligence (BIGAI)
Research Intern | Beijing, CN | Mar 2022 - Jun 2023
- Individual Research in Cognitive Reasoning Lab, under supervision of Prof. Wei Wang and Dr. Lifeng Fan
- Project on concept disentanglement, causal transition, symbolic reasoning, and visual planning
- Project on 3-D spatial reasoning
- Member of Artificial Social Intelligence group
Research & Publications
Learning Concept-based Causal Transition and Symbolic Reasoning for Visual Planning
- Yilue Qian, Peiyu Yu, Ying Nian Wu, Yao Su, Wei Wang, Lifeng Fan
- Published at IROS 2024 as Oral Pitch
- [Paper] [Web] [Video] [Poster]
Projects
Efficient Linear Algebra Solver with GPU Acceleration
Oct 2024 - Nov 2024
- Keywords: Cuda, Parallel Computation
- Developed a high-performance Linear Algebra Solver using C++ and Cuda, with advanced parallel techniques such as blocking, tiling, shared memory, and data shuffling within threads
- Achieved 82% of the performance efficiency of NVIDIA cutlass
- Conducted comprehensive benchmarking and profiling with NVIDIA Nsight
Efficient Compilation and Scheduling for CNNs on Compute-in-memory Units
Feb 2024 - Mar 2024
- Keywords: CNN, Compilation, Compute-in-memory
- Designed a highly efficient compilation approach for CNN inference on compute-in-memory chips
- Achieved 98% reduced memory access and 67% faster inference speed
- Implemented network reconstruction, memory access management, and computational operation scheduling
Evaluating and Mitigating Human-like Cognitive Biases in LLMs
Feb 2024 - Mar 2024
- Keywords: Large Language Models, Cognitive Bias, Benchmarking, Cognitive Hierarchy
- Established CoBAL, a comprehensive cognitive bias evaluation system
- Evaluated 7 cutting-edge language models (GPT-4, Gemini-1.5, Claude-3, etc.)
- Proposed HiTHer, a novel cognitive bias mitigating method with 40-85% higher performance
Playing Brick Breaker with A Robotic Arm System
Feb 2023 - Mar 2023
- Keywords: Robotic arms, Game AI, Inverse Kinematics
- Developed trajectory planning and inverse kinematics solutions
- Created a C++ library for inverse kinematics with 82% acceleration
- Implemented Pygame visualization with 60 fps performance
Playing Hanabi with ToM and Intrinsic Rewards
Sep 2022 - Dec 2022
- Keywords: Reinforcement Learning, Theory of Mind
- Designed goal-oriented intrinsic reward module improving game scores
- Implemented hand card inference module based on Theory of Mind
- Enhanced performance by 0.47 and 2.62 points in different game modes