Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. CHAMELEON: A Flexible Data-mixing Framework for Language Model Pretraining and Finetuning
 
conference paper

CHAMELEON: A Flexible Data-mixing Framework for Language Model Pretraining and Finetuning

Xie, Wanyun  
•
Tonin, Francesco
•
Cevher, Volkan  orcid-logo
July 2025
Proceedings of the 42 nd International Conference on Machine Learning
Forty-Second International Conference on Machine Learning

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 introduce a flexible and efficient data mixing framework, CHAMELEON, that employs leverage scores to quantify domain importance within a learned embedding space. We first construct a domain affinity matrix over domain embeddings. The induced leverage scores determine a mixture that upweights domains sharing common representations in embedding space. This formulation allows direct transfer to new data by computing the new domain embeddings. In experiments, we demonstrate improvements over three key scenarios: (i) our computed weights improve performance on pretraining domains with a fraction of the compute of existing methods; (ii) CHAMELEON can adapt to data changes without proxy retraining, boosting few-shot reasoning accuracies when transferred to new data; (iii) our method enables efficient domain reweighting in finetuning, consistently improving test perplexity on all finetuning domains over uniform mixture. Our code is available at https:// github.com/LIONS-EPFL/Chameleon.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

11654_Chameleon_A_Flexible_Dat.pdf

Type

Main Document

Version

Accepted version

Access type

openaccess

License Condition

N/A

Size

1.16 MB

Format

Adobe PDF

Checksum (MD5)

c2023d85a0cfae872fb84100de27db6d

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés