A Novel Ontology-guided Attribute Partitioning Ensemble Learning Model for Early Prediction of Cognitive Deficits using Quantitative Structural MRI in Very Preterm Infants

02/08/2022
by   Zhiyuan Li, et al.
0

Structural magnetic resonance imaging studies have shown that brain anatomical abnormalities are associated with cognitive deficits in preterm infants. Brain maturation and geometric features can be used with machine learning models for predicting later neurodevelopmental deficits. However, traditional machine learning models would suffer from a large feature-to-instance ratio (i.e., a large number of features but a small number of instances/samples). Ensemble learning is a paradigm that strategically generates and integrates a library of machine learning classifiers and has been successfully used on a wide variety of predictive modeling problems to boost model performance. Attribute (i.e., feature) bagging method is the most commonly used feature partitioning scheme, which randomly and repeatedly draws feature subsets from the entire feature set. Although attribute bagging method can effectively reduce feature dimensionality to handle the large feature-to-instance ratio, it lacks consideration of domain knowledge and latent relationship among features. In this study, we proposed a novel Ontology-guided Attribute Partitioning (OAP) method to better draw feature subsets by considering domain-specific relationship among features. With the better partitioned feature subsets, we developed an ensemble learning framework, which is referred to as OAP Ensemble Learning (OAP-EL). We applied the OAP-EL to predict cognitive deficits at 2 year of age using quantitative brain maturation and geometric features obtained at term equivalent age in very preterm infants. We demonstrated that the proposed OAP-EL approach significantly outperformed the peer ensemble learning and traditional machine learning approaches.

READ FULL TEXT

page 5

page 8

page 17

research
05/26/2022

Machine Learning Models Are Not Necessarily Biased When Constructed Properly: Evidence from Neuroimaging Studies

Despite the great promise that machine learning has offered in many fiel...
research
09/21/2021

Comparison of single and multitask learning for predicting cognitive decline based on MRI data

The Alzheimer's Disease Assessment Scale-Cognitive subscale (ADAS-Cog) i...
research
10/29/2020

AutoAtlas: Neural Network for 3D Unsupervised Partitioning and Representation Learning

We present a novel neural network architecture called AutoAtlas for full...
research
05/02/2018

Large-Scale Unsupervised Deep Representation Learning for Brain Structure

Machine Learning (ML) is increasingly being used for computer aided diag...
research
03/27/2019

A novel framework for automatic detection of Autism: A study on Corpus Callosum and Intracranial Brain Volume

Computer vision and machine learning are the linchpin of field of automa...
research
03/30/2021

Scalable Statistical Inference of Photometric Redshift via Data Subsampling

Handling big data has largely been a major bottleneck in traditional sta...

Please sign up or login with your details

Forgot password? Click here to reset