Deep learning for automated detection of breast cancer in deep ultraviolet fluorescence images with diffusion probabilistic model
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911999850446848 |
|---|---|
| author | Ghahfarokhi, Sepehr Salem To, Tyrell Jorns, Julie Yen, Tina Yu, Bing Ye, Dong Hye |
| author_facet | Ghahfarokhi, Sepehr Salem To, Tyrell Jorns, Julie Yen, Tina Yu, Bing Ye, Dong Hye |
| contents | Data limitation is a significant challenge in applying deep learning to medical images. Recently, the diffusion probabilistic model (DPM) has shown the potential to generate high-quality images by converting Gaussian random noise into realistic images. In this paper, we apply the DPM to augment the deep ultraviolet fluorescence (DUV) image dataset with an aim to improve breast cancer classification for intraoperative margin assessment. For classification, we divide the whole surface DUV image into small patches and extract convolutional features for each patch by utilizing the pre-trained ResNet. Then, we feed them into an XGBoost classifier for patch-level decisions and then fuse them with a regional importance map computed by Grad-CAM++ for whole surface-level prediction. Our experimental results show that augmenting the training dataset with the DPM significantly improves breast cancer detection performance in DUV images, increasing accuracy from 93% to 97%, compared to using Affine transformations and ProGAN. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2407_00967 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Deep learning for automated detection of breast cancer in deep ultraviolet fluorescence images with diffusion probabilistic model Ghahfarokhi, Sepehr Salem To, Tyrell Jorns, Julie Yen, Tina Yu, Bing Ye, Dong Hye Computer Vision and Pattern Recognition Artificial Intelligence Data limitation is a significant challenge in applying deep learning to medical images. Recently, the diffusion probabilistic model (DPM) has shown the potential to generate high-quality images by converting Gaussian random noise into realistic images. In this paper, we apply the DPM to augment the deep ultraviolet fluorescence (DUV) image dataset with an aim to improve breast cancer classification for intraoperative margin assessment. For classification, we divide the whole surface DUV image into small patches and extract convolutional features for each patch by utilizing the pre-trained ResNet. Then, we feed them into an XGBoost classifier for patch-level decisions and then fuse them with a regional importance map computed by Grad-CAM++ for whole surface-level prediction. Our experimental results show that augmenting the training dataset with the DPM significantly improves breast cancer detection performance in DUV images, increasing accuracy from 93% to 97%, compared to using Affine transformations and ProGAN. |
| title | Deep learning for automated detection of breast cancer in deep ultraviolet fluorescence images with diffusion probabilistic model |
| topic | Computer Vision and Pattern Recognition Artificial Intelligence |
| url | https://arxiv.org/abs/2407.00967 |