Baby Scale: Investigating Models Trained on Individual Children's Language Input
Organizations: Stanford University
Abstract
Modern language models (LMs) must be trained on many orders of magnitude more words of training data than human children receive before they begin to produce useful behavior. Assessing the nature and origins of this "data gap" requires benchmarking LMs on human-scale datasets to understand how linguistic knowledge emerges from children's natural training data. Using transcripts from the BabyView dataset (videos from children ages 6-36 months), we investigate (1) scaling performance at child-scale data regimes, (2) variability in model performance across datasets from different children's experiences and linguistic predictors of dataset quality, and (3) relationships between model and child language learning outcomes. LMs trained on child data show acceptable scaling for grammar tasks, but lower scaling on semantic and world knowledge tasks than models trained on synthetic data; we also observe substantial variability on data from different children. Beyond dataset size, performance is most associated with a combination of distributional and interactional linguistic features, broadly consistent with what makes high-quality input for child language development. Finally, model likelihoods for individual words correlate with children's learning of those words, suggesting that properties of child-directed input may influence both model learning and human language development. Overall, understanding what properties make language data efficient for learning can enable more powerful small-scale language models while also shedding light on human language acquisition.
Figures & tables
| Model | Condition | # of Fams | Tokens | TTR | Zorro | WordSim | COMPS | EWoK |
| GPT-2 small | Mean of fams | 1 | 142,094 | – | 55.19 2.89 | 0.049 0.007 | 50.09 0.14 | 50.18 0.15 |
| GPT-2 small | Top-1 | 1 | 725,600 | 0.026 | 58.09 3.36 | 0.042 0.025 | 50.19 0.49 | 50.07 0.36 |
| GPT-2 small | Top-2 | 2 | 727,986 | 0.025 | 59.58 1.85 | 0.064 0.022 | 50.37 0.33 | 50.12 0.13 |
| GPT-2 small | Top-5 | 5 | 727,883 | 0.026 | 56.70 2.88 | 0.062 0.008 | 50.22 0.27 | 50.32 0.05 |
| GPT-2 small | Top-10 | 10 | 735,309 | 0.027 | 57.14 2.79 | 0.048 0.020 | 50.22 0.20 | 50.13 0.21 |
| GPT-2 small | Top-20 | 20 | 734,234 | 0.026 | 56.59 2.18 | 0.044 0.015 | 50.03 0.23 | 50.31 0.29 |
| Feature | Top10 | Meth. | Spear. | Lasso | XGB | Mean rank | Mean | Mean | Mean imp. |
| Bigram mutual information | 9 | 3 | 3 | 1 | 5 | 5.56 | 0.72 | 0.153 | 0.060 |
| Caregiver POS-tagged token count | 7 | 3 | 4 | 1 | 2 | 4.29 | 0.74 | 0.016 | 0.062 |
| Child POS-tagged token count | 7 | 3 | 1 | 4 | 2 | 2.00 | 0.86 | 0.604 | 0.356 |
| Child-to-caregiver semantic pair count | 6 | 3 | 2 | 2 | 2 | 6.17 | 0.80 | 0.005 | 0.050 |
| POS bigram entropy | 6 | 3 | 1 | 1 | 4 | 3.33 | 0.56 | 0.263 | 0.069 |
| Mean KL divergence from other datasets | 5 | 3 | 3 | 1 | 1 | 3.60 | 0.71 | 0.500 | 0.053 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Family | Age (months) | Tokens | Zorro | WordSim | COMPS | EWoK |
| BV-fam-S00400001 | 31 | 725,600 | 58.09 3.36 | 0.042 0.025 | 50.19 0.49 | 50.07 0.36 |
| BV-fam-S00510002 | 11 | 569,788 | 55.95 2.74 | 0.054 0.015 | 50.05 0.42 | 50.14 0.20 |
| BV-fam-S00360001 | 8 | 257,817 | 59.53 1.83 | 0.048 0.022 | 50.07 0.15 | 50.17 0.30 |
| BV-fam-S00220001 | 5 | 242,954 | 58.02 2.19 | 0.058 0.029 | 50.12 0.34 | 50.18 0.36 |
| BV-fam-S00400002 | 28 | 181,317 | 57.95 2.19 | 0.054 0.031 | 50.07 0.35 | 50.42 0.80 |
| BV-fam-S00320002 | 7 | 171,656 | 55.98 2.31 | 0.028 0.024 | 50.25 0.32 | 50.21 0.27 |
| Family | Age (months) | Tokens | Zorro | WordSim | COMPS | EWoK |
| BV-fam-S00400001 | 31 | 725,600 | 57.54 1.98 | 0.044 0.018 | 50.10 0.16 | 50.03 0.26 |
| BV-fam-S00510002 | 11 | 569,788 | 57.65 2.52 | 0.054 0.027 | 49.97 0.20 | 49.91 0.10 |
| BV-fam-S00360001 | 8 | 257,817 | 59.29 0.95 | 0.032 0.033 | 50.13 0.29 | 49.94 0.36 |
| BV-fam-S00220001 | 5 | 242,954 | 59.43 1.08 | 0.058 0.018 | 49.74 0.45 | 50.19 0.30 |
| BV-fam-S00400002 | 28 | 181,317 | 56.25 3.03 | 0.014 0.024 | 50.07 0.32 | 50.16 0.25 |
| BV-fam-S00320002 | 7 | 171,656 | 56.49 2.74 | 0.014 0.019 | 50.42 0.27 | 50.17 0.38 |
| Family | Age (months) | Tokens | Zorro | WordSim | COMPS | EWoK |
| BV-fam-S00400001 | 31 | 725,600 | 59.10 1.58 | 0.068 0.026 | 50.22 0.27 | 50.31 0.35 |
| BV-fam-S00510002 | 11 | 569,788 | 57.13 1.69 | 0.062 0.013 | 50.17 0.20 | 50.13 0.17 |
| BV-fam-S00360001 | 8 | 257,817 | 58.26 0.85 | 0.058 0.015 | 50.18 0.29 | 50.08 0.27 |
| BV-fam-S00220001 | 5 | 242,954 | 59.88 1.82 | 0.056 0.009 | 49.93 0.23 | 50.10 0.17 |
| BV-fam-S00400002 | 28 | 181,317 | 57.57 1.58 | 0.054 0.015 | 49.82 0.28 | 49.80 0.27 |
| BV-fam-S00320002 | 7 | 171,656 | 56.68 0.94 | 0.026 0.011 | 50.24 0.34 | 50.14 0.35 |
| Family | Age (months) | Tokens | Zorro | WordSim | COMPS | EWoK |
| BV-fam-S00400001 | 31 | 725,600 | 54.37 1.63 | 0.072 0.016 | 50.10 0.10 | 50.00 0.11 |
| BV-fam-S00510002 | 11 | 569,788 | 55.65 0.85 | 0.068 0.047 | 50.12 0.33 | 49.96 0.20 |
| BV-fam-S00360001 | 8 | 257,817 | 57.97 1.34 | 0.066 0.021 | 50.22 0.23 | 49.88 0.39 |
| BV-fam-S00220001 | 5 | 242,954 | 57.24 1.12 | 0.074 0.018 | 49.98 0.36 | 50.17 0.34 |
| BV-fam-S00400002 | 28 | 181,317 | 59.01 1.77 | 0.058 0.016 | 49.77 0.23 | 50.06 0.28 |
| BV-fam-S00320002 | 7 | 171,656 | 56.21 1.77 | 0.056 0.019 | 50.07 0.27 | 49.99 0.26 |
| Model | Target | NZ feat. | Lasso | Lasso RMSE | XGB | XGB RMSE | |
| GPT-2 Mini | Zorro | 0.0488 | 21 | 0.657 | 1.784 | 0.395 | 2.369 |
| GPT-2 Mini | WordSim | 0.0026 | 13 | -0.208 | 0.029 | -0.166 | 0.029 |
| GPT-2 Mini | COMPS | 0.0668 | 0 | -0.144 | 0.166 | -1.076 | 0.223 |
| GPT-2 Small | Zorro | 0.5230 | 6 | 0.289 | 2.974 | 0.172 | 3.208 |
| GPT-2 Small | WordSim | 0.0173 | 1 | -1.047 | 0.031 | -1.943 | 0.037 |
| GPT-2 Small | COMPS | 0.0713 | 3 | -1.479 | 0.257 | -0.601 | 0.207 |
| Feature name | Brief description |
|---|---|
| Lexical / Distributional / Scale | |
| avg_conversation_len_tokens | This feature measures the average number of tokens per conversation, capturing how much language is typically contained in a single interaction. |
| avg_sentence_len_tokens | This feature measures the average number of tokens per sentence, providing a simple proxy for sentence-level length and complexity. |
| avg_utterance_len_tokens | This feature measures the average number of tokens per utterance, summarizing how long individual turns tend to be. |
| bigram_entropy | This feature measures the entropy of the bigram distribution, so higher values indicate greater diversity in adjacent token sequences. |
| bigram_mutual_information | This feature measures the average pointwise mutual information of adjacent token pairs, so higher values indicate that neighboring words are more strongly associated than expected by chance. |
| Feature | Top10 | Meth. | Spear. | Lasso | XGB | Mean rank | Mean | Mean | Mean imp. |
|---|---|---|---|---|---|---|---|---|---|
| Lexical / Distributional / Scale | |||||||||
| bigram_mutual_information | 9 | 3 | 3 | 1 | 5 | 5.56 | 0.72 | 0.153 | 0.060 |
| hapax_ratio | 5 | 3 | 1 | 1 | 3 | 6.60 | 0.48 | 0.264 | 0.067 |
| trigram_entropy | 5 | 2 | 4 | 0 | 1 | 5.00 | 0.74 | 0.000 | 0.075 |
| num_conversations | 5 | 1 | 0 | 5 | 0 | 2.00 | 0.00 | 0.178 | 0.000 |
| avg_conversation_len_tokens | 4 | 3 | 2 | 1 | 1 | 8.25 | 0.57 | 0.005 | 0.036 |