Trading-off cost of deployment versus accuracy in learning predictive models

Daniel P. Robinson, Suchi Saria

Research output: Contribution to journalArticlepeer-review

Abstract

Predictive models are finding an increasing number of applications in many industries. As a result, a practical means for trading-off the cost of deploying a model versus its effectiveness is needed. Our work is motivated by risk prediction problems in healthcare. Cost-structures in domains such as healthcare are quite complex, posing a significant challenge to existing approaches. We propose a novel framework for designing cost-sensitive structured regularizers that is suitable for problems with complex cost dependencies. We draw upon a surprising connection to boolean circuits. In particular, we represent the problem costs as a multi-layer boolean circuit, and then use properties of boolean circuits to define an extended feature vector and a group regularizer that exactly captures the underlying cost structure. The resulting regularizer may then be combined with a fidelity function to perform model prediction, for example. For the challenging real-world application of risk prediction for sepsis in intensive care units, the use of our regularizer leads to models that are in harmony with the underlying cost structure and thus provide an excellent prediction accuracy versus cost tradeoff.

Original languageEnglish (US)
Pages (from-to)1974-1982
Number of pages9
JournalIJCAI International Joint Conference on Artificial Intelligence
Volume2016-January
StatePublished - 2016

ASJC Scopus subject areas

  • Artificial Intelligence

Fingerprint Dive into the research topics of 'Trading-off cost of deployment versus accuracy in learning predictive models'. Together they form a unique fingerprint.

Cite this