Rob Fergus

Automatic Data Augmentation for Generalization in Reinforcement Learning

2021 · arXiv (Cornell University)

Deep reinforcement learning (RL) agents often fail to generalize beyond their training environments. To alleviate this problem, recent work has proposed the use of data augmentation. However, different tasks tend to benefit from different types …

End-To-End Memory Networks

2015 · arXiv (Cornell University)

We introduce a neural network with a recurrent attention model over a possibly large external memory. The architecture is a form of Memory Network (Weston et al., 2015) but unlike the model in that work, …