We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Improving the compromise between accuracy, interpretability and personalization of rule-based machine learning in medical problems

Abstract: One of the key challenges when developing a predictive model is the capability to describe the domain knowledge and the cause-effect relationships in a simple way. Decision rules are a useful and important methodology in this context, justifying their application in several areas, particularly in clinical practice. Several machine-learning classifiers have exploited the advantageous properties of decision rules to build intelligent prediction models, namely decision trees and ensembles of trees (ETs). However, such methodologies usually suffer from a trade-off between interpretability and predictive performance. Some procedures consider a simplification of ETs, using heuristic approaches to select an optimal reduced set of decision rules. In this paper, we introduce a novel step to those methodologies. We create a new component to predict if a given rule will be correct or not for a particular patient, which introduces personalization into the procedure. Furthermore, the validation results using three public clinical datasets suggest that it also allows to increase the predictive performance of the selected set of rules, improving the mentioned trade-off.
Comments: Accepted for the IEEE Engineering in Medicine and Biology Society Conference (EMBC 2021)
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2106.07827 [cs.LG]
  (or arXiv:2106.07827v2 [cs.LG] for this version)

Submission history

From: Francisco Valente [view email]
[v1] Tue, 15 Jun 2021 01:19:04 GMT (571kb,D)
[v2] Fri, 15 Oct 2021 20:14:46 GMT (569kb,D)

Link back to: arXiv, form interface, contact.