Research

My current research interests include Learning to Optimize and AI4EDA.

Publications

1. Zijie Geng, Jie Wang, Ziyan Liu, Siyuan Xu, Zhentao Tang, Shixiong Kai, Mingxuan Yuan, Jianye Hao, Feng Wu. “LaMPlace: Learning to Optimize Cross-Stage Metrics in Macro Placement.” ICLR 2025 (Oral). [paper][code]
2. Zijie Geng, Jie Wang, Xijun Li, Fangzhou Zhu, Jianye Hao, Bin Li, Feng Wu. “Differentiable Integer Linear Programming.” ICLR 2025 (Spotlight). [paper][code]
3. Zijie Geng, Jie Wang, Ziyan Liu, Siyuan Xu, Zhentao Tang, Mingxuan Yuan, Jianye Hao, Yongdong Zhang, Feng Wu. “Reinforcement Learning within Tree Search for Fast Macro Placement.” ICML 2024. [paper][code]
4. Zijie Geng, Xijun Li, Jie Wang, Xiao Li, Yongdong Zhang, Feng Wu. “A Deep Instance Generative Framework for MILP Solvers Under Limited Data Availability.” NeurIPS 2023 (Spotlight). [paper] [project] [code]
5. Zijie Geng, Shufang Xie, Yingce Xia, Lijun Wu, Tao Qin, Jie Wang, Yongdong Zhang, Feng Wu, Tie-Yan Liu. “De Novo Molecular Generation via Connection-aware Motif Mining.” ICLR 2023. [paper] [code]
6. Haoyang Liu, Jie Wang, Zijie Geng, Xijun Li, Yuxuan Zong, Fangzhou Zhu, Jianye Hao, Feng Wu. “Apollo-MILP: An Alternating Prediction-Correction Neural Solving Framework for Mixed-Integer Linear Programming.” ICLR 2025.
7. Haoyang Liu, Jie Wang, Wanbo Zhang, Zijie Geng, Yufei Kuang, Xijun Li, Bin Li, Yongdong Zhang, Feng Wu. “MILP-StuDio: MILP Instance Generation via Block Structure Decomposition.” NeurIPS 2024. [paper]
8. Jian Luo, Jie Wang, Hong Wang, huanshuo dong, Zijie Geng, Hanzhu Chen, Yufei Kuang. “Neural Krylov Iteration for Accelerating Linear System Solving”. NeurIPS 2024 (Spotlight). [paper][code]
9. Hong Wang, Zhongkai Hao, Jie Wang, Zijie Geng, Zhen Wang, Bin Li, Feng Wu. “Accelerating Data Generation for Neural Operators via Krylov Subspace Recycling.” ICLR 2024 (Spotlight). [paper]
10. Rui Yang, Jie Wang, Zijie Geng, Mingxuan Ye, Shuiwang Ji, Bin Li, Feng Wu. “Learning task-relevant representations for generalization via characteristic functions of reward sequence distributions.” SIGKDD 2022. [paper] [code]

Preprint

1. Zhihai Wang*, Zijie Geng*, Zhaojie Tu*, Jie Wang, Yuxi Qian, Zhexuan Xu, Ziyan Liu, Siyuan Xu, Zhentao Tang, Shixiong Kai, Mingxuan Yuan, Jianye Hao, Bin Li, Yongdong Zhang, Feng Wu. “Benchmarking End-To-End Performance of AI-Based Chip Placement Algorithms.” [paper][code]
2. Xijun Li, Fangzhou Zhu, Hui-Ling Zhen, Weilin Luo, Meng Lu, Yimin Huang, Zhenan Fan, Zirui Zhou, Yufei Kuang, Zhihai Wang, Zijie Geng, Yang Li, Haoyang Liu, Zhiwu An, Muming Yang, Jianshu Li, Jie Wang, Junchi Yan, Defeng Sun, Tao Zhong, Yong Zhang, Jia Zeng, Mingxuan Yuan, Jianye Hao, Jun Yao, Kun Mao. “Machine Learning Insides OptVerse AI Solver: Design Principles and Applications.” [paper]
3. Jie Wang, Zijie Geng, Xijun Li, Jianye Hao, Yongdong Zhang, Feng Wu. “G2MILP: Learning to Generate Mixed-Integer Linear Programming Instances for MILP Solvers.” [paper]
4. Jie Wang, Rui Yang, Zijie Geng, Zhihao Shi, Mingxuan Ye, Qi Zhou, Shuiwang Ji, Bin Li, Yongdong Zhang, Feng Wu. “Generalization in Visual Reinforcement Learning with the Reward Sequence Distribution.” [paper]

Services

• Journal Reviewer:
  • IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)
• Conference Reviewer:
  • NeurIPS 2023 - 2025 (NeurIPS 2024 Top Reviewer)
  • ICLR 2024 - 2025
  • ICML 2024 - 2025
  • AAAI 2024 - 2025
  • KDD 2023