Formalizing the Binding Problem¶
会议: ICML 2026
arXiv: 2606.03976
代码: https://github.com/KordingLab/formalizing-the-binding-problem
领域: 可解释性 / 表示分析 / 视觉Transformer
关键词: 绑定问题、信息论探针、ViT表示、[CLS] token、空间 token
一句话总结¶
本文把"神经网络的绑定问题"形式化为表示 \(Z\) 中关于对象码 \(O\) 的互信息 \(I(O;Z)\),并设计自回归概率探针在 DINOv2 / CLIP 等 ViT 上度量绑定信息,发现 [CLS] 只编码 <50% 的绑定信息且结构近似二次型,而对全套空间 token 加注意力探针能恢复 ~92% 的绑定信息。
研究背景与动机¶
领域现状:现代视觉与视觉-语言模型大量依赖 ViT 把图像编码成 [CLS] 摘要 token 或一组空间 token,下游再做对比学习(CLIP / SimCLR / Barlow Twins)或 cross-modal 拼接(LLaVA / Qwen-VL)。已有工作只证明了 ViT "知道哪些 patch 属于同一个对象",并没有回答模型表示里到底有多少"哪些特征属于同一个对象"的信息。
现有痛点:VLM 在多对象、特征共享、遮挡场景下常常张冠李戴——把蓝色帽子和红色书包说成"红色帽子"。Campbell et al. 2025 等工作给出了大量任务级失败案例,但缺少一个可量化、跨模型可比的绑定能力度量,更没法定位绑定信息到底丢在了 [CLS] 还是空间 token 的哪一侧。
核心矛盾:绑定问题既是认知科学概念又是工程概念,两边都没给出"表示中绑定信息含量"的形式化定义。直接看下游任务准确率,会把"模型不知道"和"模型知道但 readout 输出错了"混在一起,无法分离表示本身的信息瓶颈。
本文目标:(1) 给出与具体编码方式无关的绑定信息形式化定义;(2) 提供一个可在任意预训练表示上跑通的探针估计器;(3) 用它解剖 ViT 不同组件、不同数据集下的绑定能力。
切入角度:作者把"特征是否存在"与"对象是否存在"分别建成二值随机向量 \(F\) 与 \(O\),借助互信息把"绑定 = 表示中关于对象码的信息"压缩成一个数。再用"探针的交叉熵 = 条件熵 + KL"这一经典不等式(Lemma 2.21),把不可直接计算的 \(H(O|Z)\) 用一个训练好的探针的 test loss 上界来估计。
核心 idea:用信息论把绑定信息拆成"数据集先验 \(H(O)\) - 探针残余不确定性 \(H(O|Z)\)",并用自回归分解 \(q_\theta(o|z)=\prod_k q_\theta(o_k|o_{<k},z)\) 绕过对象码组合爆炸,从而在任意 ViT 上得到可比较的绑定信息标量。
方法详解¶
整体框架¶
这套方法要回答"一个冻结 ViT 的表示 \(Z\) 里到底藏了多少绑定信息",做法是把"绑定"翻译成互信息:先把场景标注成 ground-truth 特征码 \(F\in\{0,1\}^n\) 与对象码 \(O\in\{0,1\}^K\),再用"探针的 test 交叉熵是条件熵的上界"这一不等式,把不可直接算的 \(H(O|Z)\) 用一个训练好的自回归探针的残余 loss 来估计。
具体落到四步:先用组合学/经验分布算出数据集先验 \(H(O)\)、\(H(F)\);再在冻结表示 \(Z\) 上分别训对象探针 \(q_\theta(o_k|o_{<k},z)\) 和特征探针 \(q_\theta(f_k|f_{<k},z)\);把它们在 test set 上的交叉熵当作 \(H(O|Z)\)、\(H(F|Z)\) 的上界;最后代入 \(I(O;Z)=H(O)-H(O|Z)\) 与 Theorem 2.16,产出四个标量——绑定信息 \(B_O(Z)=I(O;Z)\)、特征条件绑定信息 \(B^*_{O,F}(Z)=I(O;Z|F)\),以及除以先验得到的归一化版 \(\beta_O(Z)\)、\(\beta^*_{O,F}(Z)\)。
关键设计¶
1. 绑定信息的信息论形式化:把"会不会绑定"压成一个 bit 数
绑定问题此前只有概念辨析、没有与编码方案无关的可比尺度,本文把场景抽象成"特征集 \(F\) + 对象集 \(O\)",每个对象 \(o_i\) 对应 \(F\) 的一个子集(surjective map),于是 \(F\) 成为 \(O\) 的确定函数,绑定信息就定义成 \(B_O(Z):=I(O;Z)\)。直接用这个量还有个隐患:一个只把单特征学得很好的模型也会拿到高分,被误判成"会绑定"。为此引入条件版 \(B^*_{O,F}(Z):=I(O;Z|F)=I(O;Z)-I(F;Z)\)(Theorem 2.14),扣掉仅靠特征信息就能推出的部分,只留真正的"哪些特征属于同一对象"。再用 \(H(O)\)、\(H(O|F)\) 把两者归一化成无量纲的 \(\beta\)、\(\beta^*\),跨数据集才能公平对比。用互信息而非下游准确率,是因为准确率会被 readout 头吞掉信息,而 \(I\) 既反映软概率的不确定度,又能通过"换探针族"暴露信息的代数结构(线性 / 二次 / 高阶)。
2. 自回归对象码探针 + 探针族阶梯:在单 token 上估信息、顺带读出信息结构
要在 [CLS] 这种单 token 上估 \(H(O|Z)\),直接训 \(q_\theta(o|z)\) 得覆盖 \(2^K\) 的标签空间(实验中 \(K=64\),超过 \(10^{19}\)),完全不可行。本文用链式分解 \(q_\theta(o|z)=\prod_{k=1}^K q_\theta(o_k|o_{<k},z)\) 绕过组合爆炸:把表示 \(z\) 与已揭示的对象码 \(o_{<k}\) 拼成 \(x=[z\|o_{<k}]\) 喂给探针,每个 \(o_k\) 用一个二分类 logit \(\ell_k(x)\) 监督,总 loss 求和即式 (5)。关键的巧思在于让探针族成一个阶梯——线性 \(\ell_k=W_kx+b_k\)、二次 \(\ell_k=x^\top W_k x+b_k\)、四层 GELU DNN(约 3M 参数)——同时去逼近 Lemma 2.21 给出的 \(L_{CE}\ge H(O|Z)\) 下界。在合成 ColorShape 上三族 test loss 分别为 \(34.2/22.0/20.6\) bits(误差降幅 \(37.4\%/63.0\%/64.4\%\)),二次族与 DNN 几乎打平,说明绑定信息基本能被二阶函数读出。更进一步,把二次探针参数共享成 \(W_k=U_{\text{color}}^\top V_{\text{shape}}\) 后只多 \(2.4\) bits loss,等于把"[CLS] 的绑定信息是不是 \(\text{color}\otimes\text{shape}\) 双线性形式"做成了可证伪实验,结论是它确实就是颜色投影与形状投影的点积——一种保守的合取式编码。
3. 空间 token 上的简化注意力探针:只学 query 不学 key/value
ViT 全套空间 token \(\{s_i\}_{i=1}^N\) 的绑定信息天然分散在多个 patch 上,直接拼接喂 MLP 会让参数爆炸且过拟合。本文为每个对象 \(o_k\) 只学一个 query \(q_k=g_k(o_{<k})\)(没有 key / value 投影),按 \(a_{k,i}=\text{softmax}_i(q_k^\top s_i)\) 把空间向量路由成加权平均 \(\bar s_k=\sum_i a_{k,i} s_i\),再过二次读出 \(q_\theta(o_k=1|o_{<k},\{s_i\})=\sigma(\bar s_k^\top W_k \bar s_k+b_k)\)。刻意砍掉 key/value 是为了让 routing 完全由表示本身决定、避免探针自己"发明"信息,同时让注意力权重 \(a_{k,i}\) 直接当可视化诊断。结果 ColorShape 上 test ER 达 \(96.8\%\)(loss 仅 \(3.1\) bits),远低于 [CLS] 最强 DNN 探针的 \(20.6\) bits,且注意力几乎总能 route 到目标对象所在 patch——证明绑定信息确实以"对象在哪个 patch"这种空间索引形式存在于 patch token 里。
损失函数 / 训练策略¶
所有探针都用二元交叉熵 \(L_{CE}(\theta)=\sum_k \mathbb{E}_{(z,o)}[-\log q_\theta(o_k|o_{<k},z)]\) 训练,特征探针完全对称。报告 test loss 而非 train loss,并通过训练-验证-测试三集都使用 disjoint 的特征码 \(F\) 与对象码 \(O\) 来防止记忆(对象码组合空间 \(\sim10^{12}\),自然分裂)。DINOv2-Large、CLIP ViT-L/14(224 与 336 两个分辨率)保持冻结,只训探针。
实验关键数据¶
主实验:[CLS] 单 token 绑定信息(ColorShape,\(H(O)=39.9\) bits,\(H(F)=7.0\) bits)¶
| 探针族 | \(B_O(Z)\) (bits) | \(\beta_O(Z)\) | \(B^*_{O,F}(Z)\) (bits) | \(\beta^*_{O,F}(Z)\) |
|---|---|---|---|---|
| Linear | 5.7 | 14.3% | 0.3 | 0.8% |
| Quadratic | 17.9 | 44.9% | 12.5 | 37.9% |
| DNN (3M params) | 19.4 | 48.5% | 13.9 | 42.4% |
| Attention + 空间 token | 36.8 | 92.2% | 31.0 | 94.1% |
最后一行(来自 Table 5)是把表示换成全套空间 token + 注意力探针的结果,直接把绑定信息从 48.5% 抬到 92.2%。
消融:数据复杂度、遮挡、自然数据集¶
| 实验 | 关键发现 |
|---|---|
| ColorShape 颜色×形状从 1→7 (49 组合) | 随特征空间指数增长,\(\beta^*_{O,F}(Z)\) 单调下降但没有指数衰减(CLIP-L/14 224px) |
| CLEVR 不同遮挡(相机高度 0.6→3.2) | \(\beta^*_{O,F}(Z)\) 从 45.0% 单调升到 58.7%,遮挡每加一档掉 ~3 pp |
| 参数共享二次探针 vs 标准二次探针 | 共享 \(U_{\text{color}}^\top V_{\text{shape}}\) 仅多 2.4 bits loss → [CLS] 绑定信息几乎是纯二次合取码 |
| Visual Genome 自然数据 | DINOv2 / CLIP 在 VG:Color / VG:TopAttr 上 \(\beta^*_{O,F}\in[39.9\%,47.0\%]\),与合成数据相当 |
| CLIP 224px → 336px | ColorShape 47.7%→56.4%;细粒度空间表示帮助绑定 |
关键发现¶
[CLS]是绑定瓶颈:信息量上界 ~48.5%,再叠 DNN 也只多 1.5 pp;并且这点信息几乎可被一个 \(\text{color}\otimes\text{shape}\) 双线性形式完整解释,说明 ViT 摘要 token 学到的不是"对象槽",而是"特征对的合取统计"。- 空间 token 几乎无信息损失:注意力探针把绑定信息恢复到 ~92%,意味着 ViT 的绑定能力其实保存在 patch token 的空间索引里,对比学习类预训练(只用
[CLS])的目标函数没有暴露这部分能力。 - 绑定难度可分解:复杂度(特征空间大小)、遮挡、自然性是三个正交维度;用 \(\beta^*_{O,F}\) 都能稳定单调反映。
- 分辨率比模型本身更重要:同样 CLIP-L/14,从 224 到 336 像素的提升超过换 DINOv2 这种不同自监督算法的差距。
亮点与洞察¶
- 把一个哲学/认知概念变成可测标量:先前对"binding"的讨论几乎都停留在概念辨析层面,这篇把"我有多少绑定信息"压成一个 bit 数,并给了可复现的探针管线,把绑定从理论问题搬到工程基准上。
- 探针族阶梯当作"信息结构显微镜":linear/quadratic/DNN 的 loss 差距 + 参数共享变体,直接告诉你"
[CLS]里的绑定信息是颜色×形状的双线性映射",远比单看准确率信息量大;同样思路可迁移到任何"信息存不存在 / 是不是低阶"的探针实验。 - 简化注意力探针的妙用:只学 query 不学 key/value 的极小注意力,既保证 routing 是被表示本身决定的(避免探针自己发明信息),又能复用 \(a_{k,i}\) 做可解释性可视化。
- 量化"对比学习+
[CLS]"路线的代价:用一个数告诉社区——CLIP 系列只看[CLS]的下游头其实丢掉了一半绑定信息,VLM 用 patch token 的做法被这套度量背书。
局限与展望¶
- 依赖预定义离散特征/对象词典:所有公式都建立在 \(F\in\{0,1\}^n\)、\(O\in\{0,1\}^K\) 上,连续特征(如 velocity=3m/s)只能离散化处理;作者承认未来需要连续探针。
- 测的是"可解码"信息,不是"模型用了的"信息:高 \(\beta\) 只说明信息在 \(Z\) 里能被探针挖出来,并不保证下游 readout(如 LLaVA 的 LLM 头)真的用到了——所以高分模型在 VLM 任务上仍可能绑定失败。
- 条件版 \(B^*_{O,F}\) 假设 \(F\) 可由 \(O\) 推出:在生物视觉或噪声场景下并不严格成立,会让 Theorem 2.14 的分解失真。
- 实验主要在 ColorShape + CLEVR + VG 子集:分类对象数仍有限(\(K=64\) 量级),数千类级别(如 LVIS)下二次探针是否仍接近 DNN 是开问题。
- 改进思路:把探针 loss 反向当作训练目标(binding-aware pretraining);把度量推广到视频里的时序绑定;研究"信息结构"是否能预测下游 compositional generalization 的成功率。
相关工作与启发¶
- vs Campbell et al. 2025 / Zhang et al. 2024:他们用下游 VLM 任务的错误率说明绑定失败,但无法定位失败发生在表示阶段还是 readout 阶段;本文把诊断下沉到表示层,并能区分"信息没有"与"信息有但 readout 用不上"。
- vs Greff et al. 2020(On the binding problem in artificial neural networks):那是一篇 position paper,给出概念分类(segregation / representation / composition);本文把其中的"representation binding"做了可度量化定义,是对它的工程化补完。
- vs 槽注意力 / Capsule / Tensor-Product 等显式绑定架构:这些架构提出新编码方案以获得绑定能力,但缺乏统一比较尺度;本文的 \(\beta^*_{O,F}\) 提供了一个跨架构的公共基准,未来可用它公平比较 slot attention vs ViT vs JEPA。
- vs 标准 probing(如 BERT linguistic probes):传统 probing 主要测"分类能力";本文把 probing 的 loss 直接当作信息论上界,让 probing 从"性能测试"升级为"信息量估计",并通过探针族对比读出信息的代数结构。
评分¶
- 新颖性: ⭐⭐⭐⭐⭐ 第一篇把绑定问题与互信息严格对齐并给出可计算估计的工作。
- 实验充分度: ⭐⭐⭐⭐ 覆盖合成 / CLEVR / VG 三类数据 + DINOv2 / CLIP 双架构 + 49 组复杂度扫描,但模型规模和对象空间仍偏小。
- 写作质量: ⭐⭐⭐⭐⭐ 定义—定理—探针—实验—解释的链条清晰,每个度量都附了可计算的方法。
- 价值: ⭐⭐⭐⭐⭐ 给社区一个标量基准,能直接指导"用
[CLS]还是 patch token"、"上不上更高分辨率"这类落地决策,也为未来 binding-aware 预训练打好了评估底座。