Apr 20 – 30, 2020
Virtual/Digital only workshop
Europe/Paris timezone

Learned Representations from Lower-Order Interactions for Efficient Clustering

Apr 21, 2020, 8:30 PM
Virtual/Digital only workshop

Virtual/Digital only workshop


Mr Nicholas Choma (Lawrence Berkeley National Laboratory)


Efficient agglomerative clustering is reliant on the ability to exploit useful lower-order information contained within data, yet many real-world datasets do not consist of features which are naturally amenable to metric functions as required by these algorithms. In this work, we present a framework for learning representations which contain such metric structure, allowing for efficient clustering and neighborhood queries of data points. We demonstrate how this framework fits in with both traditional clustering pipelines, and more advanced approaches such as graph neural networks. Finally, we present numerical results on the TrackML particle tracking challenge dataset, where our framework shows favorable results in both physics-based tracking methods, and new end-to-end deep learning approaches with graph neural networks developed in the context of the Exa.TrkX project.

Second most appropriate track (if necessary) Enhanced performance of tracking algorithms
Consider for young scientist forum (Student or postdoc speaker) No

Primary author

Mr Nicholas Choma (Lawrence Berkeley National Laboratory)

Presentation materials