A data scientist training an RNN on long text sequences finds gradients shrinking toward zero in early timesteps. Which architecture change best addresses this?
-
A
Replace vanilla RNN cells with LSTM or GRU cells
-
B
Increase the batch size
-
C
Use mean squared error instead of cross-entropy
-
D
Remove the embedding layer