lightvector/KataGo

GTP engine and self-play learning in Go

解决的问题

KataGo 是一个高性能的开源围棋AI,旨在成为目前可用的最强选择之一。它解决了提供超人级围棋AI的问题,该AI可通过自对弈从零开始训练,无需依赖庞大的外部数据集,并可用于高水平对局和详细的游戏分析。

工作原理

KataGo 使用类似 AlphaZero 的训练流程,利用神经网络(包括对 Transformer 模型的最新支持)和蒙特卡洛图搜索(MCTS 的扩展)。它注重效率,仅用数个 GPU 运行数天即可在 19x19 棋盘上达到高级业余段位水平,或在单个消费级 GPU 上运行数月后达到超人水平。支持多种硬件后端,包括 OpenCL、CUDA、TensorRT、ROCm、ONNX Runtime 和 Eigen(CPU)。

适用人群

  • 希望拥有强劲对手或游戏分析工具的围棋玩家。
  • 需要 GTP 兼容引擎的围棋软件开发者。
  • 对自对弈训练和蒙特卡洛图搜索感兴趣的 AI 研究人员。

主要亮点

  • 胜率与实地估算:与仅提供胜率的机器人不同,KataGo 能估算实地和得分,对业余对局分析非常有用。
  • 广泛支持:支持从 7x7 到 19x19 的多种棋盘尺寸和广泛的规则集。
  • 硬件灵活性:提供六种不同的后端,确保在 NVIDIA、AMD、Intel GPU 和 CPU 上的兼容性。
  • 高效训练:使用相对普通的硬件即可从零开始训练至超人水平。
  • 人类风格对弈:可通过特定的人类 SL 模型模仿不同段位的人类对弈风格。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目