PyTorch Metric Learning
Not on the map. Why?
Abstract (arXiv)
Deep metric learning algorithms have a wide variety of applications, but implementing these algorithms can be tedious and time consuming. PyTorch Metric Learning is an open source library that aims to remove this barrier for both researchers and practitioners. The modular and flexible design allows users to easily try out different combinations of algorithms in their existing code. It also comes with complete train/test workflows, for users who want results fast. Code and documentation is available at https://www.github.com/KevinMusgrave/pytorch-metric-learning.
Key facts. Cited by 57 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Aug 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 16; 2023: 11; 2024: 9; 2025: 2; 2026: 5.
57 papers.
Sort:
- 2021ABO: Dataset and Benchmarks for Real-World 3D Object Understandingcited by 360
- 2022GSV-Cities: Toward Appropriate Supervised Visual Place Recognitioncited by 92
- 2021Understanding and Mitigating Annotation Bias in Facial Expression Recognitioncited by 50
- 2021Noise-resistant Deep Metric Learning with Ranking-based Instance Selectioncited by 23
- 2022Cross-Domain Video Anomaly Detection without Target Domain Adaptationcited by 20
- 2022Graph Neural Network for Cell Tracking in Microscopy Videoscited by 19
- 2022Bispectral Neural Networkscited by 17
- 2024Pair-VPR: Place-Aware Pre-training and Contrastive Pair Classification for Visual Place Recognition with Vision Transformerscited by 15
- 2023Triple-stream Deep Metric Learning of Great Ape Behavioural Actionscited by 14
- 2021Contrastive Fine-tuning Improves Robustness for Neural Rankerscited by 12
- 2022Match Cutting: Finding Cuts with Smooth Visual Transitionscited by 12
- 2021Eye Know You: Metric Learning for End-to-end Biometric Authentication Using Eye Movements from a Longitudinal Datasetcited by 11
38 more
- 2023Efficient OCR for Building a Diverse Digital Historycited by 10
- 2024Tails Tell Tales: Chapter-Wide Manga Transcriptions with Character Namescited by 10
- 2022Spectral, Probabilistic, and Deep Metric Learning: Tutorial and Surveycited by 9
- 2022Hierarchical Average Precision Training for Pertinent Image Retrievalcited by 8
- 2023Identity-Aware Semi-Supervised Learning for Comic Character Re-Identificationcited by 8
- 2022Speech Sequence Embeddings using Nearest Neighbors Contrastive Learningcited by 7
- 2023Reveal the Unknown: Out-of-Knowledge-Base Mention Discovery with Entity Linkingcited by 7
- 2023Automatic MILP Solver Configuration By Learning Problem Similaritiescited by 5
- 2022Eye Know You Too: A DenseNet Architecture for End-to-end Eye Movement Biometricscited by 3
- 2022Was that so hard? Estimating human classification difficultycited by 3
- 2022DialAug: Mixing up Dialogue Contexts in Contrastive Learning for Robust Conversational Modelingcited by 3
- 2022AdaTriplet: Adaptive Gradient Triplet Loss with Automatic Margin Learning for Forensic Medical Image Matchingcited by 3
- 2022Guided Deep Metric Learningcited by 3
- 2023Deductive Additivity for Planning of Natural Language Proofscited by 3
- 2024Earth System Data Cubes: Avenues for advancing Earth system researchcited by 3
- 2024On dataset transferability in medical image classificationcited by 3
- 2025YOLO11-JDE: Fast and Accurate Multi-Object Tracking with Self-Supervised Re-IDcited by 3
- 2021Rethinking Interactive Image Segmentation: Feature Space Annotationcited by 2
- 2022Category-Level Pose Retrieval with Contrastive Features Learnt with Occlusion Augmentationcited by 2
- 2023Versatile User Identification in Extended Reality using Pretrained Similarity-Learningcited by 2
- 2023Quantifying Character Similarity with Vision Transformerscited by 2
- 2023A metric learning approach for endoscopic kidney stone identificationcited by 2
- 2024Temporal Persistence and Intercorrelation of Embeddings Learned by an End-to-End Deep Learning Eye Movement-driven Biometrics Pipelinecited by 2
- 2021Incremental Class Learning using Variational Autoencoders with Similarity Learningcited by 1
- 2022Matching Text and Audio Embeddings: Exploring Transfer-learning Strategies for Language-based Audio Retrievalcited by 1
- 2024Hierarchical place recognition with omnidirectional images and curriculum learning-based loss functionscited by 1
- 2026Cross-Domain Generalization in Optical Networks via Joint Contrastive and Classification Learningcited by 1
- 2021Pretraining Neural Architecture Search Controllers with Locality-based Self-Supervised Learningcited by 0
- 2021MixDefense: A Defense-in-Depth Framework for Adversarial Example Detection Based on Statistical and Semantic Analysiscited by 0
- 2021VeriMedi: Pill Identification using Proxy-based Deep Metric Learning and Exact Solutioncited by 0
- 2021Video Surveillance for Road Traffic Monitoringcited by 0
- 2021Noise-Resistant Deep Metric Learning with Probabilistic Instance Filteringcited by 0
- 2021Multi-granularity for knowledge distillationcited by 0
- 2021Deep metric learning improves lab of origin prediction of genetically engineered plasmidscited by 0
- 2022Massively Multilingual Lexical Specialization of Multilingual Transformerscited by 0
- 2023Ensemble Transfer Learning for Multilingual Coreference Resolutioncited by 0
- 2023Contrastive Learning for API Aspect Analysiscited by 0
- 2024Enhancing Understanding Through Wildlife Re-Identificationcited by 0
50 most cited of 57 All 57 (JSON)
- 2026Contrastive Learning for Authorship Verification23 Sep 2026
- 2026Automated Goldsmith's Mark Retrieval in Silverware17 Sep 2026
- 2026Face and Voice Cross-modal Association with Learning Convex Feature Embedding30 Jul 2026
- 2026Cross-Domain Generalization in Optical Networks via Joint Contrastive and Classification Learning22 Jul 2026
- 2026Patch Ensembles for Robust Salmon Re-Identification with Weak Trajectory Labels18 May 2026
- 2025Improving Generalization in MRI-Based Deep Learning Models for Total Knee Replacement Prediction27 Apr 2025
- 2025YOLO11-JDE: Fast and Accurate Multi-Object Tracking with Self-Supervised Re-ID23 Jan 2025
- 2024On dataset transferability in medical image classification28 Dec 2024
- 2024Pair-VPR: Place-Aware Pre-training and Contrastive Pair Classification for Visual Place Recognition with Vision Transformers9 Oct 2024
- 2024MARs: Multi-view Attention Regularizations for Patch-based Feature Recognition of Space Terrain7 Oct 2024
- 2024Earth System Data Cubes: Avenues for advancing Earth system research5 Aug 2024
- 2024Tails Tell Tales: Chapter-Wide Manga Transcriptions with Character Names1 Aug 2024
38 more
- 2024Explainable Metric Learning for Deflating Data Bias5 Jul 2024
- 2024Enhancing Understanding Through Wildlife Re-Identification17 May 2024
- 2024Hierarchical place recognition with omnidirectional images and curriculum learning-based loss functions22 Apr 2024
- 2024Temporal Persistence and Intercorrelation of Embeddings Learned by an End-to-End Deep Learning Eye Movement-driven Biometrics Pipeline26 Feb 2024
- 2023Identity-Aware Semi-Supervised Learning for Comic Character Re-Identification17 Aug 2023
- 2023Contrastive Learning for API Aspect Analysis31 Jul 2023
- 2023A metric learning approach for endoscopic kidney stone identification13 Jul 2023
- 2023Deductive Additivity for Planning of Natural Language Proofs5 Jul 2023
- 2023Automatic MILP Solver Configuration By Learning Problem Similarities2 Jul 2023
- 2023Quantifying Character Similarity with Vision Transformers24 May 2023
- 2023Efficient OCR for Building a Diverse Digital History5 Apr 2023
- 2023Versatile User Identification in Extended Reality using Pretrained Similarity-Learning15 Feb 2023
- 2023Reveal the Unknown: Out-of-Knowledge-Base Mention Discovery with Entity Linking14 Feb 2023
- 2023Ensemble Transfer Learning for Multilingual Coreference Resolution22 Jan 2023
- 2023Triple-stream Deep Metric Learning of Great Ape Behavioural Actions6 Jan 2023
- 2022Cross-Domain Video Anomaly Detection without Target Domain Adaptation14 Dec 2022
- 2022GSV-Cities: Toward Appropriate Supervised Visual Place Recognition19 Oct 2022
- 2022Match Cutting: Finding Cuts with Smooth Visual Transitions11 Oct 2022
- 2022Matching Text and Audio Embeddings: Exploring Transfer-learning Strategies for Language-based Audio Retrieval6 Oct 2022
- 2022Bispectral Neural Networks7 Sep 2022
- 2022Category-Level Pose Retrieval with Contrastive Features Learnt with Occlusion Augmentation12 Aug 2022
- 2022Massively Multilingual Lexical Specialization of Multilingual Transformers1 Aug 2022
- 2022Hierarchical Average Precision Training for Pertinent Image Retrieval5 Jul 2022
- 2022Guided Deep Metric Learning4 Jun 2022
- 2022AdaTriplet: Adaptive Gradient Triplet Loss with Automatic Margin Learning for Forensic Medical Image Matching5 May 2022
- 2022DialAug: Mixing up Dialogue Contexts in Contrastive Learning for Robust Conversational Modeling15 Apr 2022
- 2022Speech Sequence Embeddings using Nearest Neighbors Contrastive Learning11 Apr 2022
- 2022Was that so hard? Estimating human classification difficulty22 Mar 2022
- 2022Graph Neural Network for Cell Tracking in Microscopy Videos9 Feb 2022
- 2022Spectral, Probabilistic, and Deep Metric Learning: Tutorial and Survey23 Jan 2022
- 2022Eye Know You Too: A DenseNet Architecture for End-to-end Eye Movement Biometrics5 Jan 2022
- 2021Deep metric learning improves lab of origin prediction of genetically engineered plasmids24 Nov 2021
- 2021ABO: Dataset and Benchmarks for Real-World 3D Object Understanding12 Oct 2021
- 2021Incremental Class Learning using Variational Autoencoders with Similarity Learning4 Oct 2021
- 2021Understanding and Mitigating Annotation Bias in Facial Expression Recognition19 Aug 2021
- 2021Multi-granularity for knowledge distillation15 Aug 2021
- 2021Noise-Resistant Deep Metric Learning with Probabilistic Instance Filtering3 Aug 2021
- 2021Contrastive Fine-tuning Improves Robustness for Neural Rankers27 May 2021
50 newest of 57 All 57 (JSON)
- 2021ABO: Dataset and Benchmarks for Real-World 3D Object Understandingcited by 360
- 2022GSV-Cities: Toward Appropriate Supervised Visual Place Recognitioncited by 92
- 2021Understanding and Mitigating Annotation Bias in Facial Expression Recognitioncited by 50
- 2021Noise-resistant Deep Metric Learning with Ranking-based Instance Selectioncited by 23
- 2022Cross-Domain Video Anomaly Detection without Target Domain Adaptationcited by 20
- 2022Graph Neural Network for Cell Tracking in Microscopy Videoscited by 19
- 2022Bispectral Neural Networkscited by 17
- 2024Pair-VPR: Place-Aware Pre-training and Contrastive Pair Classification for Visual Place Recognition with Vision Transformerscited by 15
- 2023Triple-stream Deep Metric Learning of Great Ape Behavioural Actionscited by 14
- 2021Contrastive Fine-tuning Improves Robustness for Neural Rankerscited by 12
- 2022Match Cutting: Finding Cuts with Smooth Visual Transitionscited by 12
- 2021Eye Know You: Metric Learning for End-to-end Biometric Authentication Using Eye Movements from a Longitudinal Datasetcited by 11
12 most cited of 57 All 57 (JSON)
- 2026Contrastive Learning for Authorship Verification23 Sep 2026
- 2026Automated Goldsmith's Mark Retrieval in Silverware17 Sep 2026
- 2026Face and Voice Cross-modal Association with Learning Convex Feature Embedding30 Jul 2026
- 2026Cross-Domain Generalization in Optical Networks via Joint Contrastive and Classification Learning22 Jul 2026
- 2026Patch Ensembles for Robust Salmon Re-Identification with Weak Trajectory Labels18 May 2026
- 2025Improving Generalization in MRI-Based Deep Learning Models for Total Knee Replacement Prediction27 Apr 2025
- 2025YOLO11-JDE: Fast and Accurate Multi-Object Tracking with Self-Supervised Re-ID23 Jan 2025
- 2024On dataset transferability in medical image classification28 Dec 2024
- 2024Pair-VPR: Place-Aware Pre-training and Contrastive Pair Classification for Visual Place Recognition with Vision Transformers9 Oct 2024
- 2024MARs: Multi-view Attention Regularizations for Patch-based Feature Recognition of Space Terrain7 Oct 2024
- 2024Earth System Data Cubes: Avenues for advancing Earth system research5 Aug 2024
- 2024Tails Tell Tales: Chapter-Wide Manga Transcriptions with Character Names1 Aug 2024
12 newest of 57 All 57 (JSON)