Symmetric Replay Training: Enhancing Sample Efficiency in Deep Reinforcement Learning for Combinatorial Optimization
Hyeonah Kim, Minsu Kim, Sungsoo Ahn, Jinkyoo Park
Abstract
Deep reinforcement learning (DRL) has significantly advanced the field of combinatorial optimization (CO). However, its practicality is hindered by the necessity for a large number of reward evaluations, especially in scenarios involving computationally intensive function assessments. To enhance the sample efficiency, we propose a simple but effective method, called symmetric replay training (SRT), which can be easily integrated into various DRL methods. Our method leverages high-reward samples to encourage exploration of the under-explored symmetric regions without additional online interactions - free. Through replay training, the policy is trained to maximize the likelihood of the symmetric trajectories of discovered high-rewarded samples. Experimental results demonstrate the consistent improvement of our method in sample efficiency across diverse DRL methods applied to real-world tasks, such as molecular optimization and hardware design.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 5aff78bb-d56c-46f4-944b-c5694f7ba333Cited by top-tier papers3
- Genetic-guided GFlowNets for Sample Efficient Molecular OptimizationHyeonah Kim, Minsu Kim, Sanghyeok Choi, Jinkyoo ParkNeurIPS 2024 · 42 citations
- PolyNet: Learning Diverse Solution Strategies for Neural Combinatorial OptimizationAndré Hottung, Mridul Mahajan, Kevin TierneyICLR 2025
- BOPO: Neural Combinatorial Optimization via Best-anchored and Objective-guided Preference OptimizationZijun Liao, Jinbiao Chen, Debing Wang, Zizhen Zhang et al.ICML 2025
Builds on37
- Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from PixelsDenis Yarats, Ilya Kostrikov, Rob FergusICLR 2021 · 911 citations
- Reinforcement Learning with Augmented DataMichael Laskin, Kimin Lee, Adam Stooke, Lerrel Pinto et al.NeurIPS 2020 · 833 citations
- Flow Network based Generative Models for Non-Iterative Diverse Candidate GenerationEmmanuel Bengio, Moksh Jain, Maksym Korablyov, Doina Precup et al.NeurIPS 2021 · 565 citations
- Learning to Dispatch for Job Shop Scheduling via Deep Reinforcement LearningCong Zhang, Wen Song, Zhiguang Cao, Jie Zhang et al.NeurIPS 2020 · 497 citations
- Trajectory balance: Improved credit assignment in GFlowNetsNikolay Malkin, Moksh Jain, Emmanuel Bengio, Chen Sun et al.NeurIPS 2022 · 316 citations
Related papers
- Sym-NCO: Leveraging Symmetricity for Neural Combinatorial OptimizationMinsu Kim, Junyoung Park, Jinkyoo ParkNeurIPS 2022 · 200 citations
- Time Reversal Symmetry for Efficient Robotic Manipulations in Deep Reinforcement LearningYunpeng Jiang, Jianshu Hu, Paul Weng, Yutong BanNeurIPS 2025 · 1 citation
- Efficient Meta Neural Heuristic for Multi-Objective Combinatorial OptimizationJinbiao Chen, Jiahai Wang, Zizhen Zhang, Zhiguang Cao et al.NeurIPS 2023 · 35 citations
- Towards Understanding and Improving GFlowNet TrainingMax W. Shen, Emmanuel Bengio, Ehsan Hajiramezanali, Andreas Loukas et al.ICML 2023 · 81 citations
- Symmetry-Aware Actor-Critic for 3D Molecular DesignGregor N. C. Simm, Robert Pinsler, Gábor Csányi, José Miguel Hernández-LobatoICLR 2021 · 73 citations
