LearnPruner: Rethinking Attention-based Token Pruning in Vision Language Models
LearnPruner is a two-stage framework to prune Vision-Language Models by removing redundant visual tokens and then filtering text-irrelevant tokens in the LLM's middle layer. It uses learnable modules to predict token importance, achieving 95% of original performance while using only 5.5% of vision tokens and providing significant inference speedup.