← Search

Théotime Le Hellard

4 accepted papers

2026

Guided Flow Policy: Learning from High-Value Actions in Offline Reinforcement Learning

ICLR 2026poster

Offline reinforcement learning often relies on behavior regularization that enforces policies to remain close to the dataset distribution. However, such approaches fail to distinguish between high-value and low-value actions in their regularization components. We introduce Guided Flow Policy (GFP),…

Cited by 5SourcecodeScholar
2026

SVL: Goal-Conditioned Reinforcement Learning as Survival Learning

ICML 2026poster

Standard approaches to goal-conditioned reinforcement learning (GCRL) that rely on temporal-difference learning can be unstable and sample-inefficient due to bootstrapping. While recent work has explored contrastive and supervised formulations to improve stability, we present a probabilistic alterna…

Cited by 0SourceScholar
2025

HiRemate: Hierarchical Approach for Efficient Re-materialization of Neural Networks

ICML 2025poster

Training deep neural networks (DNNs) on memory-limited GPUs is challenging, as storing intermediate activations often exceeds available memory. Re-materialization, a technique that preserves exact computations, addresses this by selectively recomputing activations instead of storing them. However,…

Cited by 0SourcePDFScholar
2023

Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch

ICML 2023oral

We propose Rockmate to control the memory requirements when training PyTorch DNN models. Rockmate is an automatic tool that starts from the model code and generates an equivalent model, using a predefined amount of memory for activations, at the cost of a few re-computations. Rockmate automatically…