Learning Neural Models for End-to-End Clustering

Benjamin Bruno Meier, Ismail Elezi, Mohammadreza Amirian +2

We propose a novel end-to-end neural network architecture that, once trained, directly outputs a probabilistic clustering of a batch of input examples in one pass. It estimates a d…