The increasing computational and environmental costs associated with AI models, especially large language models (LLMs), highlight the urgent need for network optimization. These models consume vast amounts of energy and resources, requiring more efficient training strategies to balance performance with ecological responsibility. Our focus is on enhancing the efficiency of deep neural networks on embedded devices through novel pruning techniques: “evolution of weights” and “smart pruning.” These methods, compared to traditional pruning approaches using benchmark datasets, involve evaluating parameter importance during training to better preserve accuracy during compression. Our approach results in faster computations and higher compression rates with minimal accuracy loss. We have successfully applied these techniques to LLMs consisting of around 10 million parameters. The LLM experiment is publicly available on Github to facilitate replication testing.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Pruning and Optimizing Large Language Models in an Era of GPU Scarcity

  • Ashhadul Islam,
  • Samir Brahim Belhaouari,
  • Amine Bermak

摘要

The increasing computational and environmental costs associated with AI models, especially large language models (LLMs), highlight the urgent need for network optimization. These models consume vast amounts of energy and resources, requiring more efficient training strategies to balance performance with ecological responsibility. Our focus is on enhancing the efficiency of deep neural networks on embedded devices through novel pruning techniques: “evolution of weights” and “smart pruning.” These methods, compared to traditional pruning approaches using benchmark datasets, involve evaluating parameter importance during training to better preserve accuracy during compression. Our approach results in faster computations and higher compression rates with minimal accuracy loss. We have successfully applied these techniques to LLMs consisting of around 10 million parameters. The LLM experiment is publicly available on Github to facilitate replication testing.