Если так случилось что вы еще не читали про Capsule Network (Geoffrey E Hinton), то все же стоит.

Вот сам артикл https://goo.gl/EuGd5g

Вот обьяснение с примером https://goo.gl/sZWCNv

А вот код для Keras https://goo.gl/pMyfiW