arXiv · 2501.19383
Decoding-based Regression
Abstract
Language models have recently been shown capable of performing regression wherein numeric predictions are represented as decoded strings. In this work, we provide theoretical grounds for this capability and furthermore investigate the utility of causal sequence decoding models as numeric regression heads given any feature representation. We find that, despite being trained in the usual way - for next-token prediction via cross-entropy loss - decoder-based heads are as performant as standard pointwise heads when benchmarked over standard regression tasks, while being flexible enough to capture smooth numeric distributions, such as in the task of density estimation.
Explore related subjects
Keep this discovery
Xingyou Song, Dara Bahri. 2025-01-31. Decoding-based Regression. https://arxiv.org/abs/2501.19383
Cite the original work for its findings. Save a collection to share your selection of sources.