CoolFace
Modelpublic

babylm-anon/TICL

sourceHugging Facecc-by-4.0updated 1y agoView on Hugging Face
0likes2.8kdownloads
Model Card

Model Card for TICL

A RoBERTa model pre-trained on a dataset of 10M words using (Training Data) Influence-driven Curriculum Learning.

Model Details

See our paper at REDACTED for details on our method.

Model Description

This is a model submitted to the strict-small track of the 2025 BabyLM challenge.

  • Developed by: REDACTED
  • Funded by [optional]: REDACTED
  • Model type: Language model (Masked)
  • Language(s) (NLP): eng
  • License: CC-By-4.0

Model Sources

<!-- Provide the basic links for the model. -->

Uses

This model was trained to demonstrate the effectiveness of a novel curriculum learning method over training in random order.

Training Details

Training Data

We utilize this dataset built from the following existing ones:

Data mix

WordsDocuments
C1: Child Directed Speech199999920.00%36053333.68%
C2: Children's Books199999520.00%773847.23%
C3: Dialogue199998720.00%34965032.67%
C4: Educational199999920.00%16155415.09%
C5: Written English199994520.00%12120011.32%

Training Procedure

We extract training data influence estimates from models trained in random order, and sort the training data based on that information with various strategies detailed in the paper. This is the overall best performing model in our experiments, trained in order of increasing influence and re-weighted with lognormal filter, see the paper for details.

Training Hyperparameters

We employ a novel curriculum learning strategy in which the model is trained in non-random order with a total of 100M words.

Parameter
Shared Hyperparameters
Vocabulary size52k
Hidden size768
Number of layers12
Number of attention heads12
Initializer range0.02
Tie word embeddingsTrue
Model-Specific Settings
Max position embeddings514
Intermediate (FFN) size3072
Norm epsilon1e-5
Attention dropout0.1
Activation functiongelu
Hidden dropout0.1
Training Setup
FP16False
Per Device Batch Size32
Gradient Accumulation Steps16
GPUs4
Adam β₁0.9
Adam β₂0.98
Adam ε1e-6
Weight Decay ε0.01
Learning rate5e-4
Schedulerpolynomial

Evaluation

We use this evaluation pipeline of the 2025 BabyLM challange

Results

TaskMetric
(Super) GLUE0.579
blimp_filtered0.688
supplement_filtered0.559
entity_tracking0.302
ewok_filtered0.509
wugadjnominalization0.570
Macro acc 0.584

Model Card Contact

REDACTED