PRISM: Self-Pruning Intrinsic Selection Method for Training-Free Multimodal Data Selection¶
会议: ACL2026 Best Paper
arXiv: 2502.12119
代码: 有,cache 未提供明确 URL
领域: 多模态VLM / 数据选择
关键词: 多模态数据选择, 视觉指令微调, 表征各向异性, 训练免筛选, 冗余剪枝
一句话总结¶
PRISM 发现 MLLM 视觉特征的非零均值会造成 Global Semantic Drift,从而污染基于相似度的数据选择,并用训练免的均值重中心化和低相关样本筛选,在只保留约 30% 视觉样本的情况下达到 101.7% 相对性能,同时把端到端 GPU 时间降低约 70%。
研究背景与动机¶
领域现状:多模态大模型通常先做大规模图文预训练,再用视觉指令数据进行 instruction tuning。随着 LLaVA、VisionFlan 等数据池不断扩大,指令样本数量很多,但其中也包含大量重复、低信息量或噪声样本。
现有痛点:现有视觉指令数据选择方法大多依赖 proxy model、外部 scorer、训练 loss、perplexity、gradient 或 influence function。这些方法要么需要额外模型推理,要么需要迭代训练和梯度计算,筛数据本身就很贵,甚至抵消了少训练数据带来的效率收益。
核心矛盾:数据选择本来是为了省算力,但很多选择器把成本转移到了选择阶段。作者认为根因在于大家直接使用 MLLM 视觉特征的原始几何,而这些视觉 embedding 并非均匀分布在原点附近,而是被一个强全局均值方向拉进狭窄锥体,导致 cosine similarity 把共享背景漂移误当成语义相似。
本文目标:提出一种无需 proxy、无需训练、无需梯度的 multimodal instruction selection 方法,既能保持甚至超过 full fine-tuning 性能,又能真正降低 selection + tuning 的总成本。
切入角度:论文从表征几何诊断入手,先证明视觉特征存在 representation anisotropy 和奇异值集中,再把数据选择问题改写成“先去掉全局漂移,再基于样本的独特语义分量估计冗余”。
核心 idea:用目标 MLLM 自身视觉特征的均值重中心化,恢复更可靠的相似度几何,然后保留与全体样本低相关、信息更独特的样本。
方法详解¶
PRISM 的关键不在于训练一个更强选择器,而是让原本已经存在于 MLLM 中的视觉表征重新变得可用。它把选择流程做成单次特征抽取、全局均值估计、冗余评分和百分位筛选四步,因此成本主要是一次前向和线性聚合。
整体框架¶
给定视觉指令数据集 \(D=\{d_1,\dots,d_N\}\),每个样本包含图像和文本指令。PRISM 首先用目标 MLLM 的视觉编码器、投影器和 LLM 中间层抽取每个样本的视觉特征 \(F_i\),并通过 average pooling 得到一个全局图像表示。
然后,PRISM 计算整个语料的特征均值 \(\mu_F=\frac{1}{N}\sum_i F_i\)。这个均值代表所有视觉样本共享的 global drift,而不是单个样本独有语义。对于任意两个样本,PRISM 不再用 raw cosine,而是先做 \(F_i-\mu_F\) 和 \(F_j-\mu_F\),再计算归一化内积。
接着,每个样本的 Redundancy Score 是它和其他所有样本中心化后相关性的平均值。直觉上,如果一个样本和大量样本都高度相关,它更可能是重复或低边际收益样本;如果相关性低,它可能携带更独特的视觉语义。
最后,根据选择预算 \(\tau\) 取冗余分数最低的一部分样本。主实验中,PRISM 对 LLaVA-665K 的视觉样本使用 30% 预算,同时保留 text-only 样本,最终得到 PRISM-Instruct-250K。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400, 'subGraphTitleMargin': {'top': 8, 'bottom': 16}}}}%%
flowchart TD
A["视觉指令数据集 D(图像 + 指令)"] --> B["视觉特征抽取<br/>MLLM 浅层 + average pooling 得 F_i"]
B --> RC
subgraph RC["视觉表征各向异性诊断与重中心化"]
direction TB
C["估计语料均值 μ_F = (1/N)·ΣF_i<br/>剥离全局漂移方向"] --> D["中心化相关性 ρ 替代 raw cosine"]
end
RC --> RS
subgraph RS["低相关冗余分数筛选"]
direction TB
E["冗余分数 R 取平均相关强度<br/>O(Nd) 聚合,免 N×N 矩阵"] --> F["按预算 τ 留冗余最低的样本"]
end
RS --> G["PRISM-Instruct<br/>约 30% 视觉样本 + 保留 text-only"]
注:OSC(Overall Selection Cost)是评价"选择器到底真省了没有"的端到端尺子,是驱动上述训练免流程的动机,不属于选择 pipeline 的某个阶段,故不入图。
关键设计¶
1. Overall Selection Cost 评价准则:逼问一个数据选择器到底"真省了没有"
很多选择方法只报告 subset 上的性能,却把昂贵的 selection overhead 藏起来不算,于是出现"筛得很准,但加上选择成本比直接全量训练还贵"的尴尬。PRISM 先立一把诚实的尺子:把性能比和时间比相乘,
只有当 \(C<1\) 时,一个方法才同时做到了不掉性能且端到端净省算力。这个指标把比较强行拉回端到端成本,也正是 PRISM 后面坚持"训练免、单次前向"的根本动因。
2. 视觉表征各向异性诊断与重中心化:先修好相似度,再谈选数据
作者发现根因不在选择器不够强,而在大家直接用了 MLLM 视觉特征的原始几何——这些视觉 embedding 并非均匀分布在原点附近,而是被一个强全局均值方向拉进狭窄锥体,使得 cosine similarity 把"共享背景漂移"误当成"语义相似"。把视觉特征分解为 \(x_i=\mu+\delta_i\)(\(\mu\) 是全体共享的 global component,\(\delta_i\) 才是样本特有语义),当 \(\|\mu\|\) 远大于 \(\|\delta_i\|\) 时,raw cosine 会被 \(\mu\) 主导而趋近 1。PRISM 的对策是先估计语料均值 \(\mu_F=\frac{1}{N}\sum_i F_i\),再用中心化后的相关性
替代 raw cosine。值得一提的是,作者明确说这不是 full whitening,而只针对最破坏冗余估计的 corpus-level mean shift——与其引入外部 scorer,不如先修掉目标模型自身特征空间里最显著的一阶几何偏差。
3. 低相关冗余分数筛选:用一阶聚合找出语义最独特的样本
把相似度修好之后,还得在几十万样本规模上算出谁冗余、谁独特,而构造 \(N\times N\) 相似度矩阵或跑精确 greedy coverage 都贵到不可承受。PRISM 给每个样本定义冗余分数
即它在中心化语义图中与其余样本的平均连接强度:分数高说明和一大票样本都高度相关、多半是重复或低边际收益样本,分数低则意味着携带更独特的视觉语义、训练价值更高。借助 exact aggregate implementation,这个分数能以 \(O(Nd)\) 算出而无需 materialize 全量 pairwise 矩阵,最后按选择预算 \(\tau\) 保留冗余分数低于百分位阈值的那部分样本即可。
损失函数 / 训练策略¶
PRISM 本身没有训练损失,是 training-free selector。选完数据后,作者按 LLaVA-1.5 官方超参进行一轮 visual instruction tuning。主设置使用 LLaVA-665K 和 LLaVA-1.5-7B,并与 Random、Length、EL2N、Perplexity、GraNd、TIVE、InstructionGPT-4、Self-Filter、COINCIDE、ICONS、DataTailor 等方法比较。额外实验在 VisionFlan-186K、不同 MLLM 架构和 text-only benchmark 上验证泛化与知识保留。
实验关键数据¶
主实验¶
PRISM 在 LLaVA-1.5-7B 上相对 full fine-tuning 达到 101.7% 综合性能,同时在多个多模态 benchmark 上超过全量数据或强选择方法。
| 方法 | SQA | SQA-I | VizWiz | POPE-P/R/A | MM-Vet | MMBench | MME-C | MMMU | Rel. |
|---|---|---|---|---|---|---|---|---|---|
| Full-Finetune | 69.4 | 66.8 | 50.0 | 86.1 / 87.3 / 84.2 | 31.1 | 64.3 | 311.9 | 35.4 | 100% |
| TIVE | 72.2 | 70.6 | 未列出 | 85.6 / 85.6 / 85.6 | 未列出 | 63.2 | 322.1 | 未列出 | 100.6% |
| ICONS | 未列出 | 70.8 | 未列出 | 87.5 / 87.5 / 87.5 | 未列出 | 63.1 | 未列出 | 未列出 | 101.0% |
| PRISM | 71.3 | 69.1 | 50.1 | 87.7 / 88.7 / 85.5 | 32.0 | 65.2 | 330.0 | 34.7 | 101.7% |
在 VisionFlan-186K 上,PRISM 用同样 30% 预算选择 57K 样本,仍超过 full-data aggregate,并显著强于随机选择。
| 方法 | 样本量 | VizWiz | SQA-I | TextVQA | POPE | MME | MMBench | Rel. |
|---|---|---|---|---|---|---|---|---|
| Full Data | 186K | 41.7 | 60.8 | 50.4 | 83.4 | 1263.2 | 52.6 | 100.0 |
| Random | 57K | 38.8 | 56.5 | 46.9 | 83.1 | 1175.0 | 48.9 | 94.1 |
| PRISM | 57K | 42.3 | 61.1 | 50.8 | 84.1 | 1275.5 | 53.1 | 100.9 |
消融实验¶
核心消融验证了三点:浅层视觉特征最好、低相关样本最好、average pooling 优于 last token。
| 配置 | SQA | SQA-I | VizWiz | POPE-P/R/A | MM-Vet | MMBench | MME-C | Rel. |
|---|---|---|---|---|---|---|---|---|
| Deep Layer | 71.2 | 69.1 | 51.6 | 86.6 / 88.0 / 84.2 | 31.1 | 62.9 | 254.0 | 97.2% |
| Middle Layer | 70.9 | 69.1 | 47.7 | 86.5 / 87.8 / 84.2 | 31.9 | 65.0 | 276.0 | 97.9% |
| Shallow Layer | 71.3 | 69.1 | 50.1 | 87.7 / 88.7 / 85.5 | 32.0 | 65.2 | 330.0 | 100.0% |
| High Correlation | 70.6 | 68.0 | 48.1 | 85.8 / 87.6 / 83.9 | 30.7 | 64.0 | 275.3 | 96.3% |
| Low Correlation | 71.3 | 69.1 | 50.1 | 87.7 / 88.7 / 85.5 | 32.0 | 65.2 | 330.0 | 100.0% |
| Last Token | 69.9 | 67.3 | 49.4 | 87.4 / 88.3 / 85.0 | 31.6 | 62.6 | 272.0 | 97.4% |
| Avg Pooling | 71.3 | 69.1 | 50.1 | 87.7 / 88.7 / 85.5 | 32.0 | 65.2 | 330.0 | 100.0% |
跨模型泛化显示 PRISM 不只适用于 LLaVA-1.5-7B,而是在多种 LLM 和 vision encoder 组合上都略优于 full fine-tuning。
| 模型组合 | Full Rel. | PRISM Rel. | 代表性变化 |
|---|---|---|---|
| Phi2-3B | 100% | 100.1% | MME 1765.7 → 1790.5 |
| Vicuna-7B | 100% | 101.7% | MMBench 64.3 → 65.2 |
| Vicuna-13B | 100% | 100.4% | MME 1826.7 → 1846.0 |
| Qwen2.5-7B Base | 100% | 101.0% | SQA-I 76.7 → 78.9 |
| Qwen2.5-7B Instruct | 100% | 100.9% | MMBench 71.0 → 72.4 |
| Llama-3-8B | 100% | 100.8% | SQA-I 75.2 → 77.3 |
关键发现¶
- PRISM 的主结果不是“少训一点还能不掉点”,而是相对 full fine-tuning 还略有提升:LLaVA-665K 设置为 101.7%,VisionFlan-186K 设置为 100.9%。
- 低相关样本优于高相关和中等相关样本,直接支持“重中心化后低 redundancy 更有训练价值”的核心假设。
- 浅层特征优于中深层,说明用于冗余检测的几何结构在 early visual-token representation 中更干净,深层可能混入更多任务和抽象 artifacts。
- PRISM-Instruct-250K 的最终样本数为 250,557,其中包括 LLaVA 53,591、VG 28,777、VQAv2 27,567、OCRVQA 26,638、Text-Only 40,688 等,说明它不是按数据源硬编码比例,而是由全局低冗余阈值自然决定组成。
- text-only retention 也有收益:PRISM-7B 相对 101.9%,PRISM-13B 相对 130.6%,说明更干净的视觉指令数据可能减少 catastrophic forgetting。
亮点与洞察¶
- PRISM 的巧妙之处在于把数据选择的“模型评分问题”转成“几何校准问题”。如果 raw embedding 的相似度本身坏了,再复杂的 cheap distance selector 都会被误导。
- OSC 指标很实用。它要求选择器同时满足性能 fidelity 和 net efficiency gain,对数据选择论文是一个更诚实的评价约束。
- 只做 corpus mean re-centering 而不做 whitening 是一个务实取舍。full whitening 需要高维协方差估计、正则和 rank 选择;PRISM 放弃完全 isotropize,换来稳定、无超参、可大规模部署。
- 视觉优先的选择也很有启发。附录指出 text features 相对更接近中心,联合 multimodal feature 选择反而只有 97.8% 相对性能,低于 visual-only PRISM 的 101.7%。这说明多模态选择不一定要把所有模态直接拼起来。
局限与展望¶
- PRISM 只针对基于 feature correlation 的语义冗余剪枝,不检测事实错误、伦理偏差、有害内容或标注质量问题。因此它适合作为效率选择器,不应被当作完整数据治理工具。
- 方法主要修正一阶全局均值漂移,而不是完整 whitening。若某些数据池的主要问题来自更复杂的二阶协方差结构,单纯重中心化可能不足。
- 主任务集中在视觉-语言 instruction tuning,作者提到未来可以扩展到其他模态,但语音、视频、机器人轨迹等场景是否同样存在可利用的一阶 drift 还需要验证。
- PRISM 依赖目标 MLLM 的中间视觉表示;当目标模型不可访问或视觉 token 表征不稳定时,部署会受到限制。
- 当前选择目标是保留低冗余样本,但未显式建模任务覆盖、公平性、罕见类别或安全关键样本,后续可以把几何冗余分数和质量/安全过滤器组合起来。
相关工作与启发¶
- vs Random / Length / Perplexity: 这些方法便宜但语义信号弱,PRISM 同样便宜,却利用目标 MLLM 的中心化视觉几何来估计冗余。
- vs Proxy-Based Selection: InstructionGPT-4、Self-Filter、TIVE 等方法依赖外部模型或 scorer,可能引入 proxy bias 和推理开销;PRISM 不需要外部评估器。
- vs Training-Based Selection: EL2N、GraNd、ICONS 等使用训练动态或梯度信号,信息强但成本高;PRISM 用一次特征抽取替代迭代训练信号。
- vs whitening / top-PC removal: 这些几何修正更彻底但需要额外超参或高维协方差估计;PRISM 选择最简单的一阶重中心化,在性能和可扩展性之间取平衡。
评分¶
- 新颖性: ⭐⭐⭐⭐☆ 从视觉表征各向异性解释数据选择低效,角度清晰且有理论支撑。
- 实验充分度: ⭐⭐⭐⭐⭐ 主结果、消融、VisionFlan、跨模型、语言保持和效率分析都比较扎实。
- 写作质量: ⭐⭐⭐⭐☆ 方法逻辑完整,附录补充充分,但正文公式和图表较多,阅读密度偏高。
- 价值: ⭐⭐⭐⭐⭐ 对多模态指令数据筛选很实用,尤其适合需要真正降低端到端训练成本的场景。