Cross-Domain Data Augmentation for Deep-Learning-Based Male Pelvic Organ Segmentation in Cone Beam CT

Applied Sciences

Jean Léger¹  Eliott Brion¹  JohnA. Lee¹  Benoit Macq¹  Christophe De Vleeschouwer¹  Paul Desbordes¹ 

[1] Institute of Information and Communication Technologies, Electronics and Applied Mathematics, Electrical Engineering Department, Université catholique de Louvain, 1348 Louvain-la-Neuve, Belgium;

关键词: segmentation;    deep-learning;    deformable image registration;    cone beam ct;    pelvis;    prostate cancer;    radiotherapy;    cnn;    u-net;   

DOI  :  10.3390/app10031154

来源: DOAJ

【 摘 要 】

For prostate cancer patients, large organ deformations occurring between radiotherapy treatment sessions create uncertainty about the doses delivered to the tumor and surrounding healthy organs. Segmenting those regions on cone beam CT (CBCT) scans acquired on treatment day would reduce such uncertainties. In this work, a 3D U-net deep-learning architecture was trained to segment bladder, rectum, and prostate on CBCT scans. Due to the scarcity of contoured CBCT scans, the training set was augmented with CT scans already contoured in the current clinical workflow. Our network was then tested on 63 CBCT scans. The Dice similarity coefficient (DSC) increased significantly with the number of CBCT and CT scans in the training set, reaching $0.874$ ± $0.096$ , $0.814$ ± $0.055$ , and $0.758$ ± $0.101$ for bladder, rectum, and prostate, respectively. This was about 10% better than conventional approaches based on deformable image registration between planning CT and treatment CBCT scans, except for prostate. Interestingly, adding 74 CT scans to the CBCT training set allowed maintaining high DSCs, while halving the number of CBCT scans. Hence, our work showed that although CBCT scans included artifacts, cross-domain augmentation of the training set was effective and could rely on large datasets available for planning CT scans.

【 授权许可】

Unknown