How to Efficiently Annotate Images for Best-Performing Deep Learning Based Segmentation Models: An Empirical Study with Weak and Noisy Annotations and Segment Anything Model
Yixin Zhang, Shen Zhao, Hanxue Gu, Maciej A. Mazurowski
TL;DR
This study investigates the cost-effectiveness of weak and noisy image annotations for training deep segmentation models across four datasets (VOC2012, WATER, LiTS17, BraTS20). By combining a three-step methodology—annotation time/quality estimation, programmatic imperfect annotation generation, and annotation-budget-restricted training—with four model-dataset pairings, the authors show that precise pixel-level labels are rarely the best use of a limited labeling budget. They demonstrate that noisy and weak annotations, particularly when paired with tools like the Segment Anything Model (SAM) or CRF-based regularization, can achieve comparable $mIoU$ performance at substantially lower cost, with the best option depending on the domain and annotation style. The work provides practical guidance for budget-conscious annotation strategies and introduces a framework and codebase for evaluating annotation-cost trade-offs across segmentation tasks.
Abstract
Deep neural networks (DNNs) have demonstrated exceptional performance across various image segmentation tasks. However, the process of preparing datasets for training segmentation DNNs is both labor-intensive and costly, as it typically requires pixel-level annotations for each object of interest. To mitigate this challenge, alternative approaches such as using weak labels (e.g., bounding boxes or scribbles) or less precise (noisy) annotations can be employed. Noisy and weak labels are significantly quicker to generate, allowing for more annotated images within the same time frame. However, the potential decrease in annotation quality may adversely impact the segmentation performance of the resulting model. In this study, we conducted a comprehensive cost-effectiveness evaluation on six variants of annotation strategies (9~10 sub-variants in total) across 4 datasets and conclude that the common practice of precisely outlining objects of interest is virtually never the optimal approach when annotation budget is limited. Both noisy and weak annotations showed usage cases that yield similar performance to the perfectly annotated counterpart, yet had significantly better cost-effectiveness. We hope our findings will help researchers be aware of the different available options and use their annotation budgets more efficiently, especially in cases where accurately acquiring labels for target objects is particularly costly. Our code will be made available on https://github.com/yzluka/AnnotationEfficiency2D.
