Skip to yearly menu bar Skip to main content


Poster
in
Workshop: ICML 2024 Workshop on Foundation Models in the Wild

Projected Language Models: A Large Model Pre-Segmented Into Smaller Ones

David Grangier · Angelos Katharopoulos · Pierre Ablin · Awni Hannun

Keywords: [ language model pretraining ] [ projected networks ] [ efficient inference ]


Abstract:

Large language models are versatile tools but are not suitable for small inference budgets. Small models have more efficient inference but their lower capacity means that their performance can be good only if one limits their scope to a specialized domain. This paper explores how to get a small language model with good specialized accuracy, even when specialization data is unknown during pretraining. We propose a novel architecture, projected networks (PN). PN is a high capacity network whose parameters can be linearly projected into a small network for fine tuning. We assess the empirical effectiveness of our solution compared to small model training, distillation and hard mixture of experts.

Chat is not available.