CLIMB: Clustering-gebaseerde Iteratieve Data Mengsel Bootstrapping voor Vooraf Trainen van TaalmodellenCLIMB: CLustering-based Iterative Data Mixture Bootstrapping for
Language Model Pre-training
Pre-trainingsdatasets worden doorgaans verzameld uit webinhoud en missen inherente domeinindelingen. Bijvoorbeeld, veelgebruikte datasets zoals Common Crawl bevatten geen expliciete domeinlabels, terwijl het handmatig samenstellen van gelabelde datasets zoals The Pile arbeidsintensief is. Als gevolg hiervan blijft het identificeren van een optimale pre-trainingsdatamix een uitdagend probleem, ondanks de aanzienlijke voordelen voor de pre-trainingsprestaties. Om deze uitdagingen aan te pakken, stellen wij CLustering-based Iterative Data Mixture Bootstrapping (CLIMB) voor, een geautomatiseerd framework dat datamixen ontdekt, evalueert en verfijnt in een pre-trainingsomgeving. Specifiek embedt en clustert CLIMB grootschalige datasets in een semantische ruimte en zoekt vervolgens iteratief naar optimale mixen met behulp van een kleiner proxy-model en een voorspeller. Wanneer continu getraind op 400B tokens met deze mix, overtreft ons 1B-model de state-of-the-art Llama-3.2-1B met 2,0%. Bovendien observeren we dat optimalisatie voor een specifiek domein (bijvoorbeeld Sociale Wetenschappen) een verbetering van 5% oplevert ten opzichte van willekeurige steekproeven. Ten slotte introduceren we ClimbLab, een gefilterd corpus van 1,2 biljoen tokens met 20 clusters als onderzoeksspeelveld, en ClimbMix, een compact maar krachtige dataset van 400 miljard tokens die is ontworpen voor efficiënte pre-training en superieure prestaties levert binnen een gelijk tokenbudget. We analyseren de uiteindelijke datamix en verduidelijken de kenmerken van een optimale datamix. Onze data is beschikbaar op: https://research.nvidia.com/labs/lpr/climb/