About
My research focuses on the systems foundations of modern artificial intelligence, with an emphasis on efficiency, scalability, and reliability.
I am a researcher in the Systems and Networking Research Group at Microsoft Research Asia. My work advances distributed systems, AI infrastructure, and large-scale graph computing.
I study critical challenges in scaling AI systems, from optimizing distributed training algorithms to developing efficient storage and retrieval mechanisms for massive structural datasets. My goal is to build robust and efficient systems that enable the next generation of AI applications.
I received my Ph.D. in Computer Science from the University of Science and Technology of China (USTC) in 2015. Previously, I studied at the School of the Gifted Young and received my B.S. in Computer Science from USTC in 2009.
Research Interests
- Systems for AI Distributed training, parallelization, collective communication, and inference infrastructure
- Large-Scale Graph Systems Storage, retrieval, and computation for large and evolving graphs
- Distributed Systems and Networking Scalable and dependable infrastructure for data-intensive workloads
- Programming Systems and Software Engineering Tools and abstractions for building efficient, reliable software
Selected Publications
-
TrainVerify: Equivalence-Based Verification for Distributed LLM Training
Proceedings of the 31st ACM Symposium on Operating Systems Principles (SOSP), pp. 237-253, 2025.
-
22nd USENIX Symposium on Networked Systems Design and Implementation (NSDI), pp. 667-683, 2025.
-
nnScaler: Constraint-Guided Parallelization Plan Generation for Deep Learning Training
18th USENIX Symposium on Operating Systems Design and Implementation (OSDI), pp. 347-363, 2024.
-
Aceso: Efficient Parallel DNN Training through Iterative Bottleneck Alleviation
19th European Conference on Computer Systems (EuroSys), pp. 163-181, 2024.
-
Tessel: Boosting Distributed Execution of Large DNN Models via Flexible Schedule Search
30th IEEE International Symposium on High-Performance Computer Architecture (HPCA), pp. 803-816, 2024.
-
27th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS), pp. 402-416, 2022.
-
Rammer: Enabling Holistic Deep Learning Compiler Optimizations with rTasks
14th USENIX Symposium on Operating Systems Design and Implementation (OSDI), pp. 881-897, 2020.
-
NeuGraph: Parallel Deep Neural Network Computation on Large Graphs
USENIX Annual Technical Conference (ATC), pp. 443-458, 2019.
-
ImmortalGraph: A System for Storage and Analysis of Temporal Graphs
ACM Transactions on Storage, 11(3), pp. 1-34, 2015.
-
Kineograph: Taking the Pulse of a Fast-Changing and Connected World
7th European Conference on Computer Systems (EuroSys), pp. 85-98, 2012.
Contact
Research correspondence
Research questions and collaboration
yomia (at) microsoft (dot) com
Internship applications
Applications are welcome; please send a CV
msra-srg-hire (at) microsoft (dot) com
Office
Microsoft Research Asia
No. 5 Danling Street, HaidianBeijing, China