Large Language Models Play StarCraft II: Benchmarks and A Chain of Summarization Approach

StarCraft II is a challenging benchmark for AI agents due to micro-level operations and macro-awareness. Previous works, such as Alphastar and SCC, achieve impressive performance on tackling StarCraft II , however, still exhibit deficiencies in long-term strategic planning and strategy interpretability. Emerging large language model (LLM) agents, presents the immense potential in solving intricate tasks.

Motivated by this, we aim to validate the capabilities of LLMs on StarCraft II. We first develop textual StratCraft II environment, called TextStarCraft II. Secondly, we propose a Chain of Summarization method, including single-frame summarization for processing raw observations and multi-frame summarization for analyzing game information, providing command recommendations, and generating strategic decisions. Our experiment demonstrates that LLM agents are capable of defeating the built-in AI at the Harder(Lv5) difficulty level.

Our paper:

Large Language Models Play StarCraft II: Benchmarks and A Chain of Summarization Approach https://arxiv.org/abs/2312.11865

Our website:

Large Language Models Play StarCraft II: Benchmarks and A Chain of Summarization Approach

Our demo video:

Large Language Models Play StarCraft II: Benchmarks and A Chain of Summarization Approach

Performance of LLMs in TextStarCraft II

Comparing models using either the full CoS or CoS without CoT.

Win Rate Comparison of LLM Agents Against TextStarCraft II's Built-in AI

Install StarCraft II and setup maps

Install StarCraft II

StatCraft II is a classic game developed by BLZ, and has some professional leagues such as IEM, WTL....You can download Battle.net from:https://us.shop.battle.net/en-us, or here:https://www.blizzard.com/zh-tw/

If you are Chinese, due to the Bobby Kotick, CN play cant own their sever again. So we must download StarCraft II by this video :video or you can search in the internet.

Download maps

First , we should use StarCraft II Editor.exe to download the newest ladder map

when we open this, please log in your blz account and search the map which you want. Then you should put maps to your StarCrafrt2 file in StarCraft II\Maps(If the 'Maps' file dont exist, please create it).

Or you can download maps in here:

Setup environment

Create environment

• OS, We used Windows 11 to develop this demo, because BLZ didnt release the latest sc2 on liunx, so please run our repo on Windows OS!
• python: python 3.10.
• cuda: cuda 12.1.
• torch: 2.1.0
• openai: 0.27.9, very important. This is crucial as versions above 0.28 have altered API functionalities. Install all necessary packages with pip install -r requirements.txt.

Tips

• burnysc2: This is our core package, offering an easy-to-use API for project development. Find more information here:Python-sc2
• chromadb: We utilize the Chroma vector database. Due to package conflicts, install Chromadb first, followed by burnysc2.
• Huggingface and sentence-transformers: we used the embedding model sentence-transformers/all-mpnet-base-v2, in our github version, it will automatically download. We also provide the release zip, you can just download and unzip that(with embedding model).

Run demo

Game mode

• Agent vs Botai: You can test in test_the_env.py & multiprocess_test.py

• Human vs Agent: You can try in our Human_LLM_agent_test.py

• Agent vs Agent: You can try in our 2agent_test.py

  Single process

  You can run test_the_env.py to try our agent. Here is some parameters you need to set.

• player_race: Currently, only Protoss is supported. Zerg and Terran are under development.

• opposite_race: Typically set to Zerg, but Terran and Protoss are also compatible.

• difficulty: We offer 10 difficulty levels, ranging from Level 1 (VeryEasy) to Level 10 (CheatInsane). Note that these names differ from those in the StarCraft2 client, but the AI difficulty remains unchanged.

• replay_folder: Specify the folder for saving demo replays.
• LLM_model_name: We used gpt-3.5-turbo-16k in our experiments.
• LLM_temperature: Set between 0 and 1 as per your preference.
• LLM_api_key: Your API key.
• LLM_api_base: Your API base URL.

Note: Using LLM to play StarCraft2 can take approximately 7 hours for a single game.

Multi process

To save time, you can run multiple demos simultaneously using multiprocess_test.py. Configure the following parameter:

• num_processes: The number of processes to spawn.

Other parameters are the same as in the Single Process setup.

Other settings

In our experiments, we have added some more settings, but due to several reasons these settings will coming soon.

• num_agents : This environment will support LLM agent vs LLM agent or RL agent.
• env_type: This environment will support Text or MultiModal
• 'player_race': This environment will support Zerg and Terran
• opposite_type: This env will support some human designed botai.

Create your LLM Agent

If you want to use other llm to create your own llm agent, the following things you should to know.

Component of LLM Agent

• LLM: In our repo, you should request llm from ChatBot_SingleTurn function in TextStarCraft2_2/LLM/gpt_test
• 'L1_summarize': Our level-1 summarization method is here: generate_summarize_L1 in TextStarCraft2_2/summarize/L1_summarize.py
• L2_summarize: Our level-2 summarization method is here : L2_summary in TextStarCraft2_2/summarize/gpt_test/L2_summarize.py
• action dict: The actions that llm agent can use. Here we can set TextStarCraft2_2/utils/action_info.py . You can add more actions for llm agent.
• action extractor : We can extract decisions by TextStarCraft2_2/utils/action_extractor.py

Env

The core of our TextStarCraft II env is TextStarCraft2_2/env/bot. Here you can add more settings for environment. So if you want to realise Terran and Zerg bot, you can modify our code about this dictionary.

• State: In Protoss_bot.py, the State of Env is generate from get_information function. This is what we said Obs to Text adaptor
• Action: In Protoss_bot.py, the Action space of Agent is designed by these handle_action function. This is what we said Text to Action adaptor.

Support Models

We have tested several LLMs in our experiments. The usage is in sc2_rl_agent/starcraftenv_test/LLM file

• Online LLM: GPT3.5-tubor,GLM4,Gemini-pro,Claude2.
• Local LLM: GLM3,QWEN,QWEN1.5.

Evaluation Metrics Overview

Our framework in TextStarCraft II extends traditional StarCraft II analytics to evaluate LLM agents’ strategies with metrics tailored for AI gameplay performance:

• Win Rate: Reflects the agent's performance, calculated as the percentage of games won out of total games played.

• Population Block Ratio (PBR): Indicates macro-management effectiveness, focusing on resource allocation and population growth. A higher PBR suggests less effective macro-strategy due to more time spent at population cap.

• Resource Utilization Ratio (RUR): Measures how efficiently the agent manages resources throughout the game. Higher RUR indicates underutilization of resources.

• Average Population Utilization (APU): Assesses efficiency in utilizing population capacity. Higher APU indicates better macro-management.

• Technology Rate (TR): Evaluates the agent's use of the technology tree, showing the proportion of technologies and buildings completed. It reflects the agent’s technological advancement.