Graph Edit Distance Computation via Graph Neural Networks. - Yunsheng Bai; Hao Ding 003; Song Bian 002; Ting Chen 007; Yizhou Sun; Wei Wang 010 (CoRR, 2018)

Convolutional Neural Networks for Fast Approximation of Graph Edit Distance. - Yunsheng Bai; Hao Ding 003; Yizhou Sun; Wei Wang 010 (CoRR, 2018)

Convolutional Set Matching for Graph Similarity. - Yunsheng Bai; Hao Ding 003; Yizhou Sun; Wei Wang 010 (CoRR, 2018)

Unsupervised Inductive Graph-Level Representation Learning via Graph-Graph Proximity. - Yunsheng Bai; Hao Ding 003; Yang Qiao; Agustin Marinovic; Ken Gu; Ting Chen 007; Yizhou Sun; Wei Wang 010 (IJCAI, 2019)

SimGNN: A Neural Network Approach to Fast Graph Similarity Computation. - Yunsheng Bai; Hao Ding 003; Song Bian 002; Ting Chen 007; Yizhou Sun; Wei Wang 010 (WSDM, 2019)

Unsupervised Inductive Whole-Graph Embedding by Preserving Graph Proximity. - Yunsheng Bai; Hao Ding 003; Yang Qiao; Agustin Marinovic; Ken Gu; Ting Chen 007; Yizhou Sun; Wei Wang 010 (CoRR, 2019)

Learning-Based Efficient Graph Similarity Computation via Multi-Scale Convolutional Set Matching. - Yunsheng Bai; Hao Ding 003; Ken Gu; Yizhou Sun; Wei Wang 010 (AAAI, 2020)

GHashing: Semantic Graph Hashing for Approximate Similarity Search in Graph Databases. - Zongyue Qin; Yunsheng Bai; Yizhou Sun (KDD, 2020)

Fast Detection of Maximum Common Subgraph via Deep Q-Learning. - Yunsheng Bai; Derek Xu; Alex Wang; Ken Gu; Xueqing Wu 001; Agustin Marinovic; Christopher Ro; Yizhou Sun; Wei Wang 010 (CoRR, 2020)

Hierarchical and Fast Graph Similarity Computation via Graph Coarsening and Deep Graph Learning. - Haoyan Xu; Runjian Chen; Yunsheng Bai; Jie Feng 006; Ziheng Duan; Ke Luo; Yizhou Sun; Wei Wang 010 (CoRR, 2020)

Bi-Level Graph Neural Networks for Drug-Drug Interaction Prediction. - Yunsheng Bai; Ken Gu; Yizhou Sun; Wei Wang 010 (CoRR, 2020)

Multivariate Time Series Classification with Hierarchical Variational Graph Pooling. - Haoyan Xu; Ziheng Duan; Yunsheng Bai; Yida Huang; Anni Ren; Qianru Yu; Qianru Zhang; Yueyang Wang; Xiaoqian Wang 001; Yizhou Sun; Wei Wang 010 (CoRR, 2020)

GLSearch: Maximum Common Subgraph Detection via Learning to Search. - Yunsheng Bai; Derek Xu; Yizhou Sun; Wei Wang 010 (ICML, 2021)

Enabling Automated FPGA Accelerator Optimization Using Graph Neural Networks. - Atefeh Sohrabizadeh; Yunsheng Bai; Yizhou Sun; Jason Cong (CoRR, 2021)

Improving GNN-based accelerator design automation with meta learning. - Yunsheng Bai; Atefeh Sohrabizadeh; Yizhou Sun; Jason Cong (DAC, 2022)

Automated accelerator optimization aided by graph neural networks. - Atefeh Sohrabizadeh; Yunsheng Bai; Yizhou Sun; Jason Cong (DAC, 2022)

Dual-Geometric Space Embedding Model for Two-View Knowledge Graphs. - Roshni G. Iyer; Yunsheng Bai; Wei Wang 010; Yizhou Sun (KDD, 2022)

Subgraph Matching via Query-Conditioned Subgraph Matching Neural Networks and Bi-Level Tree Search. - Yunsheng Bai; Derek Xu; Yizhou Sun; Wei Wang 010 (CoRR, 2022)

Robust GNN-Based Representation Learning for HLS. - Atefeh Sohrabizadeh; Yunsheng Bai; Yizhou Sun; Jason Cong (ICCAD, 2023)

Towards a Comprehensive Benchmark for High-Level Synthesis Targeted to FPGAs. - Yunsheng Bai; Atefeh Sohrabizadeh; Zongyue Qin; Ziniu Hu; Yizhou Sun; Jason Cong (NeurIPS, 2023)

Code Recommendation for Open Source Software Developers. - Yiqiao Jin; Yunsheng Bai; Yanqiao Zhu 001; Yizhou Sun; Wei Wang 010 (WWW, 2023)

Neural Network-based Graph-Level Operator Learning - Yunsheng Bai (2023)

Learning to Compare Hardware Designs for High-Level Synthesis. - Yunsheng Bai; Atefeh Sohrabizadeh; Zijian Ding; Rongjian Liang; Weikai Li 002; Ding Wang; Haoxing Ren; Yizhou Sun; Jason Cong (MLCAD, 2024)

Cross-Modality Program Representation Learning for Electronic Design Automation with High-Level Synthesis. - Zongyue Qin; Yunsheng Bai; Atefeh Sohrabizadeh; Zijian Ding; Ziniu Hu; Yizhou Sun; Jason Cong (MLCAD, 2024)

Does Few-Shot Learning Help LLM Performance in Code Synthesis? - Derek Xu; Tong Xie; Botao Xia; Haoyu Li; Yunsheng Bai; Yizhou Sun; Wei Wang 010 (CoRR, 2024)

ChipAlign: Instruction Alignment in Large Language Models for Chip Design via Geodesic Interpolation. - Chenhui Deng; Yunsheng Bai; Haoxing Ren (CoRR, 2024)

AssertionForge: Enhancing Formal Verification Assertion Generation with Structured Representation of Specifications and RTL. - Yunsheng Bai; Ghaith Bany Hamad; Syed Suhaib; Haoxing Ren (CoRR, 2025)

Marco: Configurable Graph-Based Task Solving and Multi-AI Agents Framework for Hardware Design. - Chia-Tung Ho; Jing Gong; Yunsheng Bai; Chenhui Deng; Haoxing Ren; Brucek Khailany (CoRR, 2025)

Polymath: A Self-Optimizing Agent with Dynamic Hierarchical Workflow. - Chia-Tung Ho; Jing Gong; Xufeng Yao; Yunsheng Bai; Abhishek B. Akkur; Haoxing Ren (CoRR, 2025)