This repository contains the implementation code for our research paper, "Ant Colony Sampling with GFlowNets for Combinatorial Optimization." We have developed our codes based on the DeepACO codebase.
- Python 3.11.5
- PyTorch 2.1.1
- PyTorch Geometric 2.4.0
For the complete list of dependencies, please refer to the requirements.txt file.
For the usage of the code, please refer to each folder's README.md file.
If you find our work useful in your research, please consider citing our paper:
@article{kim2024ant,
title={Ant Colony Sampling with GFlowNets for Combinatorial Optimization},
author={Kim, Minsu and Choi, Sanghyeok and Son, Jiwoo and Kim, Hyeonah and Park, Jinkyoo and Bengio, Yoshua},
journal={arXiv preprint arXiv:2403.07041},
year={2024}
}