Multi-Objective Optimization of Performance and Interpretability of Tabular Supervised Machine Learning Models

Schneider, Lennart ; Bischl, Bernd und Thomas, Janek (2023): Multi-Objective Optimization of Performance and Interpretability of Tabular Supervised Machine Learning Models. GECCO '23: Genetic and Evolutionary Computation Conference, Lisbon Portugal, July 15 - 19, 2023. Silva, Sara und Paquete, Luís (Hrsg.): In: GECCO '23: Proceedings of the Genetic and Evolutionary Computation Conference, New York, NY, United States: Association for Computing Machinery. S. 538-547

Abstract

We present a model-agnostic framework for jointly optimizing the predictive performance and interpretability of supervised machine learning models for tabular data. Interpretability is quantified via three measures: feature sparsity, interaction sparsity of features, and sparsity of non-monotone feature effects. By treating hyperparameter optimization of a machine learning algorithm as a multi-objective optimization problem, our framework allows for generating diverse models that trade off high performance and ease of interpretability in a single optimization run. Efficient optimization is achieved via augmentation of the search space of the learning algorithm by incorporating feature selection, interaction and monotonicity constraints into the hyperparameter search space. We demonstrate that the optimization problem effectively translates to finding the Pareto optimal set of groups of selected features that are allowed to interact in a model, along with finding their optimal monotonicity constraints and optimal hyperparameters of the learning algorithm itself. We then introduce a novel evolutionary algorithm that can operate efficiently on this augmented search space. In benchmark experiments, we show that our framework is capable of finding diverse models that are highly competitive or outperform state-of-the-art XGBoost or Explainable Boosting Machine models, both with respect to performance and interpretability.

• Dokumententyp: Konferenzbeitrag (Paper)
• Fakultät: Mathematik, Informatik und Statistik > Statistik
• Themengebiete: 000 Informatik, Informationswissenschaft, allgemeine Werke > 004 Informatik ; 300 Sozialwissenschaften > 310 Statistiken
• ISBN: 979-8-4007-0119-1
• Ort: New York, NY, United States
• Sprache: Englisch