Search papers, labs, and topics across Lattice.
This paper introduces Tiered Language Models (TLMs), which allow for the release of open-weight LLMs while maintaining control over sensitive capabilities through a novel key-based mechanism. By training a single set of weights to support multiple capability levels, TLMs can operate in a public configuration while enabling additional functionalities through a compact secret key that alters the computation graph. The results show that TLMs can effectively learn new languages and retain private knowledge without compromising the public model's performance, marking a significant advancement in balancing open access with security concerns.
TLMs can unlock sensitive capabilities in open-weight models without risking public exposure, enabling a new paradigm for LLM deployment.
Open-weight Large Language Models (LLMs) enable scientific progress and broad deployment. However, they make it difficult to control access to sensitive capabilities. Current practice either suppresses dangerous capabilities before release or mediates access through closed services that use specialized model variants, input/output monitors, and API permissions. The former is susceptible to jailbreaks while sacrificing capability for all users to mitigate the risks posed by a few, and the latter is fundamentally incompatible with open-weight release. In this paper, we propose Tiered Language Models (TLMs), where a single set of released weights supports multiple capability levels. In its default public configuration, a TLM behaves as a conventional LLM. A compact secret key specifies a permutation over a small parameter subset, inducing an alternative computation graph over the same weights that exposes additional capabilities. We develop a training protocol that jointly pretrains both configurations from scratch, then fine-tunes the keyed configuration on private data with regularization to preserve the public model's behavior. We pretrain 180M- and 650M-parameter TLMs and demonstrate that the keyed configuration can acquire a new language, gain instruction-following ability, and memorize private factual knowledge, whereas the public configuration exhibits none of these capabilities. Moreover, we show that our approach extends naturally to multiple hierarchical tiers. Because authorization operates on the model's weight structure rather than in the input space, the mechanism resists fine-tuning-based extraction and partial key compromise. In general, TLMs take a step toward reconciling open-weight release with selective capability control.
