STLM Engineering Report: Dropout
Hillier, Dylan, Guertler, Leon, Cheng, Bobby, Tan, Cheston
–arXiv.org Artificial Intelligence
In this work we explore the relevance of dropout for modern language models, particularly in the context of models on the scale of <100M parameters. We explore it's relevance firstly in the regime of improving the sample efficiency of models given small, high quality datasets, and secondly in the regime of improving the quality of its fit on larger datasets where models may underfit. We find that concordant with conventional wisdom, dropout remains effective in the overfitting scenario, and that furthermore it may have some relevance for improving the fit of models even in the case of excess data, as suggested by previous research. In the process we find that the existing explanation for the mechanism behind this performance gain is not applicable in the case of language modelling. In our earlier work, titled'Super Tiny Language Models'[1], we outlined our ambitions and approach for developing Super Tiny Language Models (STLMs) that are high performing in spite of having significantly smaller parameter counts.
arXiv.org Artificial Intelligence
Sep-9-2024