Segmentation of Macular Edema Datasets with Small Residual 3D U-Net Architectures

Frawley, J; Willcocks, CG; Habib, M; Geenen, C; Steel, DH; Obara, B

doi:10.1109/BIBE50027.2020.00100

Segmentation of Macular Edema Datasets with Small Residual 3D U-Net Architectures

Lookup NU author(s): Dr Chris Willcocks, Maged Habib, Professor David Steel ORCiD, Professor Boguslaw Obara ORCiD

Downloads

Full text for this publication is not currently held within this repository. Alternative links are provided below where available.

Abstract

© 2020 IEEE. This paper investigates the application of deep convolutional neural networks with prohibitively small datasets to the problem of macular edema segmentation. In particular, we investigate several different heavily regularized architectures. We find that, contrary to popular belief, neural architectures within this application setting are able to achieve close to human-level performance on unseen test images without requiring large numbers of training examples. Annotating these 3D datasets is difficult, with multiple criteria required. It takes an experienced clinician two days to annotate a single 3D image, whereas our trained model achieves similar performance in less than a second. We found that an approach which uses targeted dataset augmentation, alongside architectural simplification with an emphasis on residual design, has acceptable generalization performance- despite relying on fewer than 15 training examples.