Decorrelation using Optimal Transport

Abstract

Being able to decorrelate a feature space from protected attributes is an area of active research and study in ethics, fairness, and also natural sciences. We introduce a novel decorrelation method using Convex Neural Optimal Transport Solvers (Cnots) that is able to decorrelate a continuous feature space against protected attributes with optimal transport. We demonstrate how well it performs in the context of jet classification in high energy physics, where classifier scores are desired to be decorrelated from the mass of a jet. The decorrelation achieved in binary classification approaches the levels achieved by the state-of-the-art using conditional normalising flows. When moving to multiclass outputs the optimal transport approach performs significantly better than the state-of-the-art, suggesting substantial gains at decorrelating multidimensional feature spaces.

References in corpus (15)

• The automated computation of tree-level and next-to-leading order differential cross sections, and their matching to parton shower simulations
• The anti-k_t jet clustering algorithm
• A Brief Introduction to PYTHIA 8.1
• FastJet user manual
• DELPHES 3, A modular framework for fast simulation of a generic collider experiment
• Parton distributions with LHC data
• Automatic spin-entangled decays of heavy resonances in Monte Carlo simulations
• The Machine Learning Landscape of Top Taggers
• What is the Machine Learning?
• Decorrelated Jet Substructure Tagging using Adversarial Neural Networks
• The Dark Machines Anomaly Score Challenge: Benchmark Data and Model Independent Event Classification for the Large Hadron Collider
• DisCo Fever: Robust Networks Through Distance Correlation
• Enhancing searches for resonances with machine learning and moment decomposition
• Transport away your problems: Calibrating stochastic simulations with optimal transport
• Preserving physically important variables in optimal event selections: A case study in Higgs physics

Cited by in corpus (1)

• A continuous calibration of the ATLAS flavour-tagging classifiers via optimal transportation maps