arrow
Return

Holistic deep learning

delete2023-12-07
delete2
delete
OA
AI
D
Dimitris Bertsimas *
K
Kimberly Villalobos Carballo
L
Léonard Boussioux
M
Michael Lingzhi Li
A
Alex Paskov
I
Ivan Paskov
DOI:10.1007/s10994-023-06482-ydelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
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:
Deep learning
Optimization
Robustness
Sparsity
Stability
Regularization

Journal

Machine Learning cover
Machine Learning
IF:
2.9
Papers:
2.6K
Citations:
3.4W

Organization

H
Harvard University
Scholars:
26.2W
Papers: 21.9W
Citations: 28.7W