tingyingwu2010 / LRCO

Learning for Robust Combinatorial Optimization: Algorithm and Application

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Learning for Robust Combinatorial Optimization: Algorithm and Application

Learning for Robust Combinatorial Optimization: Algorithm and Application

Zhihui Shao, Jianyi Yang, Cong Shen, Shaolei Ren, IEEE International Conference on Computer Communications. (INFOCOM 2022)

@article{
  lrco_shao_2022,
  title={One Proxy Device Is Enough for Hardware-Aware Neural Architecture Search},
  author={Zhihui Shao, Jianyi Yang, Cong Shen and Shaolei Ren},
  journal = {IEEE International Conference on Computer Communications},
  month = May,
  year = 2022
}

scripts

  • train_maximizer.py is the script to pre-train a set of maximizer using learning-to-optimize (L2O).
  • train_minimizer.py is the script to train minimizer using policy gradient method.
  • inference_LRCO.py is the inference script to provide robust solution given the pre-trained LRCO framework.

author

@Zhihui Shao (zshao006@ucr.edu)

About

Learning for Robust Combinatorial Optimization: Algorithm and Application


Languages

Language:Jupyter Notebook 60.4%Language:Python 39.6%