An adaptive binary particle swarm optimization for solving multi-objective convolutional filter pruning problem

Sawant SS, Erick FX, Göb S, Holzer N, Lang EW, Götz T (2023)

Publication Type: Journal article

Publication year: 2023

Journal

Journal of Supercomputing Springer Verlag (Germany)

Abstract

In recent years, deep convolutional neural networks (DCNN) have evolved significantly in order to demonstrate remarkable performance in various computer vision tasks. However, their excess storage requirements and heavy computational burden restrict their scope of application, particularly on embedded platforms. This problem has motivated the research community to investigate effective approaches that can reduce computational burden without compromising its performance. Filter pruning is one of the popular ways to reduce the computational burden, where weak or unimportant convolutional filters are eliminated. In this paper, we propose a novel approach for filter pruning based on an adaptive multi-objective particle swarm optimization (AMPSO) to compress and accelerate DCNN. The proposed approach searches for an optimal solution while maintaining the trade-off between network’s performance and computational cost. Extensive experiments on TernausNet and U-Net for high-resolution aerial image segmentation tasks demonstrate the superiority of AMPSO in finding a compact network model.

Involved external institutions

Universität Regensburg

Germany (DE) Fraunhofer-Institut für Integrierte Schaltungen (IIS)

Germany (DE)

How to cite

APA:

Sawant, S.S., Erick, F.X., Göb, S., Holzer, N., Lang, E.W., & Götz, T. (2023). An adaptive binary particle swarm optimization for solving multi-objective convolutional filter pruning problem. Journal of Supercomputing. https://doi.org/10.1007/s11227-023-05150-1

MLA:

Sawant, Shrutika S., et al. "An adaptive binary particle swarm optimization for solving multi-objective convolutional filter pruning problem." Journal of Supercomputing (2023).

BibTeX: Download