Variational autoencoders understand knot topology
Anna Braghetto, Sumanta Kundu, Marco Baiesi, Enzo Orlandini
TL;DR
The paper develops a variational autoencoder augmented with a knot-type classifier (VAEC) to identify and generate knotted polymer configurations under spherical confinement. By training on 3D coordinates, the model learns a topology-aware latent space where knots cluster by family and complexity, and chirality can be distinguished even for knots not seen during training. The approach yields high classification accuracy and enables generation of topologically faithful configurations, with latent axes correlating to unknotting number and braid index. It also demonstrates robust generalization to longer chains and enhanced chirality recognition beyond traditional polynomial invariants, highlighting a promising route for topology-aware generative ML in polymer physics.
Abstract
Supervised machine learning (ML) methods are emerging as valid alternatives to standard mathematical methods for identifying knots in long, collapsed polymers. Here, we introduce a hybrid supervised/unsupervised ML approach for knot classification based on a variational autoencoder enhanced with a knot type classifier (VAEC). The neat organization of knots in its latent representation suggests that the VAEC, only based on an arbitrary labeling of three-dimensional configurations, has grasped complex topological concepts such as chirality, unknotting number, braid index, and the grouping in families such as achiral, torus, and twist knots. The understanding of topological concepts is confirmed by the ability of the VAEC to distinguish the chirality of knots $9_{42}$ and $10_{71}$ not used for its training and with a notoriously undetected chirality to standard tools. The well-organized latent space is also key for generating configurations with the decoder that reliably preserves the topology of the input ones. Our findings demonstrate the ability of a hybrid supervised-generative ML algorithm to capture different topological features of entangled filaments and to exploit this knowledge to faithfully reconstruct or produce new knotted configurations without simulations.
