๐ค AI Summary
The mechanisms by which information is encoded and utilized within residual streams of multi-stream Transformers remain poorly understood, particularly from a causal perspective. This work proposes the first open-source mHC language model and introduces a novel โablation-rescueโ causal analysis framework to systematically investigate the distribution and functional roles of information in multi-stream residual connections. By integrating stream-level ablation and restoration interventions, representational similarity analyses, and controlled causal experiments, the study reveals for the first time the fundamental distinction between asymmetric cross-stream information utilization and functional redundancy. It further demonstrates that representational similarity alone is insufficient to capture dynamic inter-stream information flow, thereby advancing causal understanding of the internal mechanisms underlying multi-stream architectures.
๐ Abstract
Multi-stream transformer architectures have recently been proposed as a promising direction for managing representation collapse and the vanishing gradient problem for residual connections, yet their internal mechanisms remain unexplored. In particular, the recently introduced Manifold-Constrained Hyper-Connections (mHC) architecture posits multiple residual streams with constrained interaction, but lacks in-depth mechanistic analysis. We present the first open-source mHC language model (https://huggingface.co/wgpeng/mhc-780m) and analyze the multiple-stream architecture with a suite of representation-level metrics and causal interventions to probe how parallel streams encode and utilize information. Specifically, we introduce a systematic stream ablation-and-rescue framework that enables direct causal comparison of residual streams during inference. Through targeted pairwise interventions and controlled recovery experiments, we distinguish functional redundancy from asymmetric utilization and reveal how information is distributed across streams beyond what is observable from representational similarity alone.