Partial Soft-Matching Distance for Neural Representational Comparison with Partial Unit Correspondence

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kapoor, Chaitanya, Williams, Alex H., Khosla, Meenakshi
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917287824457728
author Kapoor, Chaitanya
Williams, Alex H.
Khosla, Meenakshi
author_facet Kapoor, Chaitanya
Williams, Alex H.
Khosla, Meenakshi
contents Representational similarity metrics typically force all units to be matched, making them susceptible to noise and outliers common in neural representations. We extend the soft-matching distance to a partial optimal transport setting that allows some neurons to remain unmatched, yielding rotation-sensitive but robust correspondences. This partial soft-matching distance provides theoretical advantages -- relaxing strict mass conservation while maintaining interpretable transport costs -- and practical benefits through efficient neuron ranking in terms of cross-network alignment without costly iterative recomputation. In simulations, it preserves correct matches under outliers and reliably selects the correct model in noise-corrupted identification tasks. On fMRI data, it automatically excludes low-reliability voxels and produces voxel rankings by alignment quality that closely match computationally expensive brute-force approaches. It achieves higher alignment precision across homologous brain areas than standard soft-matching, which is forced to match all units regardless of quality. In deep networks, highly matched units exhibit similar maximally exciting images, while unmatched units show divergent patterns. This ability to partition by match quality enables focused analyses, e.g., testing whether networks have privileged axes even within their most aligned subpopulations. Overall, partial soft-matching provides a principled and practical method for representational comparison under partial correspondence.
format Preprint
id arxiv_https___arxiv_org_abs_2602_19331
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Partial Soft-Matching Distance for Neural Representational Comparison with Partial Unit Correspondence
Kapoor, Chaitanya
Williams, Alex H.
Khosla, Meenakshi
Machine Learning
Neural and Evolutionary Computing
Representational similarity metrics typically force all units to be matched, making them susceptible to noise and outliers common in neural representations. We extend the soft-matching distance to a partial optimal transport setting that allows some neurons to remain unmatched, yielding rotation-sensitive but robust correspondences. This partial soft-matching distance provides theoretical advantages -- relaxing strict mass conservation while maintaining interpretable transport costs -- and practical benefits through efficient neuron ranking in terms of cross-network alignment without costly iterative recomputation. In simulations, it preserves correct matches under outliers and reliably selects the correct model in noise-corrupted identification tasks. On fMRI data, it automatically excludes low-reliability voxels and produces voxel rankings by alignment quality that closely match computationally expensive brute-force approaches. It achieves higher alignment precision across homologous brain areas than standard soft-matching, which is forced to match all units regardless of quality. In deep networks, highly matched units exhibit similar maximally exciting images, while unmatched units show divergent patterns. This ability to partition by match quality enables focused analyses, e.g., testing whether networks have privileged axes even within their most aligned subpopulations. Overall, partial soft-matching provides a principled and practical method for representational comparison under partial correspondence.
title Partial Soft-Matching Distance for Neural Representational Comparison with Partial Unit Correspondence
topic Machine Learning
Neural and Evolutionary Computing
url https://arxiv.org/abs/2602.19331