让多卡大模型可解释且可控,性能提升41倍
Distributed Interpretability and Control for Large Language Models

- 多卡部署下实现激活层可解释与控制
- 吞吐提升41倍,内存减少7倍,支持1500词长序列分析
- 无需微调即可实现输出的单调可控,适合模型调试与安全研究
需要多张GPU卡部署的大语言模型通常具备最强能力,但现有技术难以在多卡环境下实现模型的可解释性与行为控制。本文提出一种可扩展至多卡环境的激活级可解释性(对数透镜)与控制(控制向量)方案。系统通过设计优化,使激活内存降低达7倍,吞吐提升最高41倍,相比基线在相同硬件上表现优异。我们在LLaMA-3.1(8B、70B)和Qwen-3(4B、14B、32B)上验证方法,可在收集全层激活轨迹的同时保持每秒20至100个词的处理速度。通过在层归一化后注入标签位置控制向量,实现了无需微调或额外前向传播的可控、单调输出变化,跨数据集平均控制斜率为0.702。相关基准测试、消融实验与可复现的集成方案已开源。
原文摘要 · Abstract (English)
Large language models that require multiple GPU cards to host are usually the most capable models. It is necessary to understand and steer these models, but the current technologies do not support the interpretability and steering of these models in the multi-GPU setting as well as the single-GPU setting. We present a practical implementation of activation-level interpretability (logit lens) and steering (steering vector) that scales up to multi-GPU language models. Our system implements design choices that reduce the activation memory by up to 7x and increase the throughput by up to 41x compared to a baseline on identical hardware. We demonstrate the method across LLaMA-3.1 (8B, 70B) and Qwen-3 (4B, 14B, 32B), sustaining 20-100 tokens/s while collecting full layer-wise activation trajectories for sequences of 1,500 tokens. Using label-position steering vectors injected post-LayerNorm, we show controllable, monotonic shifts in model outputs with a mean steerability slope of 0.702 across evaluated datasets, without fine-tuning or additional forward passes. We release detailed benchmarks, ablations, and a reproducible instrumentation recipe to enable practical interpretability and real-time behavioral control for frontier LLMs at https://github.com/Devdesai1901/LogitLense.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。