jacopomauro/abs_optimizer

ABS参数优化工具（ABS POPT）

镜像概述

ABS POPT是一款用于优化ABS模型参数的工具，通过改变模型参数并运行模拟，结合SMAC3算法寻找最优参数组合。相比传统网格搜索，它能避免模拟数量爆炸问题；当前版本采用SMAC3，并通过HTTP POST请求实现模拟任务的分布式执行。

核心功能

• 基于SMAC3算法的高效参数优化，降低调优时间成本；
• 支持HTTP微服务（WORKER）分布式执行ABS模拟任务；
• 提供并行运行SMAC3的能力，进一步提升调优速度；
• 允许自定义参数范围、模拟输出解析逻辑及优化指标。

使用场景

• ABS模型开发者需优化参数以提升模型性能或达成特定目标；
• 需要分布式执行模拟任务以缩短参数调优周期；
• 希望通过自动化工具替代手动参数调整的场景。

配置说明与部署示例

部署模拟微服务（WORKER）

ABS POPT将模拟任务委托给WORKER微服务。本地部署命令：

bash
sudo docker pull jacopomauro/abs_optimizer:latest
sudo docker run -d -p <PORT>:9001 --name worker_container jacopomauro/abs_optimizer:latest

替换<PORT>为服务端口。验证服务状态：

bash
curl http://localhost:<PORT>/health

返回OK表示服务正常。

部署ABS POPT控制器

部署控制器容器：

bash
sudo docker pull jacopomauro/abs_optimizer:main
sudo docker run --net="host" -d --name controller_container jacopomauro/abs_optimizer:main \
  /bin/bash -c "while true; do sleep 60; done"

工具使用示例

1. 进入控制器容器Shell：

bash
sudo docker exec -i -t controller_container bash

2. 运行示例任务（限制5次模拟或1小时超时）：

bash
python abs_opt.py run \
 --param-file examples/new_years_eve/param_spec.json \
 --abs-file examples/new_years_eve/NewYearsEve.abs \
 --output-log-parser examples/new_years_eve/solution_quality.py \
 --global-simulation-limit 5 \
 --global-timeout 3600 \
 --abs-option-l 310 

自定义文件挂载

若需使用自定义ABS文件或配置，通过Docker卷挂载：

bash
sudo docker run --net="host" -v <本地路径>:/files_directory -d --name controller_container jacopomauro/abs_optimizer:main \
  /bin/bash -c "while true; do sleep 60; done"

自定义文件将位于容器内/files_directory目录。

参数范围定义

参数范围通过JSON文件配置，示例：

json
{  
  "parameters":{  
    "p1":{  
      "type":"integer",
      "values":[1,10],
      "default":1
    },
    "p2":{  
      "type":"ordinal",
      "values":[4,5,7],
      "default":5
    }
  }
}

支持integer（整数区间）和ordinal（离散值集合）两种类型。

并行运行SMAC3

通过--parallel-executions设置并行数，建议每个并行任务对应一个WORKER。使用Kubernetes部署多WORKER及负载均衡：

1. 执行部署脚本：kubernetes/deploy.sh（需kubectl权限）；
2. 扩展WORKER数量：kubectl scale deployment http-myapp --replicas=<NUM>；
3. 获取负载均衡端口：kubectl --namespace=myapp-namespace get services | grep haproxy-ingress；
4. 并行运行命令：

bash
python abs_opt.py run \
 --param-file ... \
 --abs-file ... \
 --output-log-parser ... \
 --parallel-executions <NUM> \
 --global-simulation-limit ... \
 --server-url http://<IP> \
 --server-port <PORT> \
 --server-host myapp

清理资源

停止并删除容器与镜像：

bash
sudo docker stop controller_container && sudo docker rm controller_container
sudo docker stop worker_container && sudo docker rm worker_container
sudo docker rmi jacopomauro/abs_optimizer:latest
sudo docker rmi jacopomauro/abs_optimizer:main