Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2404.12450 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866929320440627200 |
|---|---|
| author | Wei, Ting-Ruen Hell, Michele Le, Dang Bich Thuy Vierra, Aren Pang, Ran Patel, Mahesh Kang, Young Yan, Yuling |
| author_facet | Wei, Ting-Ruen Hell, Michele Le, Dang Bich Thuy Vierra, Aren Pang, Ran Patel, Mahesh Kang, Young Yan, Yuling |
| contents | This study presents an unsupervised domain adaptation method aimed at autonomously generating image masks outlining regions of interest (ROIs) for differentiating breast lesions in breast ultrasound (US) imaging. Our semi-supervised learning approach utilizes a primitive model trained on a small public breast US dataset with true annotations. This model is then iteratively refined for the domain adaptation task, generating pseudo-masks for our private, unannotated breast US dataset. The dataset, twice the size of the public one, exhibits considerable variability in image acquisition perspectives and demographic representation, posing a domain-shift challenge. Unlike typical domain adversarial training, we employ downstream classification outcomes as a benchmark to guide the updating of pseudo-masks in subsequent iterations. We found the classification precision to be highly correlated with the completeness of the generated ROIs, which promotes the explainability of the deep learning classification model. Preliminary findings demonstrate the efficacy and reliability of this approach in streamlining the ROI annotation process, thereby enhancing the classification and localization of breast lesions for more precise and interpretable diagnoses. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2404_12450 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Enhancing AI Diagnostics: Autonomous Lesion Masking via Semi-Supervised Deep Learning Wei, Ting-Ruen Hell, Michele Le, Dang Bich Thuy Vierra, Aren Pang, Ran Patel, Mahesh Kang, Young Yan, Yuling Computer Vision and Pattern Recognition Artificial Intelligence Machine Learning This study presents an unsupervised domain adaptation method aimed at autonomously generating image masks outlining regions of interest (ROIs) for differentiating breast lesions in breast ultrasound (US) imaging. Our semi-supervised learning approach utilizes a primitive model trained on a small public breast US dataset with true annotations. This model is then iteratively refined for the domain adaptation task, generating pseudo-masks for our private, unannotated breast US dataset. The dataset, twice the size of the public one, exhibits considerable variability in image acquisition perspectives and demographic representation, posing a domain-shift challenge. Unlike typical domain adversarial training, we employ downstream classification outcomes as a benchmark to guide the updating of pseudo-masks in subsequent iterations. We found the classification precision to be highly correlated with the completeness of the generated ROIs, which promotes the explainability of the deep learning classification model. Preliminary findings demonstrate the efficacy and reliability of this approach in streamlining the ROI annotation process, thereby enhancing the classification and localization of breast lesions for more precise and interpretable diagnoses. |
| title | Enhancing AI Diagnostics: Autonomous Lesion Masking via Semi-Supervised Deep Learning |
| topic | Computer Vision and Pattern Recognition Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2404.12450 |