Self-Healing Machine Learning: A Framework for Autonomous Adaptation in Real-World Environments

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Rauba, Paulius, Seedat, Nabeel, Kacprzyk, Krzysztof, van der Schaar, Mihaela
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916464336830464
author Rauba, Paulius
Seedat, Nabeel
Kacprzyk, Krzysztof
van der Schaar, Mihaela
author_facet Rauba, Paulius
Seedat, Nabeel
Kacprzyk, Krzysztof
van der Schaar, Mihaela
contents Real-world machine learning systems often encounter model performance degradation due to distributional shifts in the underlying data generating process (DGP). Existing approaches to addressing shifts, such as concept drift adaptation, are limited by their reason-agnostic nature. By choosing from a pre-defined set of actions, such methods implicitly assume that the causes of model degradation are irrelevant to what actions should be taken, limiting their ability to select appropriate adaptations. In this paper, we propose an alternative paradigm to overcome these limitations, called self-healing machine learning (SHML). Contrary to previous approaches, SHML autonomously diagnoses the reason for degradation and proposes diagnosis-based corrective actions. We formalize SHML as an optimization problem over a space of adaptation actions to minimize the expected risk under the shifted DGP. We introduce a theoretical framework for self-healing systems and build an agentic self-healing solution H-LLM which uses large language models to perform self-diagnosis by reasoning about the structure underlying the DGP, and self-adaptation by proposing and evaluating corrective actions. Empirically, we analyze different components of H-LLM to understand why and when it works, demonstrating the potential of self-healing ML.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00186
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Healing Machine Learning: A Framework for Autonomous Adaptation in Real-World Environments
Rauba, Paulius
Seedat, Nabeel
Kacprzyk, Krzysztof
van der Schaar, Mihaela
Machine Learning
Artificial Intelligence
Real-world machine learning systems often encounter model performance degradation due to distributional shifts in the underlying data generating process (DGP). Existing approaches to addressing shifts, such as concept drift adaptation, are limited by their reason-agnostic nature. By choosing from a pre-defined set of actions, such methods implicitly assume that the causes of model degradation are irrelevant to what actions should be taken, limiting their ability to select appropriate adaptations. In this paper, we propose an alternative paradigm to overcome these limitations, called self-healing machine learning (SHML). Contrary to previous approaches, SHML autonomously diagnoses the reason for degradation and proposes diagnosis-based corrective actions. We formalize SHML as an optimization problem over a space of adaptation actions to minimize the expected risk under the shifted DGP. We introduce a theoretical framework for self-healing systems and build an agentic self-healing solution H-LLM which uses large language models to perform self-diagnosis by reasoning about the structure underlying the DGP, and self-adaptation by proposing and evaluating corrective actions. Empirically, we analyze different components of H-LLM to understand why and when it works, demonstrating the potential of self-healing ML.
title Self-Healing Machine Learning: A Framework for Autonomous Adaptation in Real-World Environments
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2411.00186