Yunsheng Bai  

 



  ![](/sites/default/files/person/IMG_0575_1.JPG)

  

 Yunsheng Bai earned his PhD in Computer Science from the University of California, Los Angeles (UCLA) in June 2023, under the guidance of Professors Yizhou Sun and Wei Wang, and in collaboration with Professor Jason Cong's team. His research is dedicated to developing innovative approaches for graph-related tasks, with a keen interest in graph neural networks and large language models applied to graph-level tasks, including graph similarity, graph matching, and HLS design modeling. Yunsheng's work has been featured in prestigious conferences across various domains, from data mining to artificial intelligence and machine learning, such as WSDM, KDD, IJCAI, AAAI, ICML, and NeurIPS.



   Research Area(s)

[Artificial Intelligence and Machine Learning ](/index.php/research-area/machine-learning-artificial-intelligence)

[Circuits and VLSI Design](/index.php/research-area/circuits)

 

 

  

 Main Field of Interest

[Artificial Intelligence and Machine Learning ](/index.php/research-area/machine-learning-artificial-intelligence)

 

  

 Google Scholar

[https://scholar.google.com/citations?user=ExgSObQAAAAJ&hl=en](https://scholar.google.com/citations?user=ExgSObQAAAAJ&hl=en)

 

  

 

 

 



 ### Publications

 

### 2025 

[FVDebug: An LLM-Driven Debugging Assistant for Automated Root Cause Analysis of Formal Verification Failures](/index.php/publication/2025-09_fvdebug-llm-driven-debugging-assistant-automated-root-cause-analysis-formal)

[Yunsheng Bai](/index.php/person/yunsheng-bai), Ghaith Bany Hamad, [Chia-Tung (Mark) Ho](/index.php/person/chia-tung-mark-ho), Syed Suhaib, Mark Haoxing Ren













[AssertionForge: Enhancing Formal Verification Assertion Generation with Structured Representation of Specifications and RTL](/publication/2025-06_assertionforge-enhancing-formal-verification-assertion-generation-structured)

[Yunsheng Bai](/person/yunsheng-bai), Ghaith Bany Hamad, Syed Suhaib, Mark Haoxing Ren













[Marco: Configurable Graph-Based Task Solving and Multi-AI Agents Framework for Hardware Design](/publication/2025-06_marco-configurable-graph-based-task-solving-and-multi-ai-agents-framework)

[Chia-Tung (Mark) Ho](/person/chia-tung-mark-ho), Jing Gong, [Yunsheng Bai](/person/yunsheng-bai), [Chenhui Deng](/person/chenhui-deng), Mark Haoxing Ren, [Brucek Khailany](/person/brucek-khailany)













### 2024 

[Learning to Compare Hardware Designs for High-Level Synthesis](/publication/2024-05_learning-compare-hardware-designs-high-level-synthesis)

[Yunsheng Bai](/person/yunsheng-bai), Atefeh Sohrabizadeh, Zijian Ding, [Rongjian Liang](/person/rongjian-liang), Weikai Li, Ding Wang, Mark Haoxing Ren, Yizhou Sun, Jason Cong