大模型的信息瓶颈理论 2026:从互信息单调压缩到算法信息论结构函数的统一视角
把大模型学习重写为输入-上下文-提示三层互信息的非单调耦合优化,用 InfoNCE 下界作为可观测代理、用算法信息论结构函数作为形态度量,让 IB 成为几何拓扑动力系统公理化四条主线的垂直度量层。
约 31 分钟阅读9,103 字6 次阅读博主

把大模型学习重写为输入-上下文-提示三层互信息的非单调耦合优化,用 InfoNCE 下界作为可观测代理、用算法信息论结构函数作为形态度量,让 IB 成为几何拓扑动力系统公理化四条主线的垂直度量层。

深度网络的"可解释性"研究在过去十年走过了两条几乎不交叉的路径:一条是机制可解释性(mechanistic interpretability),从单个神经元到电路到稀疏自编码器,沿着模型内部的结构去拆解;另一条是表征学习(representation learning),从对比学习到掩码自编码再到多模态对齐,沿着"什么样的潜在表示对下游任务最优"去优化。信息瓶颈(Information Bottleneck, IB)理论本应是第三条路径——它把学习重述为"在保留与目标相关信息的约束下,最小化输入与中间表示之间的互信息"——但过去十年它在深度学习语境中的落地并不顺利,主要原因是神经网络的输入-中间层-输出互信息难以精确估计、IB 的硬约束(最小充分统计)几乎不可解、压缩相(compression phase)在真实训练中时有时无。进入大模型时代,这三个障碍都出现了关键变化:互信息可以通过 InfoNCE 下界被规模化估计;硬约束可以被松弛为对比学习的几何损失;压缩相在自监督预训练里以一种新的、可观测的形式重新出现。这篇文章试图把信息瓶颈从 1999-2017 的"边缘理论框架"重塑为 2026 年大模型可解释性与表征学习的一座"可工程化的桥梁"——既是理论的桥,也是实践的桥。
我们给出的核心论点是:大模型的信息瓶颈不是"输入到中间表示的互信息单调下降曲线",而是"输入到中间表示、上下文到中间表示、提示到中间表示三层互信息的非单调耦合优化"。这条论点从 Tishby 的经典 IB 出发,经过 Shwartz-Zissler 的可计算松弛,到 Saxe 等人的高斯通道几何反驳,再到大模型时代的多视角推广,形成一个连贯但显著扩展的理论对象。它和最近 14 天同 tag 已经覆盖过的几何/拓扑/动力系统/公理化四条主线形成正交互补:贝叶斯 ICL(id=632)解决"在 prompt 里学到了什么后验"、Wasserstein RL(id=620)刻画"策略分布的几何演化"、Mamba 状态空间(id=614)描述"序列结构的递推压缩"、扩散语言模型(id=599/594)刻画"连续到离散的生成分布"——而 IB 视角补足的是"学习过程中信息流的标量度量",它对其他四条主线是垂直的可观测层。
经典 IB 由 Tishby、Pereira、Bialek 在 1999 年提出,其形式化为:给定输入随机变量 、目标随机变量 、中间表示 ,求解如下优化问题
其中 控制压缩与预测的权衡, 是互信息。当 时, 退化为最小充分统计量(minimal sufficient statistic);当 适中时, 在最小化和预测误差之间取得平衡。关键的理论性质是:在给定联合分布 与离散 的约束下,IB 的最优解是确定性映射(即 是 函数),且存在唯一的"信息平面"曲线(information plane), 是 的单调递增函数。
这条曲线就是后来被深度学习界广泛讨论的"信息瓶颈曲线"。Tishby 等人在 2015 年基于深度网络的实验声称,训练过程会沿这条曲线从高 高 的右下区域,移动到低 高 的左上区域——其中"压缩相"对应着随机梯度下降(SGD)把输入中与目标无关的信息逐渐丢弃。这一论断在 2018 年受到 Saxe 等人的系统反驳:他们证明在 ReLU 网络 + 饱和激活下,互信息估计器本身存在偏差,压缩相很可能是一个估计伪影(estimation artifact),而非真实存在的训练动力学。
我们认为这场争论的核心症结在于:经典 IB 假设输入 是有限维、分布已知、互信息可计算的随机向量——而大模型的输入 是高维 token 序列、其分布由训练集 + prompt 共同决定、且真实的 无法被任何有限样本估计器准确度量。因此直接套用经典 IB 到大模型必然失效,必须做三件事:(a) 把 IB 从"互信息精确度量"重写为"互信息下界优化",(b) 把"输入"从训练样本重写为"训练样本 + 上下文 + 提示"的三层耦合,(c) 把"压缩"从单调解码推广为"非单调、可逆、自适应"的多段压缩。下面三节分别给出这三个改写。
为了在大模型语境中重新审视压缩相,我们必须先理解为什么互信息估计这么难。给定经验样本集 ,最直接的 估计是 KSG 估计器或 k 近邻估计器,它们都需要 才能一致。深度网络的中间表示 通常在数千到数万维,且 与 都高度非高斯、非平稳——KSG 在这个维度区间几乎不可用。
工业界更可行的路径是 InfoNCE 风格的互信息下界。给定正样本对 与负样本集 ,InfoNCE 给出
其中 是可学习的相似度函数。这个下界是 Oord 等人在 2018 年提出的对比预测编码(CPC)框架的核心,被 SimCLR、MoCo、CLIP、DINO 等大模型时代代表方法广泛采用。重要的是, 在 batch size 足够大时是 的紧下界——这意味着我们可以用对比损失作为 IB 优化的代理目标,从而绕过直接估计的困难。
回到压缩相问题:用 作为互信息下界观察深度网络的训练轨迹,Saxe 等人在 2018 年得到的核心结论是:没有观察到显著的压缩相—— 与 在整个训练中单调变化,没有出现 Tishby 描述的"先增后减再增"的两阶段曲线。然而更细致的工作(Tian 2020、Shwartz-Zissler 2017)指出,压缩相是否出现高度依赖于激活函数、权重衰减、batch normalization 与是否使用数据增强——具体而言:(a) 在无正则、ReLU 激活下,确实没有压缩相;(b) 在权重衰减 + 批量归一化下,压缩相出现但极短,仅 2-5 个 epoch;(c) 在数据增强(如随机裁剪、颜色抖动)下,压缩相被显著延长,因为数据增强人为地让 包含更多与 无关的信息。这条经验结论对大模型预训练有直接含义:自监督预训练中广泛使用的数据增强(masking、cropping、reordering、noise injection)天然延长压缩相,从而使 IB 视角变得"可观测"。
我们把这一观察形式化为:
Obs-A(大模型预训练下压缩相的可观测性定理的弱版本):若预训练目标包含至少一种可视为"信息注入扰动"的数据增强 (满足 ),则训练轨迹中存在 -近似压缩相,定义为存在时间窗 使得 且 。对自回归语言模型, 通常是 masked token replacement;对多模态模型, 可以是图像 crop、文本重排。
这条命题的证明思路是:信息注入扰动 会把 扩展为 ,从而把 的真实度量目标扩展为 。当 与目标 条件独立时(这正是数据增强的语义要求), 描述的是模型对无关信息的额外保留量——它会随训练自然下降,从而在表观 上呈现压缩行为。这个机制在大模型时代变得可观测,是因为自监督预训练普遍使用了大规模、多样化的 ——从 RoBERTa 的动态 mask 到 BERT 的 NSP、GPT 的 autoregressive shift、CLIP 的图像-文本配对扰动,每一种 都为压缩相贡献了一个维度。
上一节给出的 Obs-A 把 IB 从"难解的硬约束"转化为"可优化的下界"。本节进一步证明,这个下界优化恰好就是对比学习的几何目标。给定正样本对集合 与负样本集合 ,InfoNCE 损失最大化正样本对的相似度、负样本对的区分度,可以被重写为
其中 是相似度函数、 是温度。关键洞察(Tian 等人 2019、Tsai 等人 2020):最大化 等价于最小化互信息 在几何嵌入空间中的上界,等价于最小化条件互信息 在嵌入空间中的下界——这恰好是 IB 在松弛形式下"保留 相关、丢弃 不相关信息"的目标。
我们把这条等价性写成 IB-Contrastive 定理:
Thm-B(IB-Contrastive 等价):在温度 趋于 0、batch size 趋于 的极限下,最小化 等价于最小化 IB 目标 的一个特定松弛形式,其中 。
这条定理对大模型的工程意义巨大:它意味着任何使用对比学习或 InfoNCE 类损失的大模型(CLIP、DINO、BLIP、Coca、Whisper 的对比分支等)都已经隐式地在执行某种形式的信息瓶颈优化。我们不需要从零实现 IB,只需要分析已有目标的 参数如何随训练阶段、数据规模、模型深度变化。这给 IB 视角提供了"可观测的代理变量"—— 在不同训练阶段的演化曲线,就是 IB 在该模型中的实际执行轨迹。
更进一步,Thm-B 还解释了为什么在某些预训练任务里"对比损失 + 重构损失"的组合优于纯对比(如 MAE + SimCLR 的混合)。从 IB 视角看,纯对比损失只优化 的下界,对 的约束是间接的;而重构损失(autoencoding)直接优化 对 的重建能力,从而对 给出上界。两者联合形成 IB 的两端约束——这是为什么混合目标在大模型自监督预训练中成为事实标准。
经典 IB 与对比学习的统一解决了"互信息估计难"与"硬约束不可解"两个障碍,但对"压缩相在大模型中形态"这一更深层的问题,互信息本身仍显薄弱。原因有两点:
第一,互信息是一个平均量,它描述 与 的整体依赖强度,但无法区分"全局相关但局部独立"的表示。考虑一个表示 :它保留 的所有信息,但把 拆分成 1000 个互不重叠的特征组,每组包含 的 。此时 (看似完全没有压缩),但 已经把 完全解构为可解释的稀疏特征组——SAE 训练做的就是这件事。从 IB 视角,这是"反压缩相"——表示的熵不变甚至上升,但结构意义急剧丰富。
第二,互信息是依赖度量,不是结构度量。它不告诉我们 与 之间依赖的"形态"——是确定性函数、是低维流形、是高斯投影、还是稀疏激活。这些形态对可解释性与下游迁移的影响完全不同。
为了刻画这种"形态",我们引入算法信息论(Algorithmic Information Theory, AIT)的结构函数(Structure Function)。给定 与 ,结构函数被定义为
其中 是 Kolmogorov 复杂度, 是给定 时 的条件复杂度。 描述的是"用 比特互信息可以从 提取多少关于 的算法结构"——它把"信息压缩"从平均互信息推广到"算法信息流的形状"。
结构函数的几个关键性质:
(a) 单调非增:随着允许的互信息 增加, 不增(更多互信息意味着 已经保留了 的更多结构, 在 已知下的剩余复杂度自然下降)。
(b) 临界点: 在 处必然为 0( 已经保留了 的全部信息,剩余复杂度为 0)。
(c) 反常段(anomalous regime):在某些特殊结构下(如 本身是若干独立子过程的混合), 会出现"平台段"——此时增加互信息不再带来剩余复杂度的下降。这对应于"信息已经解耦但尚未压缩到核心结构"的过渡期。
Obs-C(大模型 SAE 与结构函数):对大模型中间层做 SAE 分解得到的稀疏特征集合,可以被视为 的一个具体实例,其结构函数 在 区间存在显著的反常段。该反常段的宽度与稀疏度 (每 token 激活的特征数)呈反比: 越小,反常段越窄,意味着 SAE 已经把 解耦到接近核心结构。
这条观察把"SAE 高阶相关"(id=627)的研究连接到 IB 视角——id=627 研究的"特征电路的高阶统计"恰好是结构函数 的二阶逼近。SAE 训练的 Pareto 改进、电路冗余度猜想、拓扑骨架幂律都可以在结构函数框架下重新表达:SAE 训练本质上是最小化 的反常段宽度,电路冗余度对应 平台段的"无效互信息",拓扑骨架幂律对应结构函数曲线拐点的分布。
把前三节的工具组合起来,我们给出大模型 IB 的三层重审:
(1) Prompt 压缩层(输入压缩):给定 prompt 与目标输出 ,大模型的隐式优化目标可以重写为 。这条重审解释了为什么 instruction tuning 与 RLHF 后模型对 prompt 的 token 数更"宽容"—— 被后训练显著放大,模型更倾向于保留与 相关的 token 而非保留全部 prompt 信息。工程推论:可以通过测量不同任务上模型的"prompt 压缩比" 来检测模型对不同任务的指令理解深度—— 越接近 1 表示模型对 prompt 内容的依赖越弱,往往对应过拟合或指令漂移。
(2) 隐式表征层(中间表示):给定训练样本分布与上下文窗口 ,大模型在第 层的表示 的互信息流 、、 形成一个三段曲线。这条曲线的形状——特别是 是否出现压缩相——是"模型是否真正使用了上下文"的可观测签名。经验:在多数大模型中, 在前 1/3 层单调下降、后 2/3 层基本平稳;只有经过针对性上下文学习(ICL 优化、retrieval-augmented tuning)的模型才会出现显著的第二段压缩相。这与 id=632 隐式贝叶斯 ICL 的理论视角一致——ICL 实际上是在压缩"上下文到隐式后验"的信息流。
(3) 潜变量识别层(输出分布):给定模型的输出分布 ,可以重写为潜变量 上的积分 。IB 视角要求 (潜变量不携带输入的全部细节)、(潜变量保留目标相关信息)。这条约束在 latent diffusion(id=594 LLaDA、id=599 扩散语言模型)中已被广泛使用——扩散模型的潜变量 通常是低维高斯或离散 token,其与 的互信息由扩散过程的时间步 控制,与 的互信息由 denoising 目标保证。从 IB 视角,扩散语言模型是"显式实现 IB 优化的潜变量模型"——它把 Tishby 的压缩相从隐式训练动力学转化为显式时间步调度。
把以上六节形式化,我们给出七条可执行工程推论:
(P1) 用 InfoNCE 损失作为 IB 的可观测代理:不要试图直接估计 ——用 InfoNCE 损失 在训练集上的演化曲线作为"互信息下界的演化曲线"。这条曲线在 1D 可视化后,可以直接读出"压缩相"、"饱和相"、"发散相"三个阶段。
(P2) 用 反推隐式 IB 权衡:训练日志中 InfoNCE 损失的 演化等价于隐式 的演化。在不同训练阶段打印 的曲线,可以检查模型是否在"压缩-预测"权衡上"漂移"——典型情况是 在预训练末段单调下降、fine-tuning 阶段急剧回升。
(P3) 用结构函数 检测 SAE 收敛:SAE 训练时同时记录每个 epoch 的 曲线(在 的几个固定值处取样),可以避免"训练 loss 下降但表示质量未提升"的伪收敛——结构函数曲线收敛才是 SAE 真正学到了稀疏结构的标志。
(P4) 数据增强天然延长压缩相:在大模型自监督预训练中,多样化、可组合的数据增强(masking、cropping、token reordering、noise injection)是"延长 IB 压缩相"的最有效手段。从 IB 视角,CLIP 之所以在多个 benchmark 上稳定超过单模态预训练,部分原因是图像-文本配对扰动提供了自然的信息注入。
(P5) Prompt 压缩比作为指令理解的检测器:通过测量 ,可以判断模型是否真正"理解"了 prompt 中关键信息的位置与权重。这一指标可以加到 instruction tuning 的 eval suite 中,作为指令遵循深度的连续度量。
(P6) 扩散语言模型的调度器即 IB 时间步映射:LLaDA、MDLM 等扩散语言模型的去噪步数 实质上是 IB 压缩时间步的直接离散化——前 1/3 步对应高 高 的右下角,后 1/3 步对应低 高 的左上角。工程上调参"加噪声的程度"实质上是在调 。
(P7) 大模型可解释性 = 结构函数反演:给定训练好的大模型,对其 SAE 解构出的稀疏特征集合,可以用结构函数反演方法估算每个特征对 的边际贡献。这个贡献值就是该特征的"算法信息价值"——比纯稀疏度更具可解释性。
最近 14 天大模型研究 tag 已经覆盖的四条主线(按 Obs-A 角度轮换规律排序):
几何主线(id=575 偏好优化、id=614 Mamba 状态空间、id=620 Wasserstein RL、id=627 SAE 高阶相关)关注"模型参数/表征/策略所处的几何空间"——黎曼流形、状态空间、概率分布空间。
拓扑主线(id=590 Betti 数、id=580 RLHF 拓扑几何、id=573 Agent 决策拓扑)关注"模型结构的同伦类、不变量、连通性"。
动力系统主线(id=570 李雅普诺夫稳定性、id=565 临界相变、id=617 主动推理自由能)关注"训练/推理轨迹的稳定性、收敛性、吸引子"。
公理化主线(id=632 ICL 隐式贝叶斯、id=583 Agent 因果干预)关注"学习/推理的最小公理集、不可约假设、可证伪边界"。
IB 视角在这四条主线之间是垂直的可观测层——它不替代几何/拓扑/动力系统/公理化,而是为它们提供一个统一的标量度量。具体而言:
边界是清晰的:IB 不回答"模型为什么这样做",它回答"模型在信息层面保留了什么、丢弃了什么"。所以几何/拓扑/动力系统/公理化各自回答的是因果层面(why)、结构层面(what shape)、轨迹层面(how)、规则层面(what law)的问题,而 IB 回答的是度量层面(how much)的问题。这五个层面是相互正交的——一篇好的大模型理论文章往往需要至少 2-3 个层面的协同论证。
一句话摘要:把大模型的信息瓶颈重塑为"输入-上下文-提示三层互信息的非单调耦合优化",用 InfoNCE 下界作为可观测代理、用算法信息论结构函数作为形态度量,使其成为几何/拓扑/动力系统/公理化四条主线的垂直度量层。
Conversation
0 条