LLM Reasoning 相关度: 7/10

Transducing Language Models

Vésteinn Snæbjarnarson, Samuel Kiegeland, Tianyu Liu, Reda Boumasmoud, Ryan Cotterell, Tim Vieira
arXiv: 2603.05193v1 发布: 2026-03-05 更新: 2026-03-05

AI 摘要

提出了一种基于有限状态转换器(FST)的通用语言模型转换框架,用于适应特定输出需求。

主要贡献

  • 提出基于FST的语言模型转换框架
  • 开发用于边缘化和条件化的算法
  • 在三种不同领域验证框架有效性

方法论

将确定性字符串转换视为新的语言模型,使用FST表示转换,开发算法实现概率传播和条件化。

原文摘要

Modern language models define distributions over strings, but downstream tasks often require different output formats. For instance, a model that generates byte-pair strings does not directly produce word-level predictions, and a DNA model does not directly produce amino-acid sequences. In such cases, a deterministic string-to-string transformation can convert the model's output to the desired form. This is a familiar pattern in probability theory: applying a function $f$ to a random variable $X\sim p$ yields a transformed random variable $f(X)$ with an induced distribution. While such transformations are occasionally used in language modeling, prior work does not treat them as yielding new, fully functional language models. We formalize this perspective and introduce a general framework for language models derived from deterministic string-to-string transformations. We focus on transformations representable as finite-state transducers -- a commonly used state-machine abstraction for efficient string-to-string mappings. We develop algorithms that compose a language model with an FST to *marginalize* over source strings mapping to a given target, propagating probabilities through the transducer without altering model parameters and enabling *conditioning* on transformed outputs. We present an exact algorithm, an efficient approximation, and a theoretical analysis. We conduct experiments in three domains: converting language models from tokens to bytes, from tokens to words, and from DNA to amino acids. These experiments demonstrate inference-time adaptation of pretrained language models to match application-specific output requirements.

标签

language model finite-state transducer string transformation

arXiv 分类

cs.CL