Imitation Learning for Combinatorial Optimisation under Uncertainty
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Imitation learning (IL) provides a data-driven framework for approximating policies for large-scale combinatorial optimisation problems formulated as sequential decision problems (SDPs), where exact solution methods are computationally intractable. A central but underexplored aspect of IL in this context is the role of the expert that generates training demonstrations. Existing studies employ a wide range of expert constructions, yet lack a unifying framework to characterise their modelling assumptions, computational properties, and impact on learning performance. This paper introduces a systematic taxonomy of experts for imitation learning in combinatorial optimisation under uncertainty. The literature is classified along three principal dimensions: (i) treatment of uncertainty; (ii) level of optimality, distinguishing task-optimal and approximate experts; and (iii) interaction mode with the learner, ranging from one-shot supervision to iterative, interactive schemes. We further identify additional categories capturing other relevant expert characteristics. Building on this taxonomy, we propose a generalised Dataset Aggregation (DAgger) framework that accommodates multiple expert queries, expert aggregation, and flexible interaction strategies. The proposed framework is evaluated on a dynamic physician-to-patient assignment problem with stochastic arrivals and capacity constraints. Computational experiments compare learning outcomes across expert types and interaction regimes. The results show that policies learned from stochastic experts consistently outperform those learned from deterministic or full-information experts, while interactive learning improves solution quality using fewer expert demonstrations. Aggregated deterministic experts provide an effective alternative when stochastic optimisation becomes computationally challenging.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. Builds on 61 of its 122 references that are in the map. First posted on arXiv 8 Jan 2026.
Builds on
61 of 122 references.
Closest ancestors · 8
Same direction, newest first
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challenges5 Sep
- 2023Selective Sampling and Imitation Learning via Online Regression11 Jul
- 2023MEGA-DAgger: Imitation Learning with Multiple Imperfect Experts1 Mar
- 2022Discriminator-Weighted Offline Imitation Learning from Suboptimal Demonstrations20 Jul
- 2022Learning To Cut By Looking Ahead: Cutting Plane Selection via Imitation Learning27 Jun
- 2021Imitation Learning: Progress, Taxonomies and Challenges23 Jun
- 2019Imitation Learning from Imperfect Demonstration27 Jan
- 2018EnsembleDAgger: A Bayesian Approach to Safe Imitation Learning22 Jul
Other references · 53
- 2015Pointer Networkscited by 895
- 2018Machine Learning for Combinatorial Optimization: a Methodological Tour d'Horizoncited by 623
- 2017Smart "Predict, then Optimize"cited by 339
- 2019An Efficient Graph Convolutional Network Technique for the Travelling Salesman Problemcited by 269
- 2018Learning a SAT Solver from Single-Bit Supervisioncited by 200
- 2018Combinatorial Optimization with Graph Convolutional Networks and Guided Tree Searchcited by 183
- 2020Solving Mixed Integer Programs Using Neural Networkscited by 182
- 2021End-to-End Constrained Optimization Learning: A Surveycited by 157
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 130
- 2016Query-Efficient Imitation Learning for End-to-End Autonomous Drivingcited by 84
- 2023A Survey of Contextual Optimization Methods for Decision Making under Uncertaintycited by 82
- 2018Learning to Branchcited by 78
38 more
- 2020Parameterizing Branch-and-Bound Search Trees to Learn Branching Policiescited by 63
- 2021LazyDAgger: Reducing Context Switching in Interactive Imitation Learningcited by 51
- 2022MIP-GNN: A Data-Driven Framework for Guiding Combinatorial Solverscited by 49
- 2021Learning to Select Cuts for Efficient Mixed-Integer Programmingcited by 48
- 2019Neural Network Branching for Neural Network Verificationcited by 45
- 2023A GNN-Guided Predict-and-Search Framework for Mixed-Integer Linear Programmingcited by 41
- 2019Learning to Solve Large-Scale Security-Constrained Unit Commitment Problemscited by 40
- 2023Searching Large Neighborhoods for Integer Linear Programs with Contrastive Learningcited by 37
- 2022Learning to Compare Nodes in Branch and Bound with Graph Neural Networkscited by 31
- 2022Lookback for Learning to Branchcited by 27
- 2022Hindsight Learning for MDPs with Exogenous Inputscited by 26
- 2024Machine Learning Augmented Branch and Bound for Mixed Integer Linear Programmingcited by 24
- 2018Learning the Multiple Traveling Salesmen Problem with Permutation Invariant Pooling Networkscited by 22
- 2023Learning to Solve Constraint Satisfaction Problems with Recurrent Transformercited by 21
- 2023Machine Learning for Cutting Planes in Integer Programming: A Surveycited by 20
- 2017Deep Learning Assisted Heuristic Tree Search for the Container Pre-marshalling Problemcited by 18
- 2020Generalization of Machine Learning for Problem Reduction: A Case Study on Travelling Salesman Problemscited by 17
- 2020Teaching the Old Dog New Tricks: Supervised Learning with Constraintscited by 16
- 2020A Study of Learning Search Approximation in Mixed Integer Branch and Bound: Node Selection in SCIPcited by 16
- 2023Learning-based Online Optimization for Autonomous Mobility-on-Demand Fleet Controlcited by 14
- 2022Machine-learning-based arc selection for constrained shortest path problems in column generationcited by 13
- 2021Enhancing Column Generation by a Machine-Learning-Based Pricing Heuristic for Graph Coloringcited by 12
- 2022Predict+Optimize for Packing and Covering LPs with Unknown Parameters in Constraintscited by 11
- 2021Learning to Sparsify Travelling Salesman Problem Instancescited by 9
- 2019Fine-grained Search Space Classification for Hard Enumeration Variants of Subset Problemscited by 7
- 2023Electricity Price Prediction for Energy Storage System Arbitrage: A Decision-focused Approachcited by 7
- 2024Multi-task Representation Learning for Mixed Integer Linear Programmingcited by 7
- 2025Learning-Guided Rolling Horizon Optimization for Long-Horizon Flexible Job-Shop Schedulingcited by 7
- 2019Learning Multi-Stage Sparsification for Maximum Clique Enumerationcited by 6
- 2025Feasibility-Aware Decision-Focused Learning for Predicting Parameters in the Constraintscited by 6
- 2019Learning to Handle Parameter Perturbations in Combinatorial Optimization: an Application to Facility Locationcited by 5
- 2024Combinatorial Optimization and Machine Learning for Dynamic Inventory Routingcited by 5
- 2021Learning to solve the single machine scheduling problem with release times and sum of completion timescited by 4
- 2023Two-Stage Learning For the Flexible Job Shop Scheduling Problemcited by 4
- 2020Flight-connection Prediction for Airline Crew Scheduling to Construct Initial Clusters for OR Optimizercited by 3
- 2023Optimization-based Learning for Dynamic Load Planning in Trucking Service Networkscited by 3
- 2024Learning Joint Models of Prediction and Optimizationcited by 3
- 2022Machine Learning for K-adaptability in Two-stage Robust Optimizationcited by 2
50 most cited of 53 All 53 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2022Imitation Learning by Estimating Expertise of Demonstrators
- 2025Learning Equilibria from Data: Provably Efficient Multi-Agent Imitation Learning
- 2023A Strong Baseline for Batch Imitation Learning
- 2022A Ranking Game for Imitation Learning
- 2026Split the Differences, Pool the Rest: Provably Efficient Multi-Objective Imitation
- 2024Self-Improvement for Neural Combinatorial Optimization: Sample without Replacement, but Improvement
- 2025TubeDAgger: Reducing the Number of Expert Interventions with Stochastic Reach-Tubes
- 2020Provable Representation Learning for Imitation Learning via Bi-level Optimization