Categories
IPMI 2021 poster

Quantile Regression for Uncertainty Estimation in VAEs with Applications to Brain Lesion Detection

Authors: Haleh Akrami (Signal and Image Processing Institute at University of Southern California)*; Anand Joshi (University of Southern California); Sergul Aydore (Amazon Web Services); Richard Leahy (Signal and Image Processing Institute at University of Southern California)

Poster

Abstract: The Variational AutoEncoder (VAE) has become one of the most popular models for anomaly detection in applications such as lesion detection in medical images. The VAE is a generative graphical model that is used to learn the data distribution from samples and then generate new samples from this distribution. By training on normal samples, the VAE can be used to detect inputs that deviate from this learned distribution. The VAE models the output as a conditionally independent Gaussian characterized by means and variances for each output dimension. VAEs can therefore use reconstruction probability instead of reconstruction error for anomaly detection. Unfortunately, joint optimization of both mean and variance in the VAE leads to the well-known problem of shrinkage or underestimation of variance. We describe an alternative VAE model, Quantile-Regression VAE (QR-VAE), that avoids this variance shrinkage problem by estimating conditional quantiles for the given input image.
Using the estimated quantiles, we compute the conditional mean and variance for input images under the Gaussian model. We then compute reconstruction probability using this model as a principled approach to outlier or anomaly detection.
We also show how our approach can be used for heterogeneous thresholding of images for detecting lesions in brain images.

Categories
IPMI 2021 oral

Variational Knowledge Distillation for Disease Classification in Chest X-Rays

Authors: Tom J van Sonsbeek (University of Amsterdam)*; Xiantong Zhen (University of Amsterdam); Marcel Worring (University of Amsterdam); Ling Shao (Inception Institute of Artificial Intelligence)

Poster

Abstract: Disease classification relying solely on imaging data attracts great interest in medical image analysis. Current models could be further improved, however, by also employing Electronic Health Records (EHRs), which contain rich information on patients and findings from clinicians. It is challenging to incorporate this information into disease classification due to the high reliance on clinician input in EHRs, limiting the possibility for automated diagnosis. In this paper, we propose variational knowledge distillation (VKD), which is a new probabilistic inference framework for disease classification based on X-rays that leverages knowledge from EHRs. Specifically, we introduce a conditional latent variable model, where we infer the latent representation of the X-ray image with the variational posterior conditioning on the associated EHR text. By doing so, the model acquires the ability to extract the visual features relevant to the disease during learning and can therefore perform more accurate classification for unseen patients at inference based solely on their X-ray scans. We demonstrate the effectiveness of our method on three public benchmark datasets with paired X-ray images and EHRs. The results show that the proposed variational knowledge distillation can consistently improve the performance of medical image classification and significantly surpasses current methods.

Categories
IPMI 2021 poster

Geodesic B-Score for Improved Assessment of Knee Osteoarthritis

Authors: Felix Ambellan (Zuse Institute Berlin)*; Stefan Zachow (Zuse Institute Berlin); Christoph von Tycowicz (Zuse Institute Berlin)

Poster

Abstract: Three-dimensional medical imaging enables detailed understanding of osteoarthritis structural status. However, there remains a vast need for automatic, thus, reader-independent measures that provide reliable assessment of subject-specific clinical outcomes. To this end, we derive a consistent generalization of the recently proposed B-score to Riemannian shape spaces. We further present an algorithmic treatment yielding simple, yet efficient computations allowing for analysis of large shape populations with several thousand samples. Our intrinsic formulation exhibits improved discrimination ability over its Euclidean counterpart, which we demonstrate for predictive validity on assessing risks of total knee replacement. This result highlights the potential of the geodesic B-score to enable improved personalized assessment and stratification for interventions.

Categories
IPMI 2021 oral

Hypermorph: Amortized Hyperparameter Learning for Image Registration

Authors: Andrew Hoopes (MGH)*; Malte Hoffmann (Harvard Medical School); Bruce Fischl (Massachusetts General Hospital / Harvard Medical School); John Guttag (MIT); Adrian V Dalca (MIT)

Poster

Abstract: We present HyperMorph, a learning-based strategy for deformable image registration that removes the need to tune important registration hyperparameters during training. Classical registration methods solve an optimization problem to find a set of spatial correspondences between two images, while learning-based methods leverage a training dataset to learn a function that generates these correspondences. The quality of the results for both types of techniques depends greatly on the choice of hyperparameters. Unfortunately, hyperparameter tuning is time-consuming and typically involves training many separate models with various hyperparameter values, potentially leading to suboptimal results. To address this inefficiency, we introduce amortized hyperparameter learning for image registration, a novel strategy to learn the effects of hyperparameters on deformation fields. The proposed framework learns a hypernetwork that takes in an input hyperparameter and modulates a registration network to produce the optimal deformation field for that hyperparameter value. In effect, this strategy trains a single, rich model that enables rapid, fine-grained discovery of hyperparameter values from a continuous interval at test-time. We demonstrate that this approach can be used to optimize multiple hyperparameters considerably faster than existing search strategies, leading to a reduced computational and human burden as well as increased flexibility. We also show several important benefits, including increased robustness to initialization and the ability to rapidly identify optimal hyperparameter values specific to a registration task, dataset, or even a single anatomical region, all without retraining the HyperMorph model. Our code is publicly available at http://voxelmorph.mit.edu.

Categories
IPMI 2021 poster

Information-based Disentangled Representation Learning for Unsupervised MR Harmonization

Authors: Lianrui Zuo (Johns Hopkins University)*; Blake E Dewey (Johns Hopkins University); Aaron Carass (Johns Hopkins University, USA); Yihao Liu (Johns Hopkins University); Yufan He (Johns Hopkins University); Peter Calabresi (johns hopkins university); Jerry L Prince (Johns Hopkins University)

Poster

Abstract: Accuracy and consistency are two key factors in computer-assisted magnetic resonance (MR) image analysis. However, contrast variation from site to site caused by lack of standardization in MR acquisition impedes consistent measurements. In recent years, image harmonization approaches have been proposed to compensate for contrast variation in MR images. Current harmonization approaches either require cross-site traveling subjects for supervised training or heavily rely on site-specific harmonization models to encourage harmonization accuracy. These requirements potentially limit the application of current harmonization methods in large-scale multi-site studies. In this work, we propose an unsupervised MR harmonization framework, CALAMITI (Contrast Anatomy Learning and Analysis for MR Intensity Translation and Integration), based on information bottleneck theory. CALAMITI learns a disentangled latent space using a unified structure for multi-site harmonization without the need for traveling subjects. Our model is also able to adapt itself to harmonize MR images from a new site with fine tuning solely on images from the new site. Both qualitative and quantitative results show that the proposed method achieves superior performance compared with other unsupervised harmonization approaches.

Categories
IPMI 2021 poster

A Multi-Scale Spatial and Temporal Attention Network on Dynamic Connectivity to Localize The Eloquent Cortex in Brain Tumor Patients

Authors: Naresh Nandakumar (JHU)*; Komal Manzoor (JHU); Shruti Agarwal (JHU); Sachin Gujar (Johns Hopkins University); Jay Pillai (Johns Hopkins RAIL); Haris Sair (Johns Hopkins RAIL); Archana Venkataraman (Johns Hopkins University)

Poster

Abstract: We present a deep neural network architecture that combines multi-scale spatial attention with temporal attention to simultaneously localize the language and motor areas of the eloquent cortex from dynamic functional connectivity data. Our multi-scale spatial attention operates on graph-based features extracted from the connectivity matrices, thus honing in on the inter-regional interactions that collectively define the eloquent cortex. At the same time, our temporal attention model selects the intervals during which these interactions are most pronounced.
The final stage of our model employs multi-task learning to differentiate between the eloquent subsystems. Our training strategy enables us to handle missing eloquent class labels by freezing the weights in those branches while updating the rest of the network weights. We evaluate our method on resting-state fMRI data from one synthetic dataset and one in-house brain tumor dataset while using task fMRI activations as ground-truth labels for the eloquent cortex. Our model achieves higher localization accuracies than conventional deep learning approaches. It also produces interpretable spatial and temporal attention features which can provide further insights for presurgical planning. Thus, our model shows translational promise for improving the safety of brain tumor resections.

Categories
IPMI 2021 poster

Learning transition times in event sequences: the temporal event-based model of disease progression

Authors: Peter A Wijeratne (University College London)*; Daniel Alexander (University College London)

Poster

Abstract: Progressive diseases worsen over time and can be characterised by sequences of events that correspond to changes in observable features of disease progression. Here we connect ideas from two formerly separate methodologies — event-based and hidden Markov modelling — to derive a new generative model of disease progression: the Temporal Event-Based Model (TEBM). TEBM can uniquely infer the most likely group-level sequence and timing of events (natural history) from mixed data types. Moreover, it can infer and predict individual-level trajectories (prognosis) even when data are missing, giving it high clinical utility. Here we derive TEBM and provide an inference scheme based on the expectation maximisation algorithm. We use imaging, clinical and biofluid data from the Alzheimer’s Disease Neuroimaging Initiative to demonstrate the validity and utility of our model. First, we train TEBM to uncover a new sequence and timing of events in Alzheimer’s disease, which are inferred to occur over a period of ~17.6 years. Next, we demonstrate the utility of TEBM in predicting clinical progression, and that TEBM provides improved utility over a comparative disease progression model. Finally, we demonstrate that TEBM maintains predictive accuracy with up to 50% missing data. These results support the clinical validity of TEBM and its broader utility in real-world medical applications.

Categories
IPMI 2021 poster

Cortical Morphometry Analysis based on Worst Transportation Theory

Authors: Xianfeng GU (Stony Brook University)*; Yalin Wang (Arizona State University); Dongsheng An (Stony Brook University); Min Zhang (Brigham and Women’s Hospital, Harvard Medical School); Na Lei (Dalian University of Technology); Tong Zhao (inria); Jianfeng Wu (Arizona State University); Xiaoyin Xu (Brigham and Women’s Hospital, Harvard Medical School)

Poster

Abstract: Biomarkers play an important role in preclinical/early detection and intervention in Alzheimer’s disease (AD). However, obtaining effective biomarkers for AD is still a big challenge. In this work, we propose to use the worst transportation cost as the biomarker for the A beta brain surfaces.
The worst transportation (WT) aims to find the least economical way to transport one measure to the other, which contrasts to the optimal transportation (OT) that find the most economical way between measures. In contrast, maps find the The transportation cost of an OT map is the Wasserstein distance between the measures, which has been broadly applied for shape analysis.
To compute the WT cost, we generalize the Brenier theorem for the OT map to the WT map, and show that the WT map is the gradient of a concave function satisfying the Monge-Ampere equation. We also develop an algorithm to compute the WT map based on computational geometry. Finally, we successfully use the WT cost to study the group difference between the A beta positive AD patients and A beta negative cognitively unimpaired subjects.

Categories
IPMI 2021 poster

Discovering Spreading Pathways of Neuropathological Events in Alzheimer’s Disease Using Harmonic Wavelets

Authors: Jiazhou Chen (South China University of Technology)*; Defu Yang (University of North Carolina at Chapel Hill); Hongmin Cai (South China University of Technology); Martin Styner (University of North Carolina); Guorong Wu (UNC-CH)

Poster

Abstract: A plethora of neuroscience studies show that neurodegenerative diseases such as Alzheimer’s disease (AD) manifest network dysfunction much earlier before the onset of clinical symptoms, where neuropathological burdens often propagate across brain networks in a prion-like manner. In this context, characterizing the in-vivo spreading pathway of neuropathological events provides a new window to understand the pathophysiological mechanism of AD progression. However, little attention has been paid to the intrinsic geometry associated with the spreading pathway of neuropathological events, which indeed requires a network-specific algebra to quantify the associated propagation profile across individuals. To address this challenge, we propose a novel manifold harmonic approach to construct a set of region-adaptive harmonic wavelets, which allow us to capture diverse local network geometry in the aging population using the “swiss army knife” of the network topology. The learned common harmonic wavelets constitute the most representative spreading pathways that can be used to characterize the local propagation patterns of neuropathology on an individual basis. We have evaluated the power of our novel harmonic wavelets in identifying AD-related spreading pathways that lead to cognitive decline. Compared with other popular empirical biomarkers, the harmonic-based propagation patterns show much high sensitivity and specificity in stratifying cognitive normal (CN), early-stage mild cognitive impairment (MCI), and late-stage MCI, which indicates the great potential of being a putative biomarker in predicting the risk of developing AD at the pre-clinical stage.

Categories
IPMI 2021 oral

Nested Grassmanns for Dimensionality Reduction with Applications to Shape Analysis

Authors: Chun-Hao Yang (University of Florida)*; Baba Vemuri (University of Florida)

Poster

Abstract: Grassmann manifolds have been widely used to represent the geometry of feature spaces in a variety of problems in medical imaging and computer vision including but not limited to shape analysis, action recognition, subspace clustering and motion segmentation. For these problems, the features usually lie in a very high-dimensional Grassmann manifold and hence an appropriate dimensionality reduction technique is called for in order to curtail the computational burden. To this end, the Principal Geodesic Analysis (PGA), a nonlinear extension of the well known principal component analysis, is applicable as a general tool to many Riemannian manifolds. In this paper, we propose a novel framework for dimensionality reduction of data in Riemannian homogeneous spaces and then focus on the Grassman manifold which is an example of a homogeneous space. Our framework explicitly exploits the geometry of the homogeneous space yielding reduced dimensional nested sub-manifolds that need not be geodesic submanifolds and thus are more expressive. Specifically, we project points in a Grassmann manifold to an embedded lower dimensional Grassmann manifold. A salient feature of our method is that it leads to higher expressed variance compared to PGA which we demonstrate via synthetic and real data experiments.