← Search

Aidan Scannell

7 accepted papers

2026

Efficient Reinforcement Learning by Guiding World Models with Non-Curated Data

ICLR 2026poster

Leveraging offline data is a promising way to improve the sample efficiency of online reinforcement learning (RL). This paper expands the pool of usable data for offline-to-online RL by leveraging abundant non-curated data that is reward-free, of mixed quality, and collected across multiple embodime…

Cited by 0SourcecodeScholar
2026

Kalman Linear Attention: Parallel Bayesian Filtering For Efficient Language Modeling and State Tracking

ICML 2026poster

State-space language models such as Mamba and gated linear attention (GLA) offer efficient alternatives to transformers due to their linear complexity and parallel training, but often lack the expressivity and robust state-tracking needed for complex reasoning. We address these limitations by refram…

Cited by 0SourceScholar
2025

Discrete Codebook World Models for Continuous Control

ICLR 2025poster

In reinforcement learning (RL), world models serve as internal simulators, enabling agents to predict environment dynamics and future outcomes in order to make informed decisions. While previous approaches leveraging discrete latent spaces, such as DreamerV3, have demonstrated strong performance in…

2025

Entropy Regularized Task Representation Learning for Offline Meta-Reinforcement Learning

AAAI 2025technical

Offline meta-reinforcement learning aims to equip agents with the ability to rapidly adapt to new tasks by training on data from a set of different tasks. Context-based approaches utilize a history of state-action-reward transitions – referred to as the context – to infer a representation of the cur…

2024

Function-space Parameterization of Neural Networks for Sequential Learning

ICLR 2024poster

Sequential learning paradigms pose challenges for gradient-based deep learning due to difficulties incorporating new data and retaining prior knowledge. While Gaussian processes elegantly tackle these problems, they struggle with scalability and handling rich inputs, such as images. To address these…

2023

Mode-constrained Model-based Reinforcement Learning via Gaussian Processes

AISTATS 2023poster

Model-based reinforcement learning (RL) algorithms do not typically consider environments with multiple dynamic modes, where it is beneficial to avoid inoperable or undesirable modes. We present a model-based RL algorithm that constrains training to a single dynamic mode with high probability. This…

2021

Trajectory Optimisation in Learned Multimodal Dynamical Systems via Latent-ODE Collocation

ICRA 2021poster

This paper presents a two-stage method to perform trajectory optimisation in multimodal dynamical systems with unknown nonlinear stochastic transition dynamics. The method finds trajectories that remain in a preferred dynamics mode where possible and in regions of the transition dynamics model that…

Cited by 12SourceScholar