PyramidDrop: Acelerando seus Modelos de Visão-Linguagem Grandes através da Redução de Redundância Visual em PirâmidePyramidDrop: Accelerating Your Large Vision-Language Models via Pyramid
Visual Redundancy Reduction
Nos grandes modelos de visão e linguagem (LVLMs), as imagens servem como entradas que carregam uma riqueza de informações. Como o ditado "Uma imagem vale mais que mil palavras" sugere, representar uma única imagem nos LVLMs atuais pode exigir centenas ou até milhares de tokens. Isso resulta em custos computacionais significativos, que crescem de forma quadrática à medida que a resolução da imagem de entrada aumenta, impactando severamente a eficiência tanto do treinamento quanto da inferência. Abordagens anteriores tentaram reduzir o número de tokens de imagem antes ou nas camadas iniciais dos LVLMs. No entanto, essas estratégias inevitavelmente resultam na perda de informações cruciais da imagem, diminuindo, em última instância, o desempenho do modelo. Para enfrentar esse desafio, realizamos um estudo empírico que revela que todos os tokens visuais são necessários para os LVLMs nas camadas rasas, e a redundância de tokens aumenta progressivamente nas camadas mais profundas do modelo. Para isso, propomos o PyramidDrop, uma estratégia de redução de redundância visual para os LVLMs a fim de impulsionar sua eficiência tanto no treinamento quanto na inferência, com uma perda de desempenho negligenciável. Especificamente, dividimos o LVLM em várias etapas e eliminamos parte dos tokens de imagem no final de cada etapa com uma proporção pré-definida, criando tokens visuais em forma de pirâmide através das camadas do modelo. A eliminação é baseada em um cálculo de similaridade leve com um tempo de execução negligenciável. Experimentos extensos demonstram que o PyramidDrop pode alcançar uma aceleração de 40% no tempo de treinamento e 55% nas FLOPs de inferência do LLaVA-NeXT com desempenho comparável. Além disso, o PyramidDrop também poderia servir como uma estratégia plug-and-play para aceleração de inferência sem treinamento, com melhor desempenho e menor custo de inferência do que os concorrentes. Esperamos que as ideias e abordagem introduzidas pelo PyramidDrop inspirem pesquisas futuras para investigar ainda mais o papel dos tokens de imagem nos LVLMs.