energy-mcq-harder-lighteval-rag
收藏资源简介:
该数据集由Cemig发布,用于评估能源领域(包括巴西国家电力局ANEEL相关主题)的问答系统,特别是检索增强生成(RAG)管道的性能。包含645个多选题样本,每个问题提供7个选项,属于hard子集,随机猜测的基线准确率为0.1429。语言为葡萄牙语。评估时采用answer_extraction指标,即从模型生成的答案中提取选项字母。该数据集适用于测试模型在能源监管领域复杂多选题上的推理能力。
This dataset, released by Cemig, is used to evaluate question answering systems in the energy domain (including topics related to the Brazilian National Electric Energy Agency - ANEEL), particularly the performance of retrieval-augmented generation (RAG) pipelines. It contains 645 multiple-choice question samples, each with 7 options, belonging to the hard subset, with a random guess baseline accuracy of 0.1429. The language is Portuguese. Evaluation uses the answer_extraction metric, which extracts the option letter from the model-generated answer. This dataset is suitable for testing the reasoning ability of models on complex multiple-choice questions in the energy regulatory domain.
数据集概述
该数据集记录了巴西Cemig公司RAG管线在LightEval基准上的评估结果,数据以葡萄牙语呈现。
核心内容
- 任务类型: 多选题问答(MCQ),评估RAG管线在能源领域(ANEEL)的知识检索与生成能力
- 数据来源: RAG管线通过OpenAI兼容端点接受LightEval评估,使用与纯模型相同的runner、任务和指标
实验设计
- 基线(decoder-only): 无检索增强的模型(energy-gpt-regulatorio-v2)
- RAG变体(rag-<encoder>[-<reranker>]): 结合检索器与可选的重新排序器(encoder为multilingual-e5-large-ft-v2,未使用reranker)
- 检索参数: retriever_k=5,查询模式为task,上下文置于问题前
主要结论
| 实验 | 准确率 |
|---|---|
| RAG + e5-large-ft-v2 | 77.83% |
| 无RAG(基线) | 38.60% |
RAG显著提升了准确率(+39.23个百分点)。
关键参数
- 模型: cemig-nlp-releases/energy-gpt-regulatorio-v2
- 编码器: cemig-nlp-releases/multilingual-e5-large-ft-v2
- 基准数据集: cemig-ceia/benchmark-energy-mcq-harder(子集:hard)
- 样本数: 645,每个问题7个选项,随机基线准确率0.1429
- 推理后端: vllm,bfloat16,GPU为DGX-H100-02(SLURM任务30157/30158)
- 生成大小: 512
指标说明
- 采用answer_extraction指标,从模型输出中提取带分隔符的字母(如“(D)”)作为答案
- 注意:exact/prefix匹配指标不适用,因为标注答案为“ D”,而模型生成“(D)”,括号会导致匹配失效
附加信息
- 结果数据存放在
details/目录下的Parquet文件,包含每个样本的完整问题、提示、模型响应及提取结果,便于审计




