跳到正文
arXiv cs.CV· Xiaoyang Ma, Chi Wang, Hong Sun·· 2 小时前AI 评分22

SheafPrune:用层论引导图传播做视觉 token 剪枝

SheafPrune: Sheaf-Guided Graph Propagation for Visual Token Pruning

AI 导读

SheafPrune 用模态专属映射和稀疏文本-视觉图传递相关性,再在独立视觉图上传播分数并做节点度校正,在冻结语言模型前一次性保留硬 top-k 子集。该选择器约 4.5M 参数,仅用语言建模目标端到端训练,显式保留率条件让单个 checkpoint 适配所有预算。

来源:arXiv cs.CV · arxiv.org