← Search

Yufei Kuang

12 accepted papers

2025

A Graph Enhanced Symbolic Discovery Framework For Efficient Logic Optimization

ICLR 2025poster

The efficiency of Logic Optimization (LO) has become one of the key bottlenecks in chip design. To prompt efficient LO, previous studies propose using a key scoring function to predict and prune a large number of ineffective nodes of the LO heuristics. However, the existing scoring functions struggl…

Cited by 0SourcePDFScholar
2025

Dynamic Configuration for Cutting Plane Separators via Reinforcement Learning on Incremental Graph

NeurIPS 2025poster

Cutting planes (cuts) are essential for solving mixed-integer linear programming (MILP) problems, as they tighten the feasible solution space and accelerate the solving process. Modern MILP solvers offer diverse cutting plane separators to generate cuts, enabling users to leverage their potential co…

Cited by 0SourceScholar
2025

OptiTree: Hierarchical Thoughts Generation with Tree Search for LLM Optimization Modeling

NeurIPS 2025poster

Optimization modeling is one of the most crucial but technical parts of operations research (OR). To automate the modeling process, existing works have leveraged large language models (LLMs), prompting them to break down tasks into steps for generating variables, constraints, and objectives. How…

Cited by 0SourceScholar
2024

MILP-StuDio: MILP Instance Generation via Block Structure Decomposition

NeurIPS 2024poster

Mixed-integer linear programming (MILP) is one of the most popular mathematical formulations with numerous applications. In practice, improving the performance of MILP solvers often requires a large amount of high-quality data, which can be challenging to collect. Researchers thus turn to generation…

Cited by 11SourcePDFScholar
2024

Neural Krylov Iteration for Accelerating Linear System Solving

NeurIPS 2024spotlight

Solving large-scale sparse linear systems is essential in fields like mathematics, science, and engineering. Traditional numerical solvers, mainly based on the Krylov subspace iteration algorithm, suffer from the low-efficiency problem, which primarily arises from the less-than-ideal iteration. To t…

Cited by 3SourcePDFScholar
2024

Rethinking Branching on Exact Combinatorial Optimization Solver: The First Deep Symbolic Discovery Framework

ICLR 2024poster

Machine learning (ML) has been shown to successfully accelerate solving NP-hard combinatorial optimization (CO) problems under the branch and bound framework. However, the high training and inference cost and limited interpretability of ML approaches severely limit their wide application to modern…

Cited by 11SourcePDFScholar
2024

Towards General Algorithm Discovery for Combinatorial Optimization: Learning Symbolic Branching Policy from Bipartite Graph

ICML 2024poster

Machine learning (ML) approaches have been successfully applied to accelerating exact combinatorial optimization (CO) solvers. However, many of them fail to explain what patterns they have learned that accelerate the CO algorithms due to the black-box nature of ML models like neural networks, and th…

Cited by 6SourcePDFScholar
2023

Learning Cut Selection for Mixed-Integer Linear Programming via Hierarchical Sequence Model

ICLR 2023poster

Cutting planes (cuts) are important for solving mixed-integer linear programs (MILPs), which formulate a wide range of important real-world applications. Cut selection---which aims to select a proper subset of the candidate cuts to improve the efficiency of solving MILPs---heavily depends on (P1) wh…

Cited by 62SourcePDFScholar
2023

Learning robust representation for reinforcement learning with distractions by reward sequence prediction

UAI 2023poster

Reinforcement learning algorithms have achieved remarkable success in acquiring behavioral skills directly from pixel inputs. However, their application in real-world scenarios presents challenges due to their sensitivity to visual distractions (e.g., changes in viewpoint and light). A key factor co…

2023

State Sequences Prediction via Fourier Transform for Representation Learning

NeurIPS 2023spotlight

While deep reinforcement learning (RL) has been demonstrated effective in solving complex control tasks, sample efficiency remains a key challenge due to the large amounts of data required for remarkable performance. Existing research explores the application of representation learning for data-effi…

2022

Learning Robust Policy against Disturbance in Transition Dynamics via State-Conservative Policy Optimization

AAAI 2022technical

Deep reinforcement learning algorithms can perform poorly in real-world tasks due to the discrepancy between source and target environments. This discrepancy is commonly viewed as the disturbance in transition dynamics. Many existing algorithms learn robust policies by modeling the disturbance and a…

Cited by 24SourcePDFScholar
2020

Promoting Stochasticity for Expressive Policies via a Simple and Efficient Regularization Method

NeurIPS 2020poster

Many recent reinforcement learning (RL) methods learn stochastic policies with entropy regularization for exploration and robustness. However, in continuous action spaces, integrating entropy regularization with expressive policies is challenging and usually requires complex inference procedures. To…

Cited by 8SourcePDFScholar