Holistic deep learning

Dimitris Bertsimas, Kimberly Villalobos Carballo, Léonard Boussioux, Michael Lingzhi Li, Alex Paskov, Ivan Paskov

Research output: Contribution to journalArticlepeer-review

Abstract

This paper presents a novel holistic deep learning framework that simultaneously addresses the challenges of vulnerability to input perturbations, overparametrization, and performance instability from different train-validation splits. The proposed framework holistically improves accuracy, robustness, sparsity, and stability over standard deep learning models, as demonstrated by extensive experiments on both tabular and image data sets. The results are further validated by ablation experiments and SHAP value analysis, which reveal the interactions and trade-offs between the different evaluation metrics. To support practitioners applying our framework, we provide a prescriptive approach that offers recommendations for selecting an appropriate training loss function based on their specific objectives. All the code to reproduce the results can be found at https://github.com/kimvc7/HDL .

Original languageEnglish (US)
Pages (from-to)159-183
Number of pages25
JournalMachine Learning
Volume113
Issue number1
DOIs
StatePublished - Jan 2024

Keywords

  • Deep learning
  • Optimization
  • Regularization
  • Robustness
  • Sparsity
  • Stability

ASJC Scopus subject areas

  • Software
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Holistic deep learning'. Together they form a unique fingerprint.

Cite this