activity
20222026
most citedLearning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributions

14 citations · 25 across the 13 of their papers we have counts for

collaborators

17 papers

cs.AI2026

Efficient Test-Time Optimization for Multi-Agent Proof Autoformalization

Tian-Shuo Liu, Shiyuan Zhang, Zijie Geng +5

Full-proof autoformalization bridges extensive mathematical proofs in natural language with formally validated reasoning, offering a pathway to elevate the ceiling of verifiable ma…

cs.AI2026

Opt-Verifier: Unleashing the Power of LLMs for Optimization Modeling via Dual-Side Verification

Haoyang Liu, Jie Wang, Boxuan Niu +8

Building mathematical optimization models is critical in operations research (OR), while it requires substantial human expertise. Recent advancements have utilized large language m…

cs.AI2026

PAPO: Stabilizing Rubric Integration Training via Decoupled Advantage Normalization

Zelin Tan, Zhouliang Yu, Bohan Lin +9

We propose Process-Aware Policy Optimization (PAPO), a method that integrates process-level evaluation into Group Relative Policy Optimization (GRPO) through decoupled advantage no…

cs.LG2025

CoCo-MILP: Inter-Variable Contrastive and Intra-Constraint Competitive MILP Solution Prediction

Tianle Pu, Jianing Li, Yingying Gao +5

Mixed-Integer Linear Programming (MILP) is a cornerstone of combinatorial optimization, yet solving large-scale instances remains a significant computational challenge. Recently, G…

cs.LG2025

RoME: Domain-Robust Mixture-of-Experts for MILP Solution Prediction across Domains

Tianle Pu, Zijie Geng, Haoyang Liu +5

Mixed-Integer Linear Programming (MILP) is a fundamental and powerful framework for modeling complex optimization problems across diverse domains. Recently, learning-based methods…

cs.AI2025

Accelerating IC Thermal Simulation Data Generation via Block Krylov and Operator Action

Hong Wang, Wenkai Yang, Jie Wang +6

Recent advances in data-driven approaches, such as neural operators (NOs), have shown substantial efficacy in reducing the solution time for integrated circuit (IC) thermal simulat…

cs.AR2025

Piano: A Multi-Constraint Pin Assignment-Aware Floorplanner

Zhexuan Xu, Kexin Zhou, Jie Wang +5

Floorplanning is a critical step in VLSI physical design, increasingly complicated by modern constraints such as fixed-outline requirements, whitespace removal, and the presence of…

cs.RO2025

One Step Beyond: Feedthrough & Placement-Aware Rectilinear Floorplanner

Zhexuan Xu, Jie Wang, Siyuan Xu +3

Floorplanning determines the shapes and locations of modules on a chip canvas and plays a critical role in optimizing the chip's Power, Performance, and Area (PPA) metrics. However…

cs.LG2025

Apollo-MILP: An Alternating Prediction-Correction Neural Solving Framework for Mixed-Integer Linear Programming

Haoyang Liu, Jie Wang, Zijie Geng +5

Leveraging machine learning (ML) to predict an initial solution for mixed-integer linear programming (MILP) has gained considerable popularity in recent years. These methods predic…

cs.LG2024

MILP-StuDio: MILP Instance Generation via Block Structure Decomposition

Haoyang Liu, Jie Wang, Wanbo Zhang +6

Mixed-integer linear programming (MILP) is one of the most popular mathematical formulations with numerous applications. In practice, improving the performance of MILP solvers ofte…

cs.AR2024

Benchmarking End-To-End Performance of AI-Based Chip Placement Algorithms

Zhihai Wang, Zijie Geng, Zhaojie Tu +12

The increasing complexity of modern very-large-scale integration (VLSI) design highlights the significance of Electronic Design Automation (EDA) technologies. Chip placement is a c…

cs.AI2024

Machine Learning Insides OptVerse AI Solver: Design Principles and Applications

Xijun Li, Fangzhou Zhu, Hui-Ling Zhen +23

In an era of digital ubiquity, efficient resource management and decision-making are paramount across numerous industries. To this end, we present a comprehensive study on the inte…

cs.LG2024

Accelerating Data Generation for Neural Operators via Krylov Subspace Recycling

Hong Wang, Zhongkai Hao, Jie Wang +4

Learning neural operators for solving partial differential equations (PDEs) has attracted great attention due to its high inference efficiency. However, training such operators req…

cs.LG2023★ 3 cited

A Deep Instance Generative Framework for MILP Solvers Under Limited Data Availability

Zijie Geng, Xijun Li, Jie Wang +3

In the past few years, there has been an explosive surge in the use of machine learning (ML) techniques to address combinatorial optimization (CO) problems, especially mixed-intege…

cs.LG2023★ 8 cited

De Novo Molecular Generation via Connection-aware Motif Mining

Zijie Geng, Shufang Xie, Yingce Xia +6

De novo molecular generation is an essential task for science discovery. Recently, fragment-based deep generative models have attracted much research attention due to their flexibi…

cs.LG2023

Generalization in Visual Reinforcement Learning with the Reward Sequence Distribution

Jie Wang, Rui Yang, Zijie Geng +7

Generalization in partially observed markov decision processes (POMDPs) is critical for successful applications of visual reinforcement learning (VRL) in real scenarios. A widely u…

cs.LG2022★ 14 cited

Learning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributions

Rui Yang, Jie Wang, Zijie Geng +4

Generalization across different environments with the same tasks is critical for successful applications of visual reinforcement learning (RL) in real scenarios. However, visual di…