← 返回 2026-04-21

MNAFT:面向图像翻译任务的多模态大语言模型模态神经元感知微调 MNAFT: modality neuron-aware fine-tuning of multimodal large language models for image translation

Bo Li, Ningyuan Deng, Tianyu Dong, Shaobo Wang, Shaolin Zhu, Lijie Wen 📅 2026-04-18 👍 2 2026-07-13 08:36
参数高效微调 图像翻译 多模态大模型 神经元可解释性 跨语言OCR

用一阶 Taylor 展开精准识别 MLLM 中语言特异与跨模态共享神经元,仅微调关键子集实现图像翻译 SOTA。

前置知识

多模态大语言模型 (MLLM)

由视觉编码器(如 CLIP/ViT)、连接投影层和预训练 LLM 拼接而成的统一模型,能同时理解图像与文本。代表模型包括 Qwen2.5-VL、LLaVA-NeXT、InternVL2 等。视觉特征经连接器映射到 LLM 的词嵌入空间后,与文本指令一起做自回归生成。

本文的核心对象就是 MLLM,提出的方法依赖其内部既有视觉模块也有语言模块的结构,并对二者内部的神经元分别做重要性分析,因此理解 MLLM 架构是看懂 MNAFT 切分视觉/语言层的前提。

图像翻译 (Image Translation, IT)

输入一张含有源语言文字的图像,直接输出目标语言译文。它与传统机器翻译不同,文本仅以像素形式存在,模型需先'看清'再'翻译'。典型范式有级联(OCR+MT)和端到端两类。

这是 MNAFT 的目标任务。论文的对照实验、低资源语言表现和案例分析都围绕 IT 展开,理解任务定义能帮助把握为何需要专门保留'语言特异'神经元。

Taylor 展开近似神经元重要性

对损失函数 $\mathcal{L}$ 在神经元激活 $h_i$ 处做一阶展开得到 $\Delta L(h_i) \approx -\frac{\partial \mathcal{L}}{\partial h_i} h_i$,其绝对值即近似为屏蔽该神经元后损失的变化量,可作为显著性/重要性分数。

MNAFT 的核心打分函数 $\Phi(i) = |\Delta L(h_i)| = |\frac{\partial \mathcal{L}}{\partial h_i} h_i|$ 直接源于此式,是筛选语言特异/共享神经元、挑选关键层的数学基础。

参数高效微调 (PEFT) 与 LoRA

PEFT 通过冻结大部分预训练参数、仅更新少量新增/选定参数来降低微调成本。LoRA 在权重矩阵旁注入低秩分解 $W + BA$;DoRA 把权重拆成 magnitude 和 direction;MixLoRA/M2PT 在多模态场景下做了扩展。

MNAFT 是一类新型 PEFT,但它选择的不是 LoRA 这种参数结构,而是基于神经元功能本身做剪枝式选择,需要读者把它的设计与 LoRA/DoRA/MixLoRA/M2PT 放在同一坐标下比较。

梯度掩码 (gradient mask)

在前向/反向传播后,仅对被选中的参数位置应用梯度(即与一个 0/1 掩码 $M_l$ 做 element-wise 乘积),其余位置梯度被置零,从而实现'冻结'。形式上 $W_{t+1} = W_t - \alpha \cdot (\nabla_{W_l} L_t \odot M_l)$。

MNAFT 正是通过逐层掩码 $M_l[i] = 1$ 当且仅当 $i \in A_l \cup S_{l,t}$ 来实现'只更新语言特异+跨模态共享神经元'。理解掩码机制能解释为何 MNAFT 既减少更新量,又不破坏原始前向计算。

研究动机

现有多模态大模型(InternVL2、LLaVA-NeXT、Qwen2.5-VL 等)在图像翻译上仍受限于一个根本问题——视觉文本与文本输入/输出之间的模态鸿沟。传统级联方案(OCR + NMT,如 EasyOCR + NLLB-200、PP-OCRv3 + NLLB)因为 OCR 错识直接传导到翻译阶段,在 ECOIT(中文电商图片)上 METEOR 仅 13.1~13.7,与端到端模型 70+ 的水平相差悬殊。即便换成 MLLM 直接做端到端,主流做法(如 LLaVA-NeXT 路线)仍然是对全部参数做 full fine-tuning,没有利用神经元在不同语言/模态间的功能分化,导致两类后果:一是破坏预训练学到的通用知识,二是不同语言间出现参数干扰,多语言互相'抢'容量,论文在 MIT-10M (EN-DE) 上 full FT 反而比 Qwen2.5-VL-3B 的 zero-shot 还低 13 METEOR。LoRA/DoRA/MixLoRA/M2PT 等 PEFT 虽缓解了参数量,但它们共享同一套低秩参数去拟合所有任务,论文作者认为这也是一种'参数冗余'。

本文的目标是MNAFT 的目标是把 MLLM 内部的视觉层和语言层都纳入考量,先精确识别'对某图像翻译任务真正重要的神经元',再只更新这一小部分,从而在六个任务、四个数据集上同时取得比 full FT、LoRA、DoRA、MixLoRA、M2PT 以及六种专用 IT 模型(ItNet、PEIT、Translatotron-V、UMTIT、E2ETIT、DIMTDA)更优的 BLEU/METEOR,同时在 ECOIT 上把训练时间从 9.2 h 压到 7.0 h,显存从 94.3 GB 压到 20.8 GB。

与已有工作不同的是,切入角度有三层独特性:(1) 已有工作要么做整层全更新(full FT),要么做结构化低秩注入(LoRA/DoRA),没有人在 MLLM 内部以'单神经元 × 单任务'粒度做显著性评估;(2) 神经元识别用的是'指令驱动 + 一阶 Taylor 展开',与语言神经元、模型剪枝文献的纯激活幅度或纯梯度方法都不同——前者可能高估高激活但低敏感度的节点,后者可能忽略真正有贡献的节点,$\Phi(i) = |\frac{\partial \mathcal{L}}{\partial h_i} h_i|$ 同时把激活幅度和损失敏感度乘起来;(3) 既考虑视觉层也考虑语言层,并显式区分 language-agnostic 与 language-specific 两类神经元,把它们分到两个集合 $A_l$ 和 $S_{l,t}$,再分别决定是否更新,这是 MixLoRA/M2PT 等'跨模态 prompt'路线所没有的细粒度。

核心方法

MNAFT 的整体思路是'先画功能地图,再做选择性更新'。它把图像翻译任务拆成两步:先用不同指令(如'图片里的文字是什么?''把德语翻成英语')驱动 Qwen2.5-VL 跑一遍前向+反向,在每一层对每个神经元收集激活 $h_i$ 和梯度 $\partial \mathcal{L}/\partial h_i$,据此算一阶 Taylor 重要性分数 $\Phi(i)$;再按层聚合得到 $\Theta_{TE}^{(l)}$,挑出对图像翻译最关键的若干视觉层 $L_{vision}$ 和语言层 $L_{LLM}$;进入层后,对神经元按方差排序,用阈值 $\lfloor \epsilon \times p \rfloor$ 把它们切成'通用/跨语言'和'语言特异'两组;最后在微调时只对 $A_l \cup S_{l,t}$ 应用梯度掩码,其余参数冻结。直觉上,这相当于在微调前先做一次'功能性体检',只针对真正会响应的神经元做手术,避免大动干戈。

核心创新在于把'哪些参数该动'的选择,从'所有参数/低秩增量/跨模态 prompt'升级为'按神经元功能分桶'。已有 LoRA/DoRA/MixLoRA 都假设'同一组参数要服务所有任务',M2PT 加了视觉/语言 soft prompt,但仍是在共享参数上拟合;MNAFT 显式承认'某些神经元就是为某种语言/某类视觉特征而存在',并用基于 Taylor 展开的显著性同时融合激活和梯度,把这些神经元挑出来独立微调。这本质上把'参数效率'问题从结构层面(rank、prompt 长度)下沉到了功能层面(哪些神经元是任务相关的)。

方法步骤详情

方法可分四步。第一步是任务形式化:给定数据集 $\mathcal{D}=\{(v_i, s_i, t_i)\}_{i=1}^N$,图像 $v_i$ 含源语言文字,$s_i$ 是抽出的源文(用作辅助),$t_i$ 是目标译文,损失 $\mathcal{L}_{IT} = -\frac{1}{N}\sum_i \log p(t_i|v_i;\theta)$。第二步是神经元打分:对每个神经元 $i$ 算 $\Phi(i) = |\Delta L(h_i)| = |\frac{\partial \mathcal{L}(H,0)}{\partial h_i} h_i|$,其推导基于一阶 Taylor 展开,在 $a=0$ 点展开后用 $\Delta L = -\frac{\partial \mathcal{L}}{\partial h_i} h_i$ 近似。第三步是层+神经元的双层筛选:先按层聚合 $\Theta_{TE}^{(l)} = \frac{1}{T_m} \sum_t \frac{\partial \mathcal{L}(H, h_i^{(l)})}{\partial h_i^{(l)}} h_i^{(l)}$,排序后用 top-$l$ 选出 $L_{vision}$ 和 $L_{LLM}$($l_{vision}$ 和 $l_{LLM}$ 是超参);再在选中的层内按方差排序,方差低于阈值 $\lfloor \epsilon \times p \rfloor$ 的归为通用神经元 $A_l$,高于阈值的归为语言特异神经元 $S_{l,t}$,按目标语言分桶。第四步是选择性微调:构造掩码 $M_l[i] = 1$ 当且仅当 $i \in A_l \cup S_{l,t}$,在反向时把梯度与 $M_l$ 做 element-wise 乘积后送入优化器,从而只更新 $W_l$ 和 $b_l$ 中被选中的位置,其它参数冻结。

技术新颖性

技术新颖性体现在三点。一是把一阶 Taylor 展开用在'识别'而非'剪枝',且明确区分了跨语言通用和语言特异两类神经元,这套语义切分在以往 IT 工作里没有先例。二是用同一显著性分数同时指导'挑哪一层'和'挑层内哪些神经元',把层选择和参数选择统一在一个 Taylor 框架下,避免了 LoRA 路线人工设定 target modules 的经验性。三是在端到端 IT 上首次同时对视觉层和语言层做非对称选择性微调(视觉层按层选、语言层按层+按神经元选),并通过在多个模型规模(3B/7B)和不同架构(Qwen2.5-VL、LLaVA-NeXT)上的实验证明这种'功能图谱'具有跨架构泛化能力。

The overview of MNAFT Framework.
Figure 1: The overview of MNAFT Framework.
Prompt templates.
Figure 2: Prompt templates.

实验结果

实验在 4 个数据集(ECOIT 中文电商、IIMT 合成图文、MIT-10M 真实多语、OPUS-MIT-5M)、6 个任务上展开,主表(Table 1)显示 MNAFT 在所有 6 个任务上 BLEU/METEOR 都取得最佳。在 ECOIT(ZH-EN)上 MNAFT 取得 METEOR 75.1 / BLEU 54.6,比最强 baseline Qwen2.5-VL-3B + Full FT(68.5/51.4)提升 6.6/3.2 BLEU,比 MixLoRA 提升约 14 METEOR、10 BLEU,比 cascaded EasyOCR + NLLB(13.7/8.0)提升超过 5× METEOR。IIMT(DE-EN)上 MNAFT 67.9/38.0,超过 Full FT 13.3 METEOR、1.5 BLEU,超过 LoRA 11 METEOR。IIMT(FR-EN)上 67.0/45.0,超过 Full FT 7.1 METEOR。MIT-10M(DE-EN)提升最显著,从 Full FT 的 62.6/49.6 跃升到 79.8/56.8,幅度 +17.2/+7.2,显示真实数据上神经元级选择更有效。Table 2 与 SOTA 专用 IT 模型比较:ECOIT(ZH-EN)上比 Translatotron-V(73.1/52.6)高 2.0/2.0,IIMT(DE-EN)上比 UMTIT(54.6/36.1)高 13.3/1.9。Table 3 消融显示,去掉通用神经元微调在 ECOIT 上掉到 51.8/29.1(-23.3/-25.5),只微调特异神经元 61.7/42.7(-13.4/-11.9),所有层全 FT 61.6/45.8(-13.5/-8.8),只看语言层 58.7/42.3,只看视觉层 57.3/38.2,证明'两类神经元 + 双层选择'缺一不可。Table 4 算力开销:MNAFT 在 ECOIT 上 7.0 h/20.8 GB,相比 Full FT 9.2 h/94.3 GB,训练时间 -24%、显存 -78%,相比 LoRA 8.6 h/12.2 GB 时间 -18% 但显存多 8.6 GB。Figure 6 的雷达图说明 7B/8B 上 MNAFT 仍稳定优于 Full FT 和 LoRA。Figure 7 的三个 case 分别展示 MNAFT 在避免幻觉(Case 1)、识别专业术语'Genuine Leather → Echtes Leder'(Case 2)、捕捉代词消歧(Case 3'这些路线被用来对付他们')方面的定性优势。

Quantitative comparison with three categories of baseline methods.
Table 1: Quantitative comparison with three categories of baseline methods.
Quantitative comparison with the SOTA IT models.
Table 2: Quantitative comparison with the SOTA IT models.
Results of the ablation study comparing our method with different model variants on OPUS-MIT-5M (EN-ZH) and ECOIT (ZH-EN).
Table 3: Results of the ablation study comparing our method with different model variants on OPUS-MIT-5M (EN-ZH) and ECOIT (ZH-EN).
Computational cost comparison on the ECOIT and OPUS-MIT-5M datasets.
Table 4: Computational cost comparison on the ECOIT and OPUS-MIT-5M datasets.
Average Activation. (a) Language Model Layer Average Activation. (b) Vision Model Layer Average Activation.
Figure 3: Average Activation. (a) Language Model Layer Average Activation. (b) Vision Model Layer Average Activation.
Delta Average Activation. (a) Language Model Layer Delta Average Activation. (b) Vision Model Layer Delta Average Activation.
Figure 4: Delta Average Activation. (a) Language Model Layer Delta Average Activation. (b) Vision Model Layer Delta Average Activation.
Clustering of representations. (a) Language Model Layer specific neurons. (b) Language Model Layer general neurons. (c) Vision Model Layer specific neurons. (d) Vision Model Layer general neurons.
Figure 5: Clustering of representations. (a) Language Model Layer specific neurons. (b) Language Model Layer general neurons. (c) Vision Model Layer specific neurons. (d) Vision Model Layer general neurons.
Comparison of METEOR with six datasets for fine-tuning the Qwen2.5-VL-7B (a) and LLaVA-NeXT-LLaMA3 (8B) (b) using Full Fine-tuning, LoRA and MNAFT (Ours).
Figure 6: Comparison of METEOR with six datasets for fine-tuning the Qwen2.5-VL-7B (a) and LLaVA-NeXT-LLaMA3 (8B) (b) using Full Fine-tuning, LoRA and MNAFT (Ours).
Case Study.
Figure 7: Case Study.
查看结构化数据
任务指标本文基线提升
ECOIT 中文电商图像翻译(ZH-EN) METEOR / BLEU 75.1 / 54.6 Translatotron-V 73.1 / 52.6;Qwen2.5-VL-3B + Full FT 68.5 / 51.4;EasyOCR+NLLB 13.7 / 8.0 比最强专用 IT 模型 +2.0 / +2.0,比 Full FT +6.6 / +3.2,比级联方案 +5.5× / +6.8×
IIMT 德英合成图像翻译(DE-EN) METEOR / BLEU 67.9 / 38.0 UMTIT 54.6 / 36.1;M2PT 61.0 / 34.0;Full FT 54.6 / 36.5 比 UMTIT +13.3 / +1.9,比 Full FT +13.3 / +1.5
IIMT 法英合成图像翻译(FR-EN) METEOR / BLEU 67.0 / 45.0 M2PT 64.3 / 43.2;Full FT 59.9 / 42.3 比 M2PT +2.7 / +1.8
MIT-10M 真实多语图像翻译(DE-EN) METEOR / BLEU 79.8 / 56.8 Full FT 62.6 / 49.6;LoRA 57.1 / 40.6;Text-Only 63.2 / 50.7 比 Full FT +17.2 / +7.2(提升最大)
MIT-10M(EN-DE)反向翻译 METEOR / BLEU 54.7 / 42.0 Full FT 50.0 / 37.4;M2PT 49.7 / 36.3 比 Full FT +4.7 / +4.6
OPUS-MIT-5M 英中翻译(EN-ZH) METEOR / BLEU 75.2 / 60.7 LoRA 66.7 / 51.8;DoRA 66.8 / 51.9;MixLoRA 66.6 / 53.4;M2PT 66.1 / 53.1;Full FT 65.5 / 50.3 比 MixLoRA +8.6 / +7.3,比 LoRA +8.5 / +8.9
训练效率(ECOIT) 时间 / 显存 7.0 h / 20.8 GB Full FT 9.2 h / 94.3 GB;LoRA 8.6 h / 12.2 GB 比 Full FT 时间 -24%、显存 -78%;比 LoRA 时间 -18%(显存多 8.6 GB)

局限与改进

作者在 6.5 节明确指出一项实现层面的限制:梯度掩码发生在优化器更新阶段,前向和反向计算本身仍然遍历所有参数,因此显存节省主要来自 optimizer state 减少,而不是稀疏反向传播。这意味着'冻结'的神经元在反向时仍占用计算,未来若支持稀疏反向还能再省一波。Table 3 的'All Layers FT'(即不做神经元选择)相比 MNAFT 在 OPUS-MIT-5M 上掉 8.0/7.4、在 ECOIT 上掉 13.5/8.8,说明神经元选择机制是性能核心,但同时也意味着方法对'哪些神经元被选出来'非常敏感——如果显著性分数估计不准,性能会显著退化。我自己的观察还有三点:(1) 神经元识别阶段在测试集上做(论文 5.1.1 称'不涉及参数更新,类似校准式剪枝'),但用测试集选神经元再在训练集上微调,理论上存在轻微的信息泄露风险,复现时建议替换为独立 calibration set;(2) Table 1 中 MIT-10M (EN-DE) 的 zero-shot/CoT 分数(31.3/16.9 和 16.9/11.9)异常低,明显低于 Text-Only(63.6/55.1),意味着在多语真实数据上 MLLM 的视觉编码可能反而误导翻译,但论文没有进一步分析;(3) Figure 3/4 的层激活图没有给出哪些具体层被选为 top-$l$,读者无法复现'挑哪几层'的决定,只能靠论文给定的超参 $l_{vision}$/$l_{LLM}$,可解释性还不够彻底。

独立分析的弱点

独立分析可以列出三处可改进的弱点。(1) 神经元粒度的更新引入了大量不规则访存——只在被选中的索引位置做梯度写入和 optimizer step,在通用 GPU 上反而可能降低硬件利用率,这能解释为什么 MNAFT 时间比 LoRA 短(LoRA 增加额外 matmul)但比 naive 实现未必有压倒性优势,未来可通过 block-sparse 算子或把被选神经元重排成连续子矩阵来优化。(2) Taylor 展开只用到一阶信息,没有考虑神经元之间的交互(一个神经元被屏蔽后其它神经元的激活会重分布),对强耦合的中间层可能高估单个神经元的重要性,建议引入二阶项 $\frac{1}{2}\frac{\partial^2 \mathcal{L}}{\partial h_i^2} h_i^2$ 或集成 Hessian 迹作为补充。(3) 任务特异性过强——选出的 $S_{l,t}$ 是按目标语言/任务分桶的,意味着换语言就要重新做一次神经元识别(2~3 分钟虽然短,但仍是 offline 流程),无法像 LoRA 那样'一组参数跨任务复用',限制了在快速切换多语翻译的工业场景中的实用性。改进方向:构建可迁移的'神经元路由表',让一次识别支持多个相关任务。

未来方向

作者在第 7 节明确给出了几条未来方向:把 MNAFT 推广到图像描述(image captioning)、VQA、人机对话等其它多模态任务;研究不同任务间学到的'神经元特化'是否可迁移;以及通过稀疏反向传播进一步降低算力成本。基于论文成果还可以延伸的方向包括:(1) 把 MNAFT 的功能地图机制和 MoE(Mixture of Experts)结合,让路由网络本身就用 Taylor 显著性训练,避免专家数量靠经验设定;(2) 把语言特异神经元用于多语增量学习——新增一个低资源语言时,只需在已被识别的'通用神经元'上 finetune,而把'特异神经元'留给新语言,预期能缓解 catastrophic forgetting;(3) 结合可解释性研究,分析 language-agnostic 神经元到底编码了哪些跨语言共享的语义概念(句法骨架、命名实体槽位等),为构建更结构化的多语翻译系统提供模块化基础。

复现评估

复现评估整体可行。论文给出了详尽超参:使用 Qwen2.5-VL-3B/7B 和 LLaVA-NeXT-LLaMA3(8B)作为基座模型,硬件为 NVIDIA A100-80G,CUDA 12.2、torch 2.1.0、transformers 4.45.0,微调框架用 LLaMA-Factory。代码层面论文虽未明确标注开源链接,但方法在 LLaMA-Factory 上做选择性更新只需在 trainer 的 optimizer step 中插入梯度掩码,实现门槛中等。数据方面,ECOIT/IIMT/MIT-10M/OPUS-MIT-5M 全部公开(OPUS-MIT-5M 在 HuggingFace 上可下载),每任务用 100K 训练 + 100 测试,神经元识别阶段作者在测试集上做(需注意可能的信息泄露),复现时建议替换为独立 calibration 集。算力需求:单卡 A100-80G 即可跑 3B 模型(峰值 20.8 GB),7B/8B 模型需要更多显存;Full FT 基线在 80G 卡上仅 3B 模型就要 94.3 GB。复现难度主要在两点:一是 $\epsilon$ 和 $l_{vision}$/$l_{LLM}$ 这几个超参论文没有给最优值搜索细节,需要在验证集上调;二是梯度掩码要确保在分布式训练(DeepSpeed/FSDP)下不被分片策略破坏,建议在单机单卡先验证再扩展。