arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.08172cs.LGstat.ML
收藏

Finding the Optimal Network Depth in Classification Tasks

Jagiellonian University(雅盖隆大学)

Bartosz Wójcik, Maciej Wołczyk, Klaudia Bałazy, Jacek Tabor

英文摘要:

We develop a fast end-to-end method for training lightweight neural networks using multiple classifier heads. By allowing the model to determine the importance of each head and rewarding the choice of a single shallow classifier, we are able to detect and remove unneeded components of the network. This operation, which can be seen as finding the optimal depth of the model, significantly reduces the number of parameters and accelerates inference across different hardware processing units, which is not the case for many standard pruning methods. We show the performance of our method on multiple network architectures and datasets, analyze its optimization properties, and conduct ablation studies.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑