← Search

Wen Song

21 accepted papers

2026

A General Neural Backbone for Mixed-Integer Linear Optimization via Dual Attention

ICML 2026poster

Mixed-integer linear programming (MILP) is a foundational framework for combinatorial optimization across science and engineering, but remains hard to solve at scale due to NP-hardness.Recent learning-based methods typically model MILP instances as variable–constraint bipartite graphs and use Graph …

Cited by 0SourceScholar
2026

Generalizable Heuristic Generation Through LLMs with Meta-Optimization

ICLR 2026poster

Heuristic design with large language models (LLMs) has emerged as a promising approach for tackling combinatorial optimization problems (COPs). However, existing approaches often rely on manually predefined evolutionary computation (EC) heuristic-optimizers and single-task training schemes, which ma…

Cited by 0SourcecodeScholar
2026

Instance-wise Adaptive Scheduling via Derivative-Free Meta-Learning

ICLR 2026poster

Deep Reinforcement Learning has achieved remarkable progress in solving NP-hard scheduling problems. However, existing methods primarily focus on optimizing average performance over training instances, overlooking the core objective of solving each individual instance with high quality. While severa…

Cited by 0SourceScholar
2026

RESCHED: Rethinking Flexible Job Shop Scheduling from a Transformer-based Architecture with Simplified States

ICLR 2026poster

Neural approaches to the Flexible Job Shop Scheduling Problem (FJSP), particularly those based on deep reinforcement learning (DRL), have gained growing attention in recent years. Yet existing methods often rely on cumbersome state representations (i.e. sometimes requiring more than 20 handcrafted f…

Cited by 0SourcecodeScholar
2026

Towards Efficient Constraint Handling in Neural Solvers for Routing Problems

ICLR 2026poster

Neural solvers have achieved impressive progress in addressing simple routing problems, particularly excelling in computational efficiency. However, their advantages under complex constraints remain nascent, for which current constraint-handling schemes via feasibility masking or implicit feasibilit…

Cited by 0SourcecodeScholar
2024

Collaboration! Towards Robust Neural Methods for Routing Problems

NeurIPS 2024poster

Despite enjoying desirable efficiency and reduced reliance on domain expertise, existing neural methods for vehicle routing problems (VRPs) suffer from severe robustness issues — their performance significantly deteriorates on clean instances with crafted perturbations. To enhance robustness, we pro…

2024

Cross-Problem Learning for Solving Vehicle Routing Problems

IJCAI 2024poster

Existing neural heuristics often train a deep architecture from scratch for each specific vehicle routing problem (VRP), ignoring the transferable knowledge across different VRP variants. This paper proposes the cross-problem learning to assist heuristics training for different downstream VRP varian…

2024

Deep Reinforcement Learning Guided Improvement Heuristic for Job Shop Scheduling

ICLR 2024poster

Recent studies in using deep reinforcement learning (DRL) to solve Job-shop scheduling problems (JSSP) focus on construction heuristics. However, their performance is still far from optimality, mainly because the underlying graph representation scheme is unsuitable for modelling partial solutions at…

2024

Learning Topological Representations with Bidirectional Graph Attention Network for Solving Job Shop Scheduling Problem

UAI 2024poster

Existing learning-based methods for solving job shop scheduling problems (JSSP) usually use off-the-shelf GNN models tailored to undirected graphs and neglect the rich and meaningful topological structures of disjunctive graphs (DGs). This paper proposes the topology-aware bidirectional graph attent…

2024

Learning to Handle Complex Constraints for Vehicle Routing Problems

NeurIPS 2024poster

Vehicle Routing Problems (VRPs) can model many real-world scenarios and often involve complex constraints. While recent neural methods excel in constructing solutions based on feasibility masking, they struggle with handling complex constraints, especially when obtaining the masking itself is NP-har…

2024

MVMoE: Multi-Task Vehicle Routing Solver with Mixture-of-Experts

ICML 2024poster

Learning to solve vehicle routing problems (VRPs) has garnered much attention. However, most neural solvers are only structured and trained independently on a specific problem, making them less generic and practical. In this paper, we aim to develop a unified neural solver that can cope with a range…

2023

Ensemble-based Deep Reinforcement Learning for Vehicle Routing Problems under Distribution Shift

NeurIPS 2023poster

While performing favourably on the independent and identically distributed (i.i.d.) instances, most of the existing neural methods for vehicle routing problems (VRPs) struggle to generalize in the presence of a distribution shift. To tackle this issue, we propose an ensemble-based deep reinforcement…

Cited by 21SourcePDFScholar
2023

Towards Omni-generalizable Neural Methods for Vehicle Routing Problems

ICML 2023poster

Learning heuristics for vehicle routing problems (VRPs) has gained much attention due to the less reliance on hand-crafted rules. However, existing methods are typically trained and tested on the same task with a fixed size and distribution (of nodes), and hence suffer from limited generalization pe…

2022

Efficient Neural Neighborhood Search for Pickup and Delivery Problems

IJCAI 2022poster

We present an efficient Neural Neighborhood Search (N2S) approach for pickup and delivery problems (PDPs). In specific, we design a powerful Synthesis Attention that allows the vanilla self-attention to synthesize various types of features regarding a route solution. We also exploit two customized d…

2022

Graph Learning Assisted Multi-Objective Integer Programming

NeurIPS 2022accept

Objective-space decomposition algorithms (ODAs) are widely studied for solving multi-objective integer programs. However, they often encounter difficulties in handling scalarized problems, which could cause infeasibility or repetitive nondominated points and thus induce redundant runtime. To mitigat…

Cited by 11SourcePDFScholar
2022

Learning Scenario Representation for Solving Two-stage Stochastic Integer Programs

ICLR 2022poster

Many practical combinatorial optimization problems under uncertainty can be modeled as stochastic integer programs (SIPs), which are extremely challenging to solve due to the high complexity. To solve two-stage SIPs efficiently, we propose a conditional variational autoencoder (CVAE) based method to…

Cited by 17SourcePDFScholar
2021

Learning Large Neighborhood Search Policy for Integer Programming

NeurIPS 2021spotlight

We propose a deep reinforcement learning (RL) method to learn large neighborhood search (LNS) policy for integer programming (IP). The RL policy is trained as the destroy operator to select a subset of variables at each step, which is reoptimized by an IP solver as the repair operator. However, the…

2021

Learning to Iteratively Solve Routing Problems with Dual-Aspect Collaborative Transformer

NeurIPS 2021poster

Recently, Transformer has become a prevailing deep architecture for solving vehicle routing problems (VRPs). However, it is less effective in learning improvement models for VRP because its positional encoding (PE) method is not suitable in representing VRP solutions. This paper presents a novel Dua…

2021

Multi-Decoder Attention Model with Embedding Glimpse for Solving Vehicle Routing Problems

AAAI 2021technical

We present a novel deep reinforcement learning method to learn construction heuristics for vehicle routing problems. In specific, we propose a Multi-Decoder Attention Model (MDAM) to train multiple diverse policies, which effectively increases the chance of finding good solutions compared with exist…

2021

NeuroLKH: Combining Deep Learning Model with Lin-Kernighan-Helsgaun Heuristic for Solving the Traveling Salesman Problem

NeurIPS 2021poster

We present NeuroLKH, a novel algorithm that combines deep learning with the strong traditional heuristic Lin-Kernighan-Helsgaun (LKH) for solving Traveling Salesman Problem. Specifically, we train a Sparse Graph Network (SGN) with supervised learning for edge scores and unsupervised learning for nod…

2020

Learning to Dispatch for Job Shop Scheduling via Deep Reinforcement Learning

NeurIPS 2020poster

Priority dispatching rule (PDR) is widely used for solving real-world Job-shop scheduling problem (JSSP). However, the design of effective PDRs is a tedious task, requiring a myriad of specialized knowledge and often delivering limited performance. In this paper, we propose to automatically learn PD…