Indecisive trees for classification and prediction of knee osteoarthritis

Luca Minciullo*, Paul A. Bromiley, David T. Felson, Timothy F. Cootes

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

223 Downloads (Pure)

Abstract

Random forests are widely used for classification and regression tasks in medical image analysis. Each tree in the forest contains binary decision nodes that choose whether a sample should be passed to one of two child nodes. We demonstrate that replacing this with something less decisive, where some samples may go to both child nodes, can improve performance for both individual trees and whole forests. Introducing a soft decision at each node means that a sample may propagate to multiple leaves. The tree output should thus be a weighted sum of the individual leaf values. We show how the leaves can be optimised to improve performance and how backpropagation can be used to optimise the parameters of the decision functions at each node. Finally, we show that the new method outperforms an equivalent random forest on a disease classification and prediction task.

Original languageEnglish
Title of host publicationMachine Learning in Medical Imaging - 8th International Workshop, MLMI 2017, Held in Conjunction with MICCAI 2017, Proceedings
PublisherSpringer Nature
Pages283-290
Number of pages8
Volume10541 LNCS
ISBN (Print)9783319673882
DOIs
Publication statusPublished - 2017
Event8th International Workshop on Machine Learning in Medical Imaging, MLMI 2017 held in conjunction with the 20th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2017 - Quebec City, Canada
Duration: 10 Sept 201710 Sept 2017

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume10541 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference8th International Workshop on Machine Learning in Medical Imaging, MLMI 2017 held in conjunction with the 20th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2017
Country/TerritoryCanada
CityQuebec City
Period10/09/1710/09/17

Keywords

  • Decision trees
  • Optimisation
  • Random forests

Fingerprint

Dive into the research topics of 'Indecisive trees for classification and prediction of knee osteoarthritis'. Together they form a unique fingerprint.

Cite this