监控指标解释#

本文档解释了 Trinity-RFT 中用于跟踪探索、评估和训练进度的指标类别。

指标命名规范#

大多数指标遵循分层命名规范:{category}/{taskset_name}/{metric_name}/{statistic}

  • Category(类别):广泛的功能领域(rollout、eval、time、actor、critic 等)

  • Taskset name(任务集名称):使用的任务集名称,仅适用于评估阶段的指标

  • Metric name(指标名称):正在测量的具体指标

  • Statistic(统计量):统计指标(mean、max、min、std、sum 等,如适用)

聚合类型后缀#

Metric key 可以通过 聚合类型后缀 来控制该指标在 experience、run、task、step 各层级的聚合方式。后缀通过冒号分隔附加在 key 末尾:

<metric_name>:<agg_type>

支持的聚合类型:

后缀

行为

输出 key 示例

(无)

取平均值(默认)

rollout/accuracy/mean, /max, /min

:sum

求和

rollout/total_tokens/sum

:max

取最大值

rollout/latency/max

:min

取最小值

rollout/latency/min

:last

取最后一个值

rollout/model_version/last

不带后缀的 key 默认使用 mean 聚合,这保证了与现有 workflow 和 operator 的完全向后兼容。

在 workflow 中的使用示例:

# 默认 mean 聚合(向后兼容)
exp.metrics["accuracy"] = 0.85
exp.metrics["reward"] = 1.0

# 显式 sum 聚合
exp.metrics["total_tokens:sum"] = len(tokens)
exp.metrics["time/env_interaction:sum"] = elapsed_time

# last 聚合(取最后一个值)
exp.metrics["model_version:last"] = current_version

所有聚合逻辑统一实现在 trinity.utils.metrics 中,提供以下函数:

  • parse_metric_key(key):将 metric key 解析为 (name, AggType)

  • aggregate_metrics(metric_dicts, prefix, ...):训练阶段的 batch 级别聚合(task → step)。

  • aggregate_eval_metrics(metric_dicts, prefix, ...):评估阶段的 batch 级别聚合。

  • aggregate_run_level_metrics(metric_dicts):experience 级别 → run 级别聚合。

指标类别#

以下内容按指标来源(生成位置)和指标前缀(指标名称的第一部分)对指标进行分类。

Explorer 相关指标#

Explorer 相关的指标发生在模型生成响应的 rollout 阶段,包括 rollout 指标(rollout/)、评估指标(eval/)和一些时间指标(time/)。

Rollout 指标(rollout/#

Rollout 指标跟踪模型生成响应的 rollout 阶段的性能。

  • 格式rollout/{metric_name}/{statistic}

  • 示例

    • rollout/accuracy/mean:生成响应的平均准确率

    • rollout/format_score/mean:平均格式正确性分数

    • rollout/total_tokens/sum:总 token 数量(sum 聚合)

指标计算过程

考虑一个包含 batch_size 个任务的探索步骤,其中每个任务有 repeat_times 次运行。Rollout 指标(例如,rollout/)在不同级别的计算过程如下:

  • ExperiencesRun 级别(仅对 can_repeat=False 的 workflows 有效):在 aggregate_run_level_metrics 函数中,在每次 run 生成的所有 experiences 中聚合得到 run 级别指标。聚合方式遵循 :agg 后缀——不带后缀的 key 取平均,带 :sum 的 key 求和,以此类推。

  • Run 级别Task 级别:在 calculate_task_level_metrics 函数中,指标跨同一任务的 repeat_times 次运行聚合。例如,rollout/accuracy 是该任务所有运行的平均准确率。

  • Task 级别Step 级别:在 aggregate_metrics 函数中,指标跨步骤中所有任务聚合。例如,rollout/accuracy/meanrollout/accuracy/maxrollout/accuracy/min 分别是步骤中所有任务的准确率(rollout/accuracy)的平均值、最大值和最小值。对于 :sum 类型的指标,仅输出 /sum 统计量。

以下图表说明了 rollout 指标的计算过程:

graph TD
    subgraph Step["Batch_size=3 Tasks"]
        subgraph Task1["Task 1 (repeat_times=2)"]
            Run1_1["Run 1<br/>accuracy: 0.8"]
            Run1_2["Run 2<br/>accuracy: 0.9"]
            Run1_1 --> Task1_Metric["rollout/accuracy<br/>= 0.85"]
            Run1_2 --> Task1_Metric
        end

        subgraph Task2["Task 2 (repeat_times=2)"]
            Run2_1["Run 1<br/>accuracy: 0.6"]
            Run2_2["Run 2<br/>accuracy: 0.9"]
            Run2_1 --> Task2_Metric["rollout/accuracy<br/>= 0.75"]
            Run2_2 --> Task2_Metric
        end

        subgraph TaskN["Task 3 (repeat_times=2)"]
            Run3_1["Run 1<br/>accuracy: 0.95"]
            Run3_2["Run 2<br/>accuracy: 0.85"]
            Run3_1 --> Task3_Metric["rollout/accuracy<br/>= 0.9"]
            Run3_2 --> Task3_Metric
        end

        Task1_Metric --> Step_Metrics["Step Level Metrics<br/>rollout/accuracy/mean=0.83<br/>rollout/accuracy/max=0.9<br/>rollout/accuracy/min=0.75"]
        Task2_Metric --> Step_Metrics
        Task3_Metric --> Step_Metrics
    end

评估指标(eval/)和基准测试指标(bench/#

评估指标衡量模型在保留的评估任务上的性能。这些指标在评估阶段的每次运行期间计算。

  • 格式eval/{task_name}/{metric_name}/{statistic}bench/{task_name}/{metric_name}/{statistic}

  • 示例

    • eval/gsm8k-eval/accuracy/mean@4:跨 repeat_times=4 次运行的平均准确率

    • bench/gsm8k-eval/accuracy/best@4:跨 repeat_times=4 次运行的最佳准确率值

  • 注意

    • Eval 和 bench 指标的计算方式相同,唯一的区别是指标名称的前缀。

    • 默认情况下,只返回指标的平均值。如果你想返回详细统计信息,可以在配置中将 monitor.detailed_stats 设置为 True

指标计算过程

考虑一个包含 len(eval_taskset) 个任务的评估步骤,其中每个任务有 repeat_times 次运行。评估指标(例如,eval/bench/)在不同级别计算和聚合:

  • ExperiencesRun 级别(仅对 can_repeat=False 的 workflows 有效):在 aggregate_run_level_metrics 函数中,在每次 run 生成的所有 experiences 中聚合得到 run 级别指标。

  • Run 级别Task 级别:在 calculate_task_level_metrics 函数中,指标跨同一任务的 repeat_times 次运行聚合。对于默认(mean)指标,会计算 bootstrap 统计量:mean@Nstd@Nbest@Nworst@N。对于非 mean 指标(:sum:max 等),仅输出对应的单一聚合值。

  • Task 级别Step 级别:在 aggregate_eval_metrics 函数中,指标跨步骤中所有任务聚合。例如,eval/dummy/accuracy/mean@2eval/dummy/accuracy/std@2eval/dummy/accuracy/best@2eval/dummy/accuracy/worst@2 分别是步骤中所有任务的准确率(eval/dummy/accuracy)的平均值、标准差、最佳值和最差值。

以下图表说明了在包含三个任务的虚拟数据集上评估指标的计算过程。默认情况下,报告所有评估任务中指标的 mean@kstd@kbest@kworst@k。你可以在配置中将 monitor.detailed_stats 设置为 True 以返回详细统计信息。

graph TD
    subgraph Step["len(eval_taskset)=3 Tasks"]
        subgraph Task1["Task 1 (repeat_times=2)"]
            Run1_1["Run 1<br/>accuracy: 0.8"]
            Run1_2["Run 2<br/>accuracy: 0.9"]
            Run1_1 --> Task1_Metric["eval/dummy/accuracy/mean@2=0.85<br/>eval/dummy/accuracy/std@2=0.05"]
            Run1_2 --> Task1_Metric
        end

        subgraph Task2["Task 2 (repeat_times=2)"]
            Run2_1["Run 1<br/>accuracy: 0.6"]
            Run2_2["Run 2<br/>accuracy: 0.9"]
            Run2_1 --> Task2_Metric["eval/dummy/accuracy/mean@2=0.75<br/>eval/dummy/accuracy/std@2=0.15"]
            Run2_2 --> Task2_Metric
        end

        subgraph TaskN["Task 3 (repeat_times=2)"]
            Run3_1["Run 1<br/>accuracy: 0.95"]
            Run3_2["Run 2<br/>accuracy: 0.85"]
            Run3_1 --> Task3_Metric["eval/dummy/accuracy/mean@2=0.9<br/>eval/dummy/accuracy/std@2=0.05"]
            Run3_2 --> Task3_Metric
        end

        Task1_Metric --> Step_Metrics["Step Level Metrics<br/>eval/dummy/accuracy/mean@2=0.83<br/>eval/dummy/accuracy/std@2=0.083"]
        Task2_Metric --> Step_Metrics
        Task3_Metric --> Step_Metrics
    end

当你将 monitor.detailed_stats 设置为 True 时,你会得到详细的统计信息,包括 mean、std、min、max,例如下面图表中给出的一些实例:

graph TD
    subgraph Step_Metrics_Summary["Detailed Statistics"]
        Stat1["eval/dummy/accuracy/mean@2/mean=0.83<br/>eval/dummy/accuracy/mean@2/std=0.062<br/>eval/dummy/accuracy/mean@2/max=0.9<br/>eval/dummy/accuracy/mean@2/min=0.75"]
        Stat2["eval/dummy/accuracy/std@2/mean=0.083<br/>eval/dummy/accuracy/std@2/std=0.047<br/>eval/dummy/accuracy/std@2/max=0.15<br/>eval/dummy/accuracy/std@2/min=0.05"]
        Step_Metrics --> Stat1
        Step_Metrics --> Stat2
    end

时间指标(time/#

时间指标测量整个过程中各种操作的执行持续时间。

  • 格式time/{operation_name}

  • 示例

    • time/eval:从提交评估任务开始到评估阶段结束的时间;此持续时间包括评估任务和一些 rollout 任务。

    • time/wait_explore_step:等待一次 rollout 步骤完成的时间。

注意

  • 由于 rollout 阶段的异步性质,时间测量的结果可能不太准确,但仍然可以用于监控整体训练进度。

  • 除非特殊说明,时间指标以秒为单位。

  • 一些训练操作还报告每个 token 的时间指标,前缀为 timing_per_token_ms/(例如,timing_per_token_ms/update_actortiming_per_token_ms/update_critictiming_per_token_ms/advtiming_per_token_ms/values)。这些指标通过处理的 token 数量对执行时间进行归一化,提供独立于批次大小的效率测量。

Trainer 相关指标#

此类别包括跟踪 actor 模型(actor/)和 critic 模型(critic/)的训练动态的指标,以及一些性能指标(perf/global_seqlen/response_length/prompt_length/time/)。这些指标和 veRL 中的指标一致,感兴趣的用户可以参考 veRL 文档 了解更多详细信息。

数据处理指标#

此类别包括跟踪通过各种数据处理操作(experience_pipeline/)和数据采样统计(sample/)的指标。这些指标在步骤(step)级别计算,因为 experience 处理和数据采样会在每个步骤中执行一次。

Experience Pipeline 相关指标(experience_pipeline/experience_pipeline/time/#

Experience Pipeline 相关的指标统计了和数据处理相关的值,每个指标表示一个步骤中特定操作的统计值。

  • 格式experience_pipeline/{metric_name}

  • 示例

    • experience_pipeline/experience_count:处理的经验数量

    • experience_pipeline/group_advantages/reward_mean/mean:这里 reward_mean 是每个任务的平均奖励,然后我们计算步骤中所有任务的平均奖励的平均值。

以下图表说明了数据处理指标的计算过程:

graph TD
    subgraph Step["4 Experiences in one step"]
        subgraph Task1["Experience 1"]
            Run1_1["Run 1<br/>reward_mean: 0.8"]
            Run1_2["Run 2<br/>reward_mean: 0.8"]
            Run2_1["Run 3<br/>reward_mean: 0.9"]
            Run2_2["Run 4<br/>reward_mean: 0.9"]
        end
            Run1_1 --> Task1_Metric["rollout/accuracy<br/>= 0.85"]
            Run1_2 --> Task1_Metric
            Run2_1 --> Task1_Metric
            Run2_2 --> Task1_Metric
    end

采样相关指标(sample/#

采样相关的指标发生在训练阶段,用于训练期间的数据采样统计。

  • 格式sample/{metric_name}

  • 示例

    • sample/model_version/mean:采样经验的平均模型版本

    • sample/task_count:采样批次中的任务数量