Search papers, labs, and topics across Lattice.
This paper introduces UltraX, a novel framework for refining large-scale pre-training data through adaptive programmatic editing, addressing the limitations of existing refinement methodologies. By incorporating insertion alongside deletion and modification, UltraX enables fine-grained instance-level editing, enhancing the quality and reliability of training data. Experimental results indicate that UltraX not only achieves superior performance across various corpora but also demonstrates improved data efficiency by requiring fewer training tokens compared to traditional baselines.
UltraX achieves the highest average performance across datasets while using fewer training tokens, redefining efficiency in data refinement for LLMs.
As available training data approaches its physical limit, gains from Scaling Laws have begun to diminish. Consequently, improving Large Language Models (LLMs) now depends less on data expansion and more on higher-quality data utilization. However, in the context of large-scale corpora, existing refinement methodologies face significant limitations in quality, efficiency, and reliability: Rule-based approaches are constrained by fixed heuristics and struggle with instance-level variations; LLM-based approaches improve quality but fail to meet the efficiency and reliability requirements of large-scale data processing. To address these challenges, we propose UltraX, a function-calling refinement framework for large-scale pre-training data that completes the editing function space by introducing insertion in addition to deletion and modification, enabling fine-grained instance-level editing. Specifically, UltraX builds a reliable program-supervision generation pipeline. In this pipeline, dataset-adaptive prompt optimization first guides an expert LLM to produce high-quality end-to-end refined texts, and Line Alignment Mapping and Dynamic Context Replacement then convert original-refined text pairs into structured program supervision. Meanwhile, UltraX improves supervision quality and stabilizes the training distribution with low-confidence example filtering and ratio-controlled sampling by operation combination. During inference and execution, it normalizes and validates model outputs through sliding-window prediction, global operation aggregation, and systematic post-processing, improving the stability and reliability of large-scale execution. Experiments show that UltraX achieves the highest average performance across all corpora and also matches or surpasses baselines with fewer training tokens, demonstrating stronger data efficiency and refinement reliability.
Tsinghua AI