Обрезка в Керасе

Я пытаюсь создать нейронную сеть с использованием Keras с приоритетом на производительность прогнозирования, и я не могу получить достаточно высокую точность путем дальнейшего сокращения количества слоев и узлов на слой. Я заметил, что очень большая часть моих весов фактически равна нулю (>95%). Есть ли способ обрезать плотные слои в надежде сократить время прогнозирования?

2 ответов


Не так :(

в настоящее время нет простого (выделенного) способа сделать это с Keras.

обсуждение продолжается в https://groups.google.com/forum#!темы/keras-пользователи/oEecCWayJrM.

вы также можете быть заинтересованы в этой статье: https://arxiv.org/pdf/1608.04493v1.pdf.


взгляните на хирурга Keras: https://github.com/BenWhetton/keras-surgeon

Я сам не пробовал, но в документации утверждается, что у него есть функции для удаления или вставки узлов.