Preliminary result. Let us now generate the parse trees on the CapsNet model trained on the MNIST dataset as in  [1]}, investigating their evolution during training.
To this end, we compute the distribution of the coupling coefficients on the test set for two models: the model trained after the first epoch and the one that achieved the lowest loss value on the validation set.
<FIGURE>