Hierarchical clustering with deep Q-learning

Abstract

Following up on our previous study on applying hierarchical clustering algorithms to high energy particle physics, this paper explores the possibilities to use deep learning to generate models capable of processing the clusterization themselves. The technique chosen for training is reinforcement learning, that allows the system to evolve based on interactions between the model and the underlying graph. The result is a model, that by learning on a modest dataset of 10, 000 nodes during 70 epochs can reach 83, 77% precision for hierarchical and 86, 33% for high energy jet physics datasets in predicting the appropriate clusters.