Add README (Chinese) for tuner (#106)
This commit is contained in:
Yuchang Sun
@@ -4,18 +4,13 @@ This example demonstrates how to use **AgentScope-Tuner** to enhance a math prob
|
||||
|
||||
## Task Setting
|
||||
|
||||
We use the foundational [math-agent example](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/main.py) as our baseline to demonstrate the data enhancement capabilities. Notably, these data-centric techniques are generic and customizable, making them adaptable to other agent workflows.
|
||||
We use the foundational [math-agent example](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/main.py) as our baseline. The agent is a **`ReActAgent`** that solves mathematical reasoning problems through step-by-step reasoning.
|
||||
|
||||
### Agent Goal and Type
|
||||
The agent's objective is to solve mathematical reasoning problems, learning to produce a correct final answer through a step-by-step thought process. The agent is implemented as a **`ReActAgent`**, which follows a reasoning-acting loop to solve tasks iteratively.
|
||||
|
||||
### Objective of the Data-Centric Approach
|
||||
|
||||
Training can be inefficient if tasks are too easy or too hard. This example addresses this by providing **selectors** to dynamically select tasks using **data feedback**. This empowers users to explore and implement their own data-centric strategies, such as focusing on "productively challenging" samples, to maximize training efficiency.
|
||||
Training can be inefficient if tasks are too easy or too hard. This example demonstrates how to use **task selectors** to dynamically select tasks based on **data feedback**, focusing on "productively challenging" samples to maximize training efficiency. These data-centric techniques are generic and adaptable to other agent workflows.
|
||||
|
||||
## Dataset Preparation
|
||||
|
||||
To enable difficulty-based sampling, our training data needs to include features that represent the "difficulty" of each task.
|
||||
To enable difficulty-based sampling, the training data must include difficulty features (e.g., pass rates from LLMs).
|
||||
|
||||
1. **Base Dataset**: You can use any standard math problem dataset. A good example is the math data in [LLM360/guru-RL-92k](https://huggingface.co/datasets/LLM360/guru-RL-92k), which comes pre-annotated with pass rates from different LLMs, serving as direct difficulty features.
|
||||
2. **Build Your Own Features**: If you use your own dataset, you can generate these features by pre-running several models of varying capabilities and recording their pass rates. This can be done within the [**Trinity-RFT**](https://github.com/agentscope-ai/Trinity-RFT/pull/440) framework.
|
||||
@@ -34,7 +29,7 @@ Leveraging the powerful data processing capabilities of **Trinity-RFT**, **Agent
|
||||
|
||||
#### Task Selector
|
||||
|
||||
The `Task Selector` determines how samples are selected from a dataset. It can be configured directly in `Yaml Config`.
|
||||
The `Task Selector` determines how samples are selected from a dataset. It can be configured directly in configuration YAML files.
|
||||
|
||||
- **Built-in Selectors**:
|
||||
- `sequential`: Samples are selected in a fixed order.
|
||||
@@ -43,7 +38,7 @@ The `Task Selector` determines how samples are selected from a dataset. It can b
|
||||
- `offline_easy2hard`: Samples are sorted by a predefined feature for curriculum learning.
|
||||
- `difficulty_based` (Customized): An adaptive sampler based on task difficulty.
|
||||
|
||||
> For more details on `Task Selector`, including how to implement a custom selector based on feedback signals, please refer to **Trinity-RFT**'s **[Selector Development Guide](https://github.com/agentscope-ai/Trinity-RFT/blob/main/docs/sphinx_doc/source/tutorial/develop_selector.md)**.
|
||||
> For more details on `Task Selector`, including how to implement a custom selector based on feedback signals, please refer to **Trinity-RFT**'s **[Selector Development Guide](https://agentscope-ai.github.io/Trinity-RFT/en/main/tutorial/develop_selector.html)**.
|
||||
|
||||
#### Data Processor
|
||||
|
||||
@@ -51,7 +46,7 @@ The `Data Processor` allows for real-time processing of **Task** and **Experienc
|
||||
|
||||
For example, the `difficulty_based` selector requires a `pass_rate_calculator` operator to compute the agent's success rate for each task. This feedback is then used to adjust the sampling strategy.
|
||||
|
||||
> For more details on `Data Processor`, please refer to **Trinity-RFT**'s **[Operator Development Guide](https://github.com/agentscope-ai/Trinity-RFT/blob/main/docs/sphinx_doc/source/tutorial/develop_operator.md)**.
|
||||
> For more details on `Data Processor`, please refer to **Trinity-RFT**'s **[Operator Development Guide](https://agentscope-ai.github.io/Trinity-RFT/en/main/tutorial/develop_operator.html)**.
|
||||
|
||||
|
||||
### Configuring the Experiments
|
||||
@@ -147,7 +142,9 @@ python main.py --config config_difficulty.yaml
|
||||
|
||||
The following results compare the performance of the `difficulty-based` selection strategy (red line, bots) against a standard `random` selection strategy (black line, random).
|
||||
|
||||
|
||||
<div align="center">
|
||||
<img src="./training_result.jpg" alt="Training Result Image" width="90%"/>
|
||||
</div>
|
||||
|
||||
### Training Reward Curve
|
||||
|
||||
|
||||
153
tuner/data_augment/README_zh.md
Normal file
153
tuner/data_augment/README_zh.md
Normal file
@@ -0,0 +1,153 @@
|
||||
# 使用数据增强策略训练数学智能体
|
||||
|
||||
本示例演示了如何使用 **AgentScope-Tuner** 训练数学问题求解智能体。我们将重点利用**以数据为中心**的功能，例如 `difficulty_based` 任务选择器，以提高数据利用率和训练效率。
|
||||
|
||||
## 任务设置
|
||||
|
||||
我们使用基础的 [math-agent 示例](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/main.py) 作为基线。智能体是 **`ReActAgent`**，通过逐步推理解决数学推理问题。
|
||||
|
||||
如果任务太容易或太难，训练可能会效率低下。本示例演示如何使用**任务选择器**基于**数据反馈**动态选择任务，专注于"具有挑战性"的样本以最大化训练效率。这些以数据为中心的技术是通用的，可适应其他智能体工作流。
|
||||
|
||||
## 数据集准备
|
||||
|
||||
为启用基于难度的采样，训练数据必须包含难度特征（如 LLM 的通过率）。
|
||||
|
||||
1. **基础数据集**：您可以使用任何标准的数学问题数据集。一个很好的例子是 [LLM360/guru-RL-92k](https://huggingface.co/datasets/LLM360/guru-RL-92k) 中的数学数据，它预先标注了来自不同 LLM 的通过率，作为直接的难度特征。
|
||||
2. **构建您自己的特征**：如果您使用自己的数据集，可以通过预先运行几个不同能力的模型并记录它们的通过率来生成这些特征。这可以在 [**Trinity-RFT**](https://github.com/agentscope-ai/Trinity-RFT/pull/440) 框架内完成。
|
||||
3. **数据格式**：最终数据集应为 HuggingFace 格式。在此示例中，数据将根据[工作流](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/main.py)转换为 *GSM8K 格式*。除了任务内容外，它还必须包含您定义的难度特征列（例如 `qwen2.5_7b_pass_rate`）。
|
||||
4. **示例数据准备**：我们为此示例提供了一个脚本。只需执行 `python prepare_data.py` 即可生成所需的数据集。
|
||||
|
||||
## 代码实现
|
||||
|
||||
本示例采用 [math-agent 示例](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/main.py) 的 `run_react_agent` 和 `gsm8k_judge` 作为 `workflow_func` 和 `judge_func`，说明可以在不改变核心智能体逻辑的情况下应用训练策略。
|
||||
|
||||
### 以数据为中心功能的设计
|
||||
|
||||
利用 **Trinity-RFT** 强大的数据处理能力，**AgentScope-Tuner** 为任务选择和经验处理等高级操作提供了接口。
|
||||
|
||||
#### 任务选择器
|
||||
|
||||
`Task Selector` 决定如何从数据集中选择样本。它可以直接在 YAML 配置文件中配置。
|
||||
|
||||
- **内置选择器**：
|
||||
- `sequential`：按固定顺序选择样本。
|
||||
- `shuffle`：在每个 epoch 开始时打乱数据集。
|
||||
- `random`：为每个批次随机选择样本（有放回）。
|
||||
- `offline_easy2hard`：按预定义特征对样本进行排序，用于课程学习。
|
||||
- `difficulty_based`（自定义）：基于任务难度的自适应采样器。
|
||||
|
||||
> 有关 `Task Selector` 的更多详细信息，包括如何基于反馈信号实现自定义选择器，请参阅 **Trinity-RFT** 的 **[Selector 开发指南](https://agentscope-ai.github.io/Trinity-RFT/zh/main/tutorial/develop_selector.html)**。
|
||||
|
||||
#### 数据处理器
|
||||
|
||||
`Data Processor` 允许在训练期间实时处理**任务**（task）和**经验**（experience），支持计算反馈指标、数据增强或过滤等操作。
|
||||
|
||||
例如，`difficulty_based` 选择器需要一个 `pass_rate_calculator` 操作符来计算智能体对每个任务的成功率。然后使用此反馈来调整采样策略。
|
||||
|
||||
> 有关 `Data Processor` 的更多详细信息，请参阅 **Trinity-RFT** 的 **[Operator 开发指南](https://agentscope-ai.github.io/Trinity-RFT/zh/main/tutorial/develop_operator.html)**。
|
||||
|
||||
|
||||
### 配置实验
|
||||
|
||||
为了保持清晰和简洁，我们建议在 YAML 配置文件中定义所有数据特定参数，包括数据集路径和任务选择器。
|
||||
|
||||
我们提供两个配置文件，用于比较基线 `random` 选择器与 `difficulty_based` 选择器。
|
||||
|
||||
**实验 1：使用随机选择器的基线（`config_random.yaml`）**
|
||||
|
||||
在 `config_random.yaml` 中，我们在 `buffer.explorer_input.taskset` 下配置用于随机采样的 `task_selector`。
|
||||
|
||||
```yaml
|
||||
# 在 config_random.yaml 中
|
||||
buffer:
|
||||
# ...
|
||||
explorer_input:
|
||||
taskset: # 训练数据
|
||||
path: "path/to/your/augmented/math_data"
|
||||
split: "train"
|
||||
task_selector:
|
||||
selector_type: random # 任务选择策略
|
||||
```
|
||||
|
||||
**实验 2：使用基于难度选择器的高级训练（`config_difficulty.yaml`）**
|
||||
|
||||
在 `config_difficulty.yaml` 中，我们将 `task_selector` 切换为 `difficulty_based` 并提供其特定参数。请注意，此配置还启用了反馈所需的 `pass_rate_calculator`。
|
||||
|
||||
```yaml
|
||||
# 在 config_difficulty.yaml 中
|
||||
|
||||
# 启用计算器为选择器提供反馈
|
||||
data_processor:
|
||||
experience_pipeline:
|
||||
operators:
|
||||
- name: pass_rate_calculator
|
||||
|
||||
buffer:
|
||||
# ...
|
||||
explorer_input:
|
||||
taskset: # 训练数据
|
||||
path: "path/to/your/augmented/math_data"
|
||||
split: "train"
|
||||
task_selector:
|
||||
selector_type: difficulty_based # 任务选择策略
|
||||
feature_keys: [ "qwen2.5_7b_pass_rate", "qwen3_30b_pass_rate" ]
|
||||
kwargs: # 选择算法的超参数
|
||||
m: 8
|
||||
# ...
|
||||
```
|
||||
|
||||
> 本示例中的 `difficulty_based` 选择器是 ***BOTS*** 算法的实现。有关其内部工作原理的详细信息，请参阅 [***BOTS 论文***](https://arxiv.org/abs/2510.26374) 及其 [***教程***](https://github.com/agentscope-ai/Trinity-RFT/blob/main/examples/bots/README.md)。
|
||||
|
||||
## 如何运行
|
||||
|
||||
### 步骤 1：前置要求
|
||||
|
||||
确保您已按照[指南](https://github.com/agentscope-ai/agentscope-samples/blob/main/tuner/math_agent/README_zh.md#how-to-run)安装了 **AgentScope** 和 **Trinity-RFT**。
|
||||
|
||||
### 步骤 2：准备数据集
|
||||
|
||||
运行数据准备脚本。确保之后更新 `config_random.yaml` 和 `config_difficulty.yaml` 中的数据集路径。
|
||||
|
||||
```bash
|
||||
python prepare_data.py
|
||||
```
|
||||
|
||||
### 步骤 3：启动 Ray 集群
|
||||
|
||||
对于分布式训练，启动 Ray 集群。
|
||||
|
||||
```bash
|
||||
# 单节点
|
||||
ray start --head
|
||||
```
|
||||
|
||||
### 步骤 4：运行训练
|
||||
|
||||
您现在可以运行基线或基于难度的训练实验。
|
||||
|
||||
- **使用随机选择器运行基线实验：**
|
||||
|
||||
```bash
|
||||
python main.py --config config_random.yaml
|
||||
```
|
||||
|
||||
- **使用基于难度的选择器运行实验：**
|
||||
```bash
|
||||
python main.py --config config_difficulty.yaml
|
||||
```
|
||||
|
||||
## 实验结果
|
||||
|
||||
以下结果比较了 `difficulty-based` 选择策略（红线，bots）与标准 `random` 选择策略（黑线，random）的性能。
|
||||
|
||||
<div align="center">
|
||||
<img src="./training_result.jpg" alt="训练结果图" width="90%"/>
|
||||
</div>
|
||||
|
||||
### 训练奖励曲线
|
||||
|
||||
左侧图表显示了训练期间的 rollout 准确率。可以看出，随机策略采样的任务对模型来说似乎很困难，准确率保持在 0.2 以下。相比之下，使用难度选择器会产生更高的平均准确率，表明智能体正在处理更多可以成功解决的任务。
|
||||
|
||||
### 在 AIME-24 上的评估
|
||||
|
||||
为了比较，我们在 AIME-24 基准上评估了两种选择策略。右侧图表显示，基于难度的方法在性能上表现出更好的上升趋势。
|
||||
@@ -71,4 +71,4 @@ explorer:
|
||||
trainer:
|
||||
save_interval: 100 # Save checkpoint every N steps
|
||||
use_dynamic_bsz: true # Use dynamic batch size
|
||||
ulysses_sequence_parallel_size: 1 # Sequence parallel size for Ulysses
|
||||
ulysses_sequence_parallel_size: 1 # Sequence parallel size for Ulysses
|
||||
|
||||
@@ -59,4 +59,4 @@ explorer:
|
||||
trainer:
|
||||
save_interval: 100 # Save checkpoint every N steps
|
||||
use_dynamic_bsz: true # Use dynamic batch size
|
||||
ulysses_sequence_parallel_size: 1 # Sequence parallel size for Ulysses
|
||||
ulysses_sequence_parallel_size: 1 # Sequence parallel size for Ulysses
|
||||
|
||||
Reference in New Issue
Block a user