Anatomical prior-inspired label refinement for weakly supervised liver tumor segmentation with volume-level labels
Fei Lyu (Department of Computer Science, Hong Kong Baptist University),* Andy J Ma (Sun Yat-sen University), PongChi Yuen (Department of Computer Science, Hong Kong Baptist University)
The 33rd British Machine Vision Conference
Abstract
Automatic liver tumor segmentation is important for assisting doctors in accurate diagnosis of liver cancer. Existing models for liver tumor segmentation usually require accurate pixel-level labels. However, acquiring such dense labels is laborious and costly. In this paper, we propose a weakly supervised method for liver tumor segmentation using volume-level labels, which can significantly reduce the manual annotation effort. Volume-level labels are propagated to image-level labels where all the slices in one CT volume share the same label, and pixel-level pseudo labels can be estimated from image-level labels. However, it will cause the label noise problem because not all slices contain tumors. To address this issue, we propose two label refinement strategies based on anatomical priors to reduce the training noise and improve the performance. Evaluation experiments on two public datasets demonstrate that our proposed method can achieve competitive results compared to other methods with stronger supervision.Video
Citation
@inproceedings{Lyu_2022_BMVC,
author = {Fei Lyu and Andy J Ma and PongChi Yuen},
title = {Anatomical prior-inspired label refinement for weakly supervised liver tumor segmentation with volume-level labels},
booktitle = {33rd British Machine Vision Conference 2022, {BMVC} 2022, London, UK, November 21-24, 2022},
publisher = {{BMVA} Press},
year = {2022},
url = {https://bmvc2022.mpi-inf.mpg.de/1054.pdf}
}
Copyright © 2022
The British Machine Vision Conference is organised by