← Search

Vitaly Kurin

9 accepted papers

2022

In Defense of the Unitary Scalarization for Deep Multi-Task Learning

NeurIPS 2022accept

Recent multi-task learning research argues against unitary scalarization, where training simply minimizes the sum of the task losses. Several ad-hoc multi-task optimization algorithms have instead been proposed, inspired by various hypotheses about what makes multi-task settings difficult. The majo…

2021

MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research

NeurIPS 2021poster

Progress in deep reinforcement learning (RL) is heavily driven by the availability of challenging benchmarks used for training agents. However, benchmarks that are widely adopted by the community are not explicitly designed for evaluating specific capabilities of RL methods. While there exist enviro…

Cited by 108SourcecodeScholar
2021

My Body is a Cage: the Role of Morphology in Graph-Based Incompatible Control

ICLR 2021poster

Multitask Reinforcement Learning is a promising way to obtain models with better performance, generalisation, data efficiency, and robustness. Most existing work is limited to compatible settings, where the state and action space dimensions are the same across tasks. Graph Neural Networks (GNN) are…

2021

Snowflake: Scaling GNNs to high-dimensional continuous control via parameter freezing

NeurIPS 2021poster

Recent research has shown that graph neural networks (GNNs) can learn policies for locomotion control that are as effective as a typical multi-layer perceptron (MLP), with superior transfer and multi-task performance. However, results have so far been limited to training on small agents, with the pe…

2020

Can Q-Learning with Graph Networks Learn a Generalizable Branching Heuristic for a SAT Solver?

NeurIPS 2020poster

We present Graph-Q-SAT, a branching heuristic for a Boolean SAT solver trained with value-based reinforcement learning (RL) using Graph Neural Networks for function approximation. Solvers using Graph-Q-SAT are complete SAT solvers that either provide a satisfying assignment or proof of unsatisfiabil…

2019

Fast Context Adaptation via Meta-Learning

ICML 2019oral

We propose CAVIA for meta-learning, a simple extension to MAML that is less prone to meta-overfitting, easier to parallelise, and more interpretable. CAVIA partitions the model parameters into two parts: context parameters that serve as additional input to the model and are adapted on individual tas…

2019

Fast Efficient Hyperparameter Tuning for Policy Gradient Methods

NeurIPS 2019poster

The performance of policy gradient methods is sensitive to hyperparameter settings that must be tuned for any new application. Widely used grid search methods for tuning hyperparameters are sample inefficient and computationally expensive. More advanced methods like Population Based Training that le…

2019

Learning From Demonstration in the Wild

ICRA 2019poster

Learning from demonstration (LfD) is useful in settings where hand-coding behaviour or a reward function is impractical. It has succeeded in a wide range of problems but typically relies on manually generated demonstrations or specially deployed sensors and has not generally been able to leverage th…

Cited by 80SourceScholar