arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.00523cs.LGcs.CV

Co-Seg: An Image Segmentation Framework Against Label Corruption

Columbia University(哥伦比亚大学);Imperial College London(帝国理工学院);The University of Alabama(阿拉巴马大学)

Ziyi Huang, Haofeng Zhang, Andrew Laine, Elsa Angelini, Christine Hendon, Yu Gan

英文摘要:

Supervised deep learning performance is heavily tied to the availability of high-quality labels for training. Neural networks can gradually overfit corrupted labels if directly trained on noisy datasets, leading to severe performance degradation at test time. In this paper, we propose a novel deep learning framework, namely Co-Seg, to collaboratively train segmentation networks on datasets which include low-quality noisy labels. Our approach first trains two networks simultaneously to sift through all samples and obtain a subset with reliable labels. Then, an efficient yet easily-implemented label correction strategy is applied to enrich the reliable subset. Finally, using the updated dataset, we retrain the segmentation network to finalize its parameters. Experiments in two noisy labels scenarios demonstrate that our proposed model can achieve results comparable to those obtained from supervised learning trained on the noise-free labels. In addition, our framework can be easily implemented in any segmentation algorithm to increase its robustness to noisy labels.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑