Yuanzhi Li, Sébastien Bubeck, Ronen Eldan, Allie Del Giorno, Suriya Gunasekar, Yin Tat Lee
We continue the investigation into the power of smaller Transformer-based language models as initiated by \textbf{TinyStories} -- a 10 million parameter model that can produce coherent English -- and the follow-up work on \textbf{phi-1}, a 1.3 billion parameter model with Python coding performance close to the state-of-the-art. The latter work proposed to use existing Large Language Models (LLMs) to generate ``textbook quality" data as a way to enhance the learning process compared to traditional web data. We follow the ``Textbooks Are All You Need" approach, focusing this time on common sense reasoning in natural language, and create a new 1.3 billion parameter model named \textbf{phi-1.5}, with performance on natural language tasks comparable to models 5x larger, and surpassing most non-frontier LLMs on more complex reasoning tasks such as grade-school mathematics and basic coding. More generally, \textbf{phi-1.5} exhibits many of the traits of much larger LLMs, both good -- such as the ability to ``think step by step" or perform some rudimentary in-context learning -- and bad, including hallucinations and the potential for toxic and biased generations -- encouragingly though, we are seeing improvement on that front thanks to the absence of web data. We open-source \textbf{phi-1.5} to promote further research on these urgent topics.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Transfer Learning | MML | Average (%) | 37.9 | phi-1.5-web 1.3B |
| Question Answering | SIQA | Accuracy | 53 | phi-1.5-web 1.3B (zero-shot) |
| Question Answering | SIQA | Accuracy | 52.6 | phi-1.5 1.3B (zero-shot) |
| Question Answering | PIQA | Accuracy | 77 | phi-1.5-web (1.3B) |
| Code Generation | MBPP | Accuracy | 43.5 | phi-1.5-web 1.3B |
| Common Sense Reasoning | WinoGrande | Accuracy | 74 | phi-1.5-web 1.3B (zero-shot) |
| Common Sense Reasoning | ARC (Challenge) | Accuracy | 44.9 | phi-1.5-web 1.3B (zero-shot) |
| Common Sense Reasoning | ARC (Easy) | Accuracy | 76.1 | phi-1.5-web 1.3B (0-shot) |
| Multi-Task Learning | MML | Average (%) | 37.9 | phi-1.5-web 1.3B |