VLDB 2026 Research / reviewers in the wild / expert
Wenjia Bai
dblp:20/5519
· DBLP profile ↗
71ranked-venue papers
7as first author
27since 2021 · last 2026
0000-0003-2943-7698ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Applied, interdisciplinary, general and emerging computing · 64 · 7 first-author · 22 since 2021Graphics, computer vision, multimedia, augmented reality and games · 37 · 3 first-author · 12 since 2021Artificial intelligence and machine learning · 6 · 4 since 2021
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Knowledge to Sight: Reasoning over Visual Attributes via Knowledge Decomposition for Abnormality GroundingabstractIn this work, we address the problem of grounding abnormalities in medical images, where the goal is to localize clinical findings based on textual descriptions. While generalist Vision-Language Models (VLMs) excel in natural grounding tasks, they often struggle in the medical domain due to rare, compositional, and domain-specific terms that are poorly aligned with visual patterns. Specialized medical VLMs address this challenge via large-scale domain pretraining, but at the cost of substantial annotation and computational resources. To overcome these limitations, we propose Knowledge to Sight (K2Sight), a framework that introduces structured semantic supervision by decomposing clinical concepts into interpretable visual attributes, such as shape, density, and anatomical location. These attributes are distilled from domain ontologies and encoded into concise instruction-style prompts, which guide region-text alignment during training. Unlike conventional report-level supervision, our approach explicitly bridges domain knowledge and spatial structure, enabling data-efficient training of compact models. We train compact models with 0.23B and 2B parameters using only 1.5% of the data required by state-of-the-art medical VLMs. Despite their small size and limited training data, these models achieve performance on par with or better than 7B+ medical VLMs, with up to 9.82% improvement in mAP50. Code and models: https://lijunrio.github.io/K2Sight/. Che Liu 0002, Wenjia Bai, Rossella Arcucci, Cosmin Bercea, Julia A. Schnabel |
WACV | 3 |
| 2026 | Extreme cardiac MRI analysis under respiratory motion: Results of the CMRxMotion challenge
Kang Wang 0017, Chen Qin, Zhang Shi, Haoran Wang 0009, Chen Chen 0042, Cheng Ouyang, Chengliang Dai, Yuanhan Mo, Chenchen Dai, Xutong Kuang, Ruizhe Li 0005, Xin Chen 0003, Xiuzheng Yue, Song Tian, Alejandro Mora-Rubio, Kumaradevan Punithakumar, Shizhan Gong, Qi Dou 0001, Sina Amirrajab, Yasmina Alkhalil, Cian M. Scannell, Lexiaozi Fan, Huili Yang, Xiaowu Sun, Rob J. van der Geest, Tewodros Weldebirhan Arega, Fabrice Mériaudeau, Caner Ozer, Amin Ranem, John Kalkhof, Ilkay Öksüz, Anirban Mukhopadhyay 0003, Abdul Qayyum 0002, Moona Mazher, Steven A. Niederer, Carles García-Cabrera, Eric Arazo Sanchez, Michal K. Grzeszczyk, Szymon Plotka, Wanqin Ma, Xiaomeng Li 0001, Rongjun Ge, Yongqing Kou, Xinrong Chen, He Wang 0016, Chengyan Wang, Wenjia Bai, Shuo Wang 0011 |
Medical Image Anal. | 48 |
| 2026 | Toward Modality- and Sampling-Universal Learning Strategies for Accelerating Cardiovascular Imaging: Summary of the CMRxRecon2024 ChallengeabstractCardiovascular health is vital to human well-being, and cardiac magnetic resonance (CMR) imaging is considered the clinical reference standard for diagnosing cardiovascular disease. However, its adoption is hindered by long scan times, complex contrasts, and inconsistent quality. While deep learning methods perform well on specific CMR imaging sequences, they often fail to generalize across modalities and sampling schemes. The lack of benchmarks for high-quality, fast CMR image reconstruction further limits technology comparison and adoption. The CMRxRecon2024 challenge, attracting over 200 teams from 18 countries, addressed these issues with two tasks: generalization to unseen modalities and robustness to diverse undersampling patterns. We introduced the largest public multi-modality CMR raw dataset, an open benchmarking platform, and shared code. Analysis of the best-performing solutions revealed that prompt-based adaptation and enhanced physics-driven consistency enabled strong cross-scenario performance. These findings establish principles for generalizable reconstruction models and advance clinically translatable AI in cardiovascular imaging. Fanwen Wang, Zi Wang 0005, Yan Li 0064, Chen Qin, Shuo Wang 0011, Kunyuan Guo, Mengting Sun, Mingkai Huang, Michael Tänzer, Qirong Li, Yinzhe Wu 0001, Haosen Zhang, Kian Anvari Hamedani, Yuntong Lyu, Longyu Sun, Tianxing He, Lizhen Lan, Qiong Yao, Bingyu Xin, Dimitris N. Metaxas, Narges Razizadeh, Shahabedin Nabavi, George Yiasemis, Jonas Teuwen, Daniel B. Ennis, Zhihao Xue, Ruru Xu, Ilkay Öksüz, Donghang Lyu, Yanxin Huang, Xinrui Guo, Ruqian Hao, Jaykumar H. Patel, Guanke Cai, Binghua Chen, Sha Hua, Zhensen Chen, Qi Dou 0001, Xiahai Zhuang, Wenjia Bai, Harry Qin, He Wang 0016, Claudia Prieto, Michael Markl 0001, Alistair A. Young, Hao Li 0082, Xihong Hu, Lianming Wu, Xiaobo Qu 0001, Guang Yang 0006, Chengyan Wang |
IEEE Trans. Medical Imaging | 52 |
| 2025 | CardiacFlow: 3D+t Four-Chamber Cardiac Shape Completion and Generation via Flow Matching
Qiang Ma 0004, Qingjie Meng, Mengyun Qiao, Paul M. Matthews, Declan P. O'Regan, Wenjia Bai |
MICCAI (2) | 6 |
| 2025 | Mesh4D: A Motion-Aware Multi-view Variational Autoencoder for 3D+t Mesh Reconstruction
Mengyun Qiao, Qiang Ma 0004, Liu Li 0001, Bernhard Kainz, Declan P. O'Regan, Paul M. Matthews, Steven A. Niederer, Wenjia Bai |
MICCAI (16) | 10 |
| 2025 | Multi-agent Reasoning for Cardiovascular Imaging Phenotype Analysis
Mengyun Qiao, Chengqi Zang, Steven A. Niederer, Paul M. Matthews, Wenjia Bai, Bernhard Kainz |
MICCAI (1) | 6 |
| 2025 | NOVA: A Benchmark for Rare Anomaly Localization and Clinical Reasoning in Brain MRIabstractIn many real-world applications, deployed models encounter inputs that differ from the data seen during training. Open-world recognition ensures that such systems remain robust as ever-emerging, previously _unknown_ categories appear and must be addressed without retraining.Foundation and vision-language models are pre-trained on large and diverse datasets with the expectation of broad generalization across domains, including medical imaging.However, benchmarking these models on test sets with only a few common outlier types silently collapses the evaluation back to a closed-set problem, masking failures on rare or truly novel conditions encountered in clinical use.We therefore present NOVA, a challenging, real-life _evaluation-only_ benchmark of $\sim$900 brain MRI scans that span 281 rare pathologies and heterogeneous acquisition protocols. Each case includes rich clinical narratives and double-blinded expert bounding-box annotations. Together, these enable joint assessment of anomaly localisation, visual captioning, and diagnostic reasoning. Because NOVA is never used for training, it serves as an _extreme_ stress-test of out-of-distribution generalisation: models must bridge a distribution gap both in sample appearance and in semantic space. Baseline results with leading vision-language models (GPT-4o, Gemini 2.0 Flash, and Qwen2.5-VL-72B) reveal substantial performance drops, with approximately a 65\% gap in localisation compared to natural-image benchmarks and 40\% and 20\% gaps in captioning and reasoning, respectively, compared to resident radiologists. Therefore, NOVA establishes a testbed for advancing models that can detect, localize, and reason about truly unknown anomalies. Cosmin Bercea, Philipp Raffler, Evamaria O. Riedel, Lena Schmitzer, Angela Kurz, Felix Bitzer, Paula Roßmüller, Julian Canisius, Mirjam L. Beyrle, Che Liu 0002, Wenjia Bai, Bernhard Kainz, Julia A. Schnabel, Benedikt Wiestler |
NeurIPS | 12 |
| 2025 | The state-of-the-art in cardiac MRI reconstruction: Results of the CMRxRecon challenge in MICCAI 2023
Chen Qin, Shuo Wang 0011, Fanwen Wang, Yan Li 0064, Zi Wang 0005, Kunyuan Guo, Ouyang Cheng, Michael Tänzer, Longyu Sun, Mengting Sun, Zhang Shi, Sha Hua, Hao Li 0082, Zhensen Chen, Bingyu Xin, Dimitris N. Metaxas, George Yiasemis, Jonas Teuwen, Weitian Chen, Yidong Zhao, Yanwei Pang, Artem Razumov, Dmitry V. Dylov, Quan Dou, Yuyang Xue, Yuning Du, Julia Dietlmeier, Carles García-Cabrera, Ziad Al-Haj Hemidi, Nora Vogt, Ying-Hua Chu, Weibo Chen, Wenjia Bai, Xiahai Zhuang, Harry Qin, Lianming Wu, Guang Yang 0006, Xiaobo Qu 0001, He Wang 0016, Chengyan Wang |
Medical Image Anal. | 43 |
| 2025 | IMITATE: Clinical Prior Guided Hierarchical Vision-Language Pre-TrainingabstractIn medical Vision-Language Pre-training (VLP), significant work focuses on extracting text and image features from clinical reports and medical images. Yet, existing methods may overlooked the potential of the natural hierarchical structure in clinical reports, typically divided into 'findings' for description and 'impressions' for conclusions. Current VLP approaches tend to oversimplify these reports into a single entity or fragmented tokens, ignoring this structured format. In this work, we propose a novel clinical prior guided VLP framework named IMITATE to learn the structure information from medical reports with hierarchical vision-language alignment. The framework derives multi-level visual features from the chest X-ray (CXR) images and separately aligns these features with the descriptive and the conclusive text encoded in the hierarchical medical report. Furthermore, a new clinical-informed contrastive loss is introduced for cross-modal learning, which accounts for clinical prior knowledge in formulating sample correlations in contrastive learning. The proposed model, IMITATE, outperforms baseline VLP methods across six different datasets, spanning five medical imaging downstream tasks. Experimental results show benefits of using hierarchical structures in medical reports for VLP. Code: https://github.com/cheliu-computation/IMITATE-TMI2024. Che Liu 0002, Sibo Cheng, Miaojing Shi, Anand Shah, Wenjia Bai, Rossella Arcucci |
IEEE Trans. Medical Imaging | 5 |
| 2025 | A Foundation Model for Lesion Segmentation on Brain MRI With Mixture of Modality ExpertsabstractBrain lesion segmentation is crucial for neurological disease research and diagnosis. As different types of lesions exhibit distinct characteristics on different imaging modalities, segmentation methods are typically developed in a task-specific manner, where each segmentation model is tailored to a specific lesion type and modality. However, the use of task-specific models requires predetermination of the lesion type and imaging modality, which complicates their deployment in real-world scenarios. In this work, we propose a universal foundation model for brain lesion segmentation on magnetic resonance imaging (MRI), which can automatically segment different types of brain lesions given input of various MRI modalities. We develop a novel Mixture of Modality Experts (MoME) framework with multiple expert networks attending to different imaging modalities. A hierarchical gating network is proposed to combine the expert predictions and foster expertise collaboration. Moreover, to avoid the degeneration of each expert network, we introduce a curriculum learning strategy during training to preserve the specialisation of each expert. In addition to MoME, to handle the combination of multiple input modalities, we propose MoME+, which uses a soft dispatch network for input modality routing. We evaluated the proposed method on nine brain lesion datasets, encompassing five imaging modalities and eight lesion types. The results show that our model outperforms state-of-the-art universal models for brain lesion segmentation and achieves promising generalisation performance onto unseen datasets. Xinru Zhang 0001, Ni Ou, Berke Doga Basaran, Marco Visentin, Mengyun Qiao, Renyang Gu, Paul M. Matthews, Yaou Liu, Chuyang Ye, Wenjia Bai |
IEEE Trans. Medical Imaging | 10 |
| 2024 | TIP: Tabular-Image Pre-training for Multimodal Classification with Incomplete Data
Siyi Du, Shaoming Zheng, Yinsong Wang, Wenjia Bai, Declan P. O'Regan, Chen Qin |
ECCV (15) | 4 |
| 2024 | Zero-Shot ECG Classification with Multimodal Learning and Test-time Clinical Knowledge EnhancementabstractElectrocardiograms (ECGs) are non-invasive diagnostic tools crucial for detecting cardiac arrhythmic diseases in clinical practice. While ECG Self-supervised Learning (eSSL) methods show promise in representation learning from unannotated ECG data, they often overlook the clinical knowledge that can be found in reports. This oversight and the requirement for annotated samples for downstream tasks limit eSSL’s versatility. In this work, we address these issues with the Multimodal ECG Representation Learning (MERL) framework. Through multimodal learning on ECG records and associated reports, MERL is capable of performing zero-shot ECG classification with text prompts, eliminating the need for training data in downstream tasks. At test time, we propose the Clinical Knowledge Enhanced Prompt Engineering (CKEPE) approach, which uses Large Language Models (LLMs) to exploit external expert-verified clinical knowledge databases, generating more descriptive prompts and reducing hallucinations in LLM-generated content to boost zero-shot classification. Based on MERL, we perform the first benchmark across six public ECG datasets, showing the superior performance of MERL compared against eSSL methods. Notably, MERL achieves an average AUC score of 75.2% in zero-shot classification (without training data), 3.2% higher than linear probed eSSL methods with 10% annotated training data, averaged across all six datasets. Che Liu 0002, Zhongwei Wan, Cheng Ouyang, Anand Shah, Wenjia Bai, Rossella Arcucci |
ICML | 5 |
| 2024 | A Foundation Model for Brain Lesion Segmentation with Mixture of Modality Experts
Xinru Zhang 0001, Ni Ou, Berke Doga Basaran, Marco Visentin, Mengyun Qiao, Renyang Gu, Cheng Ouyang, Yaou Liu, Paul M. Matthews, Chuyang Ye, Wenjia Bai |
MICCAI (12) | 11 |
| 2024 | G2D: From Global to Dense Radiography Representation Learning via Vision-Language Pre-trainingabstractMedical imaging tasks require an understanding of subtle and localized visual features due to the inherently detailed and area-specific nature of pathological patterns, which are crucial for clinical diagnosis. Although recent advances in medical vision-language pre-training (VLP) enable models to learn clinically relevant visual features by leveraging both medical images and their associated radiology reports, current medical VLP methods primarily focus on aligning images with entire reports. This focus hinders the learning of dense (pixel-level) visual features and is suboptimal for dense prediction tasks (e.g., medical image segmentation).
To address this challenge, we propose a novel medical VLP framework, named **Global to Dense level representation learning (G2D)**, which aims to learn global and dense visual features simultaneously using only image-text pairs without extra annotations. In particular, G2D designs a **Pseudo Segmentation (PS)** task, which enables the model to learn dense visual features during VLP. Notably, generating PS masks can be performed on the fly during VLP, which does not incur extra trainable parameters. With this simple yet effective idea, G2D achieves superior performance across 5 medical imaging tasks and 25 diseases. Particularly, in the segmentation task which requires dense visual features, **G2D surpasses existing models even with just 1% of the training data for finetuning, compared to 100% used by other models**. The code can be found in https://github.com/cheliu-computation/G2D-NeurIPS24/tree/main. Che Liu 0002, Cheng Ouyang, Sibo Cheng, Anand Shah, Wenjia Bai, Rossella Arcucci |
NeurIPS | 5 |
| 2024 | DeepMesh: Mesh-Based Cardiac Motion Tracking Using Deep Learningabstract3D motion estimation from cine cardiac magnetic resonance (CMR) images is important for the assessment of cardiac function and the diagnosis of cardiovascular diseases. Current state-of-the art methods focus on estimating dense pixel-/voxel-wise motion fields in image space, which ignores the fact that motion estimation is only relevant and useful within the anatomical objects of interest, e.g., the heart. In this work, we model the heart as a 3D mesh consisting of epi- and endocardial surfaces. We propose a novel learning framework, DeepMesh, which propagates a template heart mesh to a subject space and estimates the 3D motion of the heart mesh from CMR images for individual subjects. In DeepMesh, the heart mesh of the end-diastolic frame of an individual subject is first reconstructed from the template mesh. Mesh-based 3D motion fields with respect to the end-diastolic frame are then estimated from 2D short- and long-axis CMR images. By developing a differentiable mesh-to-image rasterizer, DeepMesh is able to leverage 2D shape information from multiple anatomical views for 3D mesh reconstruction and mesh motion estimation. The proposed method estimates vertex-wise displacement and thus maintains vertex correspondences between time frames, which is important for the quantitative assessment of cardiac function across different subjects and populations. We evaluate DeepMesh on CMR images acquired from the UK Biobank. We focus on 3D motion estimation of the left ventricle in this work. Experimental results show that the proposed method quantitatively and qualitatively outperforms other image-based and mesh-based cardiac motion tracking methods. Qingjie Meng, Wenjia Bai, Declan P. O'Regan, Daniel Rueckert |
IEEE Trans. Medical Imaging | 2 |
| 2024 | CHeart: A Conditional Spatio-Temporal Generative Model for Cardiac AnatomyabstractTwo key questions in cardiac image analysis are to assess the anatomy and motion of the heart from images; and to understand how they are associated with non-imaging clinical factors such as gender, age and diseases. While the first question can often be addressed by image segmentation and motion tracking algorithms, our capability to model and answer the second question is still limited. In this work, we propose a novel conditional generative model to describe the 4D spatio-temporal anatomy of the heart and its interaction with non-imaging clinical factors. The clinical factors are integrated as the conditions of the generative modelling, which allows us to investigate how these factors influence the cardiac anatomy. We evaluate the model performance in mainly two tasks, anatomical sequence completion and sequence generation. The model achieves high performance in anatomical sequence completion, comparable to or outperforming other state-of-the-art generative models. In terms of sequence generation, given clinical conditions, the model can generate realistic synthetic 4D sequential anatomies that share similar distributions with the real data. The code and the trained generative model are available at https://github.com/MengyunQ/CHeart. Mengyun Qiao, Shuo Wang 0011, Huaqi Qiu, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Daniel Rueckert, Wenjia Bai |
IEEE Trans. Medical Imaging | 7 |
| 2023 | M-FLAG: Medical Vision-Language Pre-training with Frozen Language Models and Latent Space Geometry Optimization
Che Liu 0002, Sibo Cheng, Chen Chen 0042, Mengyun Qiao, Anand Shah, Wenjia Bai, Rossella Arcucci |
MICCAI (1) | 7 |
| 2023 | MoCoSR: Respiratory Motion Correction and Super-Resolution for 3D Abdominal MRI
Berke Doga Basaran, Qingjie Meng, Matthew Baugh, Jonathan K. Stelter, Phillip Lung, Uday Patel, Wenjia Bai, Dimitrios C. Karampinos, Bernhard Kainz |
MICCAI (10) | 8 |
| 2023 | Generative myocardial motion tracking via latent space exploration with biomechanics-informed priorabstractMyocardial motion and deformation are rich descriptors that characterize cardiac function. Image registration, as the most commonly used technique for myocardial motion tracking, is an ill-posed inverse problem which often requires prior assumptions on the solution space. In contrast to most existing approaches which impose explicit generic regularization such as smoothness, in this work we propose a novel method that can implicitly learn an application-specific biomechanics-informed prior and embed it into a neural network-parameterized transformation model. Particularly, the proposed method leverages a variational autoencoder-based generative model to learn a manifold for biomechanically plausible deformations. The motion tracking then can be performed via traversing the learnt manifold to search for the optimal transformations while considering the sequence information. The proposed method is validated on three public cardiac cine MRI datasets with comprehensive evaluations. The results demonstrate that the proposed method can outperform other approaches, yielding higher motion tracking accuracy with reasonable volume preservation and better generalizability to varying data distributions. It also enables better estimates of myocardial strains, which indicates the potential of the method in characterizing spatiotemporal signatures for understanding cardiovascular diseases. Chen Qin, Shuo Wang 0011, Chen Chen 0042, Wenjia Bai, Daniel Rueckert |
Medical Image Anal. | 4 |
| 2023 | Causality-Inspired Single-Source Domain Generalization for Medical Image SegmentationabstractDeep learning models usually suffer from the domain shift issue, where models trained on one source domain do not generalize well to other unseen domains. In this work, we investigate the single-source domain generalization problem: training a deep network that is robust to unseen domains, under the condition that training data are only available from one source domain, which is common in medical imaging applications. We tackle this problem in the context of cross-domain medical image segmentation. In this scenario, domain shifts are mainly caused by different acquisition processes. We propose a simple causality-inspired data augmentation approach to expose a segmentation model to synthesized domain-shifted training examples. Specifically, 1) to make the deep model robust to discrepancies in image intensities and textures, we employ a family of randomly-weighted shallow networks. They augment training images using diverse appearance transformations. 2) Further we show that spurious correlations among objects in an image are detrimental to domain robustness. These correlations might be taken by the network as domain-specific clues for making predictions, and they may break on unseen domains. We remove these spurious correlations via causal intervention. This is achieved by resampling the appearances of potentially correlated objects independently. The proposed approach is validated on three cross-domain segmentation scenarios: cross-modality (CT-MRI) abdominal image segmentation, cross-sequence (bSSFP-LGE) cardiac MRI segmentation, and cross-site prostate MRI segmentation. The proposed approach yields consistent performance gains compared with competitive methods when tested on unseen domains. Cheng Ouyang, Chen Chen 0042, Surui Li, Zeju Li, Chen Qin, Wenjia Bai, Daniel Rueckert |
IEEE Trans. Medical Imaging | 6 |
| 2022 | MaxStyle: Adversarial Style Composition for Robust Medical Image Segmentation
Chen Chen 0042, Zeju Li, Cheng Ouyang, Matthew Sinclair, Wenjia Bai, Daniel Rueckert |
MICCAI (5) | 5 |
| 2022 | Mesh-Based 3D Motion Tracking in Cardiac MRI Using Deep Learning
Qingjie Meng, Wenjia Bai, Declan P. O'Regan, Daniel Rueckert |
MICCAI (6) | 2 |
| 2022 | Enhancing MR image segmentation with realistic adversarial data augmentationabstractThe success of neural networks on medical image segmentation tasks typically relies on large labeled datasets for model training. However, acquiring and manually labeling a large medical image set is resource-intensive, expensive, and sometimes impractical due to data sharing and privacy issues. To address this challenge, we propose AdvChain, a generic adversarial data augmentation framework, aiming at improving both the diversity and effectiveness of training data for medical image segmentation tasks. AdvChain augments data with dynamic data augmentation, generating randomly chained photo-metric and geometric transformations to resemble realistic yet challenging imaging variations to expand training data. By jointly optimizing the data augmentation model and a segmentation network during training, challenging examples are generated to enhance network generalizability for the downstream task. The proposed adversarial data augmentation does not rely on generative networks and can be used as a plug-in module in general segmentation networks. It is computationally efficient and applicable for both low-shot supervised and semi-supervised learning. We analyze and evaluate the method on two MR image segmentation tasks: cardiac segmentation and prostate segmentation with limited labeled data. Results show that the proposed approach can alleviate the need for labeled data while improving model generalization ability, indicating its practical value in medical imaging applications. Chen Chen 0042, Chen Qin, Cheng Ouyang, Zeju Li, Shuo Wang 0011, Huaqi Qiu, Liang Chen 0018, Giacomo Tarroni, Wenjia Bai, Daniel Rueckert |
Medical Image Anal. | 9 |
| 2022 | Suggestive annotation of brain MR images with gradient-guided sampling
Chengliang Dai, Shuo Wang 0011, Yuanhan Mo, Elsa D. Angelini, Yike Guo, Wenjia Bai |
Medical Image Anal. | 6 |
| 2022 | MulViMotion: Shape-Aware 3D Myocardial Motion Tracking From Multi-View Cardiac MRIabstractRecovering the 3D motion of the heart from cine cardiac magnetic resonance (CMR) imaging enables the assessment of regional myocardial function and is important for understanding and analyzing cardiovascular disease. However, 3D cardiac motion estimation is challenging because the acquired cine CMR images are usually 2D slices which limit the accurate estimation of through-plane motion. To address this problem, we propose a novel multi-view motion estimation network (MulViMotion), which integrates 2D cine CMR images acquired in short-axis and long-axis planes to learn a consistent 3D motion field of the heart. In the proposed method, a hybrid 2D/3D network is built to generate dense 3D motion fields by learning fused representations from multi-view images. To ensure that the motion estimation is consistent in 3D, a shape regularization module is introduced during training, where shape information from multi-view images is exploited to provide weak supervision to 3D motion estimation. We extensively evaluate the proposed method on 2D cine CMR images from 580 subjects of the UK Biobank study for 3D motion tracking of the left ventricular myocardium. Experimental results show that the proposed method quantitatively and qualitatively outperforms competing methods. Qingjie Meng, Chen Qin, Wenjia Bai, Tianrui Liu 0001, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Daniel Rueckert |
IEEE Trans. Medical Imaging | 3 |
| 2021 | Cooperative Training and Latent Space Data Augmentation for Robust Medical Image Segmentation
Chen Chen 0042, Kerstin Hammernik, Cheng Ouyang, Chen Qin, Wenjia Bai, Daniel Rueckert |
MICCAI (3) | 5 |
| 2021 | Joint Motion Correction and Super Resolution for Cardiac Segmentation via Latent Optimisation
Shuo Wang 0011, Chen Qin, Nicolò Savioli, Chen Chen 0042, Declan P. O'Regan, Stuart A. Cook, Yike Guo, Daniel Rueckert, Wenjia Bai |
MICCAI (3) | 9 |
| 2020 | Realistic Adversarial Data Augmentation for MR Image Segmentation
Chen Chen 0042, Chen Qin, Huaqi Qiu, Cheng Ouyang, Shuo Wang 0011, Liang Chen 0018, Giacomo Tarroni, Wenjia Bai, Daniel Rueckert |
MICCAI (1) | 8 |
| 2020 | Suggestive Annotation of Brain Tumour Images with Gradient-Guided Sampling
Chengliang Dai, Shuo Wang 0011, Yuanhan Mo, Kaichen Zhou, Elsa D. Angelini, Yike Guo, Wenjia Bai |
MICCAI (4) | 7 |
| 2020 | Biomechanics-Informed Neural Networks for Myocardial Motion Tracking in MRI
Chen Qin, Shuo Wang 0011, Chen Chen 0042, Huaqi Qiu, Wenjia Bai, Daniel Rueckert |
MICCAI (3) | 5 |
| 2020 | Deep Generative Model-Based Quality Control for Cardiac MRI Segmentation
Shuo Wang 0011, Giacomo Tarroni, Chen Qin, Yuanhan Mo, Chengliang Dai, Chen Chen 0042, Ben Glocker, Yike Guo, Daniel Rueckert, Wenjia Bai |
MICCAI (4) | 10 |
| 2020 | Explainable Anatomical Shape Analysis Through Deep Hierarchical Generative ModelsabstractQuantification of anatomical shape changes currently relies on scalar global indexes which are largely insensitive to regional or asymmetric modifications. Accurate assessment of pathology-driven anatomical remodeling is a crucial step for the diagnosis and treatment of many conditions. Deep learning approaches have recently achieved wide success in the analysis of medical images, but they lack interpretability in the feature extraction and decision processes. In this work, we propose a new interpretable deep learning model for shape analysis. In particular, we exploit deep generative networks to model a population of anatomical segmentations through a hierarchy of conditional latent variables. At the highest level of this hierarchy, a two-dimensional latent space is simultaneously optimised to discriminate distinct clinical conditions, enabling the direct visualisation of the classification space. Moreover, the anatomical variability encoded by this discriminative latent space can be visualised in the segmentation space thanks to the generative properties of the model, making the classification task transparent. This approach yielded high accuracy in the categorisation of healthy and remodelled left ventricles when tested on unseen segmentations from our own multi-centre dataset as well as in an external validation set, and on hippocampi from healthy controls and patients with Alzheimer's disease when tested on ADNI data. More importantly, it enabled the visualisation in three-dimensions of both global and regional anatomical features which better discriminate between the conditions under exam. The proposed approach scales effectively to large populations, facilitating high-throughput analysis of normal anatomy and pathology in large-scale studies of volumetric imaging. Carlo Biffi, Juan J. Cerrolaza, Giacomo Tarroni, Wenjia Bai, Antonio M. Simoes Monteiro de Marvao, Ozan Oktay, Christian Ledig, Loïc Le Folgoc, Konstantinos Kamnitsas, Georgia Doumou, Jinming Duan 0001, Sanjay K. Prasad, Stuart A. Cook, Declan P. O'Regan, Daniel Rueckert |
IEEE Trans. Medical Imaging | 4 |
| 2019 | VS-Net: Variable Splitting Network for Accelerated Parallel MRI Reconstruction
Jinming Duan 0001, Jo Schlemper, Chen Qin, Cheng Ouyang, Wenjia Bai, Carlo Biffi, Ghalib Bello, Ben Statton, Declan P. O'Regan, Daniel Rueckert |
MICCAI (4) | 5 |
| 2019 | Learning Shape Priors for Robust Cardiac MR Segmentation from Multi-view Images
Chen Chen 0042, Carlo Biffi, Giacomo Tarroni, Steffen E. Petersen, Wenjia Bai, Daniel Rueckert |
MICCAI (2) | 5 |
| 2019 | Self-Supervised Learning for Cardiac MR Image Segmentation by Anatomical Position Prediction
Wenjia Bai, Chen Chen 0042, Giacomo Tarroni, Jinming Duan 0001, Florian Guitton, Steffen E. Petersen, Yike Guo, Paul M. Matthews, Daniel Rueckert |
MICCAI (2) | 1 |
| 2019 | Scar shape analysis and simulated electrical instabilities in a non-ischemic dilated cardiomyopathy patient cohortabstractThis paper presents a morphological analysis of fibrotic scarring in non-ischemic dilated cardiomyopathy, and its relationship to electrical instabilities which underlie reentrant arrhythmias.Two dimensional electrophysiological simulation models were constructed from a set of 699 late gadolinium enhanced cardiac magnetic resonance images originating from 157 patients.Areas of late gadolinium enhancement (LGE) in each image were assigned one of 10 possible microstructures, which modelled the details of fibrotic scarring an order of magnitude below the MRI scan resolution.A simulated programmed electrical stimulation protocol tested each model for the possibility of generating either a transmural block or a transmural reentry.The outcomes of the simulations were compared against morphological LGE features extracted from the images.Models which blocked or reentered, grouped by microstructure, were significantly different from one another in myocardial-LGE interface length, number of components and entropy, but not in relative area and transmurality.With an unknown microstructure, transmurality alone was the best predictor of block, whereas a combination of interface length, transmurality and number of components was the best predictor of reentry in linear discriminant analysis. Author summaryNon-ischemic dilated cardiomyopathy is a disease in which the lower left chamber of the heart is abnormally large.The cause of the disease can be anything that is not a loss of blood supply to the heart.Many patients with non-ischemic dilated cardiomyopathy have scars in their hearts which can be detected with magnetic resonance imaging.These scars are thought to disrupt the flow of electricity through the heart and cause deadly rhythm Gabriel Balaban, Brian Halliday, Wenjia Bai, Bradley Porter, Carlotta Malvuccio, Pablo Lamata, C. Aldo Rinaldi, Gernot Plank, Daniel Rueckert, Sanjay K. Prasad, Martin J. Bishop 0001 |
PLoS Comput. Biol. | 3 |
| 2019 | Automatic 3D Bi-Ventricular Segmentation of Cardiac Images by a Shape-Refined Multi- Task Deep Learning ApproachabstractDeep learning approaches have achieved state-of-the-art performance in cardiac magnetic resonance (CMR) image segmentation. However, most approaches have focused on learning image intensity features for segmentation, whereas the incorporation of anatomical shape priors has received less attention. In this paper, we combine a multi-task deep learning approach with atlas propagation to develop a shape-refined bi-ventricular segmentation pipeline for short-axis CMR volumetric images. The pipeline first employs a fully convolutional network (FCN) that learns segmentation and landmark localization tasks simultaneously. The architecture of the proposed FCN uses a 2.5D representation, thus combining the computational advantage of 2D FCNs networks and the capability of addressing 3D spatial consistency without compromising segmentation accuracy. Moreover, a refinement step is designed to explicitly impose shape prior knowledge and improve segmentation quality. This step is effective for overcoming image artifacts (e.g., due to different breath-hold positions and large slice thickness), which preclude the creation of anatomically meaningful 3D cardiac shapes. The pipeline is fully automated, due to network's ability to infer landmarks, which are then used downstream in the pipeline to initialize atlas propagation. We validate the pipeline on 1831 healthy subjects and 649 subjects with pulmonary hypertension. Extensive numerical experiments on the two datasets demonstrate that our proposed method is robust and capable of producing accurate, high-resolution, and anatomically smooth bi-ventricular 3D models, despite the presence of artifacts in input CMR volumes. Jinming Duan 0001, Ghalib Bello, Jo Schlemper, Wenjia Bai, Timothy Dawes, Carlo Biffi, Antonio M. Simoes Monteiro de Marvao, Georgia Doumou, Declan P. O'Regan, Daniel Rueckert |
IEEE Trans. Medical Imaging | 4 |
| 2019 | Learning-Based Quality Control for Cardiac MR ImagesabstractThe effectiveness of a cardiovascular magnetic resonance (CMR) scan depends on the ability of the operator to correctly tune the acquisition parameters to the subject being scanned and on the potential occurrence of imaging artifacts, such as cardiac and respiratory motion. In the clinical practice, a quality control step is performed by visual assessment of the acquired images; however, this procedure is strongly operator-dependent, cumbersome, and sometimes incompatible with the time constraints in clinical settings and large-scale studies. We propose a fast, fully automated, and learning-based quality control pipeline for CMR images, specifically for short-axis image stacks. Our pipeline performs three important quality checks: 1) heart coverage estimation; 2) inter-slice motion detection; 3) image contrast estimation in the cardiac region. The pipeline uses a hybrid decision forest method-integrating both regression and structured classification models-to extract landmarks and probabilistic segmentation maps from both long- and short-axis images as a basis to perform the quality checks. The technique was tested on up to 3000 cases from the UK Biobank and on 100 cases from the UK Digital Heart Project and validated against manual annotations and visual inspections performed by expert interpreters. The results show the capability of the proposed pipeline to correctly detect incomplete or corrupted scans (e.g., on UK Biobank, sensitivity and specificity, respectively, 88% and 99% for heart coverage estimation and 85% and 95% for motion detection), allowing their exclusion from the analyzed dataset or the triggering of a new acquisition. Giacomo Tarroni, Ozan Oktay, Wenjia Bai, Andreas Schuh, Hideaki Suzuki, Jonathan Passerat-Palmbach, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Stuart A. Cook, Ben Glocker, Paul M. Matthews, Daniel Rueckert |
IEEE Trans. Medical Imaging | 3 |
| 2018 | Automatic View Planning with Multi-scale Deep Reinforcement Learning Agents
Amir Alansary, Loïc Le Folgoc, Ghislain Vaillant, Ozan Oktay, Wenjia Bai, Jonathan Passerat-Palmbach, Ricardo Guerrero, Konstantinos Kamnitsas, Benjamin Hou, Steven McDonagh 0001, Ben Glocker, Bernhard Kainz, Daniel Rueckert |
MICCAI (1) | 6 |
| 2018 | Recurrent Neural Networks for Aortic Image Sequence Segmentation with Sparse Annotations
Wenjia Bai, Hideaki Suzuki, Chen Qin, Giacomo Tarroni, Ozan Oktay, Paul M. Matthews, Daniel Rueckert |
MICCAI (4) | 1 |
| 2018 | Learning Interpretable Anatomical Features Through Deep Generative Models: Application to Cardiac Remodeling
Carlo Biffi, Ozan Oktay, Giacomo Tarroni, Wenjia Bai, Antonio M. Simoes Monteiro de Marvao, Georgia Doumou, Martin Rajchl, Reem Bedair, Sanjay K. Prasad, Stuart A. Cook, Declan P. O'Regan, Daniel Rueckert |
MICCAI (2) | 4 |
| 2018 | Deep Nested Level Sets: Fully Automated Segmentation of Cardiac MR Images in Patients with Pulmonary Hypertension
Jinming Duan 0001, Jo Schlemper, Wenjia Bai, Timothy Dawes, Ghalib Bello, Georgia Doumou, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Daniel Rueckert |
MICCAI (4) | 3 |
| 2018 | Joint Learning of Motion Estimation and Segmentation for Cardiac MR Image Sequences
Chen Qin, Wenjia Bai, Jo Schlemper, Steffen E. Petersen, Stefan K. Piechnik, Stefan Neubauer, Daniel Rueckert |
MICCAI (2) | 2 |
| 2018 | Real-Time Prediction of Segmentation Quality
Robert Robinson, Ozan Oktay, Wenjia Bai, Vanya V. Valindria, Mihir Sanghvi, Nay Aung, José Miguel Paiva, Filip Zemrak, Kenneth Fung, Elena Lukaschuk, Aaron M. Lee, Valentina Carapella, Bernhard Kainz, Stefan K. Piechnik, Stefan Neubauer, Steffen E. Petersen, Chris Page, Daniel Rueckert, Ben Glocker |
MICCAI (4) | 3 |
| 2018 | Cardiac MR Segmentation from Undersampled k-space Using Deep Latent Representation Learning
Jo Schlemper, Ozan Oktay, Wenjia Bai, Daniel C. Castro, Jinming Duan 0001, Chen Qin, Joseph V. Hajnal, Daniel Rueckert |
MICCAI (1) | 3 |
| 2018 | A Comprehensive Approach for Learning-Based Fully-Automated Inter-slice Motion Correction for Short-Axis Cine Cardiac MR Image Stacks
Giacomo Tarroni, Ozan Oktay, Matthew Sinclair, Wenjia Bai, Andreas Schuh, Hideaki Suzuki, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Stuart A. Cook, Daniel Rueckert |
MICCAI (1) | 4 |
| 2018 | Three-dimensional cardiovascular imaging-genetics: a mass univariate frameworkabstractMotivation: Left ventricular (LV) hypertrophy is a strong predictor of cardiovascular outcomes, but its genetic regulation remains largely unexplained. Conventional phenotyping relies on manual calculation of LV mass and wall thickness, but advanced cardiac image analysis presents an opportunity for high-throughput mapping of genotype-phenotype associations in three dimensions (3D). Results: High-resolution cardiac magnetic resonance images were automatically segmented in 1124 healthy volunteers to create a 3D shape model of the heart. Mass univariate regression was used to plot a 3D effect-size map for the association between wall thickness and a set of predictors at each vertex in the mesh. The vertices where a significant effect exists were determined by applying threshold-free cluster enhancement to boost areas of signal with spatial contiguity. Experiments on simulated phenotypic signals and SNP replication show that this approach offers a substantial gain in statistical power for cardiac genotype-phenotype associations while providing good control of the false discovery rate. This framework models the effects of genetic variation throughout the heart and can be automatically applied to large population cohorts. Availability and implementation: The proposed approach has been coded in an R package freely available at https://doi.org/10.5281/zenodo.834610 together with the clinical data used in this work. Contact: [email protected]. Supplementary information: Supplementary data are available at Bioinformatics online. Carlo Biffi, Antonio M. Simoes Monteiro de Marvao, Mark I Attard, Timothy Dawes, Nicola Whiffin, Wenjia Bai, Wenzhe Shi, Catherine Francis, Hannah Meyer, Rachel J. Buchan, Stuart A. Cook, Daniel Rueckert, Declan P. O'Regan |
Bioinform. | 6 |
| 2018 | Myocardial strain computed at multiple spatial scales from tagged magnetic resonance imaging: Estimating cardiac biomarkers for CRT patientsabstractAbnormal cardiac motion can indicate different forms of disease, which can manifest at different spatial scales in the myocardium. Many studies have sought to characterise particular motion abnormalities associated with specific diseases, and to utilise motion information to improve diagnoses. However, the importance of spatial scale in the analysis of cardiac deformation has not been extensively investigated. We build on recent work on the analysis of myocardial strains at different spatial scales using a cardiac motion atlas to find the optimal scales for estimating different cardiac biomarkers. We apply a multi-scale strain analysis to a 43 patient cohort of cardiac resynchronisation therapy (CRT) patients using tagged magnetic resonance imaging data for (1) predicting response to CRT, (2) identifying septal flash, (3) estimating QRS duration, and (4) identifying the presence of ischaemia. A repeated, stratified cross-validation is used to demonstrate the importance of spatial scale in our analysis, revealing different optimal spatial scales for the estimation of different biomarkers. Matthew Sinclair, Devis Peressutti, Esther Puyol-Antón, Wenjia Bai, Simone Rivolo, Jessica Webb, Simon Claridge, David Nordsletten, Myrianthi Hadjicharalambous, Eric Kerfoot, C. Aldo Rinaldi, Daniel Rueckert, Andrew P. King |
Medical Image Anal. | 4 |
| 2018 | Multi-Atlas Segmentation Using Partially Annotated Data: Methods and Annotation StrategiesabstractMulti-atlas segmentation is a widely used tool in medical image analysis, providing robust and accurate results by learning from annotated atlas datasets. However, the availability of fully annotated atlas images for training is limited due to the time required for the labelling task. Segmentation methods requiring only a proportion of each atlas image to be labelled could therefore reduce the workload on expert raters tasked with annotating atlas images. To address this issue, we first re-examine the labelling problem common in many existing approaches and formulate its solution in terms of a Markov Random Field energy minimisation problem on a graph connecting atlases and the target image. This provides a unifying framework for multi-atlas segmentation. We then show how modifications in the graph configuration of the proposed framework enable the use of partially annotated atlas images and investigate different partial annotation strategies. The proposed method was evaluated on two Magnetic Resonance Imaging (MRI) datasets for hippocampal and cardiac segmentation. Experiments were performed aimed at (1) recreating existing segmentation techniques with the proposed framework and (2) demonstrating the potential of employing sparsely annotated atlas data for multi-atlas segmentation. Lisa M. Koch, Martin Rajchl, Wenjia Bai, Christian F. Baumgartner, Tong Tong 0001, Jonathan Passerat-Palmbach, Paul Aljabar, Daniel Rueckert |
IEEE Trans. Pattern Anal. Mach. Intell. | 3 |
| 2018 | Statistical Shape Modeling of the Left Ventricle: Myocardial Infarct Classification ChallengeabstractStatistical shape modeling is a powerful tool for visualizing and quantifying geometric and functional patterns of the heart. After myocardial infarction (MI), the left ventricle typically remodels in response to physiological challenges. Several methods have been proposed in the literature to describe statistical shape changes. Which method best characterizes left ventricular remodeling after MI is an open research question. A better descriptor of remodeling is expected to provide a more accurate evaluation of disease status in MI patients. We therefore designed a challenge to test shape characterization in MI given a set of three-dimensional left ventricular surface points. The training set comprised 100 MI patients, and 100 asymptomatic volunteers (AV). The challenge was initiated in 2015 at the Statistical Atlases and Computational Models of the Heart workshop, in conjunction with the MICCAI conference. The training set with labels was provided to participants, who were asked to submit the likelihood of MI from a different (validation) set of 200 cases (100 AV and 100 MI). Sensitivity, specificity, accuracy and area under the receiver operating characteristic curve were used as the outcome measures. The goals of this challenge were to (1) establish a common dataset for evaluating statistical shape modeling algorithms in MI, and (2) test whether statistical shape modeling provides additional information characterizing MI patients over standard clinical measures. Eleven groups with a wide variety of classification and feature extraction approaches participated in this challenge. All methods achieved excellent classification results with accuracy ranges from 0.83 to 0.98. The areas under the receiver operating characteristic curves were all above 0.90. Four methods showed significantly higher performance than standard clinical measures. The dataset and software for evaluation are available from the Cardiac Atlas Project website1. Avan Suinesiaputra, Pierre Ablin, Xènia Albà, Martino Alessandrini, Jack Allen, Wenjia Bai, Serkan Çimen, Peter Claes, Brett R. Cowan, Jan D'hooge, Nicolas Duchateau, Jan Ehrhardt, Alejandro F. Frangi, Ali Gooya, Vicente Grau, Karim Lekadir, Allen Lu, Anirban Mukhopadhyay 0003, Ilkay Öksüz, Nripesh Parajuli, Xavier Pennec, Marco Pereañez, Catarina Pinto, Paolo Piras, Marc-Michel Rohé, Daniel Rueckert, Dennis Säring, Maxime Sermesant, Kaleem Siddiqi, Mahdi Tabassian, Luciano Teresi, Sotirios A. Tsaftaris, Matthias Wilms, Alistair A. Young, Pau Medrano-Gracia |
IEEE J. Biomed. Health Informatics | 6 |
| 2018 | Anatomically Constrained Neural Networks (ACNNs): Application to Cardiac Image Enhancement and SegmentationabstractIncorporation of prior knowledge about organ shape and location is key to improve performance of image analysis approaches. In particular, priors can be useful in cases where images are corrupted and contain artefacts due to limitations in image acquisition. The highly constrained nature of anatomical objects can be well captured with learning-based techniques. However, in most recent and promising techniques such as CNN-based segmentation it is not obvious how to incorporate such prior knowledge. State-of-the-art methods operate as pixel-wise classifiers where the training objectives do not incorporate the structure and inter-dependencies of the output. To overcome this limitation, we propose a generic training strategy that incorporates anatomical prior knowledge into CNNs through a new regularisation model, which is trained end-to-end. The new framework encourages models to follow the global anatomical properties of the underlying anatomy (e.g. shape, label structure) via learnt non-linear representations of the shape. We show that the proposed approach can be easily adapted to different analysis tasks (e.g. image enhancement, segmentation) and improve the prediction accuracy of the state-of-the-art models. The applicability of our approach is shown on multi-modal cardiac data sets and public benchmarks. In addition, we demonstrate how the learnt deep models of 3-D shapes can be interpreted and used as biomarkers for classification of cardiac pathologies. Ozan Oktay, Enzo Ferrante, Konstantinos Kamnitsas, Mattias P. Heinrich, Wenjia Bai, Jose Caballero, Stuart A. Cook, Antonio M. Simoes Monteiro de Marvao, Timothy Dawes, Declan P. O'Regan, Bernhard Kainz, Ben Glocker, Daniel Rueckert |
IEEE Trans. Medical Imaging | 5 |
| 2017 | Semi-supervised Learning for Network-Based Cardiac MR Image Segmentation
Wenjia Bai, Ozan Oktay, Matthew Sinclair, Hideaki Suzuki, Martin Rajchl, Giacomo Tarroni, Ben Glocker, Andrew P. King, Paul M. Matthews, Daniel Rueckert |
MICCAI (2) | 1 |
| 2017 | Automatic Quality Control of Cardiac MRI Segmentation in Large-Scale Population Imaging
Robert Robinson, Vanya V. Valindria, Wenjia Bai, Hideaki Suzuki, Paul M. Matthews, Chris Page, Daniel Rueckert, Ben Glocker |
MICCAI (1) | 3 |
| 2017 | Fully Automated Segmentation-Based Respiratory Motion Correction of Multiplanar Cardiac Magnetic Resonance Images for Large-Scale Datasets
Matthew Sinclair, Wenjia Bai, Esther Puyol-Antón, Ozan Oktay, Daniel Rueckert, Andrew P. King |
MICCAI (2) | 2 |
| 2017 | A framework for combining a motion atlas with non-motion information to learn clinically useful biomarkers: Application to cardiac resynchronisation therapy response predictionabstractWe present a framework for combining a cardiac motion atlas with non-motion data. The atlas represents cardiac cycle motion across a number of subjects in a common space based on rich motion descriptors capturing 3D displacement, velocity, strain and strain rate. The non-motion data are derived from a variety of sources such as imaging, electrocardiogram (ECG) and clinical reports. Once in the atlas space, we apply a novel supervised learning approach based on random projections and ensemble learning to learn the relationship between the atlas data and some desired clinical output. We apply our framework to the problem of predicting response to Cardiac Resynchronisation Therapy (CRT). Using a cohort of 34 patients selected for CRT using conventional criteria, results show that the combination of motion and non-motion data enables CRT response to be predicted with 91.2% accuracy (100% sensitivity and 62.5% specificity), which compares favourably with the current state-of-the-art in CRT response prediction. Devis Peressutti, Matthew Sinclair, Wenjia Bai, Jacobus Ruijsink, David Nordsletten, Liya Asner, Myrianthi Hadjicharalambous, C. Aldo Rinaldi, Daniel Rueckert, Andrew P. King |
Medical Image Anal. | 3 |
| 2017 | Stratified Decision Forests for Accurate Anatomical Landmark Localization in Cardiac ImagesabstractAccurate localization of anatomical landmarks is an important step in medical imaging, as it provides useful prior information for subsequent image analysis and acquisition methods. It is particularly useful for initialization of automatic image analysis tools (e.g. segmentation and registration) and detection of scan planes for automated image acquisition. Landmark localization has been commonly performed using learning based approaches, such as classifier and/or regressor models. However, trained models may not generalize well in heterogeneous datasets when the images contain large differences due to size, pose and shape variations of organs. To learn more data-adaptive and patient specific models, we propose a novel stratification based training model, and demonstrate its use in a decision forest. The proposed approach does not require any additional training information compared to the standard model training procedure and can be easily integrated into any decision tree framework. The proposed method is evaluated on 1080 3D high-resolution and 90 multi-stack 2D cardiac cine MR images. The experiments show that the proposed method achieves state-of-the-art landmark localization accuracy and outperforms standard regression and classification based approaches. Additionally, the proposed method is used in a multi-atlas segmentation to create a fully automatic segmentation pipeline, and the results show that it achieves state-of-the-art segmentation accuracy. Ozan Oktay, Wenjia Bai, Ricardo Guerrero, Martin Rajchl, Antonio M. Simoes Monteiro de Marvao, Declan P. O'Regan, Stuart A. Cook, Mattias P. Heinrich, Ben Glocker, Daniel Rueckert |
IEEE Trans. Medical Imaging | 2 |
| 2017 | DeepCut: Object Segmentation From Bounding Box Annotations Using Convolutional Neural NetworksabstractIn this paper, we propose DeepCut, a method to obtain pixelwise object segmentations given an image dataset labelled weak annotations, in our case bounding boxes. It extends the approach of the well-known GrabCut [1] method to include machine learning by training a neural network classifier from bounding box annotations. We formulate the problem as an energy minimisation problem over a densely-connected conditional random field and iteratively update the training targets to obtain pixelwise object segmentations. Additionally, we propose variants of the DeepCut method and compare those to a naïve approach to CNN training under weak supervision. We test its applicability to solve brain and lung segmentation problems on a challenging fetal magnetic resonance dataset and obtain encouraging results in terms of accuracy. Martin Rajchl, Matthew C. H. Lee, Ozan Oktay, Konstantinos Kamnitsas, Jonathan Passerat-Palmbach, Wenjia Bai, Mellisa Damodaram, Mary A. Rutherford, Joseph V. Hajnal, Bernhard Kainz, Daniel Rueckert |
IEEE Trans. Medical Imaging | 6 |
| 2017 | Reverse Classification Accuracy: Predicting Segmentation Performance in the Absence of Ground TruthabstractWhen integrating computational tools, such as automatic segmentation, into clinical practice, it is of utmost importance to be able to assess the level of accuracy on new data and, in particular, to detect when an automatic method fails. However, this is difficult to achieve due to the absence of ground truth. Segmentation accuracy on clinical data might be different from what is found through cross validation, because validation data are often used during incremental method development, which can lead to overfitting and unrealistic performance expectations. Before deployment, performance is quantified using different metrics, for which the predicted segmentation is compared with a reference segmentation, often obtained manually by an expert. But little is known about the real performance after deployment when a reference is unavailable. In this paper, we introduce the concept of reverse classification accuracy (RCA) as a framework for predicting the performance of a segmentation method on new data. In RCA, we take the predicted segmentation from a new image to train a reverse classifier, which is evaluated on a set of reference images with available ground truth. The hypothesis is that if the predicted segmentation is of good quality, then the reverse classifier will perform well on at least some of the reference images. We validate our approach on multi-organ segmentation with different classifiers and segmentation methods. Our results indicate that it is indeed possible to predict the quality of individual segmentations, in the absence of ground truth. Thus, RCA is ideal for integration into automatic processing pipelines in clinical routine and as a part of large-scale image analysis studies. Vanya V. Valindria, Ioannis Lavdas, Wenjia Bai, Konstantinos Kamnitsas, Eric O. Aboagye, Andrea G. Rockall, Daniel Rueckert, Ben Glocker |
IEEE Trans. Medical Imaging | 3 |
| 2016 | Multi-input Cardiac Image Super-Resolution Using Convolutional Neural Networksabstract3D cardiac MR imaging enables accurate analysis of cardiac morphology and physiology. However, due to the requirements for long acquisition and breath-hold, the clinical routine is still dominated by multi-slice 2D imaging, which hamper the visualization of anatomy and quantitative measurements as relatively thick slices are acquired. As a solution, we propose a novel image super-resolution (SR) approach that is based on a residual convolutional neural network (CNN) model. It reconstructs high resolution 3D volumes from 2D image stacks for more accurate image analysis. The proposed model allows the use of multiple input data acquired from different viewing planes for improved performance. Experimental results on 1233 cardiac short and long-axis MR image stacks show that the CNN model outperforms state-of-the-art SR methods in terms of image quality while being computationally efficient. Also, we show that image segmentation and motion tracking benefits more from SR-CNN when it is used as an initial upscaling method than conventional interpolation methods for the subsequent analysis. Ozan Oktay, Wenjia Bai, Matthew C. H. Lee, Ricardo Guerrero, Konstantinos Kamnitsas, Jose Caballero, Antonio M. Simoes Monteiro de Marvao, Stuart A. Cook, Declan P. O'Regan, Daniel Rueckert |
MICCAI (3) | 2 |
| 2015 | Prospective Identification of CRT Super Responders Using a Motion Atlas and Random Projection Ensemble Learning
Devis Peressutti, Wenjia Bai, Manav Sohal, C. Aldo Rinaldi, Daniel Rueckert, Andrew P. King |
MICCAI (3) | 2 |
| 2015 | Multi-atlas segmentation with augmented features for cardiac MR images
Wenjia Bai, Wenzhe Shi, Christian Ledig, Daniel Rueckert |
Medical Image Anal. | 1 |
| 2015 | A bi-ventricular cardiac atlas built from 1000+ high resolution MR images of healthy subjects and an analysis of shape and motion
Wenjia Bai, Wenzhe Shi, Antonio M. Simoes Monteiro de Marvao, Timothy Dawes, Declan P. O'Regan, Stuart A. Cook, Daniel Rueckert |
Medical Image Anal. | 1 |
| 2015 | Right ventricle segmentation from cardiac MRI: A collation study
Caroline Petitjean, Maria A. Zuluaga, Wenjia Bai, Jean-Nicolas Dacher, Damien Grosgeorge, Jérôme Caudron, Su Ruan, Ismail Ben Ayed, Manuel Jorge Cardoso, Hsiang-Chou Chen, Daniel Jimenez-Carretero, María J. Ledesma-Carbayo, Christos Davatzikos, Jimit Doshi, Güray Erus, Oskar M. O. Maier, Cyrus M. S. Nambakhsh, Yangming Ou, Sébastien Ourselin, Chun-Wei Peng, Nicholas S. Peters, Terry M. Peters, Martin Rajchl, Daniel Rueckert, Wenzhe Shi, Ching-Wei Wang, Haiyan Wang 0018, Jing Yuan 0001 |
Medical Image Anal. | 3 |
| 2014 | Patch-Based Evaluation of Image SegmentationabstractThe quantification of similarity between image segmentations is a complex yet important task. The ideal similarity measure should be unbiased to segmentations of different volume and complexity, and be able to quantify and visualise segmentation bias. Similarity measures based on overlap, e.g. Dice score, or surface distances, e.g. Hausdorff distance, clearly do not satisfy all of these properties. To address this problem, we introduce Patch-based Evaluation of Image Segmentation (PEIS), a general method to assess segmentation quality. Our method is based on finding patch correspondences and the associated patch displacements, which allow the estimation of segmentation bias. We quantify both the agreement of the segmentation boundary and the conservation of the segmentation shape. We further assess the segmentation complexity within patches to weight the contribution of local segmentation similarity to the global score. We evaluate PEIS on both synthetic data and two medical imaging datasets. On synthetic segmentations of different shapes, we provide evidence that PEIS, in comparison to the Dice score, produces more comparable scores, has increased sensitivity and estimates segmentation bias accurately. On cardiac magnetic resonance (MR) images, we demonstrate that PEIS can evaluate the performance of a segmentation method independent of the size or complexity of the segmentation under consideration. On brain MR images, we compare five different automatic hippocampus segmentation techniques using PEIS. Finally, we visualise the segmentation bias on a selection of the cases. Christian Ledig, Wenzhe Shi, Wenjia Bai, Daniel Rueckert |
CVPR | 3 |
| 2014 | Application-Driven MRI: Joint Reconstruction and Segmentation from Undersampled MRI Data
Jose Caballero, Wenjia Bai, Anthony N. Price, Daniel Rueckert, Joseph V. Hajnal |
MICCAI (1) | 2 |
| 2014 | Multi-atlas Spectral PatchMatch: Application to Cardiac Image Segmentation
Wenzhe Shi, Hervé Lombaert, Wenjia Bai, Christian Ledig, Xiahai Zhuang, Antonio M. Simoes Monteiro de Marvao, Timothy Dawes, Declan P. O'Regan, Daniel Rueckert |
MICCAI (1) | 3 |
| 2013 | Cardiac Image Super-Resolution with Global Correspondence Using Multi-Atlas PatchMatch
Wenzhe Shi, Jose Caballero, Christian Ledig, Xiahai Zhuang, Wenjia Bai, Kanwal K. Bhatia, Antonio M. Simoes Monteiro de Marvao, Timothy Dawes, Declan P. O'Regan, Daniel Rueckert |
MICCAI (3) | 5 |
| 2013 | Temporal sparse free-form deformations
Wenzhe Shi, Martin Jantsch, Paul Aljabar, Luis Pizarro, Wenjia Bai, Haiyan Wang 0018, Declan P. O'Regan, Xiahai Zhuang, Daniel Rueckert |
Medical Image Anal. | 5 |
| 2013 | A Probabilistic Patch-Based Label Fusion Model for Multi-Atlas Segmentation With Registration Refinement: Application to Cardiac MR ImagesabstractThe evaluation of ventricular function is important for the diagnosis of cardiovascular diseases. It typically involves measurement of the left ventricular (LV) mass and LV cavity volume. Manual delineation of the myocardial contours is time-consuming and dependent on the subjective experience of the expert observer. In this paper, a multi-atlas method is proposed for cardiac magnetic resonance (MR) image segmentation. The proposed method is novel in two aspects. First, it formulates a patch-based label fusion model in a Bayesian framework. Second, it improves image registration accuracy by utilizing label information, which leads to improvement of segmentation accuracy. The proposed method was evaluated on a cardiac MR image set of 28 subjects. The average Dice overlap metric of our segmentation is 0.92 for the LV cavity, 0.89 for the right ventricular cavity and 0.82 for the myocardium. The results show that the proposed method is able to provide accurate information for clinical diagnosis. Wenjia Bai, Wenzhe Shi, Declan P. O'Regan, Tong Tong 0001, Haiyan Wang 0018, Shahnaz Jamil-Copley, Nicholas S. Peters, Daniel Rueckert |
IEEE Trans. Medical Imaging | 1 |
| 2012 | Registration Using Sparse Free-Form Deformations
Wenzhe Shi, Xiahai Zhuang, Luis Pizarro, Wenjia Bai, Haiyan Wang 0018, Kai-Pin Tung, Philip J. Edwards, Daniel Rueckert |
MICCAI (2) | 4 |
| 2011 | Motion Correction and Attenuation Correction for Respiratory Gated PET ImagesabstractPositron emission tomography (PET) is a molecular imaging technique which provides important functional information about the human body. However, thoracic PET images are often substantially degraded by respiratory motion, which adversely impacts on subsequent diagnosis. In this paper, a motion correction and attenuation correction method is proposed to correct for motion in respiratory gated PET images and to yield an accurate distribution of the radioactivity concentration. Experimental results show that this method can effectively correct for motion and improve PET image quality. The method is able to provide improved diagnostic information without increasing the acquisition time or the radiation burden. Wenjia Bai, J. Michael Brady |
IEEE Trans. Medical Imaging | 1 |