| 2026 | VTS | FVRuleLearner: Operator-Level Reasoning Tree (Op-Tree)-Based Rules Learning for Formal Verification. | Lily Jiaxin Wan, Chia-Tung Ho, Yunsheng Bai, Cunxi Yu, Deming Chen, Haoxing Ren |
| 2025 | DAC | ChipAlign: Instruction Alignment in Large Language Models for Chip Design via Geodesic Interpolation. | Chenhui Deng, Yunsheng Bai, Haoxing Ren |
| 2023 | ICCAD | Robust GNN-Based Representation Learning for HLS. | Atefeh Sohrabizadeh, Yunsheng Bai, Yizhou Sun, Jason Cong |
| 2023 | WWW | Code Recommendation for Open Source Software Developers. | Yiqiao Jin, Yunsheng Bai, Yanqiao Zhu, Yizhou Sun, Wei Wang |
| 2022 | DAC | Improving GNN-based accelerator design automation with meta learning. | Yunsheng Bai, Atefeh Sohrabizadeh, Yizhou Sun, Jason Cong |
| 2022 | DAC | Automated accelerator optimization aided by graph neural networks. | Atefeh Sohrabizadeh, Yunsheng Bai, Yizhou Sun, Jason Cong |
| 2022 | FPGA | Automated Accelerator Optimization Aided by Graph Neural Networks. | Atefeh Sohrabizadeh, Yunsheng Bai, Yizhou Sun, Jason Cong |
| 2022 | KDD | Dual-Geometric Space Embedding Model for Two-View Knowledge Graphs. | Roshni G. Iyer, Yunsheng Bai, Wei Wang, Yizhou Sun |
| 2021 | ICML | GLSearch: Maximum Common Subgraph Detection via Learning to Search. | Yunsheng Bai, Derek Xu, Yizhou Sun, Wei Wang |
| 2020 | AAAI | Learning-Based Efficient Graph Similarity Computation via Multi-Scale Convolutional Set Matching. | Yunsheng Bai, Hao Ding, Ken Gu, Yizhou Sun, Wei Wang |
| 2020 | KDD | GHashing: Semantic Graph Hashing for Approximate Similarity Search in Graph Databases. | Zongyue Qin, Yunsheng Bai, Yizhou Sun |
| 2019 | IJCAI | Unsupervised Inductive Graph-Level Representation Learning via Graph-Graph Proximity. | Yunsheng Bai, Hao Ding, Yang Qiao, Agustin Marinovic, Ken Gu, Ting Chen, Yizhou Sun, Wei Wang |
| 2019 | WSDM | SimGNN: A Neural Network Approach to Fast Graph Similarity Computation. | Yunsheng Bai, Hao Ding, Song Bian, Ting Chen, Yizhou Sun, Wei Wang |