TAMP: Token-Adaptive Layerwise Pruning in Multimodal Large Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lee, Jaewoo, Xuan, Keyang, Ekbote, Chanakya, Polisetty, Sandeep, Fung, Yi R., Liang, Paul Pu
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913843620347904
author Lee, Jaewoo
Xuan, Keyang
Ekbote, Chanakya
Polisetty, Sandeep
Fung, Yi R.
Liang, Paul Pu
author_facet Lee, Jaewoo
Xuan, Keyang
Ekbote, Chanakya
Polisetty, Sandeep
Fung, Yi R.
Liang, Paul Pu
contents Multimodal Large Language Models (MLLMs) have shown remarkable versatility in understanding diverse multimodal data and tasks. However, these capabilities come with an increased model scale. While post-training pruning reduces model size in unimodal models, its application to MLLMs often yields limited success. Our analysis discovers that conventional methods fail to account for the unique token attributes across layers and modalities inherent to MLLMs. Inspired by this observation, we propose TAMP, a simple yet effective pruning framework tailored for MLLMs, featuring two key components: (1) Diversity-Aware Sparsity, which adjusts sparsity ratio per layer based on diversities among multimodal output tokens, preserving more parameters in high-diversity layers; and (2) Adaptive Multimodal Input Activation, which identifies representative multimodal input tokens using attention scores to guide unstructured weight pruning. We validate our method on two state-of-the-art MLLMs: LLaVA-NeXT, designed for vision-language tasks, and VideoLLaMA2, capable of processing audio, visual, and language modalities. Empirical experiments across various multimodal evaluation benchmarks demonstrate that each component of our approach substantially outperforms existing pruning techniques.
format Preprint
id arxiv_https___arxiv_org_abs_2504_09897
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TAMP: Token-Adaptive Layerwise Pruning in Multimodal Large Language Models
Lee, Jaewoo
Xuan, Keyang
Ekbote, Chanakya
Polisetty, Sandeep
Fung, Yi R.
Liang, Paul Pu
Computer Vision and Pattern Recognition
Multimodal Large Language Models (MLLMs) have shown remarkable versatility in understanding diverse multimodal data and tasks. However, these capabilities come with an increased model scale. While post-training pruning reduces model size in unimodal models, its application to MLLMs often yields limited success. Our analysis discovers that conventional methods fail to account for the unique token attributes across layers and modalities inherent to MLLMs. Inspired by this observation, we propose TAMP, a simple yet effective pruning framework tailored for MLLMs, featuring two key components: (1) Diversity-Aware Sparsity, which adjusts sparsity ratio per layer based on diversities among multimodal output tokens, preserving more parameters in high-diversity layers; and (2) Adaptive Multimodal Input Activation, which identifies representative multimodal input tokens using attention scores to guide unstructured weight pruning. We validate our method on two state-of-the-art MLLMs: LLaVA-NeXT, designed for vision-language tasks, and VideoLLaMA2, capable of processing audio, visual, and language modalities. Empirical experiments across various multimodal evaluation benchmarks demonstrate that each component of our approach substantially outperforms existing pruning techniques.
title TAMP: Token-Adaptive Layerwise Pruning in Multimodal Large Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2504.09897