跳转至

Language Reconstruction with Brain Predictive Coding from fMRI Data

会议: ACL 2026
arXiv: 2405.11597
代码: 无
领域: 脑机接口 / 语言解码
关键词: fMRI语言重建, 预测编码, 脑信号解码, 神经语言学, 侧网络

一句话总结

本文提出 PredFT,一个结合主网络(语言解码)和侧网络(脑预测编码表征)的端到端 fMRI-to-Text 解码模型,通过从大脑预测相关脑区(PTO 区域)提取前瞻性语义表征并融合到解码过程中,在 LeBel 数据集上 BLEU-1 达 34.95%(Sub-1),相比最强基线 MapGuide 提升 7.84 个百分点。

研究背景与动机

领域现状:从 fMRI 信号重建自然语言是理解人脑语言形成机制的重要窗口。近年研究利用预训练语言模型实现了开放词汇的 fMRI-to-Text 解码:Tang 等人用 GPT 生成语义候选再用脑信号选择匹配内容,Xi 等人将问题转化为序列到序列翻译。

现有痛点:现有研究专注于模型架构设计和语言模型利用,但忽略了一个关键的神经科学基础——自然语言在人脑中是如何编码的。具体来说,大脑在感知当前语音刺激时会自然地对未来内容进行多时间尺度的预测(预测编码理论),但这一信息从未被用于指导语言重建。

核心矛盾:脑信号中包含丰富的前瞻性预测信息,但现有解码模型仅利用了当前时刻的脑活动表征,浪费了大脑自然产生的预测信号。

本文目标:(1) 验证预测编码理论在 fMRI-to-Text 解码中的可行性;(2) 设计能有效利用脑预测表征的解码模型;(3) 分析不同脑区、预测距离和长度对解码性能的影响。

切入角度:预测编码理论指出大脑在听到语音时会自然预测未来词汇。Caucheteux 等人已证明如果用预测内容构建语言模型表征,语言模型激活与脑响应之间的线性映射会增强。这启发我们:能否从脑信号中提取预测表征来辅助语言重建?

核心 idea:设计双网络架构——主网络负责标准的 fMRI-to-Text 解码,侧网络从脑预测相关脑区(PTO 区域)提取前瞻性表征,通过 Predictive Coding Attention 将预测信息融合到解码过程中。

方法详解

整体框架

PredFT 是端到端模型,包含主网络 \(\mathcal{M}_\theta\)(编码器-解码器)和侧网络 \(\mathcal{M}_\phi\)(编码器-解码器)。主网络将 fMRI 序列编码为时空特征后用 Transformer 解码器生成文本;侧网络从预测相关脑区(ROIs)提取表征并通过自注意力融合,其编码器输出 \(H_{\phi_\text{Enc}}^M\) 作为预测表征注入主网络。推理时侧网络解码器被丢弃。

关键设计

  1. 主网络编码器(fMRI 特征提取 + 时序建模):

    • 功能:从原始 fMRI 信号中提取空间-时间特征
    • 核心思路:对 4D 体积 fMRI 图像 \(F_{i,j} \in \mathbb{R}^{w \times h \times d \times (k+1)}\),使用 \(L\) 层 3D-CNN(含组归一化、ReLU、残差连接)逐步降维至一维向量 \(x_{i,j}^t \in \mathbb{R}^{d_m}\);对 2D 表面 fMRI 直接用线性层降维。随后应用 FIR 模型 \(g_t\) 补偿 BOLD 信号延迟,拼接 \(k-k^*\) 个未来帧再线性融合。最后加时间位置编码送入 Transformer 编码器捕捉时间依赖
    • 设计动机:fMRI 的 BOLD 信号有约 4-6 秒延迟,FIR 模型的时间补偿对正确对齐脑活动与语音至关重要
  2. 侧网络(脑预测表征提取):

    • 功能:从预测相关脑区提取前瞻性语义表征
    • 核心思路:侧网络编码器 \(\mathcal{M}_{\phi_\text{Enc}}\) 接收预测相关 ROIs 序列 \(R_{i,j}\)(由 STS、IFG、SMG、Angular Gyrus 等脑区拼接),经全连接层降维、FIR 补偿、位置编码后送入 Transformer 编码器,输出预测表征 \(H_{\phi_\text{Enc}}^M\)。侧网络解码器以未来词 \(V_j\)(距离 \(d\)、长度 \(l\) 的预测目标)为输入,通过交叉熵损失训练编码器学习预测表征。推理时解码器丢弃
    • 设计动机:预测编码验证实验表明,PTO 区域的预测分数显著高于全脑或随机 ROIs,选择正确的脑区对提取有效预测信号至关重要
  3. 预测编码注意力(Predictive Coding Attention):

    • 功能:将侧网络的预测表征融合到主网络解码器中
    • 核心思路:在主网络 Transformer 解码器的每层中增加 PC-Attention 模块,以解码器隐状态 \(H_{\theta_\text{Dec}}^l\) 为 query,以侧网络编码器输出 \(H_{\phi_\text{Enc}}^M\) 为 key 和 value。关键是注意力掩码的设计:对文本片段 \(u_j^t\) 中的每个 token,允许关注时间步 \(t\) 之后的所有预测表征,屏蔽之前的——因为预测信息应该来自当前时刻之后的脑活动
    • 设计动机:掩码设计确保了因果性——当前词的解码只利用未来的预测信息,符合预测编码的"前瞻"本质

损失函数 / 训练策略

端到端联合训练,总损失 \(\mathcal{L} = \mathcal{L}_\text{Main} + \lambda \mathcal{L}_\text{Side}\)。两个网络共享词嵌入层(仅由主网络梯度更新),分别用左到右自回归交叉熵损失。推理时侧网络解码器被丢弃,仅保留编码器提供预测表征。

实验关键数据

主实验

LeBel 数据集 within-subject 解码(10 帧 = 20 秒)

模型 BLEU-1 BLEU-4 ROUGE1-F BERTScore
Tang's 22.25 0.00 19.44 80.84
BrainLLM 24.18 1.11 21.16 83.26
MapGuide 27.11 1.54 24.83 82.66
PredFT w/o SideNet 27.91 1.29 26.82 81.35
PredFT 34.95 1.78 32.03 82.92

Narratives 数据集 cross-subject 解码

长度 模型 BLEU-1 ROUGE1-F BERTScore
10帧 UniCoRN 20.64 19.23 75.35
10帧 PredFT 24.73 19.53 78.52
40帧 UniCoRN 21.76 25.30 74.40
40帧 PredFT 27.80 25.96 78.63

消融实验

ROIs 选择对解码性能的影响(LeBel 数据集)

ROIs 类型 说明 相对性能
BPC(预测相关脑区) STS, IFG, SMG, Angular Gyrus 最优
Whole(全脑) 整个大脑皮层 次优
Random(随机) 随机选择脑区 最差

关键发现

  • 侧网络贡献显著:PredFT 相比 w/o SideNet 在 Sub-1 上 BLEU-1 从 27.91 提升到 34.95(+7.04),证明脑预测信息对解码有实质帮助
  • ROIs 选择至关重要:BPC 区域(PTO)一致优于全脑和随机 ROIs,验证了预测编码的脑区特异性
  • 预测长度和距离存在最优区间:过短(\(l=1,2\))或过长(\(l=11,12\))的预测长度都不理想,中等长度(\(l=6,7,8\))配合适当距离(\(d=3-5\))效果最佳
  • 被试内解码显著优于跨被试解码,所有模型在长文本生成(BLEU-3/4)上仍然困难

亮点与洞察

  • 将神经科学的预测编码理论直接转化为模型设计是非常优雅的跨学科创新——侧网络的"辅助训练、推理丢弃"策略类似知识蒸馏
  • PC-Attention 的因果掩码设计简洁有力——只让当前词关注未来的预测表征,完美对应预测编码的前瞻性质
  • 预测编码验证实验本身就有独立价值——系统展示了脑区、预测距离和长度的交互关系

局限与展望

  • 仅在 fMRI 数据上验证,其他脑信号模态(MEG、EEG)的适用性未探索
  • 被试未预期的内容可能干扰大脑预测功能,影响解码效果
  • 所有模型在长精确文本生成上仍有很大提升空间(BLEU-4 普遍低于 2%)
  • 可扩展到视觉刺激的脑信号解码场景

相关工作与启发

  • vs Tang's: Tang 用 GPT beam search 生成候选再选择,PredFT 是端到端解码,且利用了脑预测信息
  • vs UniCoRN: UniCoRN 用 BART 的三阶段训练框架,PredFT 通过侧网络引入预测编码先验,在跨被试设置上 BLEU-1 提升 6+ 个百分点
  • vs BrainLLM: BrainLLM 将 fMRI 嵌入与词嵌入拼接微调 Llama2,PredFT 通过独立的预测网络提供更有针对性的辅助信号

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次将预测编码理论系统化地应用于 fMRI-to-Text 解码,跨学科创新突出
  • 实验充分度: ⭐⭐⭐⭐ 两个数据集、多被试、ROIs 分析和预测参数分析全面,但缺少人工评估
  • 写作质量: ⭐⭐⭐⭐ 预测编码验证到模型设计的逻辑推导清晰流畅
  • 价值: ⭐⭐⭐⭐ 为脑机接口领域提供了有理论支撑的新方法,验证了脑预测信息的实用价值

相关论文