English

Holistic Deep Learning

Machine Learning 2023-03-22 v5 Artificial Intelligence

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.

Keywords

Cite

@article{arxiv.2110.15829,
  title  = {Holistic Deep Learning},
  author = {Dimitris Bertsimas and Kimberly Villalobos Carballo and Léonard Boussioux and Michael Lingzhi Li and Alex Paskov and Ivan Paskov},
  journal= {arXiv preprint arXiv:2110.15829},
  year   = {2023}
}

Comments

Under review at Machine Learning