Improving ovarian cancer segmentation accuracy with transformers through AI-guided labeling
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866913617601888256 |
|---|---|
| author | Rangnekar, Aneesh Boehm, Kevin M. Aherne, Emily A. Nikolovski, Ines Gangai, Natalie Liu, Ying Zamarin, Dimitry Roche, Kara L. Shah, Sohrab P. Lakhman, Yulia Veeraraghavan, Harini |
| author_facet | Rangnekar, Aneesh Boehm, Kevin M. Aherne, Emily A. Nikolovski, Ines Gangai, Natalie Liu, Ying Zamarin, Dimitry Roche, Kara L. Shah, Sohrab P. Lakhman, Yulia Veeraraghavan, Harini |
| contents | Transformer models have demonstrated the capability to produce highly accurate segmentation of organs and tumors. However, model training requires high-quality curated datasets to ensure robust generalization to unseen datasets. Hence, we developed an artificial intelligence (AI) guided approach to assist with radiologist tumor delineation of partially segmented computed tomography datasets containing primary (adnexa) tumors and metastatic (omental) implants. AI guidance was implemented by training a 2D multiple resolution residual network trained with a dataset of 245 contrast-enhanced CTs with partially segmented examples. The same dataset curated through AI guidance was then used to refine two pretrained transformer models called SMIT and Swin UNETR. The models were independently tested on 71 publicly available multi-institutional 3D CT datasets. Segmentation accuracy was computed using the Dice similarity coefficient metric (DSC), average symmetric surface distance (ASSD), and the relative volume difference (RVD) metrics. Radiomic features reproducibility was assessed using the concordance correlation coefficient (CCC). Training with AI-guided segmentations significantly improved the accuracy of both SMIT (p = 6.2e-5) and Swin UNETR (p = 2e-4) models compared with using a partially delineated training dataset. Furthermore, SMIT-generated segmentations resulted in more reproducible features compared to Swin UNETR under multiple feature categories. Our results show that AI-guided data curation provides a more efficient approach to train AI models and that AI-generated segmentations can provide reproducible radiomics features. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2406_17666 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving ovarian cancer segmentation accuracy with transformers through AI-guided labeling Rangnekar, Aneesh Boehm, Kevin M. Aherne, Emily A. Nikolovski, Ines Gangai, Natalie Liu, Ying Zamarin, Dimitry Roche, Kara L. Shah, Sohrab P. Lakhman, Yulia Veeraraghavan, Harini Image and Video Processing Transformer models have demonstrated the capability to produce highly accurate segmentation of organs and tumors. However, model training requires high-quality curated datasets to ensure robust generalization to unseen datasets. Hence, we developed an artificial intelligence (AI) guided approach to assist with radiologist tumor delineation of partially segmented computed tomography datasets containing primary (adnexa) tumors and metastatic (omental) implants. AI guidance was implemented by training a 2D multiple resolution residual network trained with a dataset of 245 contrast-enhanced CTs with partially segmented examples. The same dataset curated through AI guidance was then used to refine two pretrained transformer models called SMIT and Swin UNETR. The models were independently tested on 71 publicly available multi-institutional 3D CT datasets. Segmentation accuracy was computed using the Dice similarity coefficient metric (DSC), average symmetric surface distance (ASSD), and the relative volume difference (RVD) metrics. Radiomic features reproducibility was assessed using the concordance correlation coefficient (CCC). Training with AI-guided segmentations significantly improved the accuracy of both SMIT (p = 6.2e-5) and Swin UNETR (p = 2e-4) models compared with using a partially delineated training dataset. Furthermore, SMIT-generated segmentations resulted in more reproducible features compared to Swin UNETR under multiple feature categories. Our results show that AI-guided data curation provides a more efficient approach to train AI models and that AI-generated segmentations can provide reproducible radiomics features. |
| title | Improving ovarian cancer segmentation accuracy with transformers through AI-guided labeling |
| topic | Image and Video Processing |
| url | https://arxiv.org/abs/2406.17666 |