Skip to content

gooriiie/MR-Pruner

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

16 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

MR-Pruner: Multi-Resolution Token Pruning Method

πŸ“Œ Introduction

We propose MR-Pruner, a training-free, graph-based token pruning framework explicitly designed for multi-resolution MLLMs. Unlike prior single-resolution pruning methods, MR-Pruner accounts for the informativeness distribution across resolutions and their mutual complementarity.

Key highlights:

  • πŸ” Resolution-aware pruning: Separately handles high- and low-resolution tokens with adaptive ratios.
  • πŸ”— Graph-based scoring: Builds intra- and cross-resolution token graphs to propagate informativeness.
  • πŸš€ Training-free: Plug-and-play, no model retraining needed.
  • 🎯 Robust: Preserves complementary tokens across resolutions, enabling resilience in extreme pruning.

πŸ“° News

  • [2025/11/11] Our paper MR-Pruner has been accepted to WACV 2026!πŸŽ‰

βš™οΈ Installation

MR-Pruner is built on Lmms-eval, which is evaluation toolkit for MLLMs.

git clone https://github.com/gooriiie/MR-Pruner.git
cd MR-Pruner

# create environment
conda create -n mrpruner python=3.10
conda activate mrpruner
pip install -r requirements.txt

# create environment with yaml
conda env create -f environment.yml

πŸš€ Run

Run evaluation with default settings.

# TextVQA
bash ./scripts/eval_lmms_eval_textvqa.sh  

# ScienceQA-IMG
bash ./scripts/eval_lmms_eval_sqa.sh

# POPE
bash ./scripts/eval_lmms_eval_pope.sh  

About

MR-Pruner: Training-free Multi-resolution Visual Token Pruning for Multi-modal Large Language Models

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors