Predicting Fine-Grained Traffic Conditions via Spatio-Temporal LSTM

Algorithm 1

For calculating each element of LSTM.

Input: The network’s input in current time ;

the initial weight matrix ,

and bias units about gates , , ;

Output: The forget gate, input gate, cell state in different

time, output gate and the cell output.

(1) ;

(2) #: represent Sigmoid function.

(3) ;

(4) ;

(5) ;

(6) ;

(7) ;

(8) return, , , , , .