arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2203.03587cs.CV

On the pitfalls of entropy-based uncertainty for multi-class semi-supervised segmentation

CentraleSupélec(中央理工-高等电力学院);CHU Sainte-Justine(圣贾斯汀大学医院中心);ETS Montreal(蒙特利尔高等技术学院)

Martin Van Waerebeke, Gregory Lodygensky, Jose Dolz

英文摘要:

Semi-supervised learning has emerged as an appealing strategy to train deep models with limited supervision. Most prior literature under this learning paradigm resorts to dual-based architectures, typically composed of a teacher-student duple. To drive the learning of the student, many of these models leverage the aleatoric uncertainty derived from the entropy of the predictions. While this has shown to work well in a binary scenario, we demonstrate in this work that this strategy leads to suboptimal results in a multi-class context, a more realistic and challenging setting. We argue, indeed, that these approaches underperform due to the erroneous uncertainty approximations in the presence of inter-class overlap. Furthermore, we propose an alternative solution to compute the uncertainty in a multi-class setting, based on divergence distances and which account for inter-class overlap. We evaluate the proposed solution on a challenging multi-class segmentation dataset and in two well-known uncertainty-based segmentation methods. The reported results demonstrate that by simply replacing the mechanism used to compute the uncertainty, our proposed solution brings substantial improvement on tested setups.

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

↑