arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2607.04563cs.CL
收藏

文本的保真度-多样性度量

Fidelity-Diversity Metrics for Text

Department of Computer Science, Cornell University(康奈尔大学计算机科学系);Statistics and Data Science Center, Massachusetts Institute of Technology(麻省理工学院统计与数据科学中心);Department of Statistics, Cornell University(康奈尔大学统计学系)

Amanda Wang, Tudor Manole, Florentina Bunea, John Thickstun

AI总结:

随着语言建模技术成熟,数据质量评估对训练模型很关键。基于生成模型精度和召回率的工作,开发一对度量,用最优传输散度函数量化文本的保真度和多样性,能区分候选文本缺陷。

摘要:

随着语言建模技术的成熟,越来越多的研究关注于用于训练这些模型的数据集的组成和管理。例如,从业者通常寻求用额外的文本扩充高质量数据集,以提高基于该数据训练的模型的性能。然而,关于数据扩充的明智决策需要对数据质量进行更细致的评估。我们基于测量生成模型的精度和召回率的工作,开发了一对度量,用于量化:(1) 保真度,即捕获候选文本与参考数据的相似程度;(2) 多样性,即捕获其覆盖参考数据集模式的程度。我们的度量基于离散文本摘要之间的最优传输散度泛函。在对M2D2文本数据集的实验中,我们表明这些度量能够区分有缺陷的候选文本中缺乏保真度和缺乏多样性的情况。在进一步的实验中,我们的度量检测到合成GSM8K风格数学数据集中的多样性缺陷,这与在该合成数据上微调的语言模型的下游准确性下降相关。

英文摘要:

As language modeling technology matures, there is an increasing research focus on the composition and curation of datasets used to train these models. For instance, practitioners commonly seek to augment high-quality datasets with additional text to enhance the performance of models trained on that data. However, informed decisions about data augmentation require more nuanced assessments about data quality. We build on work measuring the precision and recall of generative models to develop a pair of metrics that quantify (1) fidelity, capturing how closely candidate text resembles reference data, and (2) diversity, capturing how well it covers the modes of the reference dataset. Our metrics are based on optimal transport divergence functionals between discrete text summaries. In experiments on M2D2 text datasets, we show that these metrics are able to disentangle a lack of fidelity from a lack of diversity in deficient candidate text. In further experiments, our metrics detect diversity deficits in synthetic GSM8K-style math datasets, which correlate with degradations in downstream accuracy of language models finetuned on this synthetic data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑