English

The Simpler The Better: An Entropy-Based Importance Metric To Reduce Neural Networks' Depth

Machine Learning 2024-06-06 v2

Abstract

While deep neural networks are highly effective at solving complex tasks, large pre-trained models are commonly employed even to solve consistently simpler downstream tasks, which do not necessarily require a large model's complexity. Motivated by the awareness of the ever-growing AI environmental impact, we propose an efficiency strategy that leverages prior knowledge transferred by large models. Simple but effective, we propose a method relying on an Entropy-bASed Importance mEtRic (EASIER) to reduce the depth of over-parametrized deep neural networks, which alleviates their computational burden. We assess the effectiveness of our method on traditional image classification setups. Our code is available at https://github.com/VGCQ/EASIER.

Keywords

Cite

@article{arxiv.2404.18949,
  title  = {The Simpler The Better: An Entropy-Based Importance Metric To Reduce Neural Networks' Depth},
  author = {Victor Quétu and Zhu Liao and Enzo Tartaglione},
  journal= {arXiv preprint arXiv:2404.18949},
  year   = {2024}
}

Comments

arXiv admin note: text overlap with arXiv:2404.16890