VLDB 2026 Research / reviewers in the wild / expert
Wen Song 0004
dblp:50/5489-4
· DBLP profile ↗
48ranked-venue papers
8as first author
41since 2021 · last 2026
0000-0001-7624-1861ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 36 · 6 first-author · 30 since 2021Applied, interdisciplinary, general and emerging computing · 8 · 2 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 6 · 2 first-author · 4 since 2021Databases, data management, data science and information retrieval · 4 · 4 since 2021Software engineering, systems software and programming languages · 1
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Learning to Generate Preferences for Multiobjective Deep LearningabstractMultiobjective optimization (MOO) is important for deep learning applications with multiple conflicting objectives. Pareto front learning (PFL) methods learn a single model conditioned on the preference of objectives and can be applied to any preference at inference time. However, existing PFL methods use predefined strategies (e.g., uniform sampling) to generate preferences, which could result in unevenly spaced solutions since the shape of Pareto front is largely ignored. In this article, we propose a lightweight and model-agnostic method to train a preference generator for a given PFL model, which learns to generate proper preferences from uniformly sampled ones, such that the resulting solutions are evenly spaced on the Pareto front. Compared to previous works, our method enables a more rational allocation of preferences, which can either be utilized to enhance a pretrained PFL model or be seamlessly integrated into the PFL training process to improve efficiency. We apply our method to state-of-the-art PFL methods with various backbones (e.g., multilayer perceptron, convolutional neural network, transformer) and validate the significance of preference generation across various tasks, from multitask supervised learning to multiobjective reinforcement learning-based neural combinatorial optimization. Experimental results show that our method improves the backbone algorithm in most settings, showing its effectiveness and general applicability. Peixin Huang, Yu Sun 0051, Gang Wang 0014, Yaoxin Wu, Wen Song 0004, Yew-Soon Ong |
IEEE Trans. Ind. Informatics | 6 |
| 2025 | Visual-Enhanced Multimodal Framework for Flexible Job Shop Scheduling ProblemabstractMultimodal models leverage complementary information across modalities to enrich feature representations. While visual information shows potential in representing structure for some combinatorial optimization problems (COPs), its application to complex scheduling like the Flexible Job Shop Scheduling Problem (FJSP) remains underexplored. Current learning-based FJSP solvers predominantly rely on handcrafted state features. This dependence can lead to inconsistencies and may not fully capture the problem's intricate dynamics. Crucially, these methods overlook visual modalities. Visual representations offer a distinct advantage by inherently capturing the global topological structure and complex resource interactions within the FJSP state. Unlike localized handcrafted features, this holistic, structural view provides a richer foundation for understanding scheduling complexity and making informed decisions. To overcome these limitations by leveraging visual information-known for representing topological structures and providing richer state representations-we introduce the AO-framework. This multimodal feature fusion approach enhances handcrafted state features by integrating insights from visual data. Our core contribution is a novel fusion mechanism utilizing orthogonal projection and local attention. Unlike traditional methods that often rely on simple concatenation of visual data, our method uniquely reduces redundancy by projecting global image-derived features onto local handcrafted features. This process extracts distinct information inherent to the visual modality, significantly improving the quality and complementarity of the resulting state features and enabling more informed scheduling decisions. To our knowledge, the AO-framework represents the first multimodal framework applied to scheduling problems, demonstrating the significant potential of visual information in this domain. Extensive experiments across various FJSP solvers and datasets confirm that our framework yields substantial enhancements in solution quality, decision-making capabilities, and generalization. Peng Zhao 0018, Zhiguang Cao, Di Wang 0004, Wen Song 0004, Wei Pang 0001, You Zhou 0008, Yuan Jiang 0007 |
ACM Multimedia | 4 |
| 2025 | A novel local enhanced channel self-attention based on Transformer for industrial remaining useful life prediction
Zhizheng Zhang 0008, Wen Song 0004, Wenxu Sun, Qiqiang Li |
Eng. Appl. Artif. Intell. | 2 |
| 2025 | Multivariate time series generation based on dual-channel Transformer conditional GAN for industrial remaining useful life prediction
Zhizheng Zhang 0008, Wenxu Sun, Wen Song 0004, Qiqiang Li |
Knowl. Based Syst. | 4 |
| 2025 | Solving two-stage stochastic integer programs via representation learning
Yaoxin Wu, Zhiguang Cao, Wen Song 0004, Yingqian Zhang 0001 |
Neural Networks | 3 |
| 2025 | Conditional Neural Heuristic for Multiobjective Vehicle Routing ProblemsabstractExisting neural heuristics for multiobjective vehicle routing problems (MOVRPs) are primarily conditioned on instance context, which failed to appropriately exploit preference and problem size, thus holding back the performance. To thoroughly unleash the potential, we propose a novel conditional neural heuristic (CNH) that fully leverages the instance context, preference, and size with an encoder-decoder structured policy network. Particularly, in our CNH, we design a dual-attention-based encoder to relate preferences and instance contexts, so as to better capture their joint effect on approximating the exact Pareto front (PF). We also design a size-aware decoder based on the sinusoidal encoding to explicitly incorporate the problem size into the embedding, so that a single trained model could better solve instances of various scales. Besides, we customize the REINFORCE algorithm to train the neural heuristic by leveraging stochastic preferences (SPs), which further enhances the training performance. Extensive experimental results on random and benchmark instances reveal that our CNH could achieve favorable approximation to the whole PF with higher hypervolume (HV) and lower optimality gap (Gap) than those of the existing neural and conventional heuristics. More importantly, a single trained model of our CNH can outperform other neural heuristics that are exclusively trained on each size. In addition, the effectiveness of the key designs is also verified through ablation studies. Mingfeng Fan, Yaoxin Wu, Zhiguang Cao, Wen Song 0004, Guillaume Sartoretti, Huan Liu 0028, Guohua Wu 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 4 |
| 2024 | Deep Reinforcement Learning Guided Improvement Heuristic for Job Shop SchedulingabstractRecent 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 each construction step. This paper proposes a novel DRL-guided improvement heuristic for solving JSSP, where graph representation is employed to encode complete solutions. We design a Graph-Neural-Network-based representation scheme, consisting of two modules to effectively capture the information of dynamic topology and different types of nodes in graphs encountered during the improvement process. To speed up solution evaluation during improvement, we present a novel message-passing mechanism that can evaluate multiple solutions simultaneously. We prove that the computational complexity of our method scales linearly with problem size. Experiments on classic benchmarks show that the improvement policy learned by our method outperforms state-of-the-art DRL-based methods by a large margin. Zhiguang Cao, Wen Song 0004, Yaoxin Wu, Jie Zhang 0002 |
ICLR | 3 |
| 2024 | MVMoE: Multi-Task Vehicle Routing Solver with Mixture-of-ExpertsabstractLearning 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 of VRP variants simultaneously. Specifically, we propose a multi-task vehicle routing solver with mixture-of-experts (MVMoE), which greatly enhances the model capacity without a proportional increase in computation. We further develop a hierarchical gating mechanism for the MVMoE, delivering a good trade-off between empirical performance and computational complexity. Experimentally, our method significantly promotes zero-shot generalization performance on 10 unseen VRP variants, and showcases decent results on the few-shot setting and real-world benchmark instances. We further conduct extensive studies on the effect of MoE configurations in solving VRPs, and observe the superiority of hierarchical gating when facing out-of-distribution data. The source code is available at: https://github.com/RoyalSkye/Routing-MVMoE. Jianan Zhou 0002, Zhiguang Cao, Yaoxin Wu, Wen Song 0004, Yining Ma 0001, Jie Zhang 0002 |
ICML | 4 |
| 2024 | Cross-Problem Learning for Solving Vehicle Routing Problems
Zhuoyi Lin, Yaoxin Wu, Bangjian Zhou, Zhiguang Cao, Wen Song 0004, Yingqian Zhang 0001, J. Senthilnath 0001 |
IJCAI | 5 |
| 2024 | MGMatch: Fast Matchmaking with Nonlinear Objective and Constraints via Multimodal Deep Graph LearningabstractAs a core problem of online games, matchmaking is to assign players into multiple teams to maximize their gaming experience. With the rapid development of game industry, it is increasingly difficulty to explicitly model players' experiences as linear functions. Instead, it is often modeled in a data-driven way by training a neural network. Meanwhile, complex rules must be satisfied to ensure the robustness of matchmaking, which are often described using logical operators. Therefore, matchmaking in practical scenarios is a challenging combinatorial optimization problem with nonlinear objective, linear constraints and logical constraints, which receives much less attention in previous research. In this paper, we propose a novel deep learning method for high-quality matchmaking in real-time. We first cast the problem as standard mixed-integer programming (MIP) by linearizing ReLU networks and logical constraints. Then, based on supervised learning, we design and train a multi-modal graph learning architecture to predict optimal solutions end-to-end from instance data, and solve a surrogate problem to efficiently obtain feasible solutions. Evaluation results on real industry datasets show that our method can deliver near-optimal solutions within 100ms. Yu Sun 0051, Kai Wang 0064, Zhipeng Hu, Runze Wu 0001, Yaoxin Wu, Wen Song 0004, Tangjie Lv, Changjie Fan |
KDD | 6 |
| 2024 | Collaboration! Towards Robust Neural Methods for Routing ProblemsabstractDespite 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 propose an ensemble-based *Collaborative Neural Framework (CNF)* w.r.t. the defense of neural VRP methods, which is crucial yet underexplored in the literature. Given a neural VRP method, we adversarially train multiple models in a collaborative manner to synergistically promote robustness against attacks, while boosting standard generalization on clean instances. A neural router is designed to adeptly distribute training instances among models, enhancing overall load balancing and collaborative efficacy. Extensive experiments verify the effectiveness and versatility of CNF in defending against various attacks across different neural VRP methods. Notably, our approach also achieves impressive out-of-distribution generalization on benchmark instances. Jianan Zhou 0002, Yaoxin Wu, Zhiguang Cao, Wen Song 0004, Jie Zhang 0002, Zhiqi Shen 0001 |
NeurIPS | 4 |
| 2024 | Learning to Handle Complex Constraints for Vehicle Routing ProblemsabstractVehicle 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-hard. In this paper, we propose a novel Proactive Infeasibility Prevention (PIP) framework to advance the capabilities of neural methods towards more complex VRPs. Our PIP integrates the Lagrangian multiplier as a basis to enhance constraint awareness and introduces preventative infeasibility masking to proactively steer the solution construction process. Moreover, we present PIP-D, which employs an auxiliary decoder and two adaptive strategies to learn and predict these tailored masks, potentially enhancing performance while significantly reducing computational costs during training. To verify our PIP designs, we conduct extensive experiments on the highly challenging Traveling Salesman Problem with Time Window (TSPTW), and TSP with Draft Limit (TSPDL) variants under different constraint hardness levels. Notably, our PIP is generic to boost many neural methods, and exhibits both a significant reduction in infeasible rate and a substantial improvement in solution quality. Jieyi Bi, Yining Ma 0001, Jianan Zhou 0002, Wen Song 0004, Zhiguang Cao, Yaoxin Wu, Jie Zhang 0002 |
NeurIPS | 4 |
| 2024 | Learning Topological Representations with Bidirectional Graph Attention Network for Solving Job Shop Scheduling ProblemabstractExisting 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 attention network (TBGAT), a novel GNN architecture based on the attention mechanism, to embed the DG for solving JSSP in a local search framework. Specifically, TBGAT embeds the DG from a forward and a backward view, respectively, where the messages are propagated by following the different topologies of the views and aggregated via graph attention. Then, we propose a novel operator based on the message-passing mechanism to calculate the forward and backward topological sorts of the DG, which are the features for characterizing the topological structures and exploited by our model. In addition, we theoretically and experimentally show that TBGAT has linear computational complexity to the number of jobs and machines, respectively, strengthening our method’s practical value. Besides, extensive experiments on five synthetic datasets and seven classic benchmarks show that TBGAT achieves new SOTA results by outperforming a wide range of neural methods by a large margin. All the code and data are publicly available online at https://github.com/zcaicaros/TBGAT. Zhiguang Cao, Yaoxin Wu, Wen Song 0004 |
UAI | 4 |
| 2024 | Stochastic Economic Lot Scheduling via Self-Attention Based Deep Reinforcement LearningabstractThe Stochastic Economic Lot Scheduling Problem (SELSP) is a difficult dynamic optimization problem with wide industrial applications. Traditional methods such as hyper-heuristics are manually designed based on substantial expert knowledge, which may limit their optimization performance. Recently, Deep Reinforcement Learning (DRL) is shown to be promising in automatically learning scheduling policies for SELSP. However, its performance is still quite far from that of hyper-heuristics, due to the lack of suitable deep models. In this paper, we propose a novel DRL method to learn dynamic scheduling policies for SELSP in an end-to-end fashion. Based on self-attention, our method can effectively extract useful features from raw state information, and is flexible in handling different numbers of products, which is not viable for previous methods. Experiments on a complex biopharmaceutical manufacturing process show that our method outperforms a recent DRL method and state-of-the-art hyper-heuristics. Moreover, the trained policy performs better in environments different from training with demand forecast errors and varying number of products, showing its strong robustness and generalization ability.Note to Practitioners—The Stochastic Economic Lot Scheduling Problem (SELSP) is an important problem for manufacturing enterprises, which is to optimally balance the production and inventory so as to minimize the total cost. However, SELSP is very challenging to solve due to the involvement of uncertain factors such as customer demands and machine failures. Traditional methods for solving SELSP, such as heuristic policies and hyper-heuristics, heavily rely on human experiences to design and hence the performance could be limited. This paper proposes a Deep Reinforcement Learning (DRL) based method to automatically learn scheduling policy for solving SELSP, which could alleviate the above limitation through a self-attention based feature extraction mechanism and reward based training. Experimental results on a realistic manufacturing process show that our method can deliver higher revenue than conventional manual policy and an existing DRL based method. Wen Song 0004, Nan Mi, Qiqiang Li, Jing Zhuang, Zhiguang Cao |
IEEE Trans Autom. Sci. Eng. | 1 |
| 2024 | Learning Feature Embedding Refiner for Solving Vehicle Routing ProblemsabstractWhile the encoder-decoder structure is widely used in the recent neural construction methods for learning to solve vehicle routing problems (VRPs), they are less effective in searching solutions due to deterministic feature embeddings and deterministic probability distributions. In this article, we propose the feature embedding refiner (FER) with a novel and generic encoder-refiner-decoder structure to boost the existing encoder-decoder structured deep models. It is model-agnostic that the encoder and the decoder can be from any pretrained neural construction method. Regarding the introduced refiner network, we design its architecture by combining the standard gated recurrent units (GRU) cell with two new layers, i.e., an accumulated graph attention (AGA) layer and a gated nonlinear (GNL) layer. The former extracts dynamic graph topological information of historical solutions stored in a diversified solution pool to generate aggregated pool embeddings that are further improved by the GRU, and the latter adaptively refines the feature embeddings from the encoder with the guidance of the improved pool embeddings. To this end, our FER allows current neural construction methods to not only iteratively refine the feature embeddings for boarder search range but also dynamically update the probability distributions for more diverse search. We apply FER to two prevailing neural construction methods including attention model (AM) and policy optimization with multiple optima (POMO) to solve the traveling salesman problem (TSP) and the capacitated VRP (CVRP). Experimental results show that our method achieves lower gaps and better generalization than the original ones and also exhibits competitive performance to the state-of-the-art neural improvement methods. Yining Ma 0001, Zhiguang Cao, Yaoxin Wu, Wen Song 0004, Jie Zhang 0002, Yeow Meng Chee |
IEEE Trans. Neural Networks Learn. Syst. | 5 |
| 2023 | Towards Omni-generalizable Neural Methods for Vehicle Routing ProblemsabstractLearning 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 performance. This paper studies a challenging yet realistic setting, which considers generalization across both size and distribution in VRPs. We propose a generic meta-learning framework, which enables effective training of an initialized model with the capability of fast adaptation to new tasks during inference. We further develop a simple yet efficient approximation method to reduce the training overhead. Extensive experiments on both synthetic and benchmark instances of the traveling salesman problem (TSP) and capacitated vehicle routing problem (CVRP) demonstrate the effectiveness of our method. The code is available at: https://github.com/RoyalSkye/Omni-VRP. Jianan Zhou 0002, Yaoxin Wu, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
ICML | 3 |
| 2023 | Dynamic Job Shop Scheduling via Deep Reinforcement LearningabstractRecently, deep reinforcement learning (DRL) is shown to be promising in learning dispatching rules end-to-end for complex scheduling problems. However, most research is limited to deterministic problems. In this paper, we focus on the dynamic job-shop scheduling problem (DJSP), which is a complex dynamic optimization problem under uncertainty. We propose a DRL based method to learn dispatching policies for DJSP. Unlike existing DRL based dynamic scheduling methods that use a fixed number of dispatching rules as actions, our decision-making framework directly selects legitimate jobs, which is able to break the limitations imposed by priority dispatching rules. We design two training methods, including a gradient based algorithm with dense rewards, and an evolutionary strategy with sparse rewards. Extensive experiments show that our DRL method can learn high-quality DJSP dispatching policies, and can significantly outperform a state-of-the-art Genetic Programming (GP) based dispatching rule learning method. Xinjie Liang, Wen Song 0004, Pengfei Wei 0001 |
ICTAI | 2 |
| 2023 | Ensemble-based Deep Reinforcement Learning for Vehicle Routing Problems under Distribution ShiftabstractWhile 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 learning method for VRPs, which learns a group of diverse sub-policies to cope with various instance distributions. In particular, to prevent convergence of the parameters to the same one, we enforce diversity across sub-policies by leveraging Bootstrap with random initialization. Moreover, we also explicitly pursue inequality between sub-policies by exploiting regularization terms during training to further enhance diversity. Experimental results show that our method is able to outperform the state-of-the-art neural baselines on randomly generated instances of various distributions, and also generalizes favourably on the benchmark instances from TSPLib and CVRPLib, which confirmed the effectiveness of the whole method and the respective designs. Yuan Jiang 0007, Zhiguang Cao, Yaoxin Wu, Wen Song 0004, Jie Zhang 0002 |
NeurIPS | 4 |
| 2023 | Container stacking optimization based on Deep Reinforcement Learning
Zhentang Duan, Wen Song 0004, Qiqiang Li |
Eng. Appl. Artif. Intell. | 3 |
| 2023 | Instance-specific algorithm configuration via unsupervised deep graph clusteringabstractInstance-specific Algorithm Configuration (AC) methods are effective in automatically generating high-quality algorithm parameters for heterogeneous NP-hard problems from multiple sources. However, existing works rely on manually designed features to describe training instances, which are simple numerical attributes and cannot fully capture structural differences. Targeting at Mixed-Integer Programming (MIP) solvers, this paper proposes a novel instances-specific AC method based on end-to-end deep graph clustering. By representing an MIP instance as a bipartite graph, a random walk algorithm is designed to extract raw features with both numerical and structural information from the instance graph. Then an auto-encoder is designed to learn dense instance embeddings unsupervisedly, which facilitates clustering heterogeneous instances into homogeneous clusters for training instance-specific configurations. Experimental results on multiple benchmarks show that the proposed method can improve the solving efficiency of CPLEX on highly heterogeneous instances, and outperform existing instance specific AC methods. Wen Song 0004, Yi Liu 0015, Zhiguang Cao, Yaoxin Wu, Qiqiang Li |
Eng. Appl. Artif. Intell. | 1 |
| 2023 | Flexible Job-Shop Scheduling via Graph Neural Network and Deep Reinforcement LearningabstractRecently, deep reinforcement learning (DRL) has been applied to learn priority dispatching rules (PDRs) for solving complex scheduling problems. However, the existing works face challenges in dealing with flexibility, which allows an operation to be scheduled on one out of multiple machines and is often required in practice. Such one-to-many relationship brings additional complexity in both decision making and state representation. This article considers the well-known flexible job-shop scheduling problem and addresses these issues by proposing a novel DRL method to learn high-quality PDRs end to end. The operation selection and the machine assignment are combined as a composite decision. Moreover, based on a novel heterogeneous graph representation of scheduling states, a heterogeneous-graph-neural-network-based architecture is proposed to capture complex relationships among operations and machines. Experiments show that the proposed method outperforms traditional PDRs and is computationally efficient, even on instances of larger scales and different properties unseen in training. Wen Song 0004, Qiqiang Li, Zhiguang Cao |
IEEE Trans. Ind. Informatics | 1 |
| 2023 | Learning to Solve Multiple-TSP With Time Window and Rejections via Deep Reinforcement LearningabstractWe propose a manager-worker framework (the implementation of our model is publically available at:https://github.com/zcaicaros/manager-worker-mtsptwr) based on deep reinforcement learning to tackle a hard yet nontrivial variant of Travelling Salesman Problem (TSP), i.e., multiple-vehicle TSP with time window and rejections (mTSPTWR), where customers who cannot be served before the deadline are subject to rejections. Particularly, in the proposed framework, a manager agent learns to divide mTSPTWR into sub-routing tasks by assigning customers to each vehicle via a Graph Isomorphism Network (GIN) based policy network. A worker agent learns to solve sub-routing tasks by minimizing the cost in terms of both tour length and rejection rate for each vehicle, the maximum of which is then fed back to the manager agent to learn better assignments. Experimental results demonstrate that the proposed framework outperforms strong baselines in terms of higher solution quality and shorter computation time. More importantly, the trained agents also achieve competitive performance for solving unseen larger instances. Rongkai Zhang 0001, Zhiguang Cao, Wen Song 0004, Puay Siew Tan, Jie Zhang 0002, Bihan Wen, Justin Dauwels |
IEEE Trans. Intell. Transp. Syst. | 4 |
| 2023 | Learning Large Neighborhood Search for Vehicle Routing in Airport Ground HandlingabstractDispatching vehicle fleets to serve flights is a key task in airport ground handling (AGH). Due to the notable growth of flights, it is challenging to simultaneously schedule multiple types of operations (services) for a large number of flights, where each type of operation is performed by one specific vehicle fleet. To tackle this issue, we first represent the operation scheduling as a complex vehicle routing problem and formulate it as a mixed integer linear programming (MILP) model. Then given the graph representation of the MILP model, we propose a learning assisted large neighborhood search (LNS) method using data generated based on real scenarios, where we integrate imitation learning and graph convolutional network (GCN) to learn a destroy operator to automatically select variables, and employ an off-the-shelf solver as the repair operator to reoptimize the selected variables. Experimental results based on a real airport show that the proposed method allows for handling up to 200 flights with 10 types of operations simultaneously, and outperforms state-of-the-art methods. Moreover, the learned method performs consistently accompanying different solvers, and generalizes well on larger instances, verifying the versatility and scalability of our method. Jianan Zhou 0002, Yaoxin Wu, Zhiguang Cao, Wen Song 0004, Jie Zhang 0002, Zhenghua Chen |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2022 | Dynamic Transfer Gaussian Process RegressionabstractIn this paper, we work on a challenging dynamic transfer regression problem where domains come in a streaming manner. At each time stage, a new domain emerges and is taken as the target domain while all the domains in previous time stages are taken as source domains. We propose a transfer Gaussian process model GPdk with a novel dynamic transfer kernel DyTK to handle the dynamic transfer regression problem. Specifically, DyTK is with a sequential form to fit the domain stream. To adaptively control the knowledge transfer strength, DyTK is designed to be capable of modeling the inter-domain relatedness of every inter-domain pair. A theorem that ensures DyTK to be positive semi-definite is then proposed. We also theoretically analyze the transfer performance of GPdk by deriving its generalization error bounds. The error bounds further motivate us to propose a parameter reuse strategy to alleviate the scalability issue of GPdk along time. Extensive experiments on both synthetic and real-world datasets show the effectiveness of GPdk in handling dynamic transfer regression problems. Pengfei Wei 0001, Xinghua Qu, Wen Song 0004, Zejun Ma 0001 |
CIKM | 3 |
| 2022 | Learning Scenario Representation for Solving Two-stage Stochastic Integer Programs
Yaoxin Wu, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
ICLR | 2 |
| 2022 | Efficient Neural Neighborhood Search for Pickup and Delivery ProblemsabstractWe 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 decoders that automatically learn to perform removal and reinsertion of a pickup-delivery node pair to tackle the precedence constraint. Additionally, a diversity enhancement scheme is leveraged to further ameliorate the performance. Our N2S is generic, and extensive experiments on two canonical PDP variants show that it can produce state-of-the-art results among existing neural methods. Moreover, it even outstrips the well-known LKH3 solver on the more constrained PDP variant. Our implementation for N2S is available online. Yining Ma 0001, Zhiguang Cao, Wen Song 0004, Hongliang Guo 0001, Yue-Jiao Gong, Yeow Meng Chee |
IJCAI | 4 |
| 2022 | Graph Learning Assisted Multi-Objective Integer ProgrammingabstractObjective-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 mitigate the issue, we present a graph neural network (GNN) based method to learn the reduction rule in the ODA. We formulate the algorithmic procedure of generic ODAs as a Markov decision process, and parameterize the policy (reduction rule) with a novel two-stage GNN to fuse information from variables, constraints and especially objectives for better state representation. We train our model with imitation learning and deploy it on a state-of-the-art ODA. Results show that our method significantly improves the solving efficiency of the ODA. The learned policy generalizes fairly well to larger problems or more objectives, and the proposed GNN outperforms existing ones for integer programming in terms of test and generalization accuracy. Yaoxin Wu, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002, Mingyan Lin |
NeurIPS | 2 |
| 2022 | Learning variable ordering heuristics for solving Constraint Satisfaction Problems
Wen Song 0004, Zhiguang Cao, Jie Zhang 0002, Andrew Lim 0001 |
Eng. Appl. Artif. Intell. | 1 |
| 2022 | A novel concavity based method for automatic segmentation of touching cells in microfluidic chips
Qiqiang Li, Wen Song 0004, Pengfei Wei 0001, Jing Guo 0007 |
Expert Syst. Appl. | 3 |
| 2022 | Multi-objective evolutionary algorithm based on RBF network for solving the stochastic vehicle routing problem
Yunyun Niu, Wen Song 0004, Zhiguang Cao |
Inf. Sci. | 4 |
| 2022 | Deep Reinforcement Learning for Solving the Heterogeneous Capacitated Vehicle Routing ProblemabstractExisting deep reinforcement learning (DRL)-based methods for solving the capacitated vehicle routing problem (CVRP) intrinsically cope with a homogeneous vehicle fleet, in which the fleet is assumed as repetitions of a single vehicle. Hence, their key to construct a solution solely lies in the selection of the next node (customer) to visit excluding the selection of vehicle. However, vehicles in real-world scenarios are likely to be heterogeneous with different characteristics that affect their capacity (or travel speed), rendering existing DRL methods less effective. In this article, we tackle heterogeneous CVRP (HCVRP), where vehicles are mainly characterized by different capacities. We consider both min-max and min-sum objectives for HCVRP, which aim to minimize the longest or total travel time of the vehicle(s) in the fleet. To solve those problems, we propose a DRL method based on the attention mechanism with a vehicle selection decoder accounting for the heterogeneous fleet constraint and a node selection decoder accounting for the route construction, which learns to construct a solution by automatically selecting both a vehicle and a node for this vehicle at each step. Experimental results based on randomly generated instances show that, with desirable generalization to various problem sizes, our method outperforms the state-of-the-art DRL method and most of the conventional heuristics, and also delivers competitive performance against the state-of-the-art heuristic method, that is, slack induction by string removal. In addition, the results of extended experiments demonstrate that our method is also able to solve CVRPLib instances with satisfactory performance. Yining Ma 0001, Zhiguang Cao, Andrew Lim 0001, Wen Song 0004, Jie Zhang 0002 |
IEEE Trans. Cybern. | 6 |
| 2022 | Stochastic Cooperative Bidding Strategy for Multiple Microgrids With Peer-to-Peer Energy TradingabstractIt is a significant and challenging problem to coordinate multiple microgrids (MMGs) belonging to different entities and achieve their excellent energy-sharing performance to ensure the stability of electricity markets. This article studies a grid-oriented energy bidding problem for MMGs with peer-to-peer (P2P) energy trading under uncertainty. A stochastic Cartel game (SCG) based strategy is developed. A stochastic Cartel nonlinear programming model is formulated to characterize the joint energy bidding, the energy production, and P2P energy transactions while minimizing the total cost for MMGs under uncertainty. A diagonal quadratic approximation method is employed to linearize quadratic terms, and the SCG problem for MMGs is further decomposed into subproblems for individual MGs based on a surrogate Lagrangian relaxation method. The equivalence of problem transformation is proved and equilibrium solutions are derived in an iterative and distributed manner. Comparisons for different strategies, models, and solution algorithms are conducted to testify the rationality and validity of the proposed strategy. Luhao Wang, Wen Song 0004, Qiqiang Li |
IEEE Trans. Ind. Informatics | 3 |
| 2022 | Heterogeneous Attentions for Solving Pickup and Delivery Problem via Deep Reinforcement LearningabstractRecently, there is an emerging trend to apply deep reinforcement learning to solve the vehicle routing problem (VRP), where a learnt policy governs the selection of next node for visiting. However, existing methods could not handle well the pairing and precedence relationships in the pickup and delivery problem (PDP), which is a representative variant of VRP. To address this challenging issue, we leverage a novel neural network integrated with a heterogeneous attention mechanism to empower the policy in deep reinforcement learning to automatically select the nodes. In particular, the heterogeneous attention mechanism specifically prescribes attentions for each role of the nodes while taking into account the precedence constraint, i.e., the pickup node must precede the pairing delivery node. Further integrated with a masking scheme, the learnt policy is expected to find higher-quality solutions for solving PDP. Extensive experimental results show that our method outperforms the state-of-the-art heuristic and deep learning model, respectively, and generalizes well to different distributions and problem sizes. Liang Xin, Zhiguang Cao, Andrew Lim 0001, Wen Song 0004, Jie Zhang 0002 |
IEEE Trans. Intell. Transp. Syst. | 5 |
| 2022 | Learning Improvement Heuristics for Solving Routing ProblemsabstractRecent studies in using deep learning (DL) to solve routing problems focus on construction heuristics, whose solutions are still far from optimality. Improvement heuristics have great potential to narrow this gap by iteratively refining a solution. However, classic improvement heuristics are all guided by handcrafted rules that may limit their performance. In this article, we propose a deep reinforcement learning framework to learn the improvement heuristics for routing problems. We design a self-attention-based deep architecture as the policy network to guide the selection of the next solution. We apply our method to two important routing problems, i.e., the traveling salesman problem (TSP) and the capacitated vehicle routing problem (CVRP). Experiments show that our method outperforms state-of-the-art DL-based approaches. The learned policies are more effective than the traditional handcrafted ones and can be further enhanced by simple diversifying strategies. Moreover, the policies generalize well to different problem sizes, initial solutions, and even real-world data set. Yaoxin Wu, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002, Andrew Lim 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 2 |
| 2021 | Multi-Decoder Attention Model with Embedding Glimpse for Solving Vehicle Routing ProblemsabstractWe 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 existing methods that train only one policy. A customized beam search strategy is designed to fully exploit the diversity of MDAM. In addition, we propose an Embedding Glimpse layer in MDAM based on the recursive nature of construction, which can improve the quality of each policy by providing more informative embeddings. Extensive experiments on six different routing problems show that our method significantly outperforms the state-of-the-art deep learning based models. Liang Xin, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
AAAI | 2 |
| 2021 | Learning to Iteratively Solve Routing Problems with Dual-Aspect Collaborative TransformerabstractRecently, 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 Dual-Aspect Collaborative Transformer (DACT) to learn embeddings for the node and positional features separately, instead of fusing them together as done in existing ones, so as to avoid potential noises and incompatible correlations. Moreover, the positional features are embedded through a novel cyclic positional encoding (CPE) method to allow Transformer to effectively capture the circularity and symmetry of VRP solutions (i.e., cyclic sequences). We train DACT using Proximal Policy Optimization and design a curriculum learning strategy for better sample efficiency. We apply DACT to solve the traveling salesman problem (TSP) and capacitated vehicle routing problem (CVRP). Results show that our DACT outperforms existing Transformer based improvement models, and exhibits much better generalization performance across different problem sizes on synthetic and benchmark instances, respectively. Yining Ma 0001, Zhiguang Cao, Wen Song 0004, Le Zhang 0001, Zhenghua Chen, Jing Tang 0004 |
NeurIPS | 4 |
| 2021 | Learning Large Neighborhood Search Policy for Integer ProgrammingabstractWe 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 combinatorial number of variable subsets prevents direct application of typical RL algorithms. To tackle this challenge, we represent all subsets by factorizing them into binary decisions on each variable. We then design a neural network to learn policies for each variable in parallel, trained by a customized actor-critic algorithm. We evaluate the proposed method on four representative IP problems. Results show that it can find better solutions than SCIP in much less time, and significantly outperform other LNS baselines with the same runtime. Moreover, these advantages notably persist when the policies generalize to larger problems. Further experiments with Gurobi also reveal that our method can outperform this state-of-the-art commercial solver within the same time limit. Yaoxin Wu, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
NeurIPS | 2 |
| 2021 | NeuroLKH: Combining Deep Learning Model with Lin-Kernighan-Helsgaun Heuristic for Solving the Traveling Salesman ProblemabstractWe 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 node penalties, both of which are critical for improving the performance of LKH. Based on the output of SGN, NeuroLKH creates the edge candidate set and transforms edge distances to guide the searching process of LKH. Extensive experiments firmly demonstrate that, by training one model on a wide range of problem sizes, NeuroLKH significantly outperforms LKH and generalizes well to much larger sizes. Also, we show that NeuroLKH can be applied to other routing problems such as Capacitated Vehicle Routing Problem (CVRP), Pickup and Delivery Problem (PDP), and CVRP with Time Windows (CVRPTW). Liang Xin, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
NeurIPS | 2 |
| 2021 | Detecting the shuttlecock for a badminton robot: A YOLO based approach
Zhiguang Cao, Tingbo Liao, Wen Song 0004, Zhenghua Chen, Chongshou Li |
Expert Syst. Appl. | 3 |
| 2021 | Step-Wise Deep Learning Models for Solving Routing ProblemsabstractRouting problems are very important in intelligent transportation systems. Recently, a number of deep learning-based methods are proposed to automatically learn construction heuristics for solving routing problems. However, these methods do not completely follow Bellman's Principle of Optimality since the visited nodes during construction are still included in the following subtasks, resulting in suboptimal policies. In this article, we propose a novel step-wise scheme which explicitly removes the visited nodes in each node selection step. We apply this scheme to two representative deep models for routing problems, pointer network and transformer attention model (TAM), and significantly improve the performance of the original models. To reduce computational complexity, we further propose the approximate step-wise TAM model by modifying one layer of attention. It enables training on larger instances compared to step-wise TAM, and outperforms state-of-the-art deep models with greedy decoding strategy. Liang Xin, Wen Song 0004, Zhiguang Cao, Jie Zhang 0002 |
IEEE Trans. Ind. Informatics | 2 |
| 2021 | Improving the Performance of Transportation Networks: A Semi-Centralized Pricing ApproachabstractImproving the performance of transportation network is a crucial task in traffic management. In this paper, we start with a cooperative routing problem, which aims to minimize the chance of road network breakdown. To address this problem, we propose a subgradient method, which can be naturally implemented as a semi-centralized pricing approach. Particularly, each road link adopts the pricing scheme to calculate and adjust the local toll regularly, while the vehicles update their routes to minimize the toll costs by exploiting the global toll information. To prevent the potential oscillation brought by the subgradient method, we introduce a heavy-ball method to further improve the performance of the pricing approach. We then test both the basic and improved pricing approaches in a real road network, and simultaneously compare them with several baselines. The experimental results demonstrate that, our approaches significantly outperform others, by comprehensively evaluating them in terms of various metrics including average travel time and travel distance, winners and losers, potential congestion occurrence, last arrival time, toll costs and average traffic flows, with two different O-D profiles. Zhiguang Cao, Hongliang Guo 0001, Wen Song 0004, Kai-Zhou Gao, Liujiang Kang, Xuexi Zhang, Qilun Wu |
IEEE Trans. Intell. Transp. Syst. | 3 |
| 2020 | Learning to Dispatch for Job Shop Scheduling via Deep Reinforcement LearningabstractPriority 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 PDRs via an end-to-end deep reinforcement learning agent. We exploit the disjunctive graph representation of JSSP, and propose a Graph Neural Network based scheme to embed the states encountered during solving. The resulting policy network is size-agnostic, effectively enabling generalization on large-scale instances. Experiments show that the agent can learn high-quality PDRs from scratch with elementary raw features, and demonstrates strong performance against the best existing PDRs. The learned policies also perform well on much larger instances that are unseen in training. Wen Song 0004, Zhiguang Cao, Jie Zhang 0002, Puay Siew Tan |
NeurIPS | 2 |
| 2020 | Cost-sensitive deep forest for price prediction
Chao Ma 0006, Zhenbing Liu, Zhiguang Cao, Wen Song 0004, Jie Zhang 0002, Weiliang Zeng |
Pattern Recognit. | 4 |
| 2019 | A Neural Model for Method Name Generation from Functional DescriptionabstractThe names of software artifacts, e.g., method names, are important for software understanding and maintenance, as good names can help developers easily understand others’ code. However, the existing naming guidelines are difficult for developers, especially novices, to come up with meaningful, concise and compact names for the variables, methods, classes and files. With the popularity of open source, an enormous amount of project source code can be accessed, and the exhaustiveness and instability of manually naming methods could now be relieved by automatically learning a naming model from a large code repository. Nevertheless, building a comprehensive naming system is still challenging, due to the gap between natural language functional descriptions and method names. Specifically, there are three challenges: how to model the relationship between the functional descriptions and formal method names, how to handle the explosion of vocabulary when dealing with large repositories, and how to leverage the knowledge learned from large repositories to a specific project. To answer these questions, we propose a neural network to directly generate readable method names from natural language description. The proposed method is built upon the encoder-decoder framework with the attention and copying mechanisms. Our experiments show that our method can generate meaningful and accurate method names and achieve significant improvement over the state-of-the-art baseline models. We also address the cold-start problem using a training trick to utilize big data in Github for specific projects. Sa Gao, Chunyang Chen 0001, Zhenchang Xing, Wen Song 0004, Shangwei Lin 0001 |
SANER | 5 |
| 2019 | A Sampling Approach for Proactive Project Scheduling under Generalized Time-dependent Workability UncertaintyabstractIn real-world project scheduling applications, activity durations are often uncertain. Proactive scheduling can effectively cope with the duration uncertainties, by generating robust baseline solutions according to a priori stochastic knowledge. However, most of the existing proactive approaches assume that the duration uncertainty of an activity is not related to its scheduled start time, which may not hold in many real-world scenarios. In this paper, we relax this assumption by allowing the duration uncertainty to be time-dependent, which is caused by the uncertainty of whether the activity can be executed on each time slot. We propose a stochastic optimization model to find an optimal Partial-order Schedule (POS) that minimizes the expected makespan. This model can cover both the time-dependent uncertainty studied in this paper and the traditional time-independent duration uncertainty. To circumvent the underlying complexity in evaluating a given solution, we approximate the stochastic optimization model based on Sample Average Approximation (SAA). Finally, we design two efficient branch-and-bound algorithms to solve the NP-hard SAA problem. Empirical evaluation confirms that our approach can generate high-quality proactive solutions for a variety of uncertainty distributions. Wen Song 0004, Jie Zhang 0002, Zhiguang Cao, Hui Xi |
J. Artif. Intell. Res. | 1 |
| 2018 | Risk-Aware Proactive Scheduling via Conditional Value-at-RiskabstractIn this paper, we consider the challenging problem of riskaware proactive scheduling with the objective of minimizing robust makespan. State-of-the-art approaches based on probabilistic constrained optimization lead to Mixed Integer Linear Programs that must be heuristically approximated. We optimize the robust makespan via a coherent risk measure, Conditional Value-at-Risk (CVaR). Since traditional CVaR optimization approaches assuming linear spaces does not suit our problem, we propose a general branch-and-bound framework for combinatorial CVaR minimization. We then design an approximate complete algorithm, and employ resource reasoning to enable constraint propagation for multiple samples. Empirical results show that our algorithm outperforms state-of-the-art approaches with higher solution quality. Wen Song 0004, Jie Zhang 0002, Hui Xi |
AAAI | 1 |
| 2017 | A Sampling Based Approach for Proactive Project Scheduling with Time-Dependent Duration UncertaintyabstractMost of the existing proactive scheduling approaches assume the durations of activities can be described by independent random variables that have no relation with time. We deal with the more challenging problem where the duration uncertainty is related to the scheduled time period. We propose a sampling based approach by extending the Consensus method from stochastic optimization. Experimental results show the effectiveness of our approach in solution quality and stability. Wen Song 0004, Jie Zhang 0002, Hui Xi |
AAAI | 1 |
| 2017 | A multi-unit combinatorial auction based approach for decentralized multi-project scheduling
Wen Song 0004, Jie Zhang 0002, Hui Xi |
Auton. Agents Multi Agent Syst. | 1 |