I have a LSTM (Long-Short Term Memory) Neural Network that has this structure:

model = Sequential()
model.add(Masking(mask_value=0.0, input_shape=(271,2)))
adam = Adam(lr=.0000001, clipnorm = .001)
model.add(LSTM(271, activation = 'linear', input_shape=(271,2))) #return_sequences = True
#model.add(LSTM(3, activation = 'hard_sigmoid', inner_activation = 'hard_sigmoid'))
#model.add(Dense(1, activation = 'linear'))
model.compile(loss='mean_squared_logarithmic_error', optimizer=adam)
model.fit(maskingreg, maskingresp, nb_epoch=50, batch_size=500, verbose=2)

As you can see, I am clipping gradients and making the learning rate minuscule. However when the activation is linear, the net returns NaNs for the loss at every training epoch but the first. Does anyone have any other ideas why this is happening or even if there were more in depth troubleshooters in Keras to figure out why the loss is NaN?

Additionally when I use a hard sigmoid activation so that the min and max is capped, the net doesn't spit out NaNs but does not perform well. Obviously, this is not the main usage of a hard sigmoid, which is why I would prefer to stick with linear activation.

One problem I considered is that I'm doing masking incorrectly and this is causing a very low rank tensor. I have 52 samples, 271 time steps, and 2 features. I assumed that if I set both of the two features along any time step and any sample to 0. (with mask_value = 0.0), it would skip those features, but would read other samples in the same time step and any features in previous or additional time steps along the same sample. I assumed this was right, but masking seemed pretty confusing so I'm not positive.

  • $\begingroup$ Can you explain LSTM? $\endgroup$ – kjetil b halvorsen Sep 6 '16 at 14:55
  • $\begingroup$ Hi Kjetil, LSTM is a Long-Short-Term Memory Artificial Neural Network. It is basically a neural network that can learn long term dependencies due to estimating a state variable for the current system. LSTMs and related models (GRUs etc) differ from normal recurrent neural networks because set the parameter governing state transitions at a constant which prevents the gradients from disappearing. $\endgroup$ – www3 Sep 16 '16 at 15:14
  • $\begingroup$ That information belongs in the original post, not only in the comment. Can you edit it in? $\endgroup$ – kjetil b halvorsen Sep 16 '16 at 15:34
  • 2
    $\begingroup$ Not to sound snooty or anything, but it is likely that if you don't know what a LSTM is, you can't help me figure out how to improve LSTM performance. $\endgroup$ – www3 Sep 23 '16 at 14:45
  • 1
    $\begingroup$ Might be, but still it is the policy in here that acronyms should be spelt out at least once per post! That makes the website much more enjoyable for everybody. $\endgroup$ – kjetil b halvorsen Sep 23 '16 at 14:49

I had this problem before and one solution that worked for me was to use weight and activation regularization, specifically l2 regularization. Even setting the alpha values very small (0.000001) helped a lot without compromising accuracy too much.

For your code, add the following import statement:

from keras.regularizers import l2

And change line 4 from your code to the following:

model.add(LSTM(271, activation = 'linear', input_shape=(271,2), 

Another thing that I noted from my experience is that if you stack many LSTM layers with linear or ReLU activation layers this solution becomes less robust. You may need to play around with the regularization alpha constants as well as the learning rate and norm clipping to get things to work properly for your particular problem.


I am having this issue since 2 days. In this time I have changed a lot in my network without much success.

The solution for me was to adjust my optimizer, you might want to reduce the learning rate even further, if you still have problems and the network becomes unstable.

Adam(lr=0.0001, clipnorm=1.0, clipvalue=0.5)


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.