COIN: Chance-Constrained Imitation Learning for Uncertainty-aware Adaptive Resource Oversubscription Policy

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Lu, Das, Mayukh, Yang, Fangkai, Duo, Chao, Qiao, Bo, Dong, Hang, Qin, Si, Bansal, Chetan, Lin, Qingwei, Rajmohan, Saravan, Zhang, Dongmei, Zhang, Qi
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917567205998592
author Wang, Lu
Das, Mayukh
Yang, Fangkai
Duo, Chao
Qiao, Bo
Dong, Hang
Qin, Si
Bansal, Chetan
Lin, Qingwei
Rajmohan, Saravan
Zhang, Dongmei
Zhang, Qi
author_facet Wang, Lu
Das, Mayukh
Yang, Fangkai
Duo, Chao
Qiao, Bo
Dong, Hang
Qin, Si
Bansal, Chetan
Lin, Qingwei
Rajmohan, Saravan
Zhang, Dongmei
Zhang, Qi
contents We address the challenge of learning safe and robust decision policies in presence of uncertainty in context of the real scientific problem of adaptive resource oversubscription to enhance resource efficiency while ensuring safety against resource congestion risk. Traditional supervised prediction or forecasting models are ineffective in learning adaptive policies whereas standard online optimization or reinforcement learning is difficult to deploy on real systems. Offline methods such as imitation learning (IL) are ideal since we can directly leverage historical resource usage telemetry. But, the underlying aleatoric uncertainty in such telemetry is a critical bottleneck. We solve this with our proposed novel chance-constrained imitation learning framework, which ensures implicit safety against uncertainty in a principled manner via a combination of stochastic (chance) constraints on resource congestion risk and ensemble value functions. This leads to substantial ($\approx 3-4\times$) improvement in resource efficiency and safety in many oversubscription scenarios, including resource management in cloud services.
format Preprint
id arxiv_https___arxiv_org_abs_2401_07051
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle COIN: Chance-Constrained Imitation Learning for Uncertainty-aware Adaptive Resource Oversubscription Policy
Wang, Lu
Das, Mayukh
Yang, Fangkai
Duo, Chao
Qiao, Bo
Dong, Hang
Qin, Si
Bansal, Chetan
Lin, Qingwei
Rajmohan, Saravan
Zhang, Dongmei
Zhang, Qi
Machine Learning
Artificial Intelligence
We address the challenge of learning safe and robust decision policies in presence of uncertainty in context of the real scientific problem of adaptive resource oversubscription to enhance resource efficiency while ensuring safety against resource congestion risk. Traditional supervised prediction or forecasting models are ineffective in learning adaptive policies whereas standard online optimization or reinforcement learning is difficult to deploy on real systems. Offline methods such as imitation learning (IL) are ideal since we can directly leverage historical resource usage telemetry. But, the underlying aleatoric uncertainty in such telemetry is a critical bottleneck. We solve this with our proposed novel chance-constrained imitation learning framework, which ensures implicit safety against uncertainty in a principled manner via a combination of stochastic (chance) constraints on resource congestion risk and ensemble value functions. This leads to substantial ($\approx 3-4\times$) improvement in resource efficiency and safety in many oversubscription scenarios, including resource management in cloud services.
title COIN: Chance-Constrained Imitation Learning for Uncertainty-aware Adaptive Resource Oversubscription Policy
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2401.07051