Storing Is Not Remembering: LSTM-UT and Bounded Gated Memory for Looped Transformers
In OpenWebText2 language modelling, LSTM-UT outperforms CoTFormer and, at equal width, reaches slightly lower perplexity than CoTFormer while CoTFormer needs up to 91% more training time per step; against a parameter-matched CoTFormer, LSTM-UT comes within 0.6 perplexity.