Few-Shot Bayesian Imitation Learning with Logical Program Policies

Tom Silver, Kelsey R. Allen, Alex K. Lew, Leslie Kaelbling, and Josh Tenenbaum, AAAI 2020

This paper introduces a bayesian imitation learning approach to learn policies from few demonstrations. They call these policies Logical Program Policies (LPP) which are essentially policies learnt as combination of logical and programmatic policies. Logical because these are relational and programmatic because they are features are automatically learned.

The bayesian prior used here is the prior probability distribution over the Probablistic Context Free Grammer (P-CFG). Paper proposes to generate a dataset () where each state action pair is an example. Feature set of each example is obtained by initializing all the P-CFGs for each example. The target variable for each example is if , otherwise.

Next, all the features are arranged in decreasing order of their prior probabilities and iteratively decision-trees (s) are learnt with incremental feature size. So, at iteration , features used are . The DT learned are converted to logical representation (i.e. disjunction of conjunctions of the P-CFG feature) and each DT is evaluated on the dataset and finally the top- DTs are used as weighted mixture model for testing.


The paper is a difficult read and doesn't convey the actual procedure followed in the code. Algorithm in the paper suggest that the posterior is iteratively refined however digging into the code suggests that the posterior is independently computed for each tree and there is no carry forward from one iteration to another.

The paper introduced a good set of 2D grid domains with generalizable domain-specific language. Code is very neat and easy to read.

The baselines used in the paper CNN and FCN are not meant for few-shot learning so it is not a surprise that they did not work. Some comparision to meta-learning approaches would have been useful.


  1. Few-Shot Bayesian Imitation Learning with Logical Program Policies
  2. https://github.com/tomsilver/policies_logic_programs