Diversity networks: Neural network compression using determinantal point processes

Zelda Mariet, Suvrit Sra

Research output: Contribution to conferencePaperpeer-review

44 Scopus citations

Abstract

We introduce Divnet, a flexible technique for learning networks with diverse neurons. Divnet models neuronal diversity by placing a Determinantal Point Process (DPP) over neurons in a given layer. It uses this DPP to select a subset of diverse neurons and subsequently fuses the redundant neurons into the selected ones. Compared with previous approaches, Divnet offers a more principled, flexible technique for capturing neuronal diversity and thus implicitly enforcing regularization. This enables effective auto-tuning of network architecture and leads to smaller network sizes without hurting performance. Moreover, through its focus on diversity and neuron fusing, Divnet remains compatible with other procedures that seek to reduce memory footprints of networks. We present experimental results to corroborate our claims: for pruning neural networks, Divnet is seen to be notably superior to competing approaches.

Original languageEnglish
StatePublished - 2016
Externally publishedYes
Event4th International Conference on Learning Representations, ICLR 2016 - San Juan, Puerto Rico
Duration: 2 May 20164 May 2016

Conference

Conference4th International Conference on Learning Representations, ICLR 2016
Country/TerritoryPuerto Rico
CitySan Juan
Period2/05/164/05/16

Fingerprint

Dive into the research topics of 'Diversity networks: Neural network compression using determinantal point processes'. Together they form a unique fingerprint.

Cite this