Fast Differentiable Sorting and Ranking
Not on the map. Why?
Abstract (arXiv)
The sorting operation is one of the most commonly used building blocks in computer programming. In machine learning, it is often used for robust statistics. However, seen as a function, it is piecewise linear and as a result includes many kinks where it is non-differentiable. More problematic is the related ranking operator, often used for order statistics and ranking metrics. It is a piecewise constant function, meaning that its derivatives are null or undefined. While numerous works have proposed differentiable proxies to sorting and ranking, they do not achieve the O(n log n) time complexity one would expect from sorting and ranking operations. In this paper, we propose the first differentiable sorting and ranking operators with O(n log n) time and O(n) space complexity. Our proposal in addition enjoys exact computation and differentiation. We achieve this feat by constructing differentiable operators as projections onto the permutahedron, the convex hull of permutations, and using a reduction to isotonic optimization. Empirically, we confirm that our approach is an order of magnitude faster than existing approaches and showcase two novel applications: differentiable Spearman's rank correlation coefficient and least trimmed squares.
Key facts. Cited by 197 papers in the map (data through 14 Oct 2026). First posted on arXiv 20 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 24; 2022: 26; 2023: 41; 2024: 43; 2025: 28; 2026: 35.
197 papers.
Sort:
By area:
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
- 2023Blind Image Quality Assessment via Vision-Language Correspondence: A Multitask Learning Perspectivecited by 280
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 130
- 2021Learning Optimal Conformal Classifierscited by 112
- 2022Equivariance with Learned Canonicalization Functionscited by 109
- 2021Blindly Assess Quality of In-the-Wild Videos via Quality-aware Pre-training and Motion Perceptioncited by 97
- 2023Uncertainty Quantification over Graph with Conformalized Graph Neural Networkscited by 79
- 2025VisualQuality-R1: Reasoning-Induced Image Quality Assessment via Reinforcement Learning to Rankcited by 79
- 2021Differentiable Patch Selection for Image Recognitioncited by 68
- 2021GrooMeD-NMS: Grouped Mathematically Differentiable NMS for Monocular 3D Object Detectioncited by 66
- 2022Training Uncertainty-Aware Classifiers with Conformalized Deep Learningcited by 64
38 more
- 2021Universal Approximation of Functions on Setscited by 62
- 2025Enhancing Time Series Forecasting through Selective Representation Spaces: A Patch Perspectivecited by 39
- 2022Low Dimensional Invariant Embeddings for Universal Geometric Learningcited by 38
- 2021Self-Supervised Online Reward Shaping in Sparse-Reward Environmentscited by 37
- 2022SIMPLE: A Gradient Estimator for k-Subset Samplingcited by 34
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2021Differentiable Sorting Networks for Scalable Sorting and Ranking Supervisioncited by 31
- 2025JarvisEvo: Towards a Self-Evolving Photo Editing Agent with Synergistic Editor-Evaluator Optimizationcited by 31
- 2021Meta-Calibration: Learning of Model Calibration Using Differentiable Expected Calibration Errorcited by 29
- 2022Backpropagation through Combinatorial Algorithms: Identity with Projection Workscited by 29
- 2022Learning with Combinatorial Optimization Layers: a Probabilistic Approachcited by 29
- 2023A Large-Scale Study of Probabilistic Calibration in Neural Network Regressioncited by 27
- 2024Provably Robust Conformal Prediction with Improved Efficiencycited by 27
- 2024Boosted Conformal Prediction Intervalscited by 27
- 2021ICON: Learning Regular Maps Through Inverse Consistencycited by 26
- 2023Fast, Differentiable and Sparse Top-k: a Convex Analysis Perspectivecited by 26
- 2024Distributional Preference Alignment of LLMs via Optimal Transportcited by 26
- 2022Monotonic Differentiable Sorting Networkscited by 25
- 2021Techniques for Symbol Grounding with SATNetcited by 24
- 2021Ranking the information content of distance measurescited by 22
- 2023Exploring the Naturalness of AI-Generated Imagescited by 22
- 2021Learning with Algorithmic Supervision via Continuous Relaxationscited by 21
- 2022Differentiable Top-k Classification Learningcited by 21
- 2024An Information Theoretic Perspective on Conformal Predictioncited by 21
- 2023ProtoGate: Prototype-based Neural Networks with Global-to-local Feature Selection for Tabular Biomedical Datacited by 18
- 2025FlexSelect: Flexible Token Selection for Efficient Long Video Understandingcited by 18
- 2021Implicit Generative Copulascited by 17
- 2022Flexible Differentiable Optimization via Model Transformationscited by 16
- 2025Improving Equivariant Networks with Probabilistic Symmetry Breakingcited by 16
- 2023BOtied: Multi-objective Bayesian optimization with tied multivariate rankscited by 15
- 2024Video Quality Assessment: A Comprehensive Surveycited by 15
- 2025Leveraging Allophony in Self-Supervised Speech Models for Atypical Pronunciation Assessmentcited by 15
- 2025Unified Uncertainty-Aware Diffusion for Multi-Agent Trajectory Modelingcited by 15
- 2021Self-Supervised Learning of Audio Representations from Permutations with Differentiable Rankingcited by 14
- 2023Differentiable Clustering with Perturbed Spanning Forestscited by 14
- 2026Learning to Evict from Key-Value Cachecited by 14
- 2022Optimizing generalized Gini indices for fairness in rankingscited by 13
- 2023Learning by Sorting: Self-supervised Learning with Group Ordering Constraintscited by 13
50 most cited of 197 All 197 (JSON)
- 2026VQIL: Visible Quantum Interference Learning for Perturbation Response Ranking on Graphs9 Oct 2026
- 2026SLP-ProbHard: Probabilistic Hard-Constrained Learning via Structural Latent Parameterization27 Sep 2026
- 2026Witnesses Explain Anomalies3 Sep 2026
- 2026Forecast Collapse in Time-Series Foundation Models14 Aug 2026
- 2026Fast LapSum: Exact Differentiable Top-k at Million Scale7 Aug 2026
- 2026LaPrune: Controllable Differentiable Sparsity at Million Scale4 Aug 2026
- 2026Beyond Magnitude and Shape: A Direction-Aware Loss for Time Series Forecasting3 Aug 2026
- 2026Controlling Implicit Shortcut Reliance in L2 Spoken English Auto-markers17 Jul 2026
- 2026Finding Multiple Interpretations in Datasets10 Jun 2026
- 2026Can We Predict The Human Preference For Text-to-Image Content Prior To Generation And Is It Even Useful To Do So?3 Jun 2026
- 2026InfoAtlas: A Foundation Model for Zero-Shot Statistical Dependence Estimate29 May 2026
- 2026Learning Permutation from Structure Without Supervision25 May 2026
38 more
- 2026Polar probe linearly decodes semantic structures from LLMs13 May 2026
- 2026Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling12 May 2026
- 2026Heteroscedastic Diffusion for Multi-Agent Trajectory Modeling11 May 2026
- 2026Contextual Plackett-Luce: An Efficient Neural Model for Probabilistic Sequence Selection under Ambiguity9 May 2026
- 2026Super-Level-Set Regression: Conditional Quantiles via Volume Minimization7 May 2026
- 2026LambdaRankIC: Directly Optimizing Rank IC for Financial Prediction1 May 2026
- 2026Metric-agnostic Learning-to-Rank via Boosting and Rank Approximation16 Apr 2026
- 2026Learn to Rank: Visual Attribution by Learning Importance Ranking7 Apr 2026
- 2026Soft Tournament Equilibrium: Differentiable Set-Valued Inference for Non-Transitive Pairwise Comparisons6 Apr 2026
- 2026ArrowFlow: Hierarchical Machine Learning in the Space of Permutations5 Apr 2026
- 2026Learning Permutation Distributions via Reflected Diffusion on Ranks18 Mar 2026
- 2026Integrating Inductive Biases in Transformers via Distillation for Financial Time Series Forecasting17 Mar 2026
- 2026Na"ive PAINE: Lightweight Text-to-Image Generation Improvement with Prompt Evaluation12 Mar 2026
- 2026SoftJAX & SoftTorch: Empowering Automatic Differentiation Libraries with Informative Gradients9 Mar 2026
- 2026Differentiable Zero-One Loss via Hypersimplex Projections26 Feb 2026
- 2026RaCo: Ranking and Covariance for Practical Learned Keypoints17 Feb 2026
- 2026CAIRO: Decoupling Order from Scale in Regression16 Feb 2026
- 2026Learning to Evict from Key-Value Cache10 Feb 2026
- 2026GRAFT: Decoupling Ranking and Calibration for Survival Analysis8 Feb 2026
- 2026Learning to Optimize by Differentiable Programming23 Jan 2026
- 2026Reward Learning through Ranking Mean Squared Error14 Jan 2026
- 2026Rank-based Geographical Regularization: Revisiting Contrastive Self-Supervised Learning for Multispectral Remote Sensing Imagery5 Jan 2026
- 2026Learning Resilient Elections with Adversarial GNNs4 Jan 2026
- 2025Structural Incompatibility of Differentiable Sorting and Within-Vector Rank Normalization27 Dec 2025
- 2025Conformal Correction for Efficiency May be at Odds with Entropy2 Dec 2025
- 2025JarvisEvo: Towards a Self-Evolving Photo Editing Agent with Synergistic Editor-Evaluator Optimization28 Nov 2025
- 2025Comprehension of Multilingual Expressions Referring to Target Objects in Visual Inputs14 Nov 2025
- 2025Enhancing Time Series Forecasting through Selective Representation Spaces: A Patch Perspective16 Oct 2025
- 2025Conformal Prediction for Signal Temporal Logic Inference29 Sep 2025
- 2025Deceptive Risk Minimization: Out-of-Distribution Generalization by Deceiving Distribution Shift Detectors15 Sep 2025
- 2025SoftStep: Learning Sparse Similarity Powers Deep Neighbor-Based Regression9 Jun 2025
- 2025Residual Reweighted Conformal Prediction for Graph Neural Networks9 Jun 2025
- 2025XicorAttention: Time Series Transformer Using Attention with Nonlinear Correlation3 Jun 2025
- 2025FlexSelect: Flexible Token Selection for Efficient Long Video Understanding1 Jun 2025
- 2025Differentiable Generalized Sliced Wasserstein Plans28 May 2025
- 2025Deep greedy unfolding: Sorting out argsorting in greedy sparse recovery algorithms21 May 2025
- 2025VisualQuality-R1: Reasoning-Induced Image Quality Assessment via Reinforcement Learning to Rank20 May 2025
- 2025InfiGFusion: Graph-on-Logits Distillation via Efficient Gromov-Wasserstein for Model Fusion20 May 2025
50 newest of 197 All 197 (JSON)
- 2024SymmetricDiffusers: Learning Discrete Diffusion on Finite Symmetric Groupscited by 6
- 2024CHAMP: Conformalized 3D Human Multi-Hypothesis Pose Estimatorscited by 5
- 2026Heteroscedastic Diffusion for Multi-Agent Trajectory Modelingcited by 3
- 2026Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Samplingcited by 3
- 2026Na"ive PAINE: Lightweight Text-to-Image Generation Improvement with Prompt Evaluationcited by 0
- 2026Can We Predict The Human Preference For Text-to-Image Content Prior To Generation And Is It Even Useful To Do So?cited by 0
- 2026Can We Predict The Human Preference For Text-to-Image Content Prior To Generation And Is It Even Useful To Do So?3 Jun 2026
- 2026Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling12 May 2026
- 2026Heteroscedastic Diffusion for Multi-Agent Trajectory Modeling11 May 2026
- 2026Na"ive PAINE: Lightweight Text-to-Image Generation Improvement with Prompt Evaluation12 Mar 2026
- 2024SymmetricDiffusers: Learning Discrete Diffusion on Finite Symmetric Groups3 Oct 2024
- 2024CHAMP: Conformalized 3D Human Multi-Hypothesis Pose Estimators27 May 2024
- 2025JarvisEvo: Towards a Self-Evolving Photo Editing Agent with Synergistic Editor-Evaluator Optimizationcited by 31
- 2024Distributional Preference Alignment of LLMs via Optimal Transportcited by 26
- 2025Multivariate Conformal Selectioncited by 8
- 2024Holistic Utility Preference Learning for Listwise Alignmentcited by 2
- 2026Reward Learning through Ranking Mean Squared Errorcited by 0
- 2026Reward Learning through Ranking Mean Squared Error14 Jan 2026
- 2025JarvisEvo: Towards a Self-Evolving Photo Editing Agent with Synergistic Editor-Evaluator Optimization28 Nov 2025
- 2025Multivariate Conformal Selection1 May 2025
- 2024Holistic Utility Preference Learning for Listwise Alignment17 Oct 2024
- 2024Distributional Preference Alignment of LLMs via Optimal Transport9 Jun 2024
- 2025VisualQuality-R1: Reasoning-Induced Image Quality Assessment via Reinforcement Learning to Rankcited by 79
- 2022Backpropagation through Combinatorial Algorithms: Identity with Projection Workscited by 29
- 2023Learning Reliable Logical Rules with SATNetcited by 3
- 2024What Makes Two Language Models Think Alike?cited by 1
- 2025VisualQuality-R1: Reasoning-Induced Image Quality Assessment via Reinforcement Learning to Rank20 May 2025
- 2024What Makes Two Language Models Think Alike?18 Jun 2024
- 2023Learning Reliable Logical Rules with SATNet3 Oct 2023
- 2022Backpropagation through Combinatorial Algorithms: Identity with Projection Works30 May 2022
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignmentcited by 11
- 2024Efficient and Robust Knowledge Distillation from A Stronger Teacher Based on Correlation Matchingcited by 2
- 2023MonoSKD: General Distillation Framework for Monocular 3D Object Detection via Spearman Correlation Coefficientcited by 1
- 2024Efficient and Robust Knowledge Distillation from A Stronger Teacher Based on Correlation Matching9 Oct 2024
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignment19 Sep 2024
- 2023MonoSKD: General Distillation Framework for Monocular 3D Object Detection via Spearman Correlation Coefficient17 Oct 2023
- 2026Learning to Evict from Key-Value Cachecited by 14
- 2026Learning to Evict from Key-Value Cache10 Feb 2026
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
- 2023Blind Image Quality Assessment via Vision-Language Correspondence: A Multitask Learning Perspectivecited by 280
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 130
- 2021Learning Optimal Conformal Classifierscited by 112
- 2022Equivariance with Learned Canonicalization Functionscited by 109
- 2021Blindly Assess Quality of In-the-Wild Videos via Quality-aware Pre-training and Motion Perceptioncited by 97
- 2023Uncertainty Quantification over Graph with Conformalized Graph Neural Networkscited by 79
- 2021Differentiable Patch Selection for Image Recognitioncited by 68
- 2021GrooMeD-NMS: Grouped Mathematically Differentiable NMS for Monocular 3D Object Detectioncited by 66
- 2022Training Uncertainty-Aware Classifiers with Conformalized Deep Learningcited by 64
- 2021Universal Approximation of Functions on Setscited by 62
12 most cited of 176 All 176 (JSON)
- 2026VQIL: Visible Quantum Interference Learning for Perturbation Response Ranking on Graphs9 Oct 2026
- 2026SLP-ProbHard: Probabilistic Hard-Constrained Learning via Structural Latent Parameterization27 Sep 2026
- 2026Witnesses Explain Anomalies3 Sep 2026
- 2026Forecast Collapse in Time-Series Foundation Models14 Aug 2026
- 2026Fast LapSum: Exact Differentiable Top-k at Million Scale7 Aug 2026
- 2026LaPrune: Controllable Differentiable Sparsity at Million Scale4 Aug 2026
- 2026Beyond Magnitude and Shape: A Direction-Aware Loss for Time Series Forecasting3 Aug 2026
- 2026Controlling Implicit Shortcut Reliance in L2 Spoken English Auto-markers17 Jul 2026
- 2026Finding Multiple Interpretations in Datasets10 Jun 2026
- 2026InfoAtlas: A Foundation Model for Zero-Shot Statistical Dependence Estimate29 May 2026
- 2026Learning Permutation from Structure Without Supervision25 May 2026
- 2026Polar probe linearly decodes semantic structures from LLMs13 May 2026
12 newest of 176 All 176 (JSON)