Feature Selection with Distance Correlation
Not on the map. Why?
Abstract (arXiv)
Choosing which properties of the data to use as input to multivariate decision algorithms -- a.k.a. feature selection -- is an important step in solving any problem with machine learning. While there is a clear trend towards training sophisticated deep networks on large numbers of relatively unprocessed inputs (so-called automated feature engineering), for many tasks in physics, sets of theoretically well-motivated and well-understood features already exist. Working with such features can bring many benefits, including greater interpretability, reduced training and run time, and enhanced stability and robustness. We develop a new feature selection method based on Distance Correlation (DisCo), and demonstrate its effectiveness on the tasks of boosted top- and W-tagging. Using our method to select features from a set of over 7,000 energy flow polynomials, we show that we can match the performance of much deeper architectures, by using only ten features and two orders-of-magnitude fewer model parameters.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 23 of its 71 references that are in the map. First posted on arXiv 30 Nov 2022.
Builds on
23 of 71 references.
References · 23
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,040
- 2017Learning Important Features Through Propagating Activation Differencescited by 1,454
- 2019Explaining individual predictions when features are dependent: More accurate approximations to Shapley valuescited by 205
- 2018Fast inference of deep neural networks in FPGAs for particle physicscited by 132
- 2020Lorentz Group Equivariant Neural Network for Particle Physicscited by 120
- 2018Energy Flow Networks: Deep Sets for Particle Jetscited by 83
- 2019Explainable AI for Trees: From Local Explanations to Global Understandingcited by 64
- 2021Machine Learning in the Search for New Fundamental Physicscited by 50
- 2015Jet-Images -- Deep Learning Editioncited by 41
- 2022PELICAN: Permutation Equivariant and Lorentz Invariant or Covariant Aggregator Network for Particle Physicscited by 26
- 2021Point Cloud Transformers applied to Collider Physicscited by 23
11 more
- 2017QCD-Aware Recursive Neural Networks for Jet Physicscited by 22
- 2020Jet tagging in the Lund plane with graph networkscited by 22
- 2013Partial Distance Correlation with Methods for Dissimilaritiescited by 20
- 2017Jet Constituents for Deep Neural Network Based Top Quark Taggingcited by 15
- 2021Online-compatible Unsupervised Non-resonant Anomaly Detectioncited by 15
- 2020Explainable AI for ML jet taggers using expert variables and layerwise relevance propagationcited by 14
- 2017Long Short-Term Memory (LSTM) networks with jet constituents for boosted top tagging at the LHCcited by 13
- 2019Interpretable Deep Learning for Two-Prong Jet Classification with Jet Spectracited by 7
- 2022BIP: Boost Invariant Polynomials for Efficient Jet Taggingcited by 7
- 2022A Detailed Study of Interpretability of Deep Neural Network based Top Taggerscited by 6
- 2018Infrared Safety of a Neural-Net Top Tagging Algorithmcited by 2
Built on by
Citing papers per year: 2023: 1; 2024: 0; 2025: 1; 2026: 1.
3 papers.