Geng, Zijie

12 publications

NeurIPS 2025 Accurate KV Cache Eviction via Anchor Direction Projection for Efficient LLM Inference Zijie Geng, Jie Wang, Ziqi Liu, Feng Ju, Yiming Li, Xing Li, Mingxuan Yuan, Jianye Hao, Defu Lian, Enhong Chen, Feng Wu

ICLR 2025 Apollo-MILP: An Alternating Prediction-Correction Neural Solving Framework for Mixed-Integer Linear Programming Haoyang Liu, Jie Wang, Zijie Geng, Xijun Li, Yuxuan Zong, Fangzhou Zhu, Jianye Hao, Feng Wu

NeurIPS 2025 Benchmarking End-to-End Performance of AI-Based Chip Placement Algorithms 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, Feng Wu

ICLR 2025 Differentiable Integer Linear Programming Zijie Geng, Jie Wang, Xijun Li, Fangzhou Zhu, Jianye Hao, Bin Li, Feng Wu

ICLR 2025 LaMPlace: Learning to Optimize Cross-Stage Metrics in Macro Placement Zijie Geng, Jie Wang, Ziyan Liu, Siyuan Xu, Zhentao Tang, Shixiong Kai, Mingxuan Yuan, Jianye Hao, Feng Wu

NeurIPS 2025 RoME: Domain-Robust Mixture-of-Experts for MILP Solution Prediction Across Domains Tianle Pu, Zijie Geng, Haoyang Liu, Shixuan Liu, Jie Wang, Li Zeng, Chao Chen, Changjun Fan

ICLR 2024 Accelerating Data Generation for Neural Operators via Krylov Subspace Recycling Hong Wang, Zhongkai Hao, Jie Wang, Zijie Geng, Zhen Wang, Bin Li, Feng Wu

NeurIPS 2024 MILP-StuDio: MILP Instance Generation via Block Structure Decomposition Haoyang Liu, Jie Wang, Wanbo Zhang, Zijie Geng, Yufei Kuang, Xijun Li, Yongdong Zhang, Bin Li, Feng Wu

NeurIPS 2024 Neural Krylov Iteration for Accelerating Linear System Solving Jian Luo, Jie Wang, Hong Wang, Huanshuo Dong, Zijie Geng, Hanzhu Chen, Yufei Kuang

ICML 2024 Reinforcement Learning Within Tree Search for Fast Macro Placement Zijie Geng, Jie Wang, Ziyan Liu, Siyuan Xu, Zhentao Tang, Mingxuan Yuan, Jianye Hao, Yongdong Zhang, Feng Wu

NeurIPS 2023 A Deep Instance Generative Framework for MILP Solvers Under Limited Data Availability Zijie Geng, Xijun Li, Jie Wang, Xiao Li, Yongdong Zhang, Feng Wu

ICLR 2023 De Novo Molecular Generation via Connection-Aware Motif Mining Zijie Geng, Shufang Xie, Yingce Xia, Lijun Wu, Tao Qin, Jie Wang, Yongdong Zhang, Feng Wu, Tie-Yan Liu