vix.ing · top · new · best · stats · spec

Layer Pruning with Consensus: A Triple-Win Solution

2024/11/21 by Leandro Giusti Mugnaini, Mugnaini, Leandro Giusti, Carolina Tavares Duarte +5
Computer Science · #Logic, programming, and type systems #Optimization and Search Problems #Mobile Agent-Based Network Management

paper · pdf · doi:10.48550/arxiv.2411.14345

Abstract

Layer pruning offers a promising alternative to standard structured pruning, effectively reducing computational costs, latency, and memory footprint. While notable layer-pruning approaches aim to detect unimportant layers for removal, they often rely on single criteria that may not fully capture the complex, underlying properties of layers. We propose a novel approach that combines multiple similarity metrics into a single expressive measure of low-importance layers, called the Consensus criterion. Our technique delivers a triple-win solution: low accuracy drop, high-performance improvement, and increased robustness to adversarial attacks. With up to 78.80% FLOPs reduction and performance on par with state-of-the-art methods across different benchmarks, our approach reduces energy consumption and carbon emissions by up to 66.99% and 68.75%, respectively. Additionally, it avoids shortcut learning and improves robustness by up to 4 percentage points under various adversarial attacks. Overall, the Consensus criterion demonstrates its effectiveness in creating robust, efficient, and environmentally friendly pruned models.

Related