用熵调控向量强度,让小模型对话更专注不跑题。
EnSToM: Enhancing Dialogue Systems with Entropy-Scaled Steering Vectors for Topic Maintenance
- 根据输入不确定性动态调节引导向量强度,智能应对跑题干扰。
- 在小数据下显著提升话题一致性,优于微调方法。
- 适合资源受限场景的客服类对话系统,兼顾效率与可靠性。
小型大语言模型(sLLMs)因其轻量高效,适用于资源受限环境。然而,sLLMs在任务导向型对话系统中常难以保持话题一致性,这在服务聊天机器人等场景中至关重要。需确保模型能拒绝无关或恶意输入,维持预期功能,防止误用并保障可靠性。现有激活工程方法虽能在某些场景有效,但初步实验显示其在话题遵循方面存在局限。为此,我们提出一种新方法——熵尺度引导向量话题保持(EnSToM)。EnSToM基于输入不确定性动态调整引导强度,使模型能有效处理离题干扰,同时保持话题准确性。实验表明,相比微调方法,EnSToM在较小数据规模下实现显著性能提升。该方法在不牺牲效率的前提下增强sLLM对话系统的主题一致性,提供了一种稳健的解决方案。
原文摘要 · Abstract (English)
Small large language models (sLLMs) offer the advantage of being lightweight and efficient, which makes them suitable for resource-constrained environments. However, sLLMs often struggle to maintain topic consistency in task-oriented dialogue systems, which is critical for scenarios such as service chatbots. Specifically, it is important to ensure that the model denies off-topic or malicious inputs and adheres to its intended functionality so as to prevent potential misuse and uphold reliability. Towards this, existing activation engineering approaches have been proposed to manipulate internal activations during inference. While these methods are effective in certain scenarios, our preliminary experiments reveal their limitations in ensuring topic adherence. Therefore, to address this, we propose a novel approach termed Entropy-scaled Steering vectors for Topic Maintenance (EnSToM). EnSToM dynamically adjusts the steering intensity based on input uncertainty, which allows the model to handle off-topic distractors effectively while preserving on-topic accuracy. Our experiments demonstrate that EnSToM achieves significant performance gain with a relatively small data size compared to fine-tuning approaches. By improving topic adherence without compromising efficiency, our approach provides a robust solution for enhancing sLLM-based dialogue systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。