MazeExplorer: A Customisable 3D Benchmark for Assessing Generalisation in Reinforcement Learning. - Luke Harries; Sebastian Lee; Jaroslaw Rzepecki; Katja Hofmann; Sam Devlin (CoG, 2019)

Continual Learning in the Teacher-Student Setup: Impact of Task Similarity. - Sebastian Lee; Sebastian Goldt; Andrew M. Saxe (ICML, 2021)

Fast and Memory-Efficient Neural Code Completion. - Alexey Svyatkovskiy; Sebastian Lee; Anna Hadjitofi; Maik Riechert; Juliana Vicente Franco; Miltiadis Allamanis (MSR, 2021)

Maslow's Hammer in Catastrophic Forgetting: Node Re-Use vs. Node Activation. - Sebastian Lee; Stefano Sarao Mannelli; Claudia Clopath; Sebastian Goldt; Andrew M. Saxe (ICML, 2022)

The RL Perceptron: Generalisation Dynamics of Policy Learning in High Dimensions. - Nishil Patel; Sebastian Lee; Stefano Sarao Mannelli; Sebastian Goldt; Andrew M. Saxe (CoRR, 2023)

Lifelong Reinforcement Learning via Neuromodulation. - Sebastian Lee; Samuel Liebana Garcia; Claudia Clopath; Will Dabney (CoRR, 2024)

A Theory of Initialisation's Impact on Specialisation. - Devon Jarvis; Sebastian Lee; Clémentine Carla Juliette Dominé; Andrew M. Saxe; Stefano Sarao Mannelli (ICLR, 2025)

Uncertainty Prioritized Experience Replay. - Rodrigo Carrasco-Davis; Sebastian Lee; Claudia Clopath; Will Dabney (CoRR, 2025)