About

Short Bio

I am currently a Ph.D. student advised by Prof. Jing Gao in the Elmore Family School of Electrical and Computer Engineering at Purdue University. Before joining Purdue in 2022, I worked with Prof. Song Guo at the Pervasive Edge Intelligence Laboratory (PEILab). I obtained my B.Sc. degree in Computing (major) and Applied Mathematics (minor) from The Hong Kong Polytechnic University in 2020. During my doctoral studies, I have conducted research in collaboration with leading technology companies, including Alibaba and Apple. I have served as a reviewer for a number of prestigious conferences, such as ICML, NeurIPS, ICLR, and KDD. I was a recipient of the Best Student Paper Award from ACM BSCI'19.

Research Interest

Federated Learning

Large-scale Optimization

Large Language Model

LLM-based Multi-agents

News

  • [Jul. 2025] Our paper on LLM copyright protection with regularized selective unlearning is accepted by COLM'25.
  • [May 2025] I join Apple AIML as an intern this summer.
  • [Jan. 2025] Our paper on federated RLHF is accepted by ICLR'25.
  • [Oct. 2024] I receive a travel grant from SenSys'24.
  • [Sept. 2024] Our paper on model-heterogeneous federated learning is accepted by NeurIPS'24.
  • [Sept. 2024] Our paper on LLM copyright protection is accepted by EMNLP'24.
  • [Sept. 2024] Our paper on heterogeneous multi-modal federated learning is accepted by SenSys'24.
Education

Education

August 2022 - Present
Purdue University
West Lafayette, Indiana, United States
September 2020 - August 2022
The Hong Kong Polytechnic University
Hong Kong
  • Thesis: Gradient-wise Optimization for Distributed Machine Learning [PDF]
August 2016 - September 2020
The Hong Kong Polytechnic University
Hong Kong
  • Minor in Applied Mathematics
February 2019 - July 2019
The University of Twente
Enschede, Overijssel, Netherlands
Experience

Work Experience

Purdue University West Lafayette, Indiana, United States

Graduate Research Assistant (Advisor: Prof. Jing Gao)
August 2022 - Present

Alibaba Group Bellevue, Washington, United States

Research Intern (Mentor: Dr. Yaliang Li and Dr. Zitao Li)
May 2023 - August 2023

Pervasive Edge Intelligence Laboratory (PEILab), The Hong Kong Polytechnic University Hong Kong

Research Assistant (Advisor: Prof. Song Guo)
September 2020 - August 2022

The Chinese University of Hong Kong (Shenzhen) Shenzhen, Guangdong, China

Research Assistant (Advisor: Prof. Wei Cai)
July 2019 - September 2019

The University of British Columbia Vancouver, British Columbia, Canada

Research Assistant (Advisor: Prof. Victor C.M. Leung)
July 2018 - September 2018
Teaching

Teaching

ENG 2003 Information Technology

Teaching Assistant @ The Hong Kong Polytechnic University
Spring 2022

COMP 1011 Programming Fundamentals

Teaching Assistant @ The Hong Kong Polytechnic University
Fall 2021

Services

Professional Activities

Conference TPC/Reviewer
BSCI 2020, ECML-PKDD 2020, ICML 2022, NeurIPS 2022, ICML 2023, NeurIPS 2023, ICLR 2024, ICML 2024, KDD 2024, NeurIPS 2024, ACL Rolling Review 2024, KDD 2025, ICLR 2025, ICML 2025, ACL Rolling Review 2025, NeurIPS 2025
Journal Reviewer
IEEE OJ-CS, ACM TOMM, ACM TOIT, IEEE TMC
Publications

* indicates equal contribution.

* indicates equal contribution.