← Search

Maxime Gazeau

3 accepted papers

2023

A*Net: A Scalable Path-based Reasoning Approach for Knowledge Graphs

NeurIPS 2023poster

Reasoning on large-scale knowledge graphs has been long dominated by embedding methods. While path-based methods possess the inductive capacity that embeddings lack, their scalability is limited by the exponential number of paths. Here we present A\*Net, a scalable path-based method for knowledge gr…

2023

In-context Reinforcement Learning with Algorithm Distillation

ICLR 2023top-5%

We propose Algorithm Distillation (AD), a method for distilling reinforcement learning (RL) algorithms into neural networks by modeling their training histories with a causal sequence model. Algorithm Distillation treats learning to reinforcement learn as an across-episode sequential prediction prob…

Cited by 145SourcePDFScholar
2020

An Empirical Study of Stochastic Gradient Descent with Structured Covariance Noise

AISTATS 2020poster

The choice of batch-size in a stochastic optimization algorithm plays a substantial role for both optimization and generalization. Increasing the batch-size used typically improves optimization but degrades generalization. To address the problem of improving generalization while maintaining optimal…

Cited by 15SourcePDFScholar