Optimizing Rank-based Metrics with Blackbox Differentiation
Not on the map. Why?
Abstract (arXiv)
Rank-based metrics are some of the most widely used criteria for performance evaluation of computer vision models. Despite years of effort, direct optimization for these metrics remains a challenge due to their non-differentiable and non-decomposable nature. We present an efficient, theoretically sound, and general method for differentiating rank-based metrics with mini-batch gradient descent. In addition, we address optimization instability and sparsity of the supervision signal that both arise from using rank-based metrics as optimization targets. Resulting losses based on recall and Average Precision are applied to image retrieval and object detection tasks. We obtain performance that is competitive with state-of-the-art on standard image retrieval datasets and consistently improve performance of near state-of-the-art object detectors. The code is available at https://github.com/martius-lab/blackbox-backprop
Key facts. Cited by 58 papers in the map (data through 7 Oct 2026). First posted on arXiv 7 Dec 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 17; 2023: 13; 2024: 9; 2025: 3; 2026: 2.
58 papers.
Sort:
By area:
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2023LVM-Med: Learning Large-Scale Self-Supervised Vision Models for Medical Imaging via Second-order Graph Matchingcited by 60
- 2022Ordered Subgraph Aggregation Networkscited by 57
- 2022RankSim: Ranking Similarity Regularization for Deep Imbalanced Regressioncited by 52
- 2021CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraintscited by 49
- 2021Recall@k Surrogate Loss with Large Batches and Similarity Mixupcited by 49
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 34
- 2022SIMPLE: A Gradient Estimator for k-Subset Samplingcited by 34
- 2022Backpropagation through Combinatorial Algorithms: Identity with Projection Workscited by 29
- 2022Rank-based Decomposable Losses in Machine Learning: A Surveycited by 26
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspectivecited by 26
- 2021Rank & Sort Loss for Object Detection and Instance Segmentationcited by 22
38 more
- 2021Learning with Algorithmic Supervision via Continuous Relaxationscited by 21
- 2022SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problemscited by 21
- 2021Robust and Decomposable Average Precision for Image Retrievalcited by 19
- 2024RankED: Addressing Imbalance and Uncertainty in Edge Detection Using Ranking-based Lossescited by 18
- 2022Learning Where To Look -- Generative NAS is Surprisingly Efficientcited by 15
- 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
- 2021Self-Supervised Learning of Audio Representations from Permutations with Differentiable Rankingcited by 14
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximizationcited by 14
- 2023Long-tail Detection with Effective Class-Marginscited by 13
- 2021Neuro-algorithmic Policies enable Fast Combinatorial Generalizationcited by 11
- 2021Combinatorial Optimization for Panoptic Segmentation: A Fully Differentiable Approachcited by 10
- 2023Optimization of Rank Losses for Image Retrievalcited by 10
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metricscited by 9
- 2023IKOL: Inverse kinematics optimization layer for 3D human pose and shape estimation via Gauss-Newton differentiationcited by 9
- 2024Not All Pairs are Equal: Hierarchical Learning for Average-Precision-Oriented Video Retrievalcited by 9
- 2022Total Variation Optimization Layers for Computer Visioncited by 8
- 2022Hierarchical Average Precision Training for Pertinent Image Retrievalcited by 8
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 7
- 2021Set-to-Sequence Methods in Machine Learning: a Reviewcited by 6
- 2021Deep AUC Maximization for Medical Image Classification: Challenges and Opportunitiescited by 6
- 2024LPGD: A General Framework for Backpropagation through Embedded Optimization Layerscited by 6
- 2024ExGra-Med: Extended Context Graph Alignment for Medical Vision-Language Modelscited by 6
- 2022Supervised Metric Learning to Rank for Retrieval via Contextual Similarity Optimizationcited by 5
- 2022A Solver-Free Framework for Scalable Learning in Neural ILP Architecturescited by 5
- 2021Rethinking the Optimization of Average Precision: Only Penalizing Negative Instances before Positive Ones is Enoughcited by 4
- 2023Center Contrastive Loss for Metric Learningcited by 4
- 2023Divide & Conquer for Entailment-aware Multi-hop Evidence Retrievalcited by 3
- 2022Intra-class Adaptive Augmentation with Neighbor Correction for Deep Metric Learningcited by 2
- 2023SimCon Loss with Multiple Views for Text Supervised Semantic Segmentationcited by 2
- 2024Evolutionary Retrofittingcited by 2
- 2024Newton Losses: Using Curvature Information for Learning with Differentiable Algorithmscited by 2
- 2025Exact Reformulation and Optimization for Direct Metric Optimization in Binary Imbalanced Classificationcited by 2
- 2026SoftJAX & SoftTorch: Empowering Automatic Differentiation Libraries with Informative Gradientscited by 2
- 2022Improving Generalization of Metric Learning via Listwise Self-distillationcited by 1
- 2023Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive Ratecited by 1
- 2024Optimizing Automatic Speech Assessment: W-RankSim Regularization and Hybrid Feature Fusion Strategiescited by 1
- 2025Instance-Level Generation for Representation Learningcited by 1
- 2022Segment Augmentation and Differentiable Ranking for Logo Retrievalcited by 0
50 most cited of 58 All 58 (JSON)
- 2026SoftJAX & SoftTorch: Empowering Automatic Differentiation Libraries with Informative Gradients9 Mar 2026
- 2026Managing Solution Stability in Decision-Focused Learning with Cost Regularization29 Jan 2026
- 2025Instance-Level Generation for Representation Learning10 Oct 2025
- 2025MetaSynth: Multi-Agent Metadata Generation from Implicit Feedback in Black-Box Systems1 Oct 2025
- 2025Exact Reformulation and Optimization for Direct Metric Optimization in Binary Imbalanced Classification21 Jul 2025
- 2024Newton Losses: Using Curvature Information for Learning with Differentiable Algorithms24 Oct 2024
- 2024Evolutionary Retrofitting15 Oct 2024
- 2024ExGra-Med: Extended Context Graph Alignment for Medical Vision-Language Models3 Oct 2024
- 2024Realigned Softmax Warping for Deep Metric Learning28 Aug 2024
- 2024Not All Pairs are Equal: Hierarchical Learning for Average-Precision-Oriented Video Retrieval22 Jul 2024
- 2024Exploring End-to-end Differentiable Neural Charged Particle Tracking -- A Loss Landscape Perspective18 Jul 2024
- 2024LPGD: A General Framework for Backpropagation through Embedded Optimization Layers8 Jul 2024
38 more
- 2024Optimizing Automatic Speech Assessment: W-RankSim Regularization and Hybrid Feature Fusion Strategies16 Jun 2024
- 2024RankED: Addressing Imbalance and Uncertainty in Edge Detection Using Ranking-based Losses4 Mar 2024
- 2023Divide & Conquer for Entailment-aware Multi-hop Evidence Retrieval5 Nov 2023
- 2023Perturbation-based Active Learning for Question Answering4 Nov 2023
- 2023LoCUS: Learning Multiscale 3D-consistent Features from Posed Images2 Oct 2023
- 2023Optimization of Rank Losses for Image Retrieval15 Sep 2023
- 2023Center Contrastive Loss for Metric Learning1 Aug 2023
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunities25 Jul 2023
- 2023LVM-Med: Learning Large-Scale Self-Supervised Vision Models for Medical Imaging via Second-order Graph Matching20 Jun 2023
- 2023LibAUC: A Deep Learning Library for X-Risk Optimization5 Jun 2023
- 2023Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive Rate31 Mar 2023
- 2023SimCon Loss with Multiple Views for Text Supervised Semantic Segmentation7 Feb 2023
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspective2 Feb 2023
- 2023IKOL: Inverse kinematics optimization layer for 3D human pose and shape estimation via Gauss-Newton differentiation2 Feb 2023
- 2023Long-tail Detection with Effective Class-Margins23 Jan 2023
- 2022Intra-class Adaptive Augmentation with Neighbor Correction for Deep Metric Learning29 Nov 2022
- 2022Self-distillation with Online Diffusion on Batch Manifolds Improves Deep Metric Learning14 Nov 2022
- 2022SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problems22 Oct 2022
- 2022A Solver-Free Framework for Scalable Learning in Neural ILP Architectures17 Oct 2022
- 2022SIMPLE: A Gradient Estimator for k-Subset Sampling4 Oct 2022
- 2022Supervised Metric Learning to Rank for Retrieval via Contextual Similarity Optimization4 Oct 2022
- 2022Learning Deep Representations via Contrastive Learning for Instance Retrieval28 Sep 2022
- 2022Segment Augmentation and Differentiable Ranking for Logo Retrieval6 Sep 2022
- 2022Rank-based Decomposable Losses in Machine Learning: A Survey18 Jul 2022
- 2022Hierarchical Average Precision Training for Pertinent Image Retrieval5 Jul 2022
- 2022Ordered Subgraph Aggregation Networks22 Jun 2022
- 2022Improving Generalization of Metric Learning via Listwise Self-distillation17 Jun 2022
- 2022Algorithmic Foundations of Empirical X-risk Minimization1 Jun 2022
- 2022RankSim: Ranking Similarity Regularization for Deep Imbalanced Regression30 May 2022
- 2022Backpropagation through Combinatorial Algorithms: Identity with Projection Works30 May 2022
- 2022Total Variation Optimization Layers for Computer Vision7 Apr 2022
- 2022Learning Where To Look -- Generative NAS is Surprisingly Efficient16 Mar 2022
- 2021Deep AUC Maximization for Medical Image Classification: Challenges and Opportunities1 Nov 2021
- 2021Learning with Algorithmic Supervision via Continuous Relaxations11 Oct 2021
- 2021Robust and Decomposable Average Precision for Image Retrieval1 Oct 2021
- 2021Recall@k Surrogate Loss with Large Batches and Similarity Mixup25 Aug 2021
- 2021Rank & Sort Loss for Object Detection and Instance Segmentation24 Jul 2021
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximization2 Jul 2021
50 newest of 58 All 58 (JSON)
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2022Ordered Subgraph Aggregation Networkscited by 57
- 2022RankSim: Ranking Similarity Regularization for Deep Imbalanced Regressioncited by 52
- 2021CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraintscited by 49
- 2021Recall@k Surrogate Loss with Large Batches and Similarity Mixupcited by 49
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 34
- 2022SIMPLE: A Gradient Estimator for k-Subset Samplingcited by 34
- 2022Rank-based Decomposable Losses in Machine Learning: A Surveycited by 26
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspectivecited by 26
- 2021Rank & Sort Loss for Object Detection and Instance Segmentationcited by 22
- 2021Learning with Algorithmic Supervision via Continuous Relaxationscited by 21
- 2022SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problemscited by 21
12 most cited of 54 All 54 (JSON)
- 2026SoftJAX & SoftTorch: Empowering Automatic Differentiation Libraries with Informative Gradients9 Mar 2026
- 2026Managing Solution Stability in Decision-Focused Learning with Cost Regularization29 Jan 2026
- 2025Instance-Level Generation for Representation Learning10 Oct 2025
- 2025Exact Reformulation and Optimization for Direct Metric Optimization in Binary Imbalanced Classification21 Jul 2025
- 2024Newton Losses: Using Curvature Information for Learning with Differentiable Algorithms24 Oct 2024
- 2024Evolutionary Retrofitting15 Oct 2024
- 2024Realigned Softmax Warping for Deep Metric Learning28 Aug 2024
- 2024Not All Pairs are Equal: Hierarchical Learning for Average-Precision-Oriented Video Retrieval22 Jul 2024
- 2024Exploring End-to-end Differentiable Neural Charged Particle Tracking -- A Loss Landscape Perspective18 Jul 2024
- 2024LPGD: A General Framework for Backpropagation through Embedded Optimization Layers8 Jul 2024
- 2024Optimizing Automatic Speech Assessment: W-RankSim Regularization and Hybrid Feature Fusion Strategies16 Jun 2024
- 2024RankED: Addressing Imbalance and Uncertainty in Edge Detection Using Ranking-based Losses4 Mar 2024
12 newest of 54 All 54 (JSON)