arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.07831cs.LGcs.NE
收藏

Learning Associative Inference Using Fast Weight Memory

The Swiss AI Lab IDSIA / USI / SUPSI(瑞士人工智能实验室IDSIA / 瑞士意大利语区大学 / 瑞士南部应用科学大学);Microsoft Research(微软研究院)

Imanol Schlag, Tsendsuren Munkhdalai, Jürgen Schmidhuber

英文摘要:

Humans can quickly associate stimuli to solve problems in novel contexts. Our novel neural network model learns state representations of facts that can be composed to perform such associative inference. To this end, we augment the LSTM model with an associative memory, dubbed Fast Weight Memory (FWM). Through differentiable operations at every step of a given input sequence, the LSTM updates and maintains compositional associations stored in the rapidly changing FWM weights. Our model is trained end-to-end by gradient descent and yields excellent performance on compositional language reasoning problems, meta-reinforcement-learning for POMDPs, and small-scale word-level language modelling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑