Kerasを使用したトレーニングスパースモデル

Question

私はスパースモデルをトレーニングしようとしています。つまり、モデルパラメータの一部が最適化中にゼロのままでなければなりません。

Kerasでは、オプティマイザがマスクされたマスクを更新しないように、パラメータのマスクを定義することは可能ですか？

残念ながら、パラメータをより細かくマスクする必要があるため、1つのレイヤをフリーズすることはできません。

Answer 1

tf.whereを使用して、パラメータとtf.stop_gradient(parameters)の間で要素を選択することができます。