首页|Rethinking Polyp Segmentation from An Out-of-distribution Perspective

Rethinking Polyp Segmentation from An Out-of-distribution Perspective

扫码查看
Unlike existing fully-supervised approaches,we rethink colorectal polyp segmentation from an out-of-distribution perspect-ive with a simple but effective self-supervised learning approach.We leverage the ability of masked autoencoders-self-supervised vision transformers trained on a reconstruction task-to learn in-distribution representations,here,the distribution of healthy colon images.We then perform out-of-distribution reconstruction and inference,with feature space standardisation to align the latent distribution of the diverse abnormal samples with the statistics of the healthy samples.We generate per-pixel anomaly scores for each image by calcu-lating the difference between the input and reconstructed images and use this signal for out-of-distribution(i.e.,polyp)segmentation.Experimental results on six benchmarks show that our model has excellent segmentation performance and generalises across datasets.Our code is publicly available at https://github.com/GewelsJI/Polyp-OOD.

Polyp segmentationanomaly segmentationout-of-distribution segmentationmasked autoencoderabdomen

Ge-Peng Ji、Jing Zhang、Dylan Campbell、Huan Xiong、Nick Barnes

展开 >

Australian National University,Canberra8105,Australia

Mohamed bin Zayed University of Artificial Intelligence,Abu Dhabi 999041,UAE

Open Access funding enabled and organized by CAUL

2024

机器智能研究(英文)
中国科学院自动化所

机器智能研究(英文)

CSTPCDEI
影响因子:0.49
ISSN:2731-538X
年,卷(期):2024.21(4)