Dynamic Activation Pitfalls in LLaMA Models: An Empirical Study

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ma, Chi, Huang, Mincong, Wang, Chao, Wang, Yujie, Yu, Lei
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913351398850560
author Ma, Chi
Huang, Mincong
Wang, Chao
Wang, Yujie
Yu, Lei
author_facet Ma, Chi
Huang, Mincong
Wang, Chao
Wang, Yujie
Yu, Lei
contents In this work, we systematically investigate the efficacy of dynamic activation mechanisms within the LLaMA family of language models. Despite the potential of dynamic activation methods to reduce computation and increase speed in models using the ReLU activation function, our empirical findings have uncovered several inherent pitfalls in the current dynamic activation schemes. Through extensive experiments across various dynamic activation strategies, we demonstrate that LLaMA models usually underperform when compared to their ReLU counterparts, particularly in scenarios demanding high sparsity ratio. We attribute these deficiencies to a combination of factors: 1) the inherent complexity of dynamically predicting activation heads and neurons; 2) the inadequate sparsity resulting from activation functions; 3) the insufficient preservation of information resulting from KV cache skipping. Our analysis not only sheds light on the limitations of dynamic activation in the context of large-scale LLaMA models but also proposes roadmaps for enhancing the design of future sparsity schemes.
format Preprint
id arxiv_https___arxiv_org_abs_2405_09274
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dynamic Activation Pitfalls in LLaMA Models: An Empirical Study
Ma, Chi
Huang, Mincong
Wang, Chao
Wang, Yujie
Yu, Lei
Machine Learning
In this work, we systematically investigate the efficacy of dynamic activation mechanisms within the LLaMA family of language models. Despite the potential of dynamic activation methods to reduce computation and increase speed in models using the ReLU activation function, our empirical findings have uncovered several inherent pitfalls in the current dynamic activation schemes. Through extensive experiments across various dynamic activation strategies, we demonstrate that LLaMA models usually underperform when compared to their ReLU counterparts, particularly in scenarios demanding high sparsity ratio. We attribute these deficiencies to a combination of factors: 1) the inherent complexity of dynamically predicting activation heads and neurons; 2) the inadequate sparsity resulting from activation functions; 3) the insufficient preservation of information resulting from KV cache skipping. Our analysis not only sheds light on the limitations of dynamic activation in the context of large-scale LLaMA models but also proposes roadmaps for enhancing the design of future sparsity schemes.
title Dynamic Activation Pitfalls in LLaMA Models: An Empirical Study
topic Machine Learning
url https://arxiv.org/abs/2405.09274