Technische Samenvatting: Kunnen vrijgegeven LLM-vocabulaire structuren ondersteunen voor schatting op token-niveau van verborgen corpora?
1. Probleemformulering
De samenstelling van pretraining-corpora is fundamenteel voor het begrijpen van de capaciteiten van Large Language Models (LLM's), maar deze informatie blijft vaak opaak, zelfs wanneer de modelgewichten worden vrijgegeven. Hoewel vrijgegeven tokenizer-vocabularia (meestal getraind met Byte-Pair Encoding, of BPE) vaak publiek zijn, worden ze zelang niet vergezeld door de exacte statistieken van de trainingsdata.
Het kernprobleel dat wordt aangepakt is corpusratio-schatting op token-niveau. Gegeven een vrijgegeven doel-tokenizer-vocabulaire V∗ getraind op een verborgen corpus C∗, is het doel om de corpusratio ri∗ voor elke token vi in de vocabulaire te schatten. In tegenstelling tot eerder werk dat grove categorie-mengsels afleidt (bijv. taal- of domeinproporties) of specifieke tokengroepen traceert, streeft dit artikel ernaar om de ratio voor arbitraire individuele tokens te schatten.
De auteurs gaan ervan uit dat toegang beschikbaar is tot één of meer bekende corpora D met observeerbare token-ratio's. De uitdaging is om een estimator f^ te leren die token-ID's naar ratio's in het verborgen corpus mapt door gebruik te maken van de relatie tussen token-ID's en ratio's die in de bekende corpora zijn waargenomen.
2. Kernobservatie: Overdraagbaarheid van ID-ratio-distributies
Voordat ze een oplossing voorstellen, onderzoeken de auteurs of de relatie tussen token-ID's en corpus-ratio's overdraagbaar is naar verschillende corpora.
- Methode: Ze trainden BPE-tokenizers op verschillende talen (Engels, Frans, Japans, Chinees) en domeinen (Web, Wikipedia, Code, Wiskunde) en plotten de distributies van token-ID's versus hun corpus-ratio's in een log-log-ruimte.
- Bevinding: Ondanks significante verschillen in taalstructuur of domeininhoud, delen de token-ID–ratio-distributies een stabiele globale vorm.
- Kwantificering: Met behulp van een directionele transfer-gelijkenissscore gebaseerd op de Kullback–Leibler (KL) divergentie, toonden ze aan dat deze distributies breed overdraagbaar zijn. Zo zijn Engels en Frans bijvoorbeeld bijna uitwisselbaar, terwijl zelfs structureel verschillende paren zoals Code en Wikipedia substantiële gelijkenis behouden.
- Implicatie: Deze stabiliteit motiveert het overdragen van de distributieve structuur van bekende corpora naar een doel-tokenizer getraind op verborgen data.
3. Methodologie: Quantile-Guided Density Estimation (QGDE)
Om deze overdraagbaarheid uit te buiten, stellen de auteurs Quantile-Guided Density Estimation (QGDE) voor. De methode bestaat uit drie hoofdfasen:
A. Het fitten van ID-ratio-trends met kwantielen
In plaats van een enkele mediaan-curve te fitten (zoals gedaan in eerder werk zoals PoCTrace), modelleert QGDE de gedeelde ID–ratio-relatie met behulp van een familie van kwantiel-trends.
- Log-Log Transformatie: Volgens de wet van Zipf worden token-ID's (t) en ratio's (r) gemodelleerd in een log-log-ruimte, waar de relatie een lineaire trend benadert.
- Kwantielregressie: Voor een set kwantiel-niveaus τ fitte de methode log-lineaire curven qτ(x)=aτ+bτx. Dit legt niet alleen de centrale tendens vast, maar ook de volledige verticale spreiding van plausibele ratio's voor een gegeven token-ID.
- Genereren van kandidaten: Voor een doel-token met ID ti levert elke kwantiel-trend een kandidaat log-ratio schatting zi,τ op.
B. Selectie van Kwantiel-ankers
Om redundantie te voorkomen en dekking te garanderen, selecteert de methode een kleine set representatieve kwantiel-ankers TK∗.
- Optimalisatie: Ankers worden geselecteerd door de Quantile Anchor Coverage (QAC) te maximaliseren, die telt hoeveel bekende ID–ratio punten binnen een verticale bandbreedte van de geselecteerde trends vallen.
- Resultaat: Dit proces bevoordeelt ankers waarvan de trends gezamenlijk door goed ondersteunde regio's van de distributie lopen, in plaats van uniform verspreid te zijn of te focussen op ijle gebieden. Empirische analyse laat zien dat de dekking verzadigt naarmate het aantal ankers (K) toeneemt (bijv. verminderde meeropbrengst na K=14).
C. Lokale Dichtheidsgewogen Schatting
De laatste stap zet de meerdere kandidaat-schattingen om in een enkele punt-schatting met behulp van lokale dichtheidsgewichten.
- Mechanisme: Voor een doel-token-ID identificeert de methode nabijgelegen bekende ID–ratio punten binnen een lokaal venster.
- Weging: Het wijst zachte gewichten toe aan elke kandidaat-schatting op basis van hoe sterk de omliggende punten die specifieke kwantiel-trend ondersteunen, met behulp van Gaussian kernel density estimation.
- Eindschatting: De geschatte ratio is het dichtheid-gewogen gemiddelde van de kandidaat-schattingen. Dit transformeert het "bereik-stijl" signaal van eerder werk naar een fijnmazige schatting op token-niveau.
4. Experimentele Resultaten
De auteurs evalueerden QGDE in zowel gecontroleerde settings (met behulp van mC4, OSCAR, FineWeb, etc.) als in een realistische setting (met behulp van de vrijgegeven SmolLM tokenizer).
Gecontroleerde Settings
- Schatting op token-niveau: QGDE presteerde aanzienlijk beter dan twee baselines: directe ID-ratio-transfer (het kopiëren van bron-ratio's op basis van positie) en PoCTrace (enkele mediaan-trend).
- Prestaties: QGDE bereikte Mean Relative Errors (MRE) zo laag als 3,00% voor token-niveau schatting in gemengde bron-settings.
- Ablatie: Het verhogen van het aantal kwantiel-ankers (K) van 3 naar 14 verminderde de fouten scherp, vooral in domein-settings, waarna de winst verzadigde.
- Bron-mengsels: Het gebruik van gemengde bekende corpora leverde over het algemeen betere resultaten op dan single-source settings, hoewel de exacte ratio van het mengsel er minder toe deed dan de diversiteit van de gedekte componenten.
Aggregatie naar Categorie-niveau
- Mengsel-schatting: Token-niveau schattingen werden geaggregeerd om taal- of domeinproporties te schatten.
- Vergelijking: QGDE presteerde aanzienlijk beter dan de Data Mixture Inference (DMI) baseline.
- Foutreductie: In taal-settings daalde de MRE van
9,09% (DMI) naar 3,08% (QGDE). In domein-settings daalde het van ~15,14% naar **5,3%**.
- Observatie: Hoewel het verhogen van K de nauwkeurigheid op token-niveau verbeterde, waren de winsten op categorie-niveau minder uitgesproken vanwege het gladstrijkende effect van aggregatie.
Realistische Setting (SmolLM)
- Validatie: De methode werd getest op de SmolLM tokenizer, waarbij de grondwaarheid van de trainingsdata-proporties bekend waren (FineWeb-edu, Cosmopedia, Python-edu).
- Resultaten: QGDE behaalde de laagste fout voor zowel de token-niveau (5,72–5,78% MRE) als de categorie-niveau (5,93% MRE) schatting, waarmee het zowel directe transfer als PoCTrace versloeg. Dit bevestigde de effectiviteit van de methode, zelfs wanneer de exacte component-corpora niet direct als trainingsbronnen gebruikt konden worden.
5. Bijdragen en Betekenis
Het artikel levert drie primaire bijdragen:
- Ontdekking van Overdraagbaarheid: Het toont aan dat token-ID–ratio-distributies overdraagbaar zijn tussen BPE-vocabularia getraind op verschillende talen en domeinen, wat een bruikbaar signaal biedt voor het schatten van verborgen corpus-ratio's.
- Methodologische Innovatie: Het introduceert QGDE, een algemene estimator op token-niveau die gebruik maakt van meerdere kwantiel-trends en lokale dichtheidsgewichten om de overdraagbare distributie te benaderen, waarmee het verder gaat dan grove mengsel-inferentie of specifieke token-tracing.
- Empirische Validatie: Het toont aan dat QGDE een hoge nauwkeurigheid bereikt (zo laag als 3,00% MRE) in zowel gecontroleerde als realistische settings, waarbij het bestaande baselines overtreft.
Betekenis:
De auteurs stellen dat vrijgegeven tokenizer-vocabularia een nuttig signaal bieden voor fijnmazige corpus-schatting. Deze capaciteit gaat verder dan de grove samenstelling-inferentie die mogelijk is met huidige methoden, en biedt een pad om de databronnen van vrijgegeven LLM's te auditeren en te interpreteren, zelfs wanneer de trainings-corpora verborgen blijven. Het werk suggereert dat de "vingerafdruk" van de corpus-samenstelling is gecodeerd in de structuur van de tokenizer-vocabulaire op een manier die gedecodeerd kan worden met voldoende statistische modellering.
Beperkingen:
De auteurs merken op dat het gebrek aan grondwaarheid voor vrijgegeven LLM's (bijv. ChatGPT, Qwen, DeepSeek) directe evaluatie op deze modellen beperkt. Validatie rust momenteel op gecontroleerde experimenten en het zeldzame geval van SmolLM waarbij de trainingsdata publiek zijn. Bredere validatie wacht op de release van meer trainings-corpora.