mlsanityData to 7 Oct 2026

Rethinking Efficiency in Neural Combinatorial Optimization: Batched Preference Optimization with Mamba

arXiv 2602.20730 ↗First posted 24 Feb 2026cs.LGcited by 0

On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods

also in 1 more
Abstract (arXiv)

We study efficiency as a first-class objective in Neural Combinatorial Optimization (NCO) and present ECO, an efficient learning framework that combines batched preference optimization with a Mamba backbone. Instead of tightly interleaving every policy update with on-policy rollouts, ECO decouples trajectory generation from gradient updates through two stages: supervised warm-up on pre-computed solutions and iterative Direct Preference Optimization (DPO) on batched candidate sets generated by the current policy. We pair this learning pipeline with a mixed Mamba encoder-decoder that reduces memory growth on long sequences and improves hardware utilization. A local-search-guided bootstrapping strategy is further used during training to widen preference margins and stabilize iterative improvement. Importantly, local search is only used to construct stronger preference pairs during training and is never invoked at inference time. On TSP and CVRP, ECO achieves the strongest overall performance among the compared neural baselines while also delivering clear advantages in memory usage and throughput. We provide additional analysis on memory scaling, throughput, and the contribution of each design component.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 27 of its 42 references that are in the map. First posted on arXiv 24 Feb 2026.

Builds on

27 of 42 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025BOPO: Neural Combinatorial Optimization via Best-anchored and Objective-guided Preference Optimization10 Mar

Other references · 26

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  4. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  5. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
  7. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
  8. 2015Pointer Networkscited by 882
  9. 2024Self-Rewarding Language Modelscited by 654
  10. 2018Attention, Learn to Solve Routing Problems!cited by 646
  11. 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 607
  12. 2016Neural Combinatorial Optimization with Reinforcement Learningcited by 600
14 more
  1. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
  2. 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 571
  3. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
  4. 2023Reinforced Self-Training (ReST) for Language Modelingcited by 486
  5. 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
  6. 2020POMO: Policy Optimization with Multiple Optima for Reinforcement Learningcited by 341
  7. 2023Neural Combinatorial Optimization with Heavy Decoder: Toward Large Scale Generalizationcited by 140
  8. 2023GLOP: Learning Global Partition and Local Construction for Solving Large-scale Routing Problems in Real-timecited by 82
  9. 2024Self-Improved Learning for Scalable Neural Combinatorial Optimizationcited by 29
  10. 2025Preference Optimization for Combinatorial Optimization Problemscited by 11
  11. 2024Collaboration! Towards Robust Neural Methods for Routing Problemscited by 9
  12. 2025Hybrid-Balance GFlowNet for Solving Vehicle Routing Problemscited by 7
  13. 2025Fine-tuning Large Language Model for Automated Algorithm Designcited by 6
  14. 2025Efficient Neural Combinatorial Optimization Solver for the Min-max Heterogeneous Capacitated Vehicle Routing Problemcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026SSPO: Structure-Aware Similarity-Weighted Preference Optimization for Neural Combinatorial Optimization
  2. 2026AutoPref: Automatic Discovery of Task-Specific Preference Objectives for Neural Combinatorial Optimization
  3. 2024Self-Improvement for Neural Combinatorial Optimization: Sample without Replacement, but Improvement
  4. 2026HyCO: A Hybrid Neural Solver for Combinatorial Optimization
  5. 2026Baseline-Free Policy Optimization for Neural Combinatorial Optimization
  6. 2026WeCon: An Efficient Weight-Conditioned Neural Solver for Multi-Objective Combinatorial Optimization Problems
  7. 2025FrontierCO: Real-World and Large-Scale Evaluation of Machine Learning Solvers for Combinatorial Optimization
  8. 2025Value-Free Policy Optimization via Reward Partitioning