tau-1.8B

Model Details

• Model Name: tau-1.8B
• Base Model: Qwen1.5-1.8B
• Dataset: UltraTextbooks-2.0
• Model Size: 1.8B parameters
• Model Type: Language Model
• Training Procedure: Further pre-training of Qwen1.5-1.8B on UltraTextbooks-2.0.

Model Use

tau-1.8B is designed to be a general-purpose language model with enhanced capabilities in the domains of machine learning, mathematics, and coding. It can be used for a wide range of natural language processing tasks, such as:

• Educational question answering
• Text summarization
• Content generation for educational purposes
• Code understanding and generation
• Mathematical problem solving

The model's exposure to the diverse content in the UltraTextbooks-2.0 dataset makes it particularly well-suited for applications in educational technology and research.

Training Data

tau-1.8B was further pre-trained on the UltraTextbooks-2.0 dataset, which is an expanded version of the original UltraTextbooks dataset. UltraTextbooks-2.0 incorporates additional high-quality synthetic and human-written textbooks from various sources on the Hugging Face platform, with a focus on increasing the diversity of content in the domains of machine learning, mathematics, and coding.

For more details on the dataset, please refer to the UltraTextbooks-2.0 Dataset Card.

Performance and Limitations

Refer to Evaluation for evaluations. It is essential to note that the model may still exhibit biases or inaccuracies present in the training data. Users are encouraged to critically evaluate the model's outputs and report any issues to facilitate continuous improvement.

Environmental Impact

The training of tau-1.8B required computational resources that contribute to the model's overall environmental impact. However, efforts were made to optimize the training process and minimize the carbon footprint.

Ethical Considerations

tau-1.8B was trained on a diverse dataset that may contain biases and inaccuracies. Users should be aware of these potential limitations and use the model responsibly. The model should not be used for tasks that could cause harm or discriminate against individuals or groups.

Evaluation

Metric	Value
Avg.	45.73
AI2 Reasoning Challenge (25-Shot)	37.20
HellaSwag (10-Shot)	60.26
MMLU (5-Shot)	45.96
TruthfulQA (0-shot)	39.72
Winogrande (5-shot)	61.09
GSM8k (5-shot)	30.17