We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Representation Topology Divergence: A Method for Comparing Neural Network Representations

Abstract: Comparison of data representations is a complex multi-aspect problem that has not enjoyed a complete solution yet. We propose a method for comparing two data representations. We introduce the Representation Topology Divergence (RTD), measuring the dissimilarity in multi-scale topology between two point clouds of equal size with a one-to-one correspondence between points. The data point clouds are allowed to lie in different ambient spaces. The RTD is one of the few TDA-based practical methods applicable to real machine learning datasets. Experiments show that the proposed RTD agrees with the intuitive assessment of data representation similarity and is sensitive to its topological structure. We apply RTD to gain insights on neural networks representations in computer vision and NLP domains for various problems: training dynamics analysis, data distribution shift, transfer learning, ensemble learning, disentanglement assessment.
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Algebraic Topology (math.AT)
Journal reference: Proceedings of the 39th International Conference on Machine Learning (ICML 2022), PMLR 162
Cite as: arXiv:2201.00058 [cs.LG]
  (or arXiv:2201.00058v2 [cs.LG] for this version)

Submission history

From: Serguei Barannikov [view email]
[v1] Fri, 31 Dec 2021 21:08:56 GMT (5248kb,D)
[v2] Tue, 28 Jun 2022 22:35:25 GMT (6583kb,D)

Link back to: arXiv, form interface, contact.