Differentiable Optimization of Similarity Scores Between Models and Brains

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cloos, Nathan, Li, Moufan, Siegel, Markus, Brincat, Scott L., Miller, Earl K., Yang, Guangyu Robert, Cueva, Christopher J.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909442824470528
author Cloos, Nathan
Li, Moufan
Siegel, Markus
Brincat, Scott L.
Miller, Earl K.
Yang, Guangyu Robert
Cueva, Christopher J.
author_facet Cloos, Nathan
Li, Moufan
Siegel, Markus
Brincat, Scott L.
Miller, Earl K.
Yang, Guangyu Robert
Cueva, Christopher J.
contents How do we know if two systems - biological or artificial - process information in a similar way? Similarity measures such as linear regression, Centered Kernel Alignment (CKA), Normalized Bures Similarity (NBS), and angular Procrustes distance, are often used to quantify this similarity. However, it is currently unclear what drives high similarity scores and even what constitutes a "good" score. Here, we introduce a novel tool to investigate these questions by differentiating through similarity measures to directly maximize the score. Surprisingly, we find that high similarity scores do not guarantee encoding task-relevant information in a manner consistent with neural data; and this is particularly acute for CKA and even some variations of cross-validated and regularized linear regression. We find no consistent threshold for a good similarity score - it depends on both the measure and the dataset. In addition, synthetic datasets optimized to maximize similarity scores initially learn the highest variance principal component of the target dataset, but some methods like angular Procrustes capture lower variance dimensions much earlier than methods like CKA. To shed light on this, we mathematically derive the sensitivity of CKA, angular Procrustes, and NBS to the variance of principal component dimensions, and explain the emphasis CKA places on high variance components. Finally, by jointly optimizing multiple similarity measures, we characterize their allowable ranges and reveal that some similarity measures are more constraining than others. While current measures offer a seemingly straightforward way to quantify the similarity between neural systems, our work underscores the need for careful interpretation. We hope the tools we developed will be used by practitioners to better understand current and future similarity measures.
format Preprint
id arxiv_https___arxiv_org_abs_2407_07059
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Differentiable Optimization of Similarity Scores Between Models and Brains
Cloos, Nathan
Li, Moufan
Siegel, Markus
Brincat, Scott L.
Miller, Earl K.
Yang, Guangyu Robert
Cueva, Christopher J.
Neurons and Cognition
Machine Learning
How do we know if two systems - biological or artificial - process information in a similar way? Similarity measures such as linear regression, Centered Kernel Alignment (CKA), Normalized Bures Similarity (NBS), and angular Procrustes distance, are often used to quantify this similarity. However, it is currently unclear what drives high similarity scores and even what constitutes a "good" score. Here, we introduce a novel tool to investigate these questions by differentiating through similarity measures to directly maximize the score. Surprisingly, we find that high similarity scores do not guarantee encoding task-relevant information in a manner consistent with neural data; and this is particularly acute for CKA and even some variations of cross-validated and regularized linear regression. We find no consistent threshold for a good similarity score - it depends on both the measure and the dataset. In addition, synthetic datasets optimized to maximize similarity scores initially learn the highest variance principal component of the target dataset, but some methods like angular Procrustes capture lower variance dimensions much earlier than methods like CKA. To shed light on this, we mathematically derive the sensitivity of CKA, angular Procrustes, and NBS to the variance of principal component dimensions, and explain the emphasis CKA places on high variance components. Finally, by jointly optimizing multiple similarity measures, we characterize their allowable ranges and reveal that some similarity measures are more constraining than others. While current measures offer a seemingly straightforward way to quantify the similarity between neural systems, our work underscores the need for careful interpretation. We hope the tools we developed will be used by practitioners to better understand current and future similarity measures.
title Differentiable Optimization of Similarity Scores Between Models and Brains
topic Neurons and Cognition
Machine Learning
url https://arxiv.org/abs/2407.07059