A Negative Result on Cross-Model Activation Transfer in a Pythia Multi-Hop Setting
Summary
A recent study presents a negative result on cross-model activation transfer, investigating whether one language model can directly communicate intermediate reasoning states to another during inference via hidden activation injection, rather than natural language. Researchers tested this in a controlled multi-hop reasoning setting using Pythia-160M as the sender and Pythia-410M as the receiver. A linear translation layer successfully mapped sender and receiver hidden states, achieving a normalized cosine similarity near 0.97. However, injecting these translated activations into the receiver at inference time did not improve downstream answering. Low-strength additive injection remained near the no-injection baseline, with confidence intervals crossing zero, while replacement-style injection proved consistently destructive. Rescaling translated vectors to match the receiver's hidden-state norm also failed to rescue performance, indicating that offline representational alignment is insufficient for useful causal communication in this specific setting.
Key takeaway
For AI Scientists exploring advanced inter-model communication beyond natural language, you should recognize that direct hidden activation transfer, even with high representational alignment, proved ineffective in a Pythia multi-hop setting. Your efforts to inject translated activations at inference time may not improve downstream reasoning and could even be destructive. Focus instead on alternative mechanisms for knowledge distillation or communication that do not rely on direct hidden state manipulation for causal influence.
Key insights
Direct hidden activation transfer between Pythia models at inference time does not improve multi-hop reasoning.
Principles
- Offline representational alignment is not sufficient for causal communication.
- Hidden state translation can achieve high similarity (0.97).
- Direct activation injection may be destructive.
Method
A linear layer translates Pythia-160M hidden states to Pythia-410M's normalized space. These translated activations are then injected additively or by replacement into the receiver during multi-hop inference.
In practice
- Direct activation injection is ineffective for reasoning transfer.
- Rescaling translated vectors does not improve performance.
Topics
- Language Models
- Activation Transfer
- Pythia Models
- Multi-Hop Reasoning
- Hidden States
- Model Communication
Best for: Research Scientist, AI Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.