English
Related papers

Related papers: DoReMi: Optimizing Data Mixtures Speeds Up Languag…

200 papers

Pre-training large language models (LLMs) necessitates enormous diverse textual corpora, making effective data selection a key challenge for balancing computational resources and model performance. Current methodologies primarily emphasize…

Computation and Language · Computer Science 2025-04-22 Xiaoxuan Zhu , Zhouhong Gu , Baiqian Wu , Suhang Zheng , Tao Wang , Tianyu Li , Hongwei Feng , Yanghua Xiao

The coverage and composition of the pretraining data significantly impacts the generalization ability of Large Language Models (LLMs). Despite its importance, recent LLMs still rely on heuristics and trial and error to increase or reduce…

Machine Learning · Computer Science 2024-02-06 Simin Fan , Matteo Pagliardini , Martin Jaggi

The performance of large language models (LLMs) across diverse downstream applications is fundamentally governed by the quality and composition of their pretraining corpora. Existing domain reweighting algorithms primarily optimize data…

Machine Learning · Computer Science 2025-05-28 Simin Fan , Maria Ios Glarou , Martin Jaggi

Large language models have demonstrated remarkable capabilities across various tasks, primarily attributed to the utilization of diversely sourced data. However, the impact of pretraining data composition on model performance remains poorly…

Machine Learning · Computer Science 2025-01-28 Ce Ge , Zhijian Ma , Daoyuan Chen , Yaliang Li , Bolin Ding

The data mixture for large language model pre-training significantly impacts performance, yet how to determine an effective mixture remains unclear. We propose RegMix to automatically identify a high-performing data mixture by formulating…

Computation and Language · Computer Science 2025-01-24 Qian Liu , Xiaosen Zheng , Niklas Muennighoff , Guangtao Zeng , Longxu Dou , Tianyu Pang , Jing Jiang , Min Lin

Increasingly large imitation learning datasets are being collected with the goal of training foundation models for robotics. However, despite the fact that data selection has been of utmost importance in vision and natural language…

Robotics · Computer Science 2025-02-24 Joey Hejna , Chethan Bhateja , Yichen Jiang , Karl Pertsch , Dorsa Sadigh

Multi-domain fine-tuning of large language models requires improving performance on target domains while preserving performance on constrained domains, such as general knowledge, instruction following, or safety evaluations. Existing data…

Machine Learning · Computer Science 2026-05-12 Eleonora Gualdoni , Sonia Laguna , Louis Bethune , Joao Monteiro , Pierre Ablin , Marco Cuturi

Optimizing data mixtures is essential for unlocking the full potential of large language models (LLMs), yet identifying the optimal composition remains computationally prohibitive due to reliance on heuristic trials or expensive proxy…

Machine Learning · Computer Science 2026-01-27 Jiapeng Wang , Changxin Tian , Kunlong Chen , Ziqi Liu , Jiaxin Mao , Wayne Xin Zhao , Zhiqiang Zhang , Jun Zhou

Domain reweighting is an emerging research area aimed at adjusting the relative weights of different data sources to improve the effectiveness and efficiency of LLM pre-training. We show that data mixtures that perform well at smaller…

Machine Learning · Computer Science 2025-10-03 Feiyang Kang , Yifan Sun , Bingbing Wen , Si Chen , Dawn Song , Rafid Mahmood , Ruoxi Jia

Data mixing strategy is essential for large language model (LLM) training. Empirical evidence shows that inappropriate strategies can significantly reduce generalization. Although recent methods have improved empirical performance, several…

Computation and Language · Computer Science 2026-04-10 Yuanjian Xu , Tianze Sun , Changwei Xu , XinLong Zhao , Jianing Hao , Ran Chen , Yang Liu , Ruijie Xu , Stephen Chen , Guang Zhang

Large Language Models (LLMs) have shown remarkable ability to generalize effectively across numerous industry domains while executing a range of tasks. Many of these competencies are obtained from the data utilized during the pre-training…

For most languages of the world, language model pre-training operates in a data-constrained regime where models must repeat their training data many times, degrading generalization. Two remedies exist: aggressive hyperparameter tuning such…

Machine Learning · Computer Science 2026-05-14 Paul Jeha , Anastasiia Sedova , Louis Béthune , Skyler Seto , Jes Frellsen , Pierre Ablin , Natalie Schluter

Large language models (LLMs) deliver impressive results but face challenges from increasing model sizes and computational costs. Structured pruning reduces model size and speeds up inference but often causes uneven degradation across…

Computation and Language · Computer Science 2025-05-28 Hexuan Deng , Wenxiang Jiao , Xuebo Liu , Jing Li , Min Zhang , Zhaopeng Tu

Large language models (LLMs) rely on pretraining on massive and heterogeneous corpora, where training data composition has a decisive impact on training efficiency and downstream generalization under realistic compute and data budget…

Computation and Language · Computer Science 2026-04-21 Zhuo Chen , Yuxuan Miao , Supryadi , Deyi Xiong

Training large language models with data collected from various domains can improve their performance on downstream tasks. However, given a fixed training budget, the sampling proportions of these different domains significantly impact the…

Computation and Language · Computer Science 2025-05-29 Yajiao Liu , Congliang Chen , Junchi Yang , Ruoyu Sun

Reasoning has substantially improved the performance of large language models (LLMs) on complicated tasks. Central to the current reasoning studies, Process Reward Models (PRMs) offer a fine-grained evaluation of intermediate reasoning…

Machine Learning · Computer Science 2025-11-05 Qi Cao , Ruiyi Wang , Ruiyi Zhang , Sai Ashish Somayajula , Pengtao Xie

Domain-Adaptive Pre-training (DAP) has recently gained attention for its effectiveness in fine-tuning pre-trained models. Building on this, continual DAP has been explored to develop pre-trained models capable of incrementally incorporating…

Computation and Language · Computer Science 2025-07-04 Dohoon Kim , Donghun Kang , Taesup Moon

Large language models (LLMs) encode a vast amount of semantic knowledge and possess remarkable understanding and reasoning capabilities. Previous work has explored how to ground LLMs in robotic tasks to generate feasible and executable…

Robotics · Computer Science 2024-09-17 Yanjiang Guo , Yen-Jen Wang , Lihan Zha , Jianyu Chen

Adapting large language models (LLMs) to low-resource languages (LRLs) is constrained by the scarcity of task data and computational resources. Although Proxy Tuning offers a logit-level strategy for introducing scaling effects, it often…

Computation and Language · Computer Science 2026-04-21 Chen Zhang , Jiuheng Lin , Zhiyuan Liao , Yansong Feng

Pretraining datasets for large language models (LLMs) have grown to trillions of tokens composed of large amounts of CommonCrawl (CC) web scrape along with smaller, domain-specific datasets. It is expensive to understand the impact of these…

Machine Learning · Computer Science 2024-06-06 Cody Blakeney , Mansheej Paul , Brett W. Larsen , Sean Owen , Jonathan Frankle
‹ Prev 1 2 3 10 Next ›