English
Related papers

Related papers: DoGE: Domain Reweighting with Generalization Estim…

200 papers

Pre-training large language models has become a standard in the natural language processing community. Such models are pre-trained on generic data (e.g. BookCorpus and English Wikipedia) and often fine-tuned on tasks in the same domain.…

Computation and Language · Computer Science 2020-10-05 Kristjan Arumae , Qing Sun , Parminder Bhatia

Existing pretraining data mixing methods for large language models (LLMs) typically follow a domain-wise methodology, a top-down process that first determines domain weights and then performs uniform data sampling across each domain.…

Computation and Language · Computer Science 2025-03-04 Xiangyu Xi , Deyang Kong , Jian Yang , Jiawei Yang , Zhengyu Chen , Wei Wang , Jingang Wang , Xunliang Cai , Shikun Zhang , Wei Ye

In the training of large deep neural networks, there is a need for vast amounts of training data. To meet this need, data is collected from multiple domains, such as Wikipedia and GitHub. These domains are heterogeneous in both data quality…

Machine Learning · Computer Science 2025-11-11 Mahdi Salmani , Pratik Worah , Meisam Razaviyayn , Vahab Mirrokni

The effectiveness of large language models (LLMs) is often hindered by duplicated data in their extensive pre-training datasets. Current approaches primarily focus on detecting and removing duplicates, which risks the loss of valuable…

Computation and Language · Computer Science 2024-07-10 Nan He , Weichen Xiong , Hanwen Liu , Yi Liao , Lei Ding , Kai Zhang , Guohua Tang , Xiao Han , Wei Yang

In the rapidly advancing arena of large language models (LLMs), a key challenge is to enhance their capabilities amid a looming shortage of high-quality training data. Our study starts from an empirical strategy for the light continual…

Machine Learning · Computer Science 2024-03-04 Xuxi Chen , Zhendong Wang , Daouda Sow , Junjie Yang , Tianlong Chen , Yingbin Liang , Mingyuan Zhou , Zhangyang Wang

Large Language Models (LLMs) have achieved remarkable performance on a wide range of specialized tasks, exhibiting strong problem-solving capabilities. However, training these models is prohibitively expensive, and they often lack…

Machine Learning · Computer Science 2026-04-01 Eros Fanì , Oğuzhan Ersoy

Large language models (LLMs) have shown remarkable capability in numerous tasks and applications. However, fine-tuning LLMs using high-quality datasets under external supervision remains prohibitively expensive. In response, LLM…

Computation and Language · Computer Science 2024-12-13 Chunyang Jiang , Chi-min Chan , Wei Xue , Qifeng Liu , Yike Guo

The impressive capabilities of large language models (LLMs) have sparked debate over whether these models genuinely generalize to unseen tasks or predominantly rely on memorizing vast amounts of pretraining data. To explore this issue, we…

Computation and Language · Computer Science 2025-03-04 Xinyi Wang , Antonis Antoniades , Yanai Elazar , Alfonso Amayuelas , Alon Albalak , Kexun Zhang , William Yang Wang

Distribution shift presents a significant challenge in machine learning, where models often underperform during the test stage when faced with a different distribution than the one they were trained on. This paper focuses on domain shifts,…

Machine Learning · Computer Science 2024-03-19 Huaxiu Yao , Xinyu Yang , Xinyi Pan , Shengchao Liu , Pang Wei Koh , Chelsea Finn

Machine learning algorithms minimizing the average training loss usually suffer from poor generalization performance due to the greedy exploitation of correlations among the training data, which are not stable under distributional shifts.…

Machine Learning · Computer Science 2022-07-29 Xi Leng , Xiaoying Tang , Yatao Bian

Recent advances in prompt optimization have notably enhanced the performance of pre-trained language models (PLMs) on downstream tasks. However, the potential of optimized prompts on domain generalization has been under-explored. To explore…

Computation and Language · Computer Science 2024-10-22 Chengzhengxu Li , Xiaoming Liu , Zhaohan Zhang , Yichen Wang , Chen Liu , Yu Lan , Chao Shen

Leveraging datasets available to learn a model with high generalization ability to unseen domains is important for computer vision, especially when the unseen domain's annotated data are unavailable. We study a novel and practical problem…

Computer Vision and Pattern Recognition · Computer Science 2021-04-09 Yang Shu , Zhangjie Cao , Chenyu Wang , Jianmin Wang , Mingsheng Long

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…

The growing interest in Large Language Models (LLMs) for specialized applications has revealed a significant challenge: when tailored to specific domains, LLMs tend to experience catastrophic forgetting, compromising their general…

Computation and Language · Computer Science 2024-03-06 Rui Wang , Fei Mi , Yi Chen , Boyang Xue , Hongru Wang , Qi Zhu , Kam-Fai Wong , Ruifeng Xu

Large language models (LLMs) are commonly trained on multi-domain datasets, where domain sampling strategies significantly impact model performance due to varying domain importance across downstream tasks. Existing approaches for optimizing…

Computation and Language · Computer Science 2025-08-25 Weijie Shi , Jipeng Zhang , Yaguang Wu , Jingzhi Fang , Ruiyuan Zhang , Jiajie Xu , Jia Zhu , Hao Chen , Yao Zhao , Sirui Han , Xiaofang Zhou

Training data mixtures greatly impact the generalization performance of large language models. Existing domain reweighting methods often rely on costly weight computations and require retraining when new data is introduced. To this end, we…

Machine Learning · Computer Science 2025-06-02 Wanyun Xie , Francesco Tonin , Volkan Cevher

Prompt tuning, or the conditioning of a frozen pretrained language model (PLM) with soft prompts learned from data, has demonstrated impressive performance on a wide range of NLP tasks. However, prompt tuning requires a large training…

Computation and Language · Computer Science 2022-10-24 Xu Guo , Boyang Li , Han Yu

Domain generalization (DG) aims to help models trained on a set of source domains generalize better on unseen target domains. The performances of current DG methods largely rely on sufficient labeled data, which are usually costly or…

Computer Vision and Pattern Recognition · Computer Science 2022-04-13 Xingxuan Zhang , Linjun Zhou , Renzhe Xu , Peng Cui , Zheyan Shen , Haoxin Liu

Instead of directly distilling a language model, this study addresses the problem of aligning a base model with a target model in distribution by designing the domain mixture of training data for pretraining or continued pretraining as a…

Computation and Language · Computer Science 2026-03-18 Ryo Kishino , Riku Shiomi , Hiroaki Yamagiwa , Momose Oyama , Hidetoshi Shimodaira

Large Language Models (LLMs) can propose rules in natural language, sidestepping the need for a predefined predicate space in traditional rule learning. Yet many LLM-based approaches ignore interactions among rules, and the opportunity to…

Artificial Intelligence · Computer Science 2026-02-16 Yang Yang , Hua XU , Zhangyi Hu , Yutao Yue