Zhen Tong

Zhen Tong

Machine Learning and Algorithm Engineer

About Me

I'm Zhen Tong, a first-year graduate student at the Carnegie Mellon University, MSIN. My passion for machine learning has driven me to gain substantial experience in diverse areas such as game AI, image generative models, and various hands-on projects. Throughout my career, I have had the opportunity to work with a wide range of machine learning models and techniques, which has allowed me to build a solid foundation in Python and PyTorch. I have experience in implementing and optimizing algorithms for tasks including reinforcement learning for game AI agents, developing image generative models, and designing backend services for AI applications.

Email
120090694@link.cuhk.edu.cn
Telephone
+86 13760371947

Work Experience

Game AI Algorithm Intern at NETEASE, Digital Intelligence Department
Hangzhou, China | May 2024 - August 2024
Tech Stack: Python-PyTorch, Ray, Lightning, ETL, Flask
Reinforcement Learning and Imitation Learning Direction
  • Designed efficient game state and action representations for a series of poker games, enabling effective reinforcement learning.
  • Developed a distributed ETL (Extract, Transform, Load) pipeline using Ray to prepare large-scale game data in parquet format for model training.
  • Implemented a deep imitation learning algorithm that achieved an 80% win rate against human experts.
  • Deployed the trained agent and conducted extensive client-side concurrency performance tests.
Research Assistant at LOGO Lab
Shenzhen, China | January 2024 - May 2024
Proposed a novel benchmark and baseline model with Double-Stage Fusion.
  • Proposed a dataset spanning 3-4 kilometers of road segments with incomplete lane lines to address challenges in cost-effective, real-world lane-level HD map generation.
  • Designed a Double-Stage Fusion model that first uses a VQGAN to reconstruct input images, then trains a Transformer to learn the latent conditional distribution for generation.
  • Evaluated the model’s performance on our EcoMap dataset, demonstrating significant improvements in lane-level HD map generation compared to existing approaches.
  • Presented the novel benchmark and baseline model at NeurIPS 2024, contributing to the advancement of the field.
Software Engineer Intern at Siemens Mobility Technologies
Guangzhou, China | July 2022 – Sep 2022
Tech Stack: Java, Hadoop, Python (PyTorch), MySQL
Trained a deep learning model to diagnose anomalies in the big data system for metro control, embedded the model into the big data system, and achieved visualization.
  • Selected useful data in the big data system using MySQL, organized all relevant data pertaining to the problem, and successfully aligned the useful data into a new real-time dataset.
  • Trained the LSTM-RNN model using PyTorch, predicting future system states based on the collected data. The model demonstrated a performance accuracy of 92%.
  • Devised and deployed the infrastructure in the big data system through Hadoop, then created efficient algorithms to run the prediction model, enabling scalability to handle large amounts of data.
  • Set up a WebSocket back-end using Java with Maven and handled visualization requests from the front-end, resulting in the rendering of data in the user interface.

Education

Master of Science in Information Networking @ Carnegie Mellon University
2024 - present
2024 Graduate Admission Scholarship
Bachelor of Computer Science @ Chinese University of Hong Kong ShenZhen
2020 - 2024
CGPA = 3.8 (rank 9)
2020 CUHKSZ School of Data Science Annual Scholarship
2020, 2021, 2022 CUHKSZ School of Data Science Dean List
UC Berkeley Global Access
2023 January - June
cs182 Designing, Visualizing and Understanding Deep Neural Networks Grade A
cs170 Efficient Algorithms and Intractable Problems Grade A
cs161 Computer Security Grade A-
GPA = 3.9

Teaching Experience

CUHKSZ
2021 - present
2021-2022 Summer CSC3100 Data Structure
2021-2022 Term2 CSC1002 Computational Laboratory of Python
https://twitter.com/templateflip
https://www.facebook.com/templateflip
https://www.instagram.com/templateflip
https://github.com/templateflip