Hi,
first of all, really nice work! I very much enjoyed reading it.
I'm working on a similar topic and wonder if you also uploaded the code for your experiment "Generalization with less data", shown in figure 6 of your work. I looked through your code but could not find it ad-hoc.
Best,
David,