遇见数据集

LM-Polygraph/xsum

收藏
Hugging Face2025-07-16 更新2024-12-14 收录
官方服务:

资源简介:

--- language: - en dataset_info: - config_name: continuation features: - name: input dtype: string - name: output dtype: string splits: - name: train num_bytes: 494305693 num_examples: 204045 - name: test num_bytes: 27594857 num_examples: 11334 download_size: 321712135 dataset_size: 521900550 - config_name: simple_instruct features: - name: input dtype: string - name: output dtype: string splits: - name: train num_bytes: 498182548 num_examples: 204045 - name: test num_bytes: 27810203 num_examples: 11334 download_size: 321755661 dataset_size: 525992751 configs: - config_name: continuation data_files: - split: train path: continuation/train-* - split: test path: continuation/test-* - config_name: simple_instruct data_files: - split: train path: simple_instruct/train-* - split: test path: simple_instruct/test-* --- # Dataset Card for xsum <!-- Provide a quick summary of the dataset. --> This is a preprocessed version of xsum dataset for benchmarks in LM-Polygraph. ## Dataset Details ### Dataset Description <!-- Provide a longer summary of what this dataset is. --> - **Curated by:** https://huggingface.co/LM-Polygraph - **License:** https://github.com/IINemo/lm-polygraph/blob/main/LICENSE.md ### Dataset Sources [optional] <!-- Provide the basic links for the dataset. --> - **Repository:** https://github.com/IINemo/lm-polygraph ## Uses <!-- Address questions around how the dataset is intended to be used. --> ### Direct Use <!-- This section describes suitable use cases for the dataset. --> This dataset should be used for performing benchmarks on LM-polygraph. ### Out-of-Scope Use <!-- This section addresses misuse, malicious use, and uses that the dataset will not work well for. --> This dataset should not be used for further dataset preprocessing. ## Dataset Structure <!-- This section provides a description of the dataset fields, and additional information about the dataset structure such as criteria used to create the splits, relationships between data points, etc. --> This dataset contains the "continuation" subset, which corresponds to main dataset, used in LM-Polygraph. It may also contain other subsets, which correspond to instruct methods, used in LM-Polygraph. Each subset contains two splits: train and test. Each split contains two string columns: "input", which corresponds to processed input for LM-Polygraph, and "output", which corresponds to processed output for LM-Polygraph. ## Dataset Creation ### Curation Rationale <!-- Motivation for the creation of this dataset. --> This dataset is created in order to separate dataset creation code from benchmarking code. ### Source Data <!-- This section describes the source data (e.g. news text and headlines, social media posts, translated sentences, ...). --> #### Data Collection and Processing <!-- This section describes the data collection and processing process such as data selection criteria, filtering and normalization methods, tools and libraries used, etc. --> Data is collected from https://huggingface.co/datasets/xsum and processed by using build_dataset.py script in repository. #### Who are the source data producers? <!-- This section describes the people or systems who originally created the data. It should also include self-reported demographic or identity information for the source data creators if this information is available. --> People who created https://huggingface.co/datasets/xsum ## Bias, Risks, and Limitations <!-- This section is meant to convey both technical and sociotechnical limitations. --> This dataset contains the same biases, risks, and limitations as its source dataset https://huggingface.co/datasets/xsum ### Recommendations <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. --> Users should be made aware of the risks, biases and limitations of the dataset.

This is a preprocessed version of the xsum dataset, specifically designed for benchmarks in LM-Polygraph. The dataset includes two main parts: input and output, corresponding to the input and output for LM-Polygraph. The dataset is divided into training and test sets, each with respective byte counts and number of examples. The creation of this dataset aims to separate dataset creation code from benchmarking code. The source data for this dataset is from https://huggingface.co/datasets/xsum, processed using the build_dataset.py script in the repository. The dataset contains the same biases, risks, and limitations as its source dataset.

提供机构:
LM-Polygraph
搜集汇总
数据集介绍
构建方式
在自然语言处理领域,摘要生成任务对模型的事实一致性与鲁棒性提出了严苛要求。LM-Polygraph/xsum数据集正是为满足这一需求而构建的,它源自经典的XSum新闻摘要数据集,并经由LM-Polygraph项目团队精心预处理而成。具体而言,研究人员利用仓库中的build_dataset.py脚本,从原始XSum数据中提取并转换出两个核心子集:continuation与simple_instruct。每个子集均划分为训练集与测试集,其中训练集包含约20.4万条样本,测试集约1.1万条。每条样本由两个字符串字段构成——input代表经过处理的模型输入,output则对应预期的模型输出,从而为不确定性量化与基准测试提供了标准化接口。
特点
该数据集的核心特点在于其双配置设计,能够灵活适配不同的评估范式。continuation子集作为主数据集,保留了原始摘要任务的连续生成结构,适合评估模型在无额外指令下的原生性能。而simple_instruct子集则引入指令微调格式,将输入封装为简洁的提示语,使数据集能够兼容当前主流的大语言模型交互方式。此外,数据规模适中且划分清晰,训练集与测试集的比例约为18:1,既保证了模型学习的充分性,又为性能评估提供了可靠的统计基础。作为LM-Polygraph基准套件的一部分,它特别适用于研究语言模型在摘要生成中的置信度校准与错误检测。
使用方法
使用该数据集时,研究者可通过HuggingFace Datasets库直接加载,无需本地预处理。具体而言,调用load_dataset函数并指定配置名称为'continuation'或'simple_instruct',即可获取对应的训练与测试分片。每条数据中的input字段可直接作为语言模型的生成输入,output字段则作为参考摘要用于计算ROUGE等评估指标。在LM-Polygraph框架内,该数据集被设计为与不确定性估计模块无缝集成,支持对模型输出进行概率校准、熵计算及多种不确定性度量。值得注意的是,该数据集不应用于进一步的预处理或修改,其设计初衷是保持基准测试的标准化与可复现性。
背景与挑战
背景概述
LM-Polygraph/xsum数据集由LM-Polygraph团队于近年构建,旨在为大型语言模型的不确定性估计与校准提供标准化评估基准。该数据集源自经典的XSUM新闻摘要任务,原始数据由英国广播公司(BBC)的新闻文章及其对应单句摘要构成,核心研究问题聚焦于如何通过对比模型生成的延续文本与真实摘要,量化语言模型在生成任务中的置信度与可靠性。作为LM-Polygraph基准测试套件的重要组成部分,该数据集推动了不确定性量化方法在自然语言生成领域的系统性评估,为模型可解释性与安全性研究提供了关键数据支撑,对提升大语言模型在事实性生成任务中的可信度产生了显著影响。
当前挑战
当前数据集面临的核心挑战包括:首先,在领域问题层面,新闻摘要任务固有的主观性导致模型输出与真实摘要的语义等价性难以精确度量,传统似然估计方法在捕捉生成文本的多样性时存在偏差,亟需开发更鲁棒的不确定性评估指标。其次,构建过程中面临的数据预处理挑战凸显——原始XSUM数据集需通过特定脚本转换为适合LM-Polygraph框架的输入输出格式,这一转换可能引入信息损失或格式偏差,且不同配置(如continuation与simple_instruct)之间的结构差异增加了评估一致性维护的复杂性。此外,数据集的单语特性限制了其对多语言场景下不确定性估计泛化能力的验证,而训练集(204,045条)与测试集(11,334条)的规模差异也可能影响统计效力的稳定性。
常用场景
经典使用场景
在自然语言处理与生成式模型的评估领域,LM-Polygraph/xsum数据集作为经过精细预处理的文本摘要基准,被广泛用于评测大型语言模型在摘要任务中的不确定性量化与置信度校准能力。该数据集包含'continuation'与'simple_instruct'两种配置,分别对应原始续写任务和指令微调场景,为研究者提供了标准化的输入输出对,从而能够系统性地对比不同不确定性估计方法在新闻文本摘要上的表现,是衡量模型输出可靠性的重要工具。
实际应用
在实际应用中,LM-Polygraph/xsum数据集赋能了多种需要高可靠性文本生成的下游系统。例如,在新闻自动摘要工具中,利用该数据集训练的不确定性评估模块可自动标记模型可能产生事实性错误或幻觉的摘要片段,从而辅助编辑人员进行人工核查;在智能客服与报告生成场景,该数据集帮助构建置信度阈值机制,确保仅当模型对输出高度确定时才进行自动化响应,显著降低了信息误传的风险。
衍生相关工作
基于LM-Polygraph/xsum数据集,学术界衍生了一系列关于语言模型不确定性评估的经典工作。例如,LM-Polygraph框架本身即依托该数据集实现了多种不确定性度量方法的基准测试,包括熵、互信息、语义多样性等指标;后续研究进一步将其拓展至多语言场景与长文本生成任务,催生了诸如基于贝叶斯推断的校准方法以及对比解码策略等创新技术。这些工作共同推动了生成式模型从单纯追求流畅性向兼具可靠性与安全性的方向演进。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务