Annotation-Efficient Coreset Selection for Context-dependent Segmentation
Jin Zhang, Zhe Cao, Biwen Yang, Ruiheng Zhang
摘要
Context-dependent (CD) tasks demand the model to have advanced visual understanding ability, such as recognizing camouflaged objects and medical lesions. Current CD methods rely heavily on pixel-level annotated training sets, neglecting issues from redundant samples and the high annotation costs. In this paper, we address the pruning needs of CD datasets, focusing on selecting the most valuable samples for labeling and training using weak annotations. To achieve this, we decompose CD coreset selection into two steps: sample evaluation and coreset selection, proposing corresponding solutions: points-based optimal transport and a maximum distance entropy strategy. Specifically, we formulate sample evaluation as an optimal transport problem between foreground and background distributions, designing a foreground destruction-reconstruction process based on points to compute transport costs and score samples. For samples of varying importance, our selection strategy balances coreset coverage and diversity. We validate our method on six CD tasks, achieving 1% accuracy loss relative to full training under a 40% pruning rate.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper23
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 被引用 35,902 次
- Deep Learning on a Data Diet: Finding Important Examples Early in TrainingMansheej Paul, Surya Ganguli, Gintare Karolina DziugaiteNeurIPS 2021 · 被引用 806 次
- Selection via Proxy: Efficient Data Selection for Deep LearningCody Coleman, Christopher Yeh, Stephen Mussmann, Baharan Mirzasoleiman 等ICLR 2020 · 被引用 462 次
- GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model TrainingKrishnaTeja Killamsetty, Durga Sivasubramanian, Ganesh Ramakrishnan, Abir De 等ICML 2021 · 被引用 305 次
- Faster Wasserstein Distance Estimation with the Sinkhorn DivergenceLénaïc Chizat, Pierre Roussillon, Flavien Léger, François-Xavier Vialard 等NeurIPS 2020 · 被引用 164 次
相关 Paper
- Point2Mask: Point-supervised Panoptic Segmentation via Optimal TransportWentong Li, Yuqian Yuan, Song Wang, Jianke Zhu 等ICCV 2023 · 被引用 34 次
- Coverage-centric Coreset Selection for High Pruning RatesHaizhong Zheng, Rui Liu, Fan Lai, Atul PrakashICLR 2023 · 被引用 3 次
- Feature Distribution Matching by Optimal Transport for Effective and Robust Coreset SelectionWeiwei Xiao, Yongyong Chen, Qiben Shan, Yaowei Wang 等AAAI 2024 · 被引用 14 次
- Evolution-aware VAriance (EVA) Coreset Selection for Medical Image ClassificationYuxin Hong, Xiao Zhang, Xin Zhang, Joey Tianyi ZhouACM MM 2024 · 被引用 4 次
- FOCT: Few-shot Industrial Anomaly Detection with Foreground-aware Online Conditional TransportLong Tian, Hongyi Zhao, Ruiying Lu, Rongrong Wang 等ACM MM 2024 · 被引用 6 次
