Yitu Wang is a Research Scientist at Meta and working on AI and Systems Co-Design. He obtained Ph.D. degree in Computer Engineering from Duke University under the supervision of Prof. Yiran Chen. He received B.S.E. in Microelectronics from Fudan University in 2020. His research interests focus on computer architecture, including machine learning acceleration, near data processing for data-intensive applications such recommendation system, approximate nearest neighbor search and retrieval-augmented generation (RAG) for large language model (LLM), and algorithm-architecture co-design for deep learning system.

📖 Educations

• 2020.08 - 2025.03, Ph.D. in Computer Engineering at Duke University, Durham, NC, USA.
• 2016.09 - 2020.06, B.S.E in Microelectronics at Fudan University, Shanghai, China.

💻 Experiences

• 2025.04 - Present, Research Scientist at Meta, Menlo Park, California, USA.
  • AI & Systems Co-Design for Meta Training and Inference Accelerator (MTIA)
• 2024.06 - 2023.09, 2023.05 - 2023.08, 2022.05 - 2022.08, Research Intern at Samsung Memory Solution Lab, San Jose, California, USA.
  • LLM System with Compute Express Link (CXL)
  • Learning-based Cache Policy in CXL-SSD
  • Acceleration of Deep Learning Recommendation Models

📝 Publications

• Preprints

  • [In Submission] Yitu Wang*, Minxue Tang*, Hanqiu Chen, Shiyu Li, Andrew Chang, Cong “Callie” Hao, Hai “Helen” Li, Yiran Chen, “FedRepre: End-to-End Acceleration of Federated Learning System with Client Representative”. (* equal contribution)

• First-author papers

  • [MLSys’25] MinXue Tang*, Yitu Wang*, Jingyang Zhang, Louis DiValentin, Aolin Ding, Amin Hass, Yiran Chen, Hai Li, “FedProphet: Memory-Efficient Federated Adversarial Training via Robust and Consistent Cascade Learning”. 8th Annual Conference on Machine Learning and Systems. (* equal contribution)

  • [ISCA’24] Yitu Wang, Shiyu Li, Qilin Zheng, Linghao Song, Zongwang Li, Andrew Chang, Hai “Helen” Li, Yiran Chen, “NDSearch: Accelerating Graph-Traversal-Based Approximate Nearest Neighbor Search through Near Data Processing”. 2024 ACM/IEEE 51st IEEE/ACM Annual International Symposium on Computer Architecture.

  • [DAC’24] Hanqiu Chen*, Yitu Wang*, Vitorio Cargnini, Reza Soltaniyeh, Dongyang Li, Gongjin Sun, Pradeep Subedi, Andrew Chang, Yiran Chen, Cong “Callie” Hao, “ICGMM: CXL-enabled Memory Expansion with Intelligent Caching Using Gaussian Mixture Model”. 61st IEEE/ACM Annual Design Automation Conference, 2024. (* equal contribution)

  • [ESWEEK’23/TECS’23] Yitu Wang, Shiyu Li, Qilin Zheng, Andrew Chang, Hai “Helen” Li, Yiran Chen, “EMS-I: An Efficient Memory System Design with Specialized Caching Mechanism for Recommendation System”. ACM Transactions on Embedded Computing System, 2023.

  • [ICCAD’21] Yitu Wang, Zhenhua Zhu, Fan Chen, Mingyuan Ma, Guohao Dai, Yu Wang, Hai “Helen” Li, and Yiran Chen, “ReRec: In-ReRAM Acceleration with Access-Aware Mapping for Personalized Recommendation”. 2021 IEEE/ACM International Conference on Computer Aided Design.

  • [DATE’20] Yitu Wang, Fan Chen, Linghao Song, C.J. Richard Shi, Hai “Helen” Li, and Yiran Chen, “ReBoc: Accelerating Block-Circulant Neural Networks in ReRAM”. 2020 Design, Automation & Test in Europe Conference & Exhibition.

• Co-author papers

  • [GLSVLSI’25] Feng Cheng, Tunhou Zhang, Junyao Zhang, Jonathan Ku, Yitu Wang, Xiaoxuan Yang, Hai “Helen” Li, Yiran Chen, “AutoRAC: Automated Processing-in-Memory Accelerator Design for Recommender Systems” To appear at the 35th edition of Great Lakes Symposium on VLSI, 2025

  • [DAC’24] Qilin Zheng, Ziru Li, Jonathan Ku, Yitu Wang, Brady Taylor, Yiran Chen, “Improving the Efficiency of In- Memory-Computing Macro with a Hybrid Analog-Digital Computing Mode For Lossless Neural Network Inference”. 61st IEEE/ACM Annual Design Automation Conference, 2024.

  • [TC’24] Shiyu Li, Yitu Wang, Edward Hanson, Andrew Chang, Yang Seok Ki, Hai “Heleln” Li, Yiran Chen, “NDRec: Accelerating the Training of Recommendation Models through Near-Data Processing”, IEEE Transaction on Computers, 2024.

  • [ISCAS’24] Qilin Zheng, Shiyu Li, Yitu Wang, Ziru Li, Hai “Helen” Li, Yiran Chen, “Hybrid Digital/Analog Memoristorbased Computing Architecture for Sparse Deep Learning Acceleration”. 2024 International Symposium on Circuits and Systems.

  • [MICRO’23] Edward Hanson, Shiyu Li, Guanglei Zhou, Feng Cheng, Yitu Wang, Hai “Helen” Li, and Yiran Chen, “Si-kintusgi: Towards Recovering Golden-like Performance of Defective Many-core Spatial Architecture for AI”. Proceedings of the 56th Annual IEEE/ACM International Symposium on Microarchitecture, 2023.

  • [DAC’23] Qilin Zheng, Shiyu Li, Yitu Wang, Ziru Li, Hai “Helen” Li, and Yiran Chen, “Accelerating Sparse Attention with Reconfigurable Non-Volatile Processing-in-Memory Architecture”. 60th IEEE/ACM Annual Design Automation Conference, 2023.

  • [CVPR’22] Minxue Tang, Xuefei Ning, Yitu Wang, Yu Wang and Yiran Chen, “FedGP: Correlation-Based Active Client Selection for Heterogeneous Federated Learning”. Proceedings of 2022 IEEE/CVF Conference on Computer Vision and Pttern Recognition.

  • [EMC2-NeurIPS’19] Jingyang Zhang, Huanrui Yang, Fan Chen, Yitu Wang, and Hai “Helen” Li, “Exploring Bit-Slice Sparsity in Deep Neural Networks for Efficient ReRAM-Based Deployment”. Fifth Workshop on Energy Efficient Machine Learning and Cognitive Computing NeurIPS Edition, 2019.

• Patent

  • “Systems, methods, and apparatus for a cache management policy for memory caches”

• Survey

  • “A Survey: Collaborative Hardware and Software Design in the Era of Large Language Models”. IEEE Circuits and Systems Magazine ( Volume: 25, Issue: 1, Firstquarter 2025).

👨‍💼 Services

• Reviwer of CAL, CVPR, TC, TPDS, TCAD, TCAS-I, TVLSI, TODAES, GLSVLSI.
• Program committee member of GLSVLSI’25.