Brief Review — Improving the Generalizability of Convolutional Neural Network-Based Segmentation on CMR Images
U-Net for CMR Image Segmentation
Improving the Generalizability of Convolutional Neural Network-Based Segmentation on CMR Images,
Chen FCVM’20, by Imperial College London, University College London, St Bartholomew’s Hospital, Queen Mary University of London, and University of Oxford,
2020 Front. Cardiovasc. Med., Over 80 Citations (Sik-Ho Tsang @ Medium)
Medical Imaging, Medical Image Analysis, Image Segmentation
- By carefully designing data normalization and augmentation strategies, a neural network trained on a single-site single-scanner dataset from the UK Biobank can be successfully applied to segmenting cardiac MR images across different sites and different scanners without substantial loss of accuracy.
- This is the first work to explore the generalizability of CNN-based methods for cardiac MR image multi-structure segmentation.
- Proposed Approach
1. Proposed Approach
- Three datasets are used, as listed above:
- UK Biobank (UKBB): consists of over half a million voluntary participants aged between 40 and 69 from across the UK, nearly 100,000 participants, including brain, cardiac and whole-body MR imaging. Pixel-wise segmentations of three essential structures (LV, MYO, and RV) for both end-diastolic (ED) frames and end-systolic (ES) frames are provided as ground-truths.
- Automated Cardiac Diagnosis Challenge (ACDC): is a part of the MICCAI 2017 benchmark dataset for CMR image segmentation. This dataset is composed of 100 CMR images. The LV, MYO, and RV in this dataset have been manually segmented for both ED frames and ES frames.
- British Society of CardiovascularMagnetic Resonance Aortic Stenosis (BSCMR-AS): consists of CMR images of 599 patients with severe aortic stenosis (AS), who had been listed for surgery. Only the left ventricle in ED frames and ES frames, as well as the myocardium in ED frames, have been annotated manually.
- UKBB dataset for training and intra-domain testing, and use the ACDC data and BSCMRAS dataset for cross-domain testing.
- In UKBB, 3,975 subjects were used to train the neural network while 300 validation subjects were used for tracking the training progress and avoid over-fitting. The subset consisting of remaining 600 subjects was used for evaluating models’ performance in the intra-domain setting.
1.2. Model Architecture
- 2D U-Net is used, but with two main differences: (1) batch normalization (BN) is applied after each hidden convolutional layer to stabilize the training; (2) Dropout regularization is applied after each concatenating operation to avoid overfitting and encourage generalization.
- Cross entropy loss function is used.
- Image resampling and intensity normalization are employed to normalize images in both the training and testing stages.
- During training, a wide range of geometrical variations is applied in terms of the heart pose and size: Random horizontal and vertical flips, random rotation, random image scaling, and random image cropping.
- During testing, only center cropping is used.
While both methods achieve very similar Dice scores on the intra-domain UKBB test set with high accuracy, the proposed method significantly outperforms the previous approach on the two cross-domain datasets: ACDC set and BSCMR-AS set.
- UNet-16: a smaller version of U-Net where the number of filters in each convolutional layer is reduced by four times.
UNet-64 outperforms FCN-64 on all of the three test sets, while UNet-64 contains fewer parameters than FCN-64.
The model is capable of segmenting not only those with normal cardiac structures but also some abnormal cases with the cardiac morphological variations in those HCM images and AS images.
[2020 Front. Cardiovasc. Med.] [Chen FCVM’20]
Improving the Generalizability of Convolutional Neural Network-Based Segmentation on CMR Images
4.2. Biomedical Image Segmentation
2015–2020 [MultiResUNet] [UNet 3+] [Dense-Gated U-Net (DGNet)] [Non-local U-Net] SAUNet] [SDM] [DIU-Net] [Chen FCVM’20] 2021 [Expanded U-Net]