Know When To Stop: A Study of Semantic Drift in Text Generation
In this work, we explicitly show that modern LLMs tend to generate correct facts first, then "drift away" and generate incorrect facts later: this was occasionally observed but never properly measured. We develop a semantic drift score that measures the degree of separation between correct and incorrect facts in generated texts and confirm our hypothesis when generating Wikipedia-style biographies. This correct-then-incorrect generation pattern suggests that factual accuracy can be improved by knowing when to stop generation. Therefore, we explore the trade-off between information quantity and factual accuracy for several early stopping methods and manage to improve factuality by a large margin. We further show that reranking with semantic similarity can further improve these results, both compared to the baseline and when combined with early stopping. Finally, we try calling external API to bring the model back to the right generation path, but do not get positive results. Overall, our methods generalize and can be applied to any long-form text generation to produce more reliable information, by balancing trade-offs between factual accuracy, information quantity and computational cost.
Code (0)
등록된 구현이 없습니다.
Tasks
RerankingSemantic SimilaritySemantic Textual SimilarityText GenerationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Drifting Objectives for Refining Discrete Diffusion Language Models
Discrete diffusion language models (DDLMs) generate text by iteratively denoising categorical token sequences, while recent drifting methods for continuous generators suggest that part of this sampling-time correction ca…
An Optimal Dividend Problem with Capital Injections over a Finite Horizon
In this paper we propose and solve an optimal dividend problem with capital injections over a finite time horizon. The surplus dynamics obeys a linearly controlled drifted Brownian motion that is reflected at the origin,…
Asset liquidation under drift uncertainty and regime-switching volatility
Optimal liquidation of an asset with unknown constant drift and stochastic regime-switching volatility is studied. The uncertainty about the drift is represented by an arbitrary probability distribution; the stochastic v…
Optimal stopping and divestment timing under scenario ambiguity and learning
Aiming to analyze the impact of environmental transition on the value of assets and on asset stranding, we study optimal stopping and divestment timing decisions for an economic agent whose future revenues depend on the …
Decision MakingSupervised Learning Has a Geometric Blind Spot
Ordinary supervised training minimises the task loss and then stops. It never pays for how far the representation moves when the input is nudged along directions that helped fit training labels---including directions tha…