arXiv:2411.04257cs.LG2024-11被引 5

用轻量布隆过滤器替代传统索引,实现超大规模文档去重的高效低存方案。

LSHBloom: Memory-efficient, Extreme-scale Document Deduplication

  • 以布隆过滤器替代哈希索引,降低内存与存储开销。
  • 在peS2o数据集上比MinhashLSH快12倍,存储减少18倍。
  • 适合需要高精度去重的大规模文本训练数据处理场景。

当前大语言模型(LLM)训练需整合来自网络、学术和出版等多源的互联网规模文本数据。通过去重(检测并移除重复内容)预处理数据集是构建高质量训练数据的关键。然而,现有文档级去重方法要么准确率低,要么运行时间和内存消耗极高。本文提出LSHBloom,作为MinhashLSH的扩展,用轻量级布隆过滤器替代昂贵的LSHIndex。实验表明,LSHBloom在peS2o数据集上达到与MinhashLSH相当的去重性能,误报率几乎为零,同时运行速度提升12倍,磁盘占用减少18倍。基于外推分析,该优势在数十亿文档规模下依然成立。LSHBloom使高精度去重可扩展至互联网规模文本数据,让从业者能在以往仅能使用启发式方法的规模上实现高质量去重。

原文摘要 · Abstract (English)

Contemporary large language model (LLM) training pipelines require the assembly of internet-scale databases full of text data from a variety of sources (e.g., web, academic, and publishers). Preprocessing these datasets via deduplication -- detecting and eliminating additional instances of the same content -- is a major focus for assembling and curating training datasets for LLMs. Unrestrained, duplicates in the training dataset increase training costs and lead to undesirable properties such as memorization in trained models or cheating on evaluation. Unfortunately, contemporary approaches to document-level deduplication are either unreliable at accurately identifying duplicate documents or extremely expensive in terms of both runtime and memory. We propose LSHBloom, an extension to MinhashLSH, which replaces the expensive LSHIndex with lightweight Bloom filters. LSHBloom demonstrates the same state-of-the-art deduplication performance as MinhashLSH, with only a marginal increase in false positives (near zero in our experiments), while boasting competitive runtime (12$\times$ faster than MinhashLSH on peS2o) and, crucially, using 18$\times$ less disk space than MinhashLSH (as measured on peS2o). Based on extrapolation, we show that this advantage in space and runtime remains even at the extreme scale of several billion documents. LSHBloom allows practitioners to access the deduplication quality of MinHashLSH at scales that are normally only tractable for less sophisticated, heuristic solutions. As a result, LSHBloom promises to enable scaling high-quality document deduplication to internet-scale text datasets.

文档去重大模型训练内存优化布隆过滤器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。