Balancing Stylization and Truth via Disentangled Representation Steering

Chenglei Shen, Zhongxiang Sun, Teng Shi, Xiao Zhang, Jun Xu
8/6/2025
cs.CL

No confident code implementation found for this paper. 1 possible match can be shown below.

Abstract

Generating stylized large language model (LLM) responses via representation editing is a promising way for fine-grained output control. However, there exists an inherent trade-off: imposing a distinctive style often degrades truthfulness. Existing representation editing methods, by naively injecting style signals, overlook this collateral impact and frequently contaminate the model's core truthfulness representations, resulting in reduced answer correctness. We term this phenomenon stylization-induced truthfulness collapse. We attribute this issue to latent coupling between style and truth directions in certain key attention heads, and propose StyliTruth, a mechanism that preserves stylization while keeping truthfulness intact. StyliTruth separates the style-relevant and truth-relevant subspaces in the model's representation space via an orthogonal deflation process. This decomposition enables independent control of style and truth in their own subspaces, minimizing interference. By designing adaptive, token-level steering vectors within each subspace, we dynamically and precisely control the generation process to maintain both stylistic fidelity and truthfulness. We validate our method on multiple styles and languages. Extensive experiments and analyses show that StyliTruth significantly reduces stylization-induced truthfulness collapse and outperforms existing inference-time intervention methods in balancing style adherence with truthfulness.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 1 weaker match is hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{shen2025balancing,
  title  = {Balancing Stylization and Truth via Disentangled Representation Steering},
  author = {Chenglei Shen and Zhongxiang Sun and Teng Shi and Xiao Zhang and Jun Xu},
  year   = {2025},
  eprint = {2508.04530},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2508.04530v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2508.04530.svg)](https://codeofpaper.com/paper/2508.04530)

Discussion